vite-plugin-opencode-assistant 1.0.26 → 1.0.28

package/es/client/composables/useSSE.js

@@ -1,59 +1,116 @@
-import { ref, onUnmounted } from "vue";
-const MAX_SSE_RETRIES = 10;
-const SSE_RETRY_DELAY = 1e3;
-function useSSE(onStatusSync, onTaskUpdate, onClearElements, onConnected) {
-  const sseConnection = ref(null);
-  const sseRetryCount = ref(0);
-  const setupSSE = () => {
-    if (sseConnection.value) return;
+import { ref, computed, onUnmounted } from "vue";
+const DEFAULT_MAX_RETRIES = 10;
+const DEFAULT_RETRY_DELAY = 1e3;
+function useSSE(options) {
+  const {
+    endpoint,
+    autoConnect = true,
+    enabled,
+    maxRetries = DEFAULT_MAX_RETRIES,
+    retryDelay = DEFAULT_RETRY_DELAY,
+    onConnected,
+    onDisconnected,
+    onError,
+    onMessage
+  } = options;
+  const connection = ref(null);
+  const status = ref("idle");
+  const retryCount = ref(0);
+  function handleMessage(event) {
     try {
-      sseConnection.value = new EventSource("/__opencode_events__");
-      sseConnection.value.onmessage = (event) => {
-        try {
-          const data = JSON.parse(event.data);
-          if (data.type === "CONNECTED") {
-            onConnected();
-            sseRetryCount.value = 0;
-          } else if (data.type === "STATUS_SYNC") {
-            onStatusSync(data);
-          } else if (data.type === "TASK_UPDATE") {
-            onTaskUpdate(data);
-          } else if (data.type === "CLEAR_ELEMENTS") {
-            onClearElements();
-          }
-        } catch (e) {
-        }
+      const data = JSON.parse(event.data);
+      onMessage == null ? void 0 : onMessage(data);
+    } catch (e) {
+      onMessage == null ? void 0 : onMessage(event.data);
+    }
+  }
+  function connect() {
+    if (connection.value || status.value === "connecting") {
+      return;
+    }
+    if ((enabled == null ? void 0 : enabled.value) === false) {
+      status.value = "idle";
+      return;
+    }
+    status.value = "connecting";
+    retryCount.value = 0;
+    try {
+      connection.value = new EventSource(endpoint);
+      connection.value.onopen = () => {
+        status.value = "connected";
+        retryCount.value = 0;
+        onConnected == null ? void 0 : onConnected();
       };
-      sseConnection.value.onerror = () => {
+      connection.value.onmessage = handleMessage;
+      connection.value.onerror = () => {
         var _a;
-        (_a = sseConnection.value) == null ? void 0 : _a.close();
-        sseConnection.value = null;
-        if (sseRetryCount.value < MAX_SSE_RETRIES) {
-          sseRetryCount.value++;
-          setTimeout(setupSSE, SSE_RETRY_DELAY * sseRetryCount.value);
+        const wasConnected = status.value === "connected";
+        status.value = "error";
+        (_a = connection.value) == null ? void 0 : _a.close();
+        connection.value = null;
+        const error = new Error(`SSE connection error: ${endpoint}`);
+        onError == null ? void 0 : onError(error);
+        if (retryCount.value < maxRetries) {
+          retryCount.value++;
+          const delay = retryDelay * retryCount.value;
+          setTimeout(() => {
+            if ((enabled == null ? void 0 : enabled.value) !== false && !connection.value) {
+              connect();
+            }
+          }, delay);
+        } else if (wasConnected) {
+          onDisconnected == null ? void 0 : onDisconnected();
         }
       };
     } catch (e) {
-      sseConnection.value = null;
-      if (sseRetryCount.value < MAX_SSE_RETRIES) {
-        sseRetryCount.value++;
-        setTimeout(setupSSE, SSE_RETRY_DELAY * sseRetryCount.value);
+      status.value = "error";
+      const error = e instanceof Error ? e : new Error(String(e));
+      onError == null ? void 0 : onError(error);
+      if (retryCount.value < maxRetries) {
+        retryCount.value++;
+        const delay = retryDelay * retryCount.value;
+        setTimeout(() => {
+          if (!connection.value) {
+            connect();
+          }
+        }, delay);
       }
     }
-  };
-  const closeSSE = () => {
-    if (sseConnection.value) {
-      sseConnection.value.close();
-      sseConnection.value = null;
+  }
+  function disconnect() {
+    if (connection.value) {
+      connection.value.close();
+      connection.value = null;
+      status.value = "disconnected";
+      retryCount.value = 0;
+      onDisconnected == null ? void 0 : onDisconnected();
     }
-  };
+  }
+  function reconnect() {
+    disconnect();
+    connect();
+  }
+  function resetRetryCount() {
+    retryCount.value = 0;
+  }
+  if (autoConnect && (enabled == null ? void 0 : enabled.value) !== false) {
+    connect();
+  }
   onUnmounted(() => {
-    closeSSE();
+    disconnect();
   });
   return {
-    setupSSE,
-    closeSSE,
-    sseRetryCount
+    // 状态
+    connection,
+    status,
+    retryCount,
+    isConnected: computed(() => status.value === "connected"),
+    isConnecting: computed(() => status.value === "connecting"),
+    // 方法
+    connect,
+    disconnect,
+    reconnect,
+    resetRetryCount
   };
 }
 export {

package/es/client/composables/useServerSSE.d.ts

@@ -0,0 +1,53 @@
+import { ServiceStartupTask } from "@vite-plugin-opencode-assistant/shared";
+/**
+ * Server SSE 状态同步数据
+ */
+export interface ServerSSEStatusSyncData {
+    type: "STATUS_SYNC";
+    isStarted?: boolean;
+    task: ServiceStartupTask;
+    errorType?: string;
+    errorMessage?: string;
+}
+/**
+ * Server SSE 任务更新数据
+ */
+export interface ServerSSETaskUpdateData {
+    type: "TASK_UPDATE";
+    task: ServiceStartupTask;
+    errorType?: string;
+    errorMessage?: string;
+}
+/**
+ * Server SSE 消息类型
+ */
+export type ServerSSEMessage = {
+    type: "CONNECTED";
+} | ServerSSEStatusSyncData | ServerSSETaskUpdateData | {
+    type: "CLEAR_ELEMENTS";
+};
+/**
+ * Server SSE 配置选项
+ */
+export interface ServerSSEOptions {
+    /** 状态同步回调 */
+    onStatusSync?: (data: ServerSSEStatusSyncData) => void;
+    /** 任务更新回调 */
+    onTaskUpdate?: (data: ServerSSETaskUpdateData) => void;
+    /** 清除元素回调 */
+    onClearElements?: () => void;
+    /** 连接成功回调 */
+    onConnected?: () => void;
+    /** 是否启用 */
+    enabled?: boolean;
+}
+/**
+ * 监听 Vite Server SSE 事件
+ * 端点: /__opencode_events__
+ */
+export declare function useServerSSE(options?: ServerSSEOptions): {
+    status: import("vue").Ref<import("./useSSE").SSEConnectionStatus, import("./useSSE").SSEConnectionStatus>;
+    isConnected: import("vue").ComputedRef<boolean>;
+    connect: () => void;
+    disconnect: () => void;
+};

package/es/client/composables/useServerSSE.js

@@ -0,0 +1,36 @@
+import { useSSE } from "./useSSE";
+function useServerSSE(options = {}) {
+  const { onStatusSync, onTaskUpdate, onClearElements, onConnected } = options;
+  const { status, isConnected, connect, disconnect } = useSSE({
+    endpoint: "/__opencode_events__",
+    autoConnect: false,
+    onMessage: (data) => {
+      const message = data;
+      switch (message.type) {
+        case "CONNECTED":
+          onConnected == null ? void 0 : onConnected();
+          break;
+        case "STATUS_SYNC":
+          onStatusSync == null ? void 0 : onStatusSync(message);
+          break;
+        case "TASK_UPDATE":
+          onTaskUpdate == null ? void 0 : onTaskUpdate(message);
+          break;
+        case "CLEAR_ELEMENTS":
+          onClearElements == null ? void 0 : onClearElements();
+          break;
+      }
+    }
+  });
+  return {
+    // 状态
+    status,
+    isConnected,
+    // 方法
+    connect,
+    disconnect
+  };
+}
+export {
+  useServerSSE
+};

package/es/client/composables/useServiceStatus.d.ts

@@ -5,9 +5,7 @@ export declare function useServiceStatus(): {
     chromeMcpFailed: import("vue").Ref<boolean, boolean>;
     chromeMcpErrorType: import("vue").Ref<string | undefined, string | undefined>;
     chromeMcpErrorMessage: import("vue").Ref<string | undefined, string | undefined>;
-    thinking: import("vue").Ref<boolean, boolean>;
     loadingText: import("vue").ComputedRef<string>;
     updateStatusFromTask: (task: ServiceStartupTask | "", errorType?: string, errorMessage?: string) => void;
     setStarting: () => void;
-    setThinking: (value: boolean) => void;
 };

package/es/client/composables/useServiceStatus.js

@@ -6,7 +6,6 @@ function useServiceStatus() {
   const chromeMcpFailed = ref(false);
   const chromeMcpErrorType = ref(void 0);
   const chromeMcpErrorMessage = ref(void 0);
-  const thinking = ref(false);
   const loadingText = computed(() => {
     if (!currentTask.value) return "\u52A0\u8F7D\u4E2D...";
     return SERVICE_STARTUP_TASKS[currentTask.value] || "\u52A0\u8F7D\u4E2D...";
@@ -32,20 +31,15 @@ function useServiceStatus() {
   const setStarting = () => {
     serviceStatus.value = "starting";
   };
-  const setThinking = (value) => {
-    thinking.value = value;
-  };
   return {
     currentTask,
     serviceStatus,
     chromeMcpFailed,
     chromeMcpErrorType,
     chromeMcpErrorMessage,
-    thinking,
     loadingText,
     updateStatusFromTask,
-    setStarting,
-    setThinking
+    setStarting
   };
 }
 export {

package/es/client/composables/useSessions.d.ts

@@ -1,6 +1,18 @@
+import { type Ref } from "vue";
 import type { OpenCodeWidgetSession } from "@vite-plugin-opencode-assistant/shared";
-export declare function useSessions(showNotification: (msg: string) => void): {
-    sessions: import("vue").Ref<{
+export interface UseSessionsOptions {
+    showNotification: (msg: string) => void;
+    /** Session 更新回调 (从 SSE 事件接收) */
+    onSessionUpdate?: Ref<((session: {
+        id: string;
+        title?: string;
+        time?: {
+            updated?: number;
+        };
+    }) => void) | undefined>;
+}
+export declare function useSessions(options: UseSessionsOptions): {
+    sessions: Ref<{
         id: string;
         title?: string | undefined;
         updatedAt?: (string | number | Date) | undefined;
@@ -15,12 +27,19 @@ export declare function useSessions(showNotification: (msg: string) => void): {
         directory?: string | undefined;
         url?: string | undefined;
     }[]>;
-    loadingSessionList: import("vue").Ref<boolean | undefined, boolean | undefined>;
-    currentSessionId: import("vue").Ref<string | null, string | null>;
+    loadingSessionList: Ref<boolean | undefined, boolean | undefined>;
+    currentSessionId: Ref<string | null, string | null>;
     iframeSrc: import("vue").ComputedRef<string>;
-    iframeLoading: import("vue").Ref<boolean, boolean>;
+    iframeLoading: Ref<boolean, boolean>;
     loadSessions: () => Promise<void>;
     createSession: () => Promise<void>;
     deleteSession: (session: OpenCodeWidgetSession) => Promise<void>;
     selectSession: (session: OpenCodeWidgetSession) => void;
+    updateSessionInfo: (sessionUpdate: {
+        id: string;
+        title?: string;
+        time?: {
+            updated?: number;
+        };
+    }) => void;
 };

package/es/client/composables/useSessions.js

@@ -39,7 +39,8 @@ var __async = (__this, __arguments, generator) => {
 };
 import { ref, computed } from "vue";
 import { SESSIONS_API_PATH } from "@vite-plugin-opencode-assistant/shared";
-function useSessions(showNotification) {
+function useSessions(options) {
+  const { showNotification } = options;
   const sessions = ref([]);
   const loadingSessionList = ref(void 0);
   const currentSessionId = ref(null);
@@ -70,6 +71,18 @@ function useSessions(showNotification) {
       loadingSessionList.value = false;
     }
   });
+  const updateSessionInfo = (sessionUpdate) => {
+    var _a;
+    const index = sessions.value.findIndex((s) => s.id === sessionUpdate.id);
+    if (index === -1) return;
+    const session = sessions.value[index];
+    if (sessionUpdate.title && sessionUpdate.title !== session.title) {
+      sessions.value[index] = __spreadProps(__spreadValues({}, session), {
+        title: sessionUpdate.title,
+        updatedAt: ((_a = sessionUpdate.time) == null ? void 0 : _a.updated) || Date.now()
+      });
+    }
+  };
   const createSession = () => __async(null, null, function* () {
     try {
       const response = yield fetch(SESSIONS_API_PATH, { method: "POST" });
@@ -119,7 +132,8 @@ function useSessions(showNotification) {
     loadSessions,
     createSession,
     deleteSession,
-    selectSession
+    selectSession,
+    updateSessionInfo
   };
 }
 export {