library angular2.transform.common.annotation_matcher;

import 'package:analyzer/src/generated/ast.dart';
import 'package:barback/barback.dart' show AssetId;
import 'class_matcher_base.dart';

export 'class_matcher_base.dart' show ClassDescriptor;

/// [ClassDescriptor]s for the default angular annotations that can appear
/// on a class. These classes are re-exported in many places so this covers all
/// the possible libraries which could provide them.
const INJECTABLES = const [
  const ClassDescriptor(
      'Injectable', 'package:angular2/src/di/decorators.dart'),
  const ClassDescriptor('Injectable', 'package:angular2/di.dart'),
  const ClassDescriptor('Injectable', 'package:angular2/angular2.dart'),
  const ClassDescriptor('Injectable', 'package:angular2/bootstrap_static.dart'),
];

const DIRECTIVES = const [
  const ClassDescriptor(
      'Directive', 'package:angular2/src/core/annotations/annotations.dart',
      superClass: 'Injectable'),
  const ClassDescriptor(
      'Directive', 'package:angular2/src/core/annotations/decorators.dart',
      superClass: 'Injectable'),
  const ClassDescriptor('Directive',
      'package:angular2/src/core/annotations_impl/annotations.dart',
      superClass: 'Injectable'),
  const ClassDescriptor('Directive', 'package:angular2/annotations.dart',
      superClass: 'Injectable'),
  const ClassDescriptor('Directive', 'package:angular2/angular2.dart',
      superClass: 'Injectable'),
  const ClassDescriptor('Directive', 'package:angular2/core.dart',
      superClass: 'Injectable'),
  const ClassDescriptor('Directive', 'package:angular2/bootstrap_static.dart',
      superClass: 'Injectable'),
];

const COMPONENTS = const [
  const ClassDescriptor(
      'Component', 'package:angular2/src/core/annotations/annotations.dart',
      superClass: 'Directive'),
  const ClassDescriptor(
      'Component', 'package:angular2/src/core/annotations/decorators.dart',
      superClass: 'Directive'),
  const ClassDescriptor('Component',
      'package:angular2/src/core/annotations_impl/annotations.dart',
      superClass: 'Directive'),
  const ClassDescriptor('Component', 'package:angular2/annotations.dart',
      superClass: 'Directive'),
  const ClassDescriptor('Component', 'package:angular2/angular2.dart',
      superClass: 'Directive'),
  const ClassDescriptor('Component', 'package:angular2/bootstrap_static.dart',
      superClass: 'Directive'),
  const ClassDescriptor('Component', 'package:angular2/core.dart',
      superClass: 'Directive'),
];

const VIEWS = const [
  const ClassDescriptor('View', 'package:angular2/view.dart'),
  const ClassDescriptor('View', 'package:angular2/angular2.dart'),
  const ClassDescriptor('View', 'package:angular2/bootstrap_static.dart'),
  const ClassDescriptor('View', 'package:angular2/core.dart'),
  const ClassDescriptor(
      'View', 'package:angular2/src/core/annotations/view.dart'),
  const ClassDescriptor(
      'View', 'package:angular2/src/core/annotations_impl/view.dart'),
];

/// Checks if a given [Annotation] matches any of the given
/// [ClassDescriptors].
class AnnotationMatcher extends ClassMatcherBase {
  AnnotationMatcher._(classDescriptors) : super(classDescriptors);

  factory AnnotationMatcher() {
    return new AnnotationMatcher._([]
      ..addAll(COMPONENTS)
      ..addAll(DIRECTIVES)
      ..addAll(INJECTABLES)
      ..addAll(VIEWS));
  }

  bool _implementsWithWarning(Annotation annotation, AssetId assetId,
      List<ClassDescriptor> interfaces) {
    ClassDescriptor descriptor = firstMatch(annotation.name, assetId);
    if (descriptor == null) return false;
    return implements(descriptor, interfaces,
        missingSuperClassWarning: 'Missing `custom_annotation` entry for `${descriptor.superClass}`.');
  }

  /// Checks if an [Annotation] node implements [Injectable].
  bool isInjectable(Annotation annotation, AssetId assetId) =>
      _implementsWithWarning(annotation, assetId, INJECTABLES);

  /// Checks if an [Annotation] node implements [Directive].
  bool isDirective(Annotation annotation, AssetId assetId) =>
      _implementsWithWarning(annotation, assetId, DIRECTIVES);

  /// Checks if an [Annotation] node implements [Component].
  bool isComponent(Annotation annotation, AssetId assetId) =>
      _implementsWithWarning(annotation, assetId, COMPONENTS);

  /// Checks if an [Annotation] node implements [View].
  bool isView(Annotation annotation, AssetId assetId) =>
      _implementsWithWarning(annotation, assetId, VIEWS);
}