charts + simplify

Author: Lukas742

packages/cli/src/scripts/codemod/transforms/export-maps/main.cts

@@ -1,5 +1,3 @@
-/* eslint-disable @typescript-eslint/no-require-imports, @typescript-eslint/no-unsafe-return, @typescript-eslint/no-unsafe-call */
-
 import fs from 'fs';
 import path from 'path';
 import type { API, FileInfo, JSCodeshift, Collection } from 'jscodeshift';
@@ -8,9 +6,9 @@ const mainPackageName = '@ui5/webcomponents-react';
 const basePackageName = '@ui5/webcomponents-react-base';
 const chartsPackageName = '@ui5/webcomponents-react-charts';
 
-const componentPackageNames = [mainPackageName, chartsPackageName];
+const packageNames = [mainPackageName, basePackageName, chartsPackageName];
 
-// main enums
+// Enums for main package
 const libraryPath = require.resolve('@ui5/webcomponents-react/package.json');
 const enumsDir = path.join(path.dirname(libraryPath), 'dist', 'enums');
 let enumNames: Set<string> = new Set();
@@ -28,7 +26,7 @@ try {
   console.warn(`⚠️ Could not read enums directory at ${enumsDir}. Skipping enum detection.`, e);
 }
 
-// exports-map base
+// Mapping functions
 function resolveBaseExport(importedName: string): string | undefined {
   const directMap: Record<string, string> = {
     Device: `${basePackageName}/Device`,
@@ -42,13 +40,26 @@ function resolveBaseExport(importedName: string): string | undefined {
     utils: `${basePackageName}/utils`,
     addCustomCSSWithScoping: `${basePackageName}/internal/addCustomCSSWithScoping.js`,
   };
-  if (directMap[importedName]) {
-    return directMap[importedName];
-  }
-  // fallback
-  if (importedName === 'default' || importedName === 'index') {
-    return basePackageName;
-  }
+  if (directMap[importedName]) return directMap[importedName];
+  if (importedName === 'default' || importedName === 'index') return basePackageName;
+  return undefined;
+}
+
+function resolveChartsExport(importedName: string): string | undefined {
+  const directMap: Record<string, string> = {
+    TimelineChartAnnotation: `${chartsPackageName}/TimelineChartAnnotation`,
+    BarChartPlaceholder: `${chartsPackageName}/BarChartPlaceholder`,
+    BulletChartPlaceholder: `${chartsPackageName}/BulletChartPlaceholder`,
+    ColumnChartPlaceholder: `${chartsPackageName}/ColumnChartPlaceholder`,
+    ColumnChartWithTrendPlaceholder: `${chartsPackageName}/ColumnChartWithTrendPlaceholder`,
+    ComposedChartPlaceholder: `${chartsPackageName}/ComposedChartPlaceholder`,
+    LineChartPlaceholder: `${chartsPackageName}/LineChartPlaceholder`,
+    PieChartPlaceholder: `${chartsPackageName}/PieChartPlaceholder`,
+    ScatterChartPlaceholder: `${chartsPackageName}/ScatterChartPlaceholder`,
+    TimelineChartPlaceholder: `${chartsPackageName}/TimelineChartPlaceholder`,
+  };
+  if (directMap[importedName]) return directMap[importedName];
+  if (importedName === 'default' || importedName === 'index') return chartsPackageName;
   return undefined;
 }
 
@@ -58,9 +69,8 @@ export default function transform(file: FileInfo, api: API): string | undefined
 
   let isDirty = false;
 
-  // main & charts pkg
-  componentPackageNames.forEach((packageName) => {
-    root.find(j.ImportDeclaration, { source: { value: packageName } }).forEach((importPath) => {
+  packageNames.forEach((pkg) => {
+    root.find(j.ImportDeclaration, { source: { value: pkg } }).forEach((importPath) => {
       const specifiers = importPath.node.specifiers || [];
       specifiers.forEach((spec) => {
         if (spec.type !== 'ImportSpecifier') return;
@@ -69,19 +79,29 @@ export default function transform(file: FileInfo, api: API): string | undefined
         if (importPath.node.importKind === 'type') {
           if (importedName.endsWith('PropTypes')) {
             componentName = importedName.replace(/PropTypes$/, '');
-            // charts props
           } else if (importedName.endsWith('Props')) {
             componentName = importedName.replace(/Props$/, '');
           } else if (importedName.endsWith('DomRef')) {
             componentName = importedName.replace(/DomRef$/, '');
           }
         }
-        const newSource =
-          importPath.node.importKind === 'type'
-            ? `${packageName}/${componentName}`
-            : enumNames.has(importedName)
-              ? `${packageName}/enums/${importedName}`
-              : `${packageName}/${importedName}`;
+
+        let newSource: string;
+        if (pkg === mainPackageName) {
+          newSource =
+            importPath.node.importKind === 'type'
+              ? `${mainPackageName}/${componentName}`
+              : enumNames.has(importedName)
+                ? `${mainPackageName}/enums/${importedName}`
+                : `${mainPackageName}/${importedName}`;
+        } else if (pkg === basePackageName) {
+          newSource = resolveBaseExport(importedName) || basePackageName;
+        } else if (pkg === chartsPackageName) {
+          newSource = resolveChartsExport(componentName) || `${chartsPackageName}/${componentName}`;
+        } else {
+          newSource = pkg;
+        }
+
         const newImport = j.importDeclaration(
           [
             j.importSpecifier(
@@ -99,28 +119,5 @@ export default function transform(file: FileInfo, api: API): string | undefined
     });
   });
 
-  // base pkg
-  root.find(j.ImportDeclaration, { source: { value: basePackageName } }).forEach((importPath) => {
-    const specifiers = importPath.node.specifiers || [];
-    specifiers.forEach((spec) => {
-      if (spec.type !== 'ImportSpecifier') return;
-      const importedName = spec.imported.name as string;
-      const newSource = resolveBaseExport(importedName) || basePackageName;
-      const newImport = j.importDeclaration(
-        [
-          j.importSpecifier(
-            j.identifier(importedName),
-            j.identifier(spec.local && typeof spec.local.name === 'string' ? spec.local.name : importedName),
-          ),
-        ],
-        j.literal(newSource),
-      );
-      newImport.importKind = importPath.node.importKind;
-      j(importPath).insertBefore(newImport);
-      isDirty = true;
-    });
-    j(importPath).remove();
-  });
-
   return isDirty ? root.toSource({ quote: 'single' }) : undefined;
 }