From 75d15212b56120246f89844c16c7cb9d50f7a121 Mon Sep 17 00:00:00 2001
From: Lukas Spirig <lukas.spirig@sbb.ch>
Date: Thu, 25 Jan 2024 16:21:54 +0100
Subject: [PATCH] fix: react

---
 config/custom-elements-manifest.config.js | 23 +--------------
 src/react/vite.config.ts                  | 36 +++++++++++++++--------
 2 files changed, 25 insertions(+), 34 deletions(-)

diff --git a/config/custom-elements-manifest.config.js b/config/custom-elements-manifest.config.js
index 665249747c..fb19700740 100644
--- a/config/custom-elements-manifest.config.js
+++ b/config/custom-elements-manifest.config.js
@@ -1,5 +1,3 @@
-import { parse } from 'comment-parser';
-
 /**
  * Docs: https://custom-elements-manifest.open-wc.org/analyzer/getting-started/
  */
@@ -13,26 +11,7 @@ export default {
   plugins: [
     {
       analyzePhase({ ts, node, moduleDoc }) {
-        if (ts.isPropertyDeclaration(node)) {
-          const className = node.parent.name.getText();
-          const classDoc = moduleDoc.declarations.find(
-            (declaration) => declaration.name === className,
-          );
-
-          for (const jsDoc of node.jsDoc ?? []) {
-            for (const parsedJsDoc of parse(jsDoc.getFullText())) {
-              for (const tag of parsedJsDoc.tags) {
-                if (tag.tag === 'ssrchildcounter') {
-                  const member = classDoc.members.find((m) => m.name === node.name.getText());
-                  member['_ssrchildcounter'] = true;
-                }
-              }
-            }
-          }
-        } else if (
-          ts.isNewExpression(node) &&
-          node.expression.getText() === 'NamedSlotStateController'
-        ) {
+        if (ts.isNewExpression(node) && node.expression.getText() === 'NamedSlotStateController') {
           let classNode = node;
           while (classNode) {
             if (ts.isClassDeclaration(classNode)) {
diff --git a/src/react/vite.config.ts b/src/react/vite.config.ts
index 282b31d201..e4c91a2208 100644
--- a/src/react/vite.config.ts
+++ b/src/react/vite.config.ts
@@ -89,7 +89,7 @@ function generateReactWrappers(): PluginOption {
         .filter((m) => m.kind === 'javascript-module')
         .reduce((current, next) => current.concat(next.declarations ?? []), [] as Declaration[]);
       for (const module of manifest.modules) {
-        for (const declaration of module.declarations.filter(
+        for (const declaration of module.declarations?.filter(
           (d): d is CustomElementDeclaration => 'customElement' in d && d.customElement,
         ) ?? []) {
           const targetPath = new URL(`./${module.path}/index.ts`, packageRoot);
@@ -115,8 +115,8 @@ function generateReactWrappers(): PluginOption {
         }
       }
 
-      config.build.lib = {
-        ...(config.build.lib ? config.build.lib : {}),
+      config.build!.lib = {
+        ...(config.build!.lib ? config.build!.lib : {}),
         entry: globIndexMap(packageRoot),
       };
     },
@@ -189,7 +189,7 @@ function findExtensionUsage(
   if (usesSsrSlotState(declaration, declarations)) {
     extensions.set('withSsrDataSlotNames', (v) => `withSsrDataSlotNames(${v})`);
   }
-  const childTypes = usesSsrSlotChildCounter(declaration);
+  const childTypes = namedSlotListElements(declaration);
   if (childTypes.length) {
     extensions.set(
       'withSsrDataChildCount',
@@ -199,32 +199,44 @@ function findExtensionUsage(
   return extensions;
 }
 
+// eslint-disable-next-line @typescript-eslint/naming-convention
+type ClassDeclarationSSR = ClassDeclaration & { _ssrslotstate?: boolean };
 const ssrSlotStateKey = '_ssrslotstate';
-function usesSsrSlotState(declaration: ClassDeclaration, declarations: Declaration[]): boolean {
+function usesSsrSlotState(
+  declaration: ClassDeclarationSSR | undefined,
+  declarations: Declaration[],
+): boolean {
   while (declaration) {
     if (
       declaration[ssrSlotStateKey] ||
       declaration.mixins?.some((m) =>
-        declarations.find((d) => d.name === m.name && d[ssrSlotStateKey]),
+        declarations.find((d) => d.name === m.name && (d as ClassDeclarationSSR)[ssrSlotStateKey]),
       )
     ) {
       return true;
     }
 
     declaration = declarations.find(
-      (d): d is ClassDeclaration => d.name === declaration.superclass?.name,
+      (d): d is ClassDeclarationSSR => d.name === declaration!.superclass?.name,
     );
   }
 
   return false;
 }
 
-const ssrSlotChildCountKey = '_ssrchildcounter';
-function usesSsrSlotChildCounter(declaration: ClassDeclaration): string[] {
+function namedSlotListElements(declaration: ClassDeclaration): string[] {
   return (
     declaration.members
-      ?.find((m): m is ClassField => m[ssrSlotChildCountKey])
-      ?.type?.text.replace(/[()[\] ]/g, '')
-      .split('|') ?? []
+      ?.find(
+        (m): m is ClassField =>
+          m.inheritedFrom?.name === 'NamedSlotListElement' && m.name === 'listChildTagNames',
+      )
+      ?.default?.match(/([\w-]+)/g)
+      ?.map((m) =>
+        m
+          .split('-')
+          .map((s) => s[0] + s.substring(1).toLowerCase())
+          .join(''),
+      ) ?? []
   );
 }