From efeaf759def855a97fa8350cbac9269d55d8590b Mon Sep 17 00:00:00 2001
From: Bogdan Bogdanov <bbogdanov@vmware.com>
Date: Mon, 25 Oct 2021 14:21:30 +0300
Subject: [PATCH] feat(eslint-plugin-clarity-adoption): add datalist fixer

Signed-off-by: Bogdan Bogdanov <bbogdanov@vmware.com>
---
 .../src/rules/html-fixer-helpers.ts           | 13 +++-
 .../src/rules/no-clr-datalist/index.ts        | 11 ++++
 .../no-clr-datalist/no-clr-datalist.spec.ts   | 66 +++++++++++++++++++
 3 files changed, 88 insertions(+), 2 deletions(-)

diff --git a/packages/eslint-plugin-clarity-adoption/src/rules/html-fixer-helpers.ts b/packages/eslint-plugin-clarity-adoption/src/rules/html-fixer-helpers.ts
index b6ebec45f..97b130100 100644
--- a/packages/eslint-plugin-clarity-adoption/src/rules/html-fixer-helpers.ts
+++ b/packages/eslint-plugin-clarity-adoption/src/rules/html-fixer-helpers.ts
@@ -97,7 +97,13 @@ export function getAttributeNameFixes(
   return [fixer.replaceTextRange([start, end], newName)];
 }
 
-export function getTagFixes(fixer: RuleFixer, node: HTMLElement, oldTag: string, newTag: string): Array<RuleFix> {
+export function getTagFixes(
+  fixer: RuleFixer,
+  node: HTMLElement,
+  oldTag: string,
+  newTag: string,
+  newAttributes: Array<string> = []
+): Array<RuleFix> {
   const openingTag = `<${oldTag}`;
   const closingTag = `</${oldTag}>`;
 
@@ -109,7 +115,10 @@ export function getTagFixes(fixer: RuleFixer, node: HTMLElement, oldTag: string,
   const closingTagEnd = closingTagStart + closingTag.length - 1;
 
   return [
-    fixer.replaceTextRange([openingTagStart, openingTagEnd], `<${newTag}`),
+    fixer.replaceTextRange(
+      [openingTagStart, openingTagEnd],
+      newAttributes.length ? `<${newTag}` + ' ' + newAttributes?.join(' ') : `<${newTag}`
+    ),
     fixer.replaceTextRange([closingTagStart, closingTagEnd], `</${newTag}`),
   ];
 }
diff --git a/packages/eslint-plugin-clarity-adoption/src/rules/no-clr-datalist/index.ts b/packages/eslint-plugin-clarity-adoption/src/rules/no-clr-datalist/index.ts
index 9ac2a2ce9..c28368420 100644
--- a/packages/eslint-plugin-clarity-adoption/src/rules/no-clr-datalist/index.ts
+++ b/packages/eslint-plugin-clarity-adoption/src/rules/no-clr-datalist/index.ts
@@ -1,6 +1,7 @@
 import { ESLintUtils, TSESTree } from '@typescript-eslint/experimental-utils';
 import { HTMLElement } from '../../types/index';
 import { lintDecoratorTemplate } from '../decorator-template-helper';
+import { getDeprecatedClassFixes, getTagFixes } from '../html-fixer-helpers';
 
 export const createESLintRule = ESLintUtils.RuleCreator(() => ``);
 export type MessageIds = 'clrDatalistFailure';
@@ -26,9 +27,19 @@ export default createESLintRule({
   create(context) {
     return {
       [`HTMLElement[tagName="${disallowedTag}"]`](node: HTMLElement): void {
+        const classNode = node.attributes?.find(attribute => attribute.attributeName.value === 'class');
+
         context.report({
           node: node as any,
           messageId: 'clrDatalistFailure',
+          fix: fixer => {
+            const tagFixes = getTagFixes(fixer, node, 'clr-datalist-container', 'cds-datalist', [
+              `control-width="shrink"`,
+            ]);
+            const attributeFixes = getDeprecatedClassFixes(fixer, classNode, [] as any);
+
+            return [...tagFixes, ...attributeFixes];
+          },
         });
       },
       'ClassDeclaration > Decorator'(node: TSESTree.Decorator): void {
diff --git a/packages/eslint-plugin-clarity-adoption/src/rules/no-clr-datalist/no-clr-datalist.spec.ts b/packages/eslint-plugin-clarity-adoption/src/rules/no-clr-datalist/no-clr-datalist.spec.ts
index 7c5b17d0b..2191c5727 100644
--- a/packages/eslint-plugin-clarity-adoption/src/rules/no-clr-datalist/no-clr-datalist.spec.ts
+++ b/packages/eslint-plugin-clarity-adoption/src/rules/no-clr-datalist/no-clr-datalist.spec.ts
@@ -15,6 +15,27 @@ htmlRuleTester.run('no-clr-datalist', rule, {
           <option *ngFor="let item of items" [value]="item"></option>
         </datalist>
       </clr-datalist-container>`,
+      output: `<cds-datalist control-width="shrink">
+        <input clrDatalistInput [(ngModel)]="vertical" placeholder="No label" name="Option" />
+        <datalist>
+          <option *ngFor="let item of items" [value]="item"></option>
+        </datalist>
+      </cds-datalist>`,
+      locations: [{ line: 1, column: 1 }],
+    }),
+    getInvalidDatalistTest({
+      code: `<clr-datalist-container id="custom-id">
+        <input clrDatalistInput [(ngModel)]="vertical" placeholder="No label" name="Option" />
+        <datalist>
+          <option *ngFor="let item of items" [value]="item"></option>
+        </datalist>
+      </clr-datalist-container>`,
+      output: `<cds-datalist control-width="shrink" id="custom-id">
+        <input clrDatalistInput [(ngModel)]="vertical" placeholder="No label" name="Option" />
+        <datalist>
+          <option *ngFor="let item of items" [value]="item"></option>
+        </datalist>
+      </cds-datalist>`,
       locations: [{ line: 1, column: 1 }],
     }),
     getInvalidDatalistTest({
@@ -27,6 +48,15 @@ htmlRuleTester.run('no-clr-datalist', rule, {
           </datalist>
         </clr-datalist-container>
       `,
+      output: `
+        <div></div>
+        <cds-datalist control-width="shrink">
+          <input clrDatalistInput [(ngModel)]="vertical" placeholder="No label" name="Option" />
+          <datalist>
+            <option *ngFor="let item of items" [value]="item"></option>
+          </datalist>
+        </cds-datalist>
+      `,
       locations: [{ line: 3, column: 9 }],
     }),
     getInvalidDatalistTest({
@@ -40,8 +70,33 @@ htmlRuleTester.run('no-clr-datalist', rule, {
           </clr-datalist-container>
         </div>
       `,
+      output: `
+        <div>
+          <cds-datalist control-width="shrink">
+            <input clrDatalistInput [(ngModel)]="vertical" placeholder="No label" name="Option" />
+            <datalist>
+              <option *ngFor="let item of items" [value]="item"></option>
+            </datalist>
+          </cds-datalist>
+        </div>
+      `,
       locations: [{ line: 3, column: 11 }],
     }),
+    getInvalidDatalistTest({
+      code: `<clr-datalist-container id="custom-id" class="custom-class">
+        <input clrDatalistInput [(ngModel)]="vertical" placeholder="No label" name="Option" />
+        <datalist>
+          <option *ngFor="let item of items" [value]="item"></option>
+        </datalist>
+      </clr-datalist-container>`,
+      output: `<cds-datalist control-width="shrink" id="custom-id" class="custom-class">
+        <input clrDatalistInput [(ngModel)]="vertical" placeholder="No label" name="Option" />
+        <datalist>
+          <option *ngFor="let item of items" [value]="item"></option>
+        </datalist>
+      </cds-datalist>`,
+      locations: [{ line: 1, column: 1 }],
+    }),
     getInvalidDatalistTest({
       code: `
         <div>
@@ -54,6 +109,17 @@ htmlRuleTester.run('no-clr-datalist', rule, {
         </div>
         <clr-datalist-container></clr-datalist-container>
       `,
+      output: `
+        <div>
+          <cds-datalist control-width="shrink">
+            <input clrDatalistInput [(ngModel)]="vertical" placeholder="No label" name="Option" />
+            <datalist>
+              <option *ngFor="let item of items" [value]="item"></option>
+            </datalist>
+          </cds-datalist>
+        </div>
+        <cds-datalist control-width="shrink"></cds-datalist>
+      `,
       locations: [
         { line: 3, column: 11 },
         { line: 10, column: 9 },