createSassMigrator method to wrap common functionality

Author: jesstelford

.changeset/silent-spiders-poke.md

@@ -0,0 +1,5 @@
+---
+'@shopify/polaris-migrator': minor
+---
+
+Add `createSassMigrator` utility to stash common logic, starting with only parsing each event once.

polaris-migrator/README.md

@@ -235,11 +235,18 @@ Be aware that this may also create additional code changes in your codebase, we
 npx @shopify/polaris-migrator replace-sass-spacing <path>
 ```
 
-## Creating a migration
+## Creating Migrations
 
-### Setup
+Sometimes referred to as "codemods", migrations are JavaScript functions which modify some code from one form to another (eg; to move between breaking versions of `@shopify/polaris`). ASTs (Abstract Syntax Trees) are used to "walk" through the code in discreet, strongly typed steps, called "nodes". All changes made to nodes (and thus the AST) are then written out as the new/"migrated" version of the code.
 
-Run `yarn new-migration` to generate a new migration from a template.
+`polaris-migrator` supports two types of migrations:
+
+- SASS Migrations
+- Typescript Migrations
+
+### Creating a SASS migration
+
+Run `yarn new-migration` to generate a new migration from the `sass-migration` template:
 
 ```sh
 ❯ yarn new-migration
@@ -250,7 +257,7 @@ $ plop
   typescript-migration
 ```
 
-We will use the `sass-migration` and call our migration `replace-sass-function` for this example. Provide the name of your migration:
+Next, provide the name of your migration. For example; `replace-sass-function`:
 
 ```sh
 ? [PLOP] Please choose a generator. sass-migration
@@ -269,45 +276,44 @@ migrations
         └── replace-sass-function.test.ts
 ```
 
-### Writing migration function
+#### The SASS migration function
 
-A migration is simply a javascript function which serves as the entry-point for your codemod. The `replace-sass-function.ts` file defines a "migration" function. This function is named the same as the provided migration name, `replace-sass-function`, and is the default export of the file.
+Each migrator has a default export adhering to the [PostCSS Plugin API](https://github.com/postcss/postcss/blob/main/docs/writing-a-plugin.md) with one main difference: events are only executed once.
 
-Some example code has been provided for each template. You can make any migration code adjustments in the migration function. For Sass migrations, a [PostCSS plugin](https://github.com/postcss/postcss/blob/main/docs/writing-a-plugin.md) is used to parse and transform the source code provided by the [jscodeshift](https://github.com/facebook/jscodeshift).
+Continuing the example, here is what the migration may look like if our goal is to replace the Sass function `hello()` with `world()`.
 
 ```ts
 // polaris-migrator/src/migrations/replace-sass-function/replace-sass-function.ts
-
-import type {FileInfo} from 'jscodeshift';
-import postcss, {Plugin} from 'postcss';
 import valueParser from 'postcss-value-parser';
 
-const plugin = (): Plugin => ({
-  postcssPlugin: 'replace-sass-function',
-  Declaration(decl) {
-    // const prop = decl.prop;
-    const parsedValue = valueParser(decl.value);
-
-    parsedValue.walk((node) => {
-      if (!(node.type === 'function' && node.value === 'hello')) return;
-
-      node.value = 'world';
-    });
-
-    decl.value = parsedValue.toString();
-  },
+import {
+  isSassFunction,
+  StopWalkingFunctionNodes,
+  createSassMigrator,
+} from '../../utilities/sass';
+
+// options can be passed in from cli / config.
+export default createSassMigrator('replace-sass-function', (/* options */) => {
+  return {
+    Declaration(decl) {
+      const parsedValue = valueParser(decl.value);
+
+      parsedValue.walk((node) => {
+        if (isSassFunction('hello', node)) {
+          node.value = 'world';
+          return StopWalkingFunctionNodes;
+        }
+      });
+
+      decl.value = parsedValue.toString();
+    },
+  };
 });
-
-export default function replaceSassFunction(fileInfo: FileInfo) {
-  return postcss(plugin()).process(fileInfo.source, {
-    syntax: require('postcss-scss'),
-  }).css;
-}
 ```
 
-This example migration will replace the Sass function `hello()` with `world()`.
+A more complete example can be seen in [`replace-spacing-lengths.ts`](https://github.com/Shopify/polaris/blob/main/polaris-migrator/src/migrations/replace-spacing-lengths/replace-spacing-lengths.ts).
 
-### Testing
+#### Testing
 
 The template will also generate starting test files you can use to test your migration. In your migrations `tests` folder, you can see 3 files:
 
@@ -317,6 +323,8 @@ The template will also generate starting test files you can use to test your mig
 
 The main test file will load the input/output fixtures to test your migration against. You can configure additional fixtures and test migration options (see the `replace-sass-spacing.test.ts` as an example).
 
+## Running Migrations
+
 Run tests locally from workspace root by filtering to the migrations package:
 
 ```sh

polaris-migrator/src/migrations/replace-spacing-lengths/replace-spacing-lengths.ts

@@ -1,5 +1,3 @@
-import type {FileInfo, API, Options} from 'jscodeshift';
-import postcss, {Plugin} from 'postcss';
 import valueParser from 'postcss-value-parser';
 
 import {POLARIS_MIGRATOR_COMMENT} from '../../constants';
@@ -10,35 +8,17 @@ import {
   isSassFunction,
   isTransformableLength,
   namespace,
-  NamespaceOptions,
   toTransformablePx,
   StopWalkingFunctionNodes,
+  createSassMigrator,
 } from '../../utilities/sass';
 import {isKeyOf} from '../../utilities/type-guards';
 
-export default function replaceSpacingLengths(
-  fileInfo: FileInfo,
-  _: API,
-  options: Options,
-) {
-  return postcss(plugin(options)).process(fileInfo.source, {
-    syntax: require('postcss-scss'),
-  }).css;
-}
-
-const processed = Symbol('processed');
-
-interface PluginOptions extends Options, NamespaceOptions {}
-
-const plugin = (options: PluginOptions = {}): Plugin => {
+export default createSassMigrator('replace-sass-space', (options) => {
   const namespacedRem = namespace('rem', options);
 
   return {
-    postcssPlugin: 'replace-sass-space',
     Declaration(decl) {
-      // @ts-expect-error - Skip if processed so we don't process it again
-      if (decl[processed]) return;
-
       if (!spaceProps.has(decl.prop)) return;
 
       /**
@@ -125,7 +105,7 @@ const plugin = (options: PluginOptions = {}): Plugin => {
       }
     },
   };
-};
+});
 
 const globalValues = new Set(['inherit', 'initial', 'unset']);
 

polaris-migrator/src/utilities/sass.ts

@@ -1,4 +1,12 @@
-import postcss from 'postcss';
+import type {FileInfo, API, Options} from 'jscodeshift';
+import postcss, {
+  Root,
+  Plugin,
+  Declaration,
+  Rule as PostCSSRule,
+  Comment as PostCSSComment,
+  AtRule,
+} from 'postcss';
 import valueParser, {
   Node,
   ParsedValue,
@@ -251,3 +259,50 @@ export function createInlineComment(text: string, options?: {prose?: boolean}) {
 
   return comment;
 }
+
+interface PluginOptions extends Options, NamespaceOptions {}
+
+interface PostCSSWalkers {
+  AtRule?: (node: AtRule) => void;
+  Comment?: (node: PostCSSComment) => void;
+  Declaration?: (node: Declaration) => void;
+  Rule?: (node: PostCSSRule) => void;
+}
+
+export function createSassMigrator(
+  name: string,
+  ruleFn: (options: PluginOptions) => PostCSSWalkers,
+) {
+  return (fileInfo: FileInfo, _: API, options: Options) => {
+    const plugin: Plugin = {
+      postcssPlugin: name,
+      // PostCSS will rewalk the AST every time a declaration/rule/etc is
+      // mutated by a plugin. This can be useful in some cases, but in ours we
+      // only want single-pass behaviour.
+      //
+      // This can be avoided in 2 ways:
+      //
+      // 1) Flagging each declaration as we pass it, then skipping it on
+      //    subsequent passes.
+      // 2) Using postcss's Once() plugin callback.
+      //
+      // We're going with the Once() callback as it's idomatic PostCSS.
+      Once(root: Root) {
+        // NOTE: We initialise the rule here _inside_ the postcss Once function
+        // so multiple passes can be performed without rules accidentally
+        // retaining scoped variables, etc.
+        const rules = ruleFn(options);
+
+        if (rules.AtRule) root.walkAtRules((node) => rules.AtRule(node));
+        if (rules.Comment) root.walkComments((node) => rules.Comment(node));
+        if (rules.Declaration)
+          root.walkDecls((node) => rules.Declaration(node));
+        if (rules.Rule) root.walkRules((node) => rules.Rule(node));
+      },
+    };
+
+    return postcss(plugin).process(fileInfo.source, {
+      syntax: require('postcss-scss'),
+    }).css;
+  };
+}

polaris-migrator/templates/sass-migration/{{kebabCase migrationName}}/{{kebabCase migrationName}}.ts.hbs

@@ -1,33 +1,25 @@
-import type {FileInfo} from 'jscodeshift';
-import postcss, {Plugin} from 'postcss';
 import valueParser from 'postcss-value-parser';
 
-const processed = Symbol('processed');
-
-const plugin = (): Plugin => ({
-  postcssPlugin: '{{kebabCase migrationName}}',
-  Declaration(decl) {
-    // @ts-expect-error - Skip if processed so we don't process it again
-    if (decl[processed]) return;
-
-    // const prop = decl.prop;
-    const parsedValue = valueParser(decl.value);
-
-    parsedValue.walk((node) => {
-      if (!(node.type === 'function' && node.value === 'hello')) return;
-
-      node.value = 'world';
-    });
-
-    decl.value = parsedValue.toString();
-
-    // @ts-expect-error - Mark the declaration as processed
-    decl[processed] = true;
-  },
+import {
+  isSassFunction,
+  StopWalkingFunctionNodes,
+  createSassMigrator,
+} from '../../utilities/sass';
+
+// options can be passed in from cli / config.
+export default createSassMigrator('{{kebabCase migrationName}}', (/* options */) => {
+  return {
+    Declaration(decl) {
+      const parsedValue = valueParser(decl.value);
+
+      parsedValue.walk((node) => {
+        if (isSassFunction('hello', node)) {
+          node.value = 'world';
+          return StopWalkingFunctionNodes;
+        }
+      });
+
+      decl.value = parsedValue.toString();
+    },
+  };
 });
-
-export default function {{camelCase migrationName}}(fileInfo: FileInfo) {
-  return postcss(plugin()).process(fileInfo.source, {
-    syntax: require('postcss-scss'),
-  }).css;
-}

yarn.lock

@@ -16568,6 +16568,11 @@ obuf@^1.0.0, obuf@^1.1.2:
   resolved "https://registry.yarnpkg.com/obuf/-/obuf-1.1.2.tgz#09bea3343d41859ebd446292d11c9d4db619084e"
   integrity sha512-PX1wu0AmAdPqOL1mWhqmlOd8kOIZQwGZw6rh7uby9fTc5lhaOWFLX3I6R1hrF9k3zUY40e6igsLGkDXK92LJNg==
 
+on-change@^4.0.1:
+  version "4.0.1"
+  resolved "https://registry.yarnpkg.com/on-change/-/on-change-4.0.1.tgz#a3fea9971748efab3560351c8278ea1a692b7c16"
+  integrity sha512-Gxmr/8NsLTigoUUEPvnAIUGl7uxwfC3Mm1G+qTmODEZcT4ZGfeaFyQgfGCdbKVS2pICIFYB82RH6MUHToPcS4Q==
+
 on-exit-leak-free@^2.1.0:
   version "2.1.0"
   resolved "https://registry.yarnpkg.com/on-exit-leak-free/-/on-exit-leak-free-2.1.0.tgz#5c703c968f7e7f851885f6459bf8a8a57edc9cc4"