mgechev · mgechev · Mar 6, 2019 · Mar 6, 2019 · rafaelss95 · Mar 2, 2019
diff --git a/src/index.ts b/src/index.ts
@@ -7,7 +7,7 @@ export { Rule as ContextualLifecycleRule } from './contextualLifecycleRule';
 export { Rule as DirectiveClassSuffixRule } from './directiveClassSuffixRule';
 export { Rule as DirectiveSelectorRule } from './directiveSelectorRule';
 export { Rule as ImportDestructuringSpacingRule } from './importDestructuringSpacingRule';
-export { Rule as NoAttributeParameterDecoratorRule } from './noAttributeParameterDecoratorRule';
+export { Rule as NoAttributeDecoratorRule } from './noAttributeDecoratorRule';
 export { Rule as NoConflictingLifecycleRule } from './noConflictingLifecycleRule';
 export { Rule as NoForwardRefRule } from './noForwardRefRule';
 export { Rule as NoHostMetadataPropertyRule } from './noHostMetadataPropertyRule';

diff --git a/src/noAttributeDecoratorRule.ts b/src/noAttributeDecoratorRule.ts
@@ -0,0 +1,61 @@
+import { IRuleMetadata, RuleFailure, WalkContext } from 'tslint';
+import { AbstractRule } from 'tslint/lib/rules';
+import {
+  ConstructorDeclaration,
+  Decorator,
+  forEachChild,
+  isConstructorDeclaration,
+  Node,
+  ParameterDeclaration,
+  SourceFile
+} from 'typescript';
+import { getDecoratorName } from './util/utils';
+
+export class Rule extends AbstractRule {
+  static readonly metadata: IRuleMetadata = {
+    description: 'Disallows usage of @Attribute decorator.',
+    options: null,
+    optionsDescription: 'Not configurable.',
+    rationale: '@Attribute is considered bad practice. Use @Input instead.',
+    ruleName: 'no-attribute-decorator',
+    type: 'functionality',
+    typescriptOnly: true
+  };
+
+  static readonly FAILURE_STRING = '@Attribute is considered bad practice. Use @Input instead.';
+
+  apply(sourceFile: SourceFile): RuleFailure[] {
+    return this.applyWithFunction(sourceFile, walk);
+  }
+}
+
+const isAttributeDecorator = (decorator: Decorator): boolean => getDecoratorName(decorator) === 'Attribute';
+
+const validateConstructor = (context: WalkContext<void>, node: ConstructorDeclaration): void =>
+  node.parameters.forEach(parameter => validateParameter(context, parameter));
+
+const validateDecorator = (context: WalkContext<void>, decorator: Decorator): void => {
+  if (!isAttributeDecorator(decorator)) return;
+
+  context.addFailureAtNode(decorator, Rule.FAILURE_STRING);
+};
+
+const validateParameter = (context: WalkContext<void>, parameter: ParameterDeclaration): void => {
+  const { decorators } = parameter;
+
+  if (!decorators) return;
+
+  decorators.forEach(decorator => validateDecorator(context, decorator));
+};
+
+const walk = (context: WalkContext<void>): void => {
+  const { sourceFile } = context;
+
+  const callback = (node: Node): void => {
+    if (isConstructorDeclaration(node)) validateConstructor(context, node);
+
+    forEachChild(node, callback);
+  };
+
+  forEachChild(sourceFile, callback);
+};
diff --git a/src/noAttributeParameterDecoratorRule.ts b/src/noAttributeParameterDecoratorRule.ts
diff --git a/test/noAttributeDecoratorRule.spec.ts b/test/noAttributeDecoratorRule.spec.ts
@@ -0,0 +1,112 @@
+import { Rule } from '../src/noAttributeDecoratorRule';
+import { assertAnnotated, assertMultipleAnnotated, assertSuccess } from './testHelper';
+
+const {
+  FAILURE_STRING,
+  metadata: { ruleName }
+} = Rule;
+
+describe(ruleName, () => {
+  describe('failure', () => {
+    it('should fail if @Attribute decorator is used', () => {
+      const source = `
+        class Test {
+          label: string;
+
+          constructor(@Attribute('label') label) {
+                      ~~~~~~~~~~~~~~~~~~~
+            this.label = label;
+          }
+        }
+      `;
+      assertAnnotated({
+        message: FAILURE_STRING,
+        ruleName,
+        source
+      });
+    });
+
+    it('should fail if @Attribute decorator is used in a class expression', () => {
+      const source = `
+        class Test {
+          private count = 0;
+
+          InnerTestClass = new class {
+            constructor(@Attribute('label') label: string) {}
+                        ~~~~~~~~~~~~~~~~~~~
+          }(this);
+        }
+      `;
+      assertAnnotated({
+        message: FAILURE_STRING,
+        ruleName,
+        source
+      });
+    });
+
+    it('should fail if @Attribute decorator is used in a property class declaration', () => {
+      const source = `
+        class Test {
+          static InnerTestClass = class extends TestCase {
+            constructor(@Attribute('label') label: string) {}
+                        ~~~~~~~~~~~~~~~~~~~
+          };
+        }
+      `;
+      assertAnnotated({
+        message: FAILURE_STRING,
+        ruleName,
+        source
+      });
+    });
+
+    it('should fail if multiple @Attribute decorators are used', () => {
+      const source = `
+        class Test {
+          constructor(@Attribute('label') label: string) {
+                      ~~~~~~~~~~~~~~~~~~~
+          }
+
+          InnerTest1 = new class {
+            constructor(@Attribute('label') label: string) {}
+                        ^^^^^^^^^^^^^^^^^^^
+          }(this);
+
+          static InnerTestClass2 = class extends TestCase {
+            constructor(@Attribute('label') label: string) {}
+                        ###################
+          };
+        }
+      `;
+      assertMultipleAnnotated({
+        failures: [
+          {
+            char: '~',
+            msg: FAILURE_STRING
+          },
+          {
+            char: '^',
+            msg: FAILURE_STRING
+          },
+          {
+            char: '#',
+            msg: FAILURE_STRING
+          }
+        ],
+        ruleName,
+        source
+      });
+    });
+  });
+
+  describe('success', () => {
+    it('should succeed if no @Attribute decorator is used', () => {
+      const source = `
+        class Test {
+          constructor(@Property('label') label: string) {}
+        }
+      `;
+      assertSuccess(ruleName, source);
+    });
+  });
+});
diff --git a/test/noAttributeParameterDecoratorRule.spec.ts b/test/noAttributeParameterDecoratorRule.spec.ts