From 3d1708fca841f6076636b546fb734902ddfedbf9 Mon Sep 17 00:00:00 2001
From: Ayo <ramon.aycojr@gmail.com>
Date: Fri, 20 Oct 2023 09:35:14 +0200
Subject: [PATCH] allow multiple personas

---
 routes/[...index].ts | 28 ++++++++++++++++------------
 1 file changed, 16 insertions(+), 12 deletions(-)

diff --git a/routes/[...index].ts b/routes/[...index].ts
index 420a202..ef591ee 100644
--- a/routes/[...index].ts
+++ b/routes/[...index].ts
@@ -80,7 +80,8 @@ async function insertRegistry(
 
 async function buildRegistry(usedCustomElements: string[], type: "js" | "ts") {
   let registryScript = `<script type='module'>`;
-  let baseClassImported = false;
+  let isBaseClassImported = false;
+  let classesImported = [];
 
   for (const name of usedCustomElements) {
     const content = await useStorage().getItem(
@@ -91,19 +92,22 @@ async function buildRegistry(usedCustomElements: string[], type: "js" | "ts") {
     );
     const className = new evalStore().constructor.name;
 
-    if (
-      !baseClassImported &&
-      content.toString().includes("extends WebComponent")
-    ) {
-      const baseClassImport = `import { WebComponent } from "https://unpkg.com/web-component-base@1.7.1/WebComponent.mjs";`;
+    if (!classesImported.includes(className)) {
+      if (
+        !isBaseClassImported &&
+        content.toString().includes("extends WebComponent")
+      ) {
+        const baseClassImport = `import { WebComponent } from "https://unpkg.com/web-component-base@1.7.1/WebComponent.mjs";`;
 
-      registryScript += baseClassImport;
-      baseClassImported = true;
+        registryScript += baseClassImport;
+        isBaseClassImported = true;
+      }
+
+      registryScript += content;
+
+      registryScript += `customElements.define("${name}", ${className});`;
+      classesImported.push(className);
     }
-
-    registryScript += content;
-
-    registryScript += `customElements.define("${name}", ${className});`;
   }
 
   registryScript += "</script>";