@slkiser/opencode-quota 4.5.0 → 4.6.0

package/dist/tui.js

@@ -23,6 +23,61 @@ const COMPACT_ORDER = 90;
 const REFRESH_INTERVAL_MS = 60_000;
 const EVENT_REFRESH_DELAYS_MS = [150, 600];
 const MOUNT_RECOVERY_DELAYS_MS = [500, 1_500, 4_000];
+const FALLBACK_SURFACE_REGISTRATION = {
+  commandDisplay: "inline",
+  sidebar: {
+    enabled: true
+  },
+  compact: {
+    enabled: false,
+    homeBottom: false,
+    sessionPrompt: false,
+    hasNativeProviderQuota: false,
+    suppressedByNativeProviderQuota: false
+  },
+  announcements: {
+    homeBottom: false
+  },
+  homeBottom: false
+};
+function createTuiInitialLoadCoordinator(seed) {
+  let sessionAvailable = true;
+  let homeAvailable = true;
+  return {
+    takeInitialSession() {
+      if (!sessionAvailable) return undefined;
+      sessionAvailable = false;
+      return seed;
+    },
+    takeInitialHome() {
+      if (!homeAvailable) return undefined;
+      homeAvailable = false;
+      return seed;
+    }
+  };
+}
+function createTuiRegistrationGate() {
+  const [current, setCurrent] = createSignal({
+    status: "pending"
+  });
+  return {
+    current,
+    activate(registration, initialLoads) {
+      if (current().status !== "pending") return;
+      setCurrent({
+        status: "active",
+        registration,
+        initialLoads
+      });
+    },
+    dispose() {
+      if (current().status === "disposed") return;
+      setCurrent({
+        status: "disposed"
+      });
+    }
+  };
+}
 const sessionResources = new WeakMap();
 const homeResources = new WeakMap();
 function getSessionResourceMap(api) {
@@ -32,7 +87,7 @@ function getSessionResourceMap(api) {
   sessionResources.set(api, next);
   return next;
 }
-function createSessionQuotaResource(api, sessionID) {
+function createSessionQuotaResource(api, sessionID, initialLoads) {
   const [sidebar, setSidebar] = createSignal({
     status: "loading",
     lines: []
@@ -40,11 +95,19 @@ function createSessionQuotaResource(api, sessionID) {
   const [compact, setCompact] = createSignal({
     status: "loading"
   });
+  let loadOrdinal = 0;
   const lifecycle = createTuiRefreshLifecycle({
-    load: () => loadTuiSessionQuotaSurfaces({
-      api,
-      sessionID
-    }),
+    load: () => {
+      const initialRuntimeSeed = loadOrdinal === 0 ? initialLoads?.takeInitialSession() : undefined;
+      loadOrdinal += 1;
+      return loadTuiSessionQuotaSurfaces({
+        api,
+        sessionID,
+        ...(initialRuntimeSeed ? {
+          initialRuntimeSeed
+        } : {})
+      });
+    },
     apply: next => {
       setSidebar(next.sidebar);
       setCompact(next.compact);
@@ -87,15 +150,15 @@ function createSessionQuotaResource(api, sessionID) {
   };
   return resource;
 }
-function acquireSessionQuotaResource(api, sessionID) {
+function acquireSessionQuotaResource(api, sessionID, initialLoads) {
   const resources = getSessionResourceMap(api);
   const existing = resources.get(sessionID);
   if (existing) return existing.retain();
-  const next = createSessionQuotaResource(api, sessionID).retain();
+  const next = createSessionQuotaResource(api, sessionID, initialLoads).retain();
   resources.set(sessionID, next);
   return next;
 }
-function createHomeBottomResource(api, compactHomeBottomEnabled) {
+function createHomeBottomResource(api, compactHomeBottomEnabled, initialLoads) {
   const [bottom, setBottom] = createSignal({
     status: "loading",
     compact: compactHomeBottomEnabled ? {
@@ -104,10 +167,18 @@ function createHomeBottomResource(api, compactHomeBottomEnabled) {
       status: "disabled"
     }
   });
+  let loadOrdinal = 0;
   const lifecycle = createTuiRefreshLifecycle({
-    load: () => loadTuiHomeBottomStatus({
-      api
-    }),
+    load: () => {
+      const initialRuntimeSeed = loadOrdinal === 0 ? initialLoads?.takeInitialHome() : undefined;
+      loadOrdinal += 1;
+      return loadTuiHomeBottomStatus({
+        api,
+        ...(initialRuntimeSeed ? {
+          initialRuntimeSeed
+        } : {})
+      });
+    },
     apply: setBottom,
     afterApply: () => {
       // Fire-and-forget: write export file if enabled. A failed write must
@@ -135,21 +206,21 @@ function createHomeBottomResource(api, compactHomeBottomEnabled) {
   };
   return resource;
 }
-function acquireHomeBottomResource(api, compactHomeBottomEnabled) {
+function acquireHomeBottomResource(api, compactHomeBottomEnabled, initialLoads) {
   const existing = homeResources.get(api);
   if (existing) return existing.retain();
-  const next = createHomeBottomResource(api, compactHomeBottomEnabled).retain();
+  const next = createHomeBottomResource(api, compactHomeBottomEnabled, initialLoads).retain();
   homeResources.set(api, next);
   return next;
 }
-function useSessionQuotaResource(api, sessionID) {
-  let current = acquireSessionQuotaResource(api, sessionID());
+function useSessionQuotaResource(api, sessionID, initialLoads) {
+  let current = acquireSessionQuotaResource(api, sessionID(), initialLoads);
   const [resource, setResource] = createSignal(current);
   createEffect(() => {
     const nextSessionID = sessionID();
     if (current.sessionID === nextSessionID) return;
     const previous = current;
-    current = acquireSessionQuotaResource(api, nextSessionID);
+    current = acquireSessionQuotaResource(api, nextSessionID, initialLoads);
     setResource(current);
     previous.release();
   });
@@ -159,7 +230,7 @@ function useSessionQuotaResource(api, sessionID) {
   return resource;
 }
 function SidebarContentView(props) {
-  const resource = useSessionQuotaResource(props.api, () => props.sessionID);
+  const resource = useSessionQuotaResource(props.api, () => props.sessionID, props.initialLoads);
   const panel = () => resource().sidebar();
   const lines = () => getSidebarPanelLines(panel());
   const hasDetailLines = () => Boolean(panel().linesExpanded?.length);
@@ -276,7 +347,7 @@ function CompactStatusLine(props) {
   });
 }
 function SessionPromptWithCompactStatus(props) {
-  const resource = useSessionQuotaResource(props.api, () => props.sessionID);
+  const resource = useSessionQuotaResource(props.api, () => props.sessionID, props.initialLoads);
   const panel = () => resource().compact();
   return (() => {
     var _el$13 = _$createElement("box");
@@ -310,7 +381,7 @@ function SessionPromptWithCompactStatus(props) {
   })();
 }
 function HomeBottomView(props) {
-  const resource = acquireHomeBottomResource(props.api, props.compactHomeBottomEnabled);
+  const resource = acquireHomeBottomResource(props.api, props.compactHomeBottomEnabled, props.initialLoads);
   onCleanup(() => resource.release());
   const announcement = () => getHomeBottomAnnouncementText(resource.bottom());
   const compact = () => resource.bottom().compact;
@@ -629,7 +700,7 @@ async function runQuotaDialogCommandAsync(api, command, commandDisplay, rawInput
     });
   }
 }
-function registerQuotaDialogCommands(api, commandDisplay) {
+function registerQuotaDialogCommands(api, gate) {
   const commandState = {};
   const dispose = api.keymap.registerLayer({
     commands: QUOTA_DIALOG_COMMANDS.map(spec => ({
@@ -640,81 +711,111 @@ function registerQuotaDialogCommands(api, commandDisplay) {
       category: "OpenCode Quota",
       slashName: spec.slashName,
       run(input) {
-        void runQuotaDialogCommandAsync(api, spec.id, commandDisplay, input, commandState);
+        const state = gate.current();
+        if (state.status !== "active") return;
+        void runQuotaDialogCommandAsync(api, spec.id, state.registration.commandDisplay, input, commandState);
       }
     })),
     bindings: []
   });
   api.lifecycle.onDispose(dispose);
 }
-function registerSidebarSlots(api) {
+function registerStableTuiSlots(api, current) {
   api.slots.register({
     order: SIDEBAR_ORDER,
     slots: {
       sidebar_content(_ctx, props) {
+        const state = current();
+        if (state.status !== "active" || !state.registration.sidebar.enabled) return null;
         return _$createComponent(SidebarContentView, {
           api: api,
           get sessionID() {
             return props.session_id;
+          },
+          get initialLoads() {
+            return state.initialLoads;
           }
         });
       }
     }
   });
-}
-const tui = async api => {
-  api.lifecycle.onDispose(() => {
-    disposeQuotaTelemetryOwner(createTuiQuotaClient(api));
+  api.slots.register({
+    order: COMPACT_ORDER,
+    slots: {
+      session_prompt(_ctx, props) {
+        const state = current();
+        if (state.status !== "active" || !state.registration.compact.sessionPrompt) return null;
+        return _$createComponent(SessionPromptWithCompactStatus, {
+          api: api,
+          get sessionID() {
+            return props.session_id;
+          },
+          get initialLoads() {
+            return state.initialLoads;
+          },
+          get visible() {
+            return props.visible;
+          },
+          get disabled() {
+            return props.disabled;
+          },
+          get onSubmit() {
+            return props.on_submit;
+          },
+          get promptRef() {
+            return props.ref;
+          }
+        });
+      },
+      home_bottom() {
+        const state = current();
+        if (state.status !== "active" || !state.registration.homeBottom) return null;
+        return _$createComponent(HomeBottomView, {
+          api: api,
+          get compactHomeBottomEnabled() {
+            return state.registration.compact.homeBottom;
+          },
+          get initialLoads() {
+            return state.initialLoads;
+          }
+        });
+      }
+    }
   });
+}
+async function initializeTuiRegistration(api, gate) {
+  let initialRuntimeSeed;
   let surfaceRegistration;
   try {
-    surfaceRegistration = await resolveTuiSurfaceRegistration(api);
-  } catch {
-    registerQuotaDialogCommands(api, "inline");
-    registerSidebarSlots(api);
-    return;
-  }
-  registerQuotaDialogCommands(api, surfaceRegistration.commandDisplay);
-  if (surfaceRegistration.sidebar.enabled) {
-    registerSidebarSlots(api);
-  }
-  const compactRegistration = surfaceRegistration.compact;
-  if (!compactRegistration.enabled && !surfaceRegistration.homeBottom) return;
-  const compactSlots = {};
-  if (compactRegistration.sessionPrompt) {
-    compactSlots.session_prompt = (_ctx, props) => _$createComponent(SessionPromptWithCompactStatus, {
-      api: api,
-      get sessionID() {
-        return props.session_id;
-      },
-      get visible() {
-        return props.visible;
-      },
-      get disabled() {
-        return props.disabled;
-      },
-      get onSubmit() {
-        return props.on_submit;
-      },
-      get promptRef() {
-        return props.ref;
+    surfaceRegistration = resolveTuiSurfaceRegistration(api, {
+      captureInitialRuntime(seed) {
+        initialRuntimeSeed = seed;
       }
+    }).then(registration => ({
+      registration,
+      initialRuntimeSeed
+    })).catch(() => ({
+      registration: FALLBACK_SURFACE_REGISTRATION
+    }));
+  } catch {
+    surfaceRegistration = Promise.resolve({
+      registration: FALLBACK_SURFACE_REGISTRATION
     });
   }
-  if (surfaceRegistration.homeBottom) {
-    compactSlots.home_bottom = () => _$createComponent(HomeBottomView, {
-      api: api,
-      get compactHomeBottomEnabled() {
-        return surfaceRegistration.compact.homeBottom;
-      }
-    });
-  }
-  if (Object.keys(compactSlots).length > 0) {
-    api.slots.register({
-      order: COMPACT_ORDER,
-      slots: compactSlots
-    });
-  }
+  registerQuotaDialogCommands(api, gate);
+  void surfaceRegistration.then(({
+    registration,
+    initialRuntimeSeed
+  }) => gate.activate(registration, initialRuntimeSeed ? createTuiInitialLoadCoordinator(initialRuntimeSeed) : undefined));
+  registerStableTuiSlots(api, gate.current);
+}
+const tui = async api => {
+  const registrationGate = createTuiRegistrationGate();
+  api.lifecycle.onDispose(() => {
+    registrationGate.dispose();
+    disposeQuotaTelemetryOwner(createTuiQuotaClient(api));
+  });
+  void initializeTuiRegistration(api, registrationGate).catch(() => {});
 };
 const pluginModule = {
   id,