From 770c9adef0f2158d9097bfd392e182af11d3e4e3 Mon Sep 17 00:00:00 2001
From: paoloricciuti <ricciutipaolo@gmail.com>
Date: Thu, 31 Oct 2024 16:57:40 +0100
Subject: [PATCH] fix: properly transform invalid identifiers

---
 .../codemods/legacy-story.test.ts             | 26 +++++++++++--
 .../pre-transform/codemods/legacy-story.ts    |  5 +--
 .../codemods/template-to-snippet.test.ts      | 39 ++++++++++++-------
 .../codemods/template-to-snippet.ts           |  3 +-
 src/utils/identifier-utils.ts                 | 20 ++++++++++
 5 files changed, 71 insertions(+), 22 deletions(-)

diff --git a/src/compiler/pre-transform/codemods/legacy-story.test.ts b/src/compiler/pre-transform/codemods/legacy-story.test.ts
index cb140d8..517865c 100644
--- a/src/compiler/pre-transform/codemods/legacy-story.test.ts
+++ b/src/compiler/pre-transform/codemods/legacy-story.test.ts
@@ -188,6 +188,28 @@ describe(transformLegacyStory.name, () => {
     ).toMatchInlineSnapshot(`"<Story name="Default" children={someTemplate} />"`);
   });
 
+  it("transforms 'template' prop to 'children' and text expression becomes expression tag with identifier to snippet (case with invalid identifier)", async ({
+    expect,
+  }) => {
+    const code = `
+      <script context="module">
+        import { Story } from "@storybook/addon-svelte-csf";
+      </script>
+
+      <Story name="Default" template="some template with non valid idenitifier" />
+    `;
+    const component = await parseAndExtractSvelteNode<SvelteAST.Component>(code, 'Component');
+
+    expect(
+      print(
+        transformLegacyStory({
+          component,
+          state: { componentIdentifierName: {} },
+        })
+      )
+    ).toMatchInlineSnapshot(`"<Story name="Default" children={template_c0gseq} />"`);
+  });
+
   it("when directive 'let:args' is used then it wraps Story fragment with 'children' snippet block", async ({
     expect,
   }) => {
@@ -280,9 +302,7 @@ describe(transformLegacyStory.name, () => {
     `);
   });
 
-  it("leaves existing Story parameters untouched", async ({
-    expect,
-  }) => {
+  it('leaves existing Story parameters untouched', async ({ expect }) => {
     const code = `
       <script context="module">
         import { Story } from "@storybook/addon-svelte-csf";
diff --git a/src/compiler/pre-transform/codemods/legacy-story.ts b/src/compiler/pre-transform/codemods/legacy-story.ts
index 13c2d31..944b294 100644
--- a/src/compiler/pre-transform/codemods/legacy-story.ts
+++ b/src/compiler/pre-transform/codemods/legacy-story.ts
@@ -1,5 +1,3 @@
-import { camelCase } from 'es-toolkit/string';
-
 import {
   createASTArrayExpression,
   createASTAttribute,
@@ -11,6 +9,7 @@ import {
 } from '#parser/ast';
 import { InvalidTemplateAttribute } from '#utils/error/legacy-api/index';
 
+import { hashTemplateName } from '#utils/identifier-utils';
 import type { State } from '..';
 
 interface Params {
@@ -247,7 +246,7 @@ function templateToChildren(
     value: [
       createASTExpressionTag({
         type: 'Identifier',
-        name: camelCase(
+        name: hashTemplateName(
           value[0].type === 'Text'
             ? value[0].data
             : ((value[0].expression as ESTreeAST.Literal).value as string)
diff --git a/src/compiler/pre-transform/codemods/template-to-snippet.test.ts b/src/compiler/pre-transform/codemods/template-to-snippet.test.ts
index 1c1765f..1e012a4 100644
--- a/src/compiler/pre-transform/codemods/template-to-snippet.test.ts
+++ b/src/compiler/pre-transform/codemods/template-to-snippet.test.ts
@@ -20,11 +20,7 @@ describe(transformTemplateToSnippet.name, () => {
     `;
     const component = await parseAndExtractSvelteNode<SvelteAST.Component>(code, 'Component');
 
-    expect(
-      print(
-        transformTemplateToSnippet({ component })
-      )
-    ).toMatchInlineSnapshot(`
+    expect(print(transformTemplateToSnippet({ component }))).toMatchInlineSnapshot(`
       "{#snippet sb_default_template(args)}
       	<Button {...args} variant="primary" />
       {/snippet}"
@@ -43,17 +39,34 @@ describe(transformTemplateToSnippet.name, () => {
     `;
     const component = await parseAndExtractSvelteNode<SvelteAST.Component>(code, 'Component');
 
-    expect(
-      print(
-        transformTemplateToSnippet({ component })
-      )
-    ).toMatchInlineSnapshot(`
+    expect(print(transformTemplateToSnippet({ component }))).toMatchInlineSnapshot(`
 			"{#snippet coolTemplate(args)}
 				<Button {...args} variant="primary" />
 			{/snippet}"
 		`);
   });
 
+  it("covers a case with provided prop 'id' and prop `id` not being a valid identifier", async ({
+    expect,
+  }) => {
+    const code = `
+      <script context="module" lang="ts">
+        import { Template } from "${pkg.name}";
+      </script>
+
+      <Template id="cool-template" let:args>
+        <Button {...args} variant="primary" />
+      </Template>
+    `;
+    const component = await parseAndExtractSvelteNode<SvelteAST.Component>(code, 'Component');
+
+    expect(print(transformTemplateToSnippet({ component }))).toMatchInlineSnapshot(`
+      "{#snippet template_haitqt(args)}
+      	<Button {...args} variant="primary" />
+      {/snippet}"
+    `);
+  });
+
   it("works with 'let:context' directive", async ({ expect }) => {
     const code = `
       <script context="module" lang="ts">
@@ -66,11 +79,7 @@ describe(transformTemplateToSnippet.name, () => {
     `;
     const component = await parseAndExtractSvelteNode<SvelteAST.Component>(code, 'Component');
 
-    expect(
-      print(
-        transformTemplateToSnippet({ component })
-      )
-    ).toMatchInlineSnapshot(`
+    expect(print(transformTemplateToSnippet({ component }))).toMatchInlineSnapshot(`
       "{#snippet sb_default_template(_args, context)}
       	<p>{context.args}</p>
       {/snippet}"
diff --git a/src/compiler/pre-transform/codemods/template-to-snippet.ts b/src/compiler/pre-transform/codemods/template-to-snippet.ts
index 93a5e6d..3575d3a 100644
--- a/src/compiler/pre-transform/codemods/template-to-snippet.ts
+++ b/src/compiler/pre-transform/codemods/template-to-snippet.ts
@@ -1,5 +1,6 @@
 import { getStringValueFromAttribute } from '#parser/analyse/story/attributes';
 import type { SvelteAST } from '#parser/ast';
+import { hashTemplateName } from '#utils/identifier-utils';
 
 interface Params {
   component: SvelteAST.Component;
@@ -70,7 +71,7 @@ export function transformTemplateToSnippet(params: Params): SvelteAST.SnippetBlo
     type: 'SnippetBlock',
     expression: {
       type: 'Identifier',
-      name: id ?? 'sb_default_template',
+      name: id ? hashTemplateName(id) : 'sb_default_template',
     },
     parameters,
     body: fragment,
diff --git a/src/utils/identifier-utils.ts b/src/utils/identifier-utils.ts
index bcf299a..b3601fc 100644
--- a/src/utils/identifier-utils.ts
+++ b/src/utils/identifier-utils.ts
@@ -54,3 +54,23 @@ export const isValidVariableName = (str: string) => {
 
   return true;
 };
+
+/**
+ * Function to convert a non valid string template name to a valid identifier preventing
+ * clashing with other templates with similar names.
+ *
+ * Stolen with 🧡 from the svelte codebase by @paoloricciuti
+ *
+ * @param str the template name
+ * @returns a hash based on the content of the initial string which is a valid identifier
+ */
+export function hashTemplateName(str: string) {
+  if (isValidVariableName(str)) return str;
+
+  str = str.replace(/\r/g, '');
+  let hash = 5381;
+  let i = str.length;
+
+  while (i--) hash = ((hash << 5) - hash) ^ str.charCodeAt(i);
+  return `template_${(hash >>> 0).toString(36)}`;
+}