support alias for emitter

mizdra · Jun 15, 2024 · 64a6469 · 64a6469
1 parent 375b217
commit 64a6469
Showing 1 changed file with 16 additions and 9 deletions.
diff --git a/packages/happy-css-modules/src/emitter/dts.ts b/packages/happy-css-modules/src/emitter/dts.ts
@@ -28,18 +28,26 @@ function dashesCamelCase(str: string): string {
 }
 
 function formatTokens(tokens: Token[], localsConvention: LocalsConvention): Token[] {
+  function formatToken(token: Token, formatter: (str: string) => string): Token {
+    if ('importedName' in token && typeof token.importedName === 'string') {
+      return { ...token, name: formatter(token.name), importedName: formatter(token.importedName) };
+    } else {
+      return { ...token, name: formatter(token.name) };
+    }
+  }
+
   const result: Token[] = [];
   for (const token of tokens) {
     if (localsConvention === 'camelCaseOnly') {
-      result.push({ ...token, name: camelcase(token.name) });
+      result.push(formatToken(token, camelcase));
     } else if (localsConvention === 'camelCase') {
       result.push(token);
-      result.push({ ...token, name: camelcase(token.name) });
+      result.push(formatToken(token, camelcase));
     } else if (localsConvention === 'dashesOnly') {
-      result.push({ ...token, name: dashesCamelCase(token.name) });
+      result.push(formatToken(token, dashesCamelCase));
     } else if (localsConvention === 'dashes') {
       result.push(token);
-      result.push({ ...token, name: dashesCamelCase(token.name) });
+      result.push(formatToken(token, dashesCamelCase));
     } else {
       result.push(token); // asIs
     }
@@ -89,11 +97,10 @@ function generateTokenDeclarations(
           : // Imported tokens in non-external files are typed by dynamic import.
             // See https://github.com/mizdra/happy-css-modules/issues/106.
             new SourceNode(null, null, null, [
-              '& Readonly<Pick<(typeof import(',
-              `"${getRelativePath(filePath, originalLocation.filePath)}"`,
-              '))["default"], ',
-              `"${token.name}"`,
-              '>>',
+              `& Readonly<{ ${token.name}: (typeof import("${getRelativePath(
+                filePath,
+                originalLocation.filePath,
+              )}"))["default"]["${token.importedName ?? token.name}"] }>`,
             ]),
       );
     }