chore: initial commit

frontend/packages/lib-shared/method/scriptLoader.ts

@@ -0,0 +1,93 @@
+import { setupDrag } from './setupDrag';
+import { CompanyTypeEnum } from '@lib/shared/enums/commonEnum';
+
+interface ScriptOptions {
+  identifier: string; // 脚本标识
+}
+
+const scriptElementsMap = new Map<string, string>();
+
+function extractSQLBotId(input: string) {
+  const regex = /sqlbot-[^\s"']+/;
+  const match = input.match(regex);
+  return match ? match[0] : null;
+}
+
+export function loadScript(scriptContent: string, options: ScriptOptions): Promise<void> {
+  if (!scriptContent) {
+    return Promise.reject(new Error('scriptContent is empty'));
+  }
+  return new Promise((resolve, reject) => {
+    const content = scriptContent?.trim();
+    if (scriptElementsMap.has(options.identifier)) return;
+
+    // 处理IIFE格式
+    if (content.startsWith('(function')) {
+      const scriptId = extractSQLBotId(content);
+      if (scriptId) {
+        scriptElementsMap.set(options.identifier, scriptId);
+      }
+      // eslint-disable-next-line no-eval
+      eval(content);
+      setTimeout(() => {
+        const button = document.querySelector('.sqlbot-assistant-chat-button');
+        setupDrag(button as HTMLElement);
+      }, 300); // 等待DOM渲染
+      resolve();
+      return;
+    }
+
+    // 处理<script>标签
+    if (content.startsWith('<script')) {
+      const div = document.createElement('div');
+      div.innerHTML = content;
+      const originalScript = div.querySelector('script');
+      if (!originalScript) {
+        reject(new Error('无效的script标签'));
+        return;
+      }
+      const script = document.createElement('script');
+      // 复制所有属性
+      for (let i = 0; i < originalScript.attributes.length; i++) {
+        const attr = originalScript.attributes[i];
+        if (attr.name === 'id') {
+          scriptElementsMap.set(options.identifier, attr.value);
+        }
+        script.setAttribute(attr.name, attr.value);
+      }
+      script.onload = () => {
+        setTimeout(() => {
+          const button = document.querySelector('.sqlbot-assistant-chat-button');
+          setupDrag(button as HTMLElement);
+        }, 300); // 等待DOM渲染
+      };
+
+      document.body.appendChild(script);
+      resolve();
+      return;
+    }
+
+    reject(new Error('不支持的脚本格式'));
+  });
+}
+
+export function removeScript(identifier: string): void {
+  const scriptId = scriptElementsMap.get(identifier);
+  if (scriptId && identifier === CompanyTypeEnum.SQLBot) {
+    // 清理全局单例标记
+    const propName = `${scriptId}-state`;
+    delete (window as any)[propName];
+    if ((window as any).sqlbot_assistant_handler) {
+      delete (window as any).sqlbot_assistant_handler;
+    }
+    // 删除页面上渲染的
+    const floatingElements = document.querySelectorAll('[id^="sqlbot-"]');
+    floatingElements.forEach((el) => {
+      if (el.parentNode && !el.parentNode.isEqualNode(document.body) && !el.parentNode.isEqualNode(document.head)) {
+        (el.parentNode as Element).remove();
+      }
+      el.remove();
+    });
+    scriptElementsMap.delete(identifier);
+  }
+}