Convert 'no-reference' to use WalkContext (#2340)

palantir · Mar 14, 2017 · d76eb67 · d76eb67
1 parent 718b63e
commit d76eb67
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/src/rules/noReferenceRule.ts b/src/rules/noReferenceRule.ts
@@ -38,14 +38,12 @@ export class Rule extends Lint.Rules.AbstractRule {
     public static FAILURE_STRING = "<reference> is not allowed, use imports";
 
     public apply(sourceFile: ts.SourceFile): Lint.RuleFailure[] {
-        return this.applyWithWalker(new NoReferenceWalker(sourceFile, this.getOptions()));
+        return this.applyWithFunction(sourceFile, walk);
     }
 }
 
-class NoReferenceWalker extends Lint.RuleWalker {
-    public visitSourceFile(node: ts.SourceFile) {
-        for (const ref of node.referencedFiles) {
-            this.addFailureFromStartToEnd(ref.pos, ref.end, Rule.FAILURE_STRING);
-        }
+function walk(ctx: Lint.WalkContext<void>): void {
+    for (const ref of ctx.sourceFile.referencedFiles) {
+        ctx.addFailure(ref.pos, ref.end, Rule.FAILURE_STRING);
     }
 }