refactor(material/schematics): accept extra symbols for sass api migration (#22658)

Adding support for extra symbols to migrate lets us deal with symbols that we only want to migrate inside Google.

Author: jelbourn

src/material/schematics/ng-update/migrations/theming-api-v12/migration.ts

@@ -22,6 +22,12 @@ interface DetectImportResult {
   namespaces: string[];
 }
 
+/** Addition mixin and function names that can be updated when invoking migration directly. */
+interface ExtraSymbols {
+  mixins?: Record<string, string>;
+  functions?: Record<string, string>;
+}
+
 /**
  * Migrates the content of a file to the new theming API. Note that this migration is using plain
  * string manipulation, rather than the AST from PostCSS and the schematics string manipulation
@@ -40,13 +46,15 @@ export function migrateFileContent(content: string,
                                    oldMaterialPrefix: string,
                                    oldCdkPrefix: string,
                                    newMaterialImportPath: string,
-                                   newCdkImportPath: string): string {
+                                   newCdkImportPath: string,
+                                   extraMaterialSymbols: ExtraSymbols = {}): string {
   const materialResults = detectImports(content, oldMaterialPrefix);
   const cdkResults = detectImports(content, oldCdkPrefix);
 
   // Try to migrate the symbols even if there are no imports. This is used
   // to cover the case where the Components symbols were used transitively.
-  content = migrateMaterialSymbols(content, newMaterialImportPath, materialResults);
+  content = migrateMaterialSymbols(
+      content, newMaterialImportPath, materialResults, extraMaterialSymbols);
   content = migrateCdkSymbols(content, newCdkImportPath, cdkResults);
   content = replaceRemovedVariables(content, removedMaterialVariables);
 
@@ -103,16 +111,19 @@ function detectImports(content: string, prefix: string): DetectImportResult {
 
 /** Migrates the Material symbols in a file. */
 function migrateMaterialSymbols(content: string, importPath: string,
-                                detectedImports: DetectImportResult): string {
+                                detectedImports: DetectImportResult,
+                                extraMaterialSymbols: ExtraSymbols = {}): string {
   const initialContent = content;
   const namespace = 'mat';
+  const mixinsToUpdate = {...materialMixins, ...extraMaterialSymbols.mixins};
+  const functionsToUpdate = {...materialFunctions, ...extraMaterialSymbols.functions};
 
   // Migrate the mixins.
-  content = renameSymbols(content, materialMixins, detectedImports.namespaces, mixinKeyFormatter,
+  content = renameSymbols(content, mixinsToUpdate, detectedImports.namespaces, mixinKeyFormatter,
     getMixinValueFormatter(namespace));
 
   // Migrate the functions.
-  content = renameSymbols(content, materialFunctions, detectedImports.namespaces,
+  content = renameSymbols(content, functionsToUpdate, detectedImports.namespaces,
     functionKeyFormatter, getFunctionValueFormatter(namespace));
 
   // Migrate the variables.

src/material/schematics/ng-update/test-cases/v12/misc/theming-api-v12.spec.ts

@@ -1,5 +1,6 @@
 import {UnitTestTree} from '@angular-devkit/schematics/testing';
 import {createTestCaseSetup} from '@angular/cdk/schematics/testing';
+import {migrateFileContent} from '@angular/material/schematics/ng-update/migrations/theming-api-v12/migration';
 import {join} from 'path';
 import {MIGRATION_PATH} from '../../../../paths';
 
@@ -692,5 +693,27 @@ describe('v12 theming API migration', () => {
     ]);
   });
 
-
+  it('should migrate extra given mixins and functions', () => {
+    const originalContent = [
+      `@import '~@angular/material/theming';`,
+      `$something: mat-mdc-typography-config();`,
+      `@include mat-mdc-button-theme();`,
+    ].join('\n');
+
+    const migratedContent = migrateFileContent(
+        originalContent,
+        '~@angular/material/',
+        '~@angular/cdk/',
+        '~@angular/material',
+        '~@angular/cdk', {
+          mixins: {'mat-mdc-button-theme': 'mdc-button-theme'},
+          functions: {'mat-mdc-typography-config': 'mdc-typography-config'},
+        });
+
+    expect(migratedContent).toBe([
+      `@use '~@angular/material' as mat;`,
+      `$something: mat.mdc-typography-config();`,
+      `@include mat.mdc-button-theme();`,
+    ].join('\n'));
+  });
 });