fix(element-type): add autofix

d0whc3r · Sep 16, 2019 · 49d6b1b · 49d6b1b
1 parent 5c52b74
commit 49d6b1b
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/rules/element-type.ts b/src/rules/element-type.ts
@@ -15,6 +15,7 @@ const rule: Rule.RuleModule = {
 
   create(context): Rule.RuleListener {
     const stencil = stencilComponentContext();
+    const parserServices = context.parserServices;
 
     function parseTag(tag: string) {
       let result = tag[0].toUpperCase() + tag.slice(1);
@@ -34,9 +35,16 @@ const rule: Rule.RuleModule = {
           const [{ tag }] = parseDecorator(component);
           const parsedTag = `HTML${parseTag(tag)}Element`;
           if (tagType !== parsedTag) {
+            const originalNode = parserServices.esTreeNodeToTSNodeMap.get(node.parent);
+            const text = originalNode.getFullText();
+            const type = originalNode.type.typeName.escapedText;
             context.report({
-              node: node,
-              message: `@Element type is not matching tag for component (${parsedTag})`
+              node: node.parent,
+              message: `@Element type is not matching tag for component (${parsedTag})`,
+              fix(fixer) {
+                const result = text.replace(`: ${type}`, `: ${parsedTag}`);
+                return fixer.replaceText(node, result);
+              }
             });
           }
         }