pkg/linter/lib/src/rules/use_named_constants.dart

// Copyright (c) 2021, the Dart project authors. Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.

import 'package:analyzer/dart/ast/ast.dart';
import 'package:analyzer/dart/ast/visitor.dart';
import 'package:analyzer/dart/element/element2.dart';
import 'package:analyzer/dart/element/type.dart';
import 'package:analyzer/src/lint/constants.dart'; // ignore: implementation_imports

import '../analyzer.dart';

const _desc = r'Use predefined named constants.';

class UseNamedConstants extends LintRule {
  UseNamedConstants()
    : super(name: LintNames.use_named_constants, description: _desc);

  @override
  LintCode get lintCode => LinterLintCode.use_named_constants;

  @override
  void registerNodeProcessors(
    NodeLintRegistry registry,
    LinterContext context,
  ) {
    var visitor = _Visitor(this);
    registry.addInstanceCreationExpression(this, visitor);
  }
}

class _Visitor extends SimpleAstVisitor<void> {
  final LintRule rule;

  _Visitor(this.rule);

  @override
  void visitInstanceCreationExpression(InstanceCreationExpression node) {
    if (node.isConst) {
      var type = node.staticType;
      if (type is! InterfaceType) return;
      var element = type.element3;
      if (element is ClassElement2) {
        var nodeField =
            node
                .thisOrAncestorOfType<VariableDeclaration>()
                ?.declaredFragment
                ?.element;

        // avoid diagnostic for fields in the same class having the same value
        // class A {
        //   const A();
        //   static const a = A();
        //   static const b = A();
        // }
        if (nodeField?.enclosingElement2 == element) return;

        var library =
            (node.root as CompilationUnit).declaredFragment?.element.library2;
        if (library == null) return;
        var value = node.computeConstantValue().value;
        for (var field in element.fields2.where(
          (e) => e.isStatic && e.isConst,
        )) {
          if (field.isAccessibleIn2(library) &&
              field.computeConstantValue() == value) {
            rule.reportLint(
              node,
              arguments: ['${element.name3}.${field.name3}'],
            );
            return;
          }
        }
      }
    }
  }
}