saeeol 1.2.2 → 1.2.3

package/src/cli/cmd/tui/context/app/helper.tsx

@@ -0,0 +1,25 @@
+import { createContext, Show, useContext, type ParentProps } from "solid-js"
+
+export function createSimpleContext<T, Props extends Record<string, any>>(input: {
+  name: string
+  init: ((input: Props) => T) | (() => T)
+}) {
+  const ctx = createContext<T>()
+
+  return {
+    provider: (props: ParentProps<Props>) => {
+      const init = input.init(props)
+      return (
+        // @ts-expect-error
+        <Show when={init.ready === undefined || init.ready === true}>
+          <ctx.Provider value={init}>{props.children}</ctx.Provider>
+        </Show>
+      )
+    },
+    use() {
+      const value = useContext(ctx)
+      if (!value) throw new Error(`${input.name} context must be used within a context provider`)
+      return value
+    },
+  }
+}

package/src/cli/cmd/tui/context/app/project.tsx

@@ -0,0 +1,109 @@
+import { batch } from "solid-js"
+import type { Path, Workspace } from "@saeeol/sdk/v2"
+import { createStore, reconcile } from "solid-js/store"
+import { createSimpleContext } from "./helper"
+import { useSDK } from "./sdk"
+
+type WorkspaceStatus = "connected" | "connecting" | "disconnected" | "error"
+
+export const { use: useProject, provider: ProjectProvider } = createSimpleContext({
+  name: "Project",
+  init: () => {
+    const sdk = useSDK()
+
+    const defaultPath = {
+      home: "",
+      state: "",
+      config: "",
+      worktree: "",
+      directory: sdk.directory ?? "",
+    } satisfies Path
+
+    const [store, setStore] = createStore({
+      project: {
+        id: undefined as string | undefined,
+      },
+      instance: {
+        path: defaultPath,
+      },
+      workspace: {
+        current: undefined as string | undefined,
+        list: [] as Workspace[],
+        status: {} as Record<string, WorkspaceStatus>,
+      },
+    })
+
+    async function sync() {
+      const workspace = store.workspace.current
+      const [path, project] = await Promise.all([
+        sdk.client.path.get({ workspace }),
+        sdk.client.project.current({ workspace }),
+      ])
+
+      batch(() => {
+        setStore("instance", "path", reconcile(path.data || defaultPath))
+        setStore("project", "id", project.data?.id)
+      })
+    }
+
+    async function syncWorkspace() {
+      const listed = await sdk.client.experimental.workspace.list().catch(() => undefined)
+      if (!listed?.data) return
+      const status = await sdk.client.experimental.workspace.status().catch(() => undefined)
+      const next = Object.fromEntries((status?.data ?? []).map((item) => [item.workspaceID, item.status]))
+
+      batch(() => {
+        setStore("workspace", "list", reconcile(listed.data))
+        setStore("workspace", "status", reconcile(next))
+        if (!listed.data.some((item) => item.id === store.workspace.current)) {
+          setStore("workspace", "current", undefined)
+        }
+      })
+    }
+
+    sdk.event.on("event", (event) => {
+      if (event.payload.type === "workspace.status") {
+        setStore("workspace", "status", event.payload.properties.workspaceID, event.payload.properties.status)
+      }
+    })
+
+    return {
+      data: store,
+      project() {
+        return store.project.id
+      },
+      instance: {
+        path() {
+          return store.instance.path
+        },
+        directory() {
+          return store.instance.path.directory
+        },
+      },
+      workspace: {
+        current() {
+          return store.workspace.current
+        },
+        set(next?: string | null) {
+          const workspace = next ?? undefined
+          if (store.workspace.current === workspace) return
+          setStore("workspace", "current", workspace)
+        },
+        list() {
+          return store.workspace.list
+        },
+        get(workspaceID: string) {
+          return store.workspace.list.find((item) => item.id === workspaceID)
+        },
+        status(workspaceID: string) {
+          return store.workspace.status[workspaceID]
+        },
+        statuses() {
+          return store.workspace.status
+        },
+        sync: syncWorkspace,
+      },
+      sync,
+    }
+  },
+})

package/src/cli/cmd/tui/context/app/route.tsx

@@ -0,0 +1,67 @@
+import { createStore, reconcile, unwrap } from "solid-js/store" 
+import { createSimpleContext } from "./helper"
+import type { PromptInfo } from "../../component/prompt/history"
+
+export type HomeRoute = {
+  type: "home"
+  prompt?: PromptInfo
+}
+
+export type SessionRoute = {
+  type: "session"
+  sessionID: string
+  prompt?: PromptInfo
+}
+export type SaeeolClawRoute = {
+  type: "saeeolclaw"
+}
+
+export type LocalModelsRoute = {
+  type: "local-models"
+}
+
+export type PluginRoute = {
+  type: "plugin"
+  id: string
+  data?: Record<string, unknown>
+}
+
+export type Route = HomeRoute | SessionRoute | PluginRoute | SaeeolClawRoute | LocalModelsRoute 
+
+export const { use: useRoute, provider: RouteProvider } = createSimpleContext({
+  name: "Route",
+  init: (props: { initialRoute?: Route }) => {
+    const [store, setStore] = createStore<Route>(
+      props.initialRoute ??
+        (process.env["SAEEOL_ROUTE"]
+          ? JSON.parse(process.env["SAEEOL_ROUTE"])
+          : {
+              type: "home",
+            }),
+    )
+    let previous: Route | undefined
+
+    return {
+      get data() {
+        return store
+      },
+      navigate(route: Route) {
+        previous = structuredClone(unwrap(store)) 
+        setStore(reconcile(route))
+      },
+      back() {
+        const target = previous ?? ({ type: "home" } as const)
+        previous = undefined
+        console.log("navigate", target)
+        setStore(target)
+      },
+    }
+  },
+})
+
+export type RouteContext = ReturnType<typeof useRoute>
+
+export function useRouteData<T extends Route["type"]>(type: T) {
+  const route = useRoute()
+  return route.data as Extract<Route, { type: typeof type }>
+}

package/src/cli/cmd/tui/context/app/sdk.tsx

@@ -0,0 +1,142 @@
+import { createSaeeolClient } from "@saeeol/sdk/v2"
+import type { GlobalEvent } from "@saeeol/sdk/v2"
+import { createSimpleContext } from "./helper"
+import { createGlobalEmitter } from "@solid-primitives/event-bus"
+import { Flag } from "@saeeol/core/flag/flag"
+import { batch, onCleanup, onMount } from "solid-js"
+
+export type EventSource = {
+  subscribe: (handler: (event: GlobalEvent) => void) => Promise<() => void>
+}
+
+export const { use: useSDK, provider: SDKProvider } = createSimpleContext({
+  name: "SDK",
+  init: (props: {
+    url: string
+    directory?: string
+    fetch?: typeof fetch
+    headers?: RequestInit["headers"]
+    events?: EventSource
+  }) => {
+    const abort = new AbortController()
+    let sse: AbortController | undefined
+
+    function createSDK() {
+      return createSaeeolClient({
+        baseUrl: props.url,
+        signal: abort.signal,
+        directory: props.directory,
+        fetch: props.fetch,
+        headers: props.headers,
+      })
+    }
+
+    let sdk = createSDK()
+
+    const emitter = createGlobalEmitter<{
+      event: GlobalEvent
+    }>()
+
+    let queue: GlobalEvent[] = []
+    let timer: Timer | undefined
+    let last = 0
+    const retryDelay = 1000
+    const maxRetryDelay = 30000
+
+    const flush = () => {
+      if (queue.length === 0) return
+      const events = queue
+      queue = []
+      timer = undefined
+      last = Date.now()
+      // Batch all event emissions so all store updates result in a single render
+      batch(() => {
+        for (const event of events) {
+          emitter.emit("event", event)
+        }
+      })
+    }
+
+    const handleEvent = (event: GlobalEvent) => {
+      queue.push(event)
+      const elapsed = Date.now() - last
+
+      if (timer) return
+      // If we just flushed recently (within 16ms), batch this with future events
+      // Otherwise, process immediately to avoid latency
+      if (elapsed < 16) {
+        timer = setTimeout(flush, 16)
+        return
+      }
+      flush()
+    }
+
+    function startSSE() {
+      sse?.abort()
+      const ctrl = new AbortController()
+      sse = ctrl
+      ;(async () => {
+        let attempt = 0
+        while (true) {
+          if (abort.signal.aborted || ctrl.signal.aborted) break
+
+          const events = await sdk.global.event({
+            signal: ctrl.signal,
+            sseMaxRetryAttempts: 0,
+          })
+
+          if (Flag.SAEEOL_EXPERIMENTAL_WORKSPACES) {
+            // Start syncing workspaces, it's important to do this after
+            // we've started listening to events
+            await sdk.sync.start().catch(() => {})
+          }
+
+          for await (const event of events.stream) {
+            if (ctrl.signal.aborted) break
+            handleEvent(event)
+          }
+
+          if (timer) clearTimeout(timer)
+          if (queue.length > 0) flush()
+          attempt += 1
+          if (abort.signal.aborted || ctrl.signal.aborted) break
+
+          // Exponential backoff
+          const backoff = Math.min(retryDelay * 2 ** (attempt - 1), maxRetryDelay)
+          await new Promise((resolve) => setTimeout(resolve, backoff))
+        }
+      })().catch(() => {})
+    }
+
+    onMount(async () => {
+      if (props.events) {
+        const unsub = await props.events.subscribe(handleEvent)
+        onCleanup(unsub)
+
+        if (Flag.SAEEOL_EXPERIMENTAL_WORKSPACES) {
+          // Start syncing workspaces, it's important to do this after
+          // we've started listening to events
+          await sdk.sync.start().catch(() => {})
+        }
+      } else {
+        startSSE()
+      }
+    })
+
+    onCleanup(() => {
+      abort.abort()
+      sse?.abort()
+      if (timer) clearTimeout(timer)
+    })
+
+    return {
+      get client() {
+        return sdk
+      },
+      directory: props.directory,
+      event: emitter,
+      fetch: props.fetch ?? fetch,
+      url: props.url,
+    }
+  },
+})