@pi-unipi/info-screen 0.1.7 → 0.1.9

package/core-groups.ts

@@ -322,8 +322,10 @@ export function setPiApi(api: any): void {
  * Add a module to the announced list.
  */
 export function trackModule(name: string, version: string): void {
+
   if (!announcedModules.find((m) => m.name === name)) {
     announcedModules.push({ name, version });
+  
   }
 }
 
@@ -331,6 +333,7 @@ export function trackModule(name: string, version: string): void {
  * Get list of announced modules.
  */
 export function getAnnouncedModules(): Array<{ name: string; version: string }> {
+
   return [...announcedModules];
 }
 
@@ -375,16 +378,32 @@ export function registerCoreGroups(): void {
       const homeDir = process.env.HOME || process.env.USERPROFILE || homedir();
       const shortCwd = cwd.startsWith(homeDir) ? `~${cwd.slice(homeDir.length)}` : cwd;
 
-      const modules = getAnnouncedModules();
-      const moduleNames = modules.map((m) => m.name.replace(/^@[^/]+\//, ""));
+      // Get modules from announced events AND registered groups
+      const announced = getAnnouncedModules();
+      const registeredGroups = infoRegistry.getAllGroups();
+      
+      // Combine: announced modules + groups that aren't from announced modules
+      const moduleNames = new Set<string>();
+      for (const m of announced) {
+        moduleNames.add(m.name.replace(/^@[^/]+\//, ""));
+      }
+      // Add non-core groups as modules (they come from extensions)
+      const coreGroupIds = new Set(["overview", "usage", "tools", "extensions", "skills"]);
+      for (const g of registeredGroups) {
+        if (!coreGroupIds.has(g.id)) {
+          moduleNames.add(g.id);
+        }
+      }
+      
       const totalLoadTime = getTotalLoadTime();
+      const moduleList = Array.from(moduleNames);
 
       return {
         version: { value: getPiVersion(), detail: "pi" },
         cwd: { value: shortCwd },
         modules: {
-          value: String(modules.length),
-          detail: moduleNames.slice(0, 4).join(", ") + (moduleNames.length > 4 ? ` +${moduleNames.length - 4} more` : ""),
+          value: String(moduleList.length),
+          detail: moduleList.slice(0, 4).join(", ") + (moduleList.length > 4 ? ` +${moduleList.length - 4} more` : ""),
         },
         uptime: { value: formatUptime(process.uptime()) },
         loadTime: { value: `${totalLoadTime}ms` },

package/index.ts

@@ -64,6 +64,9 @@ export default function (pi: ExtensionAPI) {
       trackModule(event.name, event.version || "unknown");
       recordLoadTime(event.name, "module", event.loadTimeMs);
 
+      // Invalidate overview cache so next render shows updated modules
+      infoRegistry.invalidateCache("overview");
+
       // Track tools from this module
       if (event.tools && Array.isArray(event.tools)) {
         for (const tool of event.tools) {
@@ -96,11 +99,11 @@ export default function (pi: ExtensionAPI) {
   });
 
   // Session lifecycle
-  pi.on("session_start", async (_event, ctx) => {
+  pi.on("session_start", async (event, ctx) => {
     const settings = getInfoSettings();
 
-    // Show dashboard on boot if enabled
-    if (settings.showOnBoot) {
+    // Show dashboard only on initial startup, not on /new
+    if (settings.showOnBoot && event.reason === "startup") {
       // Wait for other modules to announce
       await waitForModules();
 
@@ -109,6 +112,7 @@ export default function (pi: ExtensionAPI) {
         (tui, _theme, _keybindings, done) => {
           const overlay = new InfoOverlay();
           overlay.onClose = () => done(undefined);
+          overlay.requestRender = () => tui.requestRender();
           // Return three-method object as per pi-tui docs
           return {
             render: (w: number) => overlay.render(w),
@@ -151,6 +155,7 @@ export default function (pi: ExtensionAPI) {
         (tui, _theme, _keybindings, done) => {
           const overlay = new InfoOverlay();
           overlay.onClose = () => done(undefined);
+          overlay.requestRender = () => tui.requestRender();
           return {
             render: (w: number) => overlay.render(w),
             invalidate: () => overlay.invalidate(),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi-unipi/info-screen",
-  "version": "0.1.7",
+  "version": "0.1.9",
   "description": "Dashboard and module registry for Unipi — configurable info overlay with tabbed groups",
   "type": "module",
   "license": "MIT",

package/tui/info-overlay.ts

@@ -54,6 +54,7 @@ export class InfoOverlay implements Component {
   onClose?: () => void;
 
   constructor() {
+    // Start loading data immediately
     this.loadData();
   }
 
@@ -69,6 +70,8 @@ export class InfoOverlay implements Component {
    */
   private async loadData(): Promise<void> {
     this.loading = true;
+    // Wait a bit for modules to announce before fetching groups
+    await new Promise(r => setTimeout(r, 500));
     // Always re-fetch ALL groups to catch late registrations
     this.groups = infoRegistry.getAllGroups();
     
@@ -84,13 +87,21 @@ export class InfoOverlay implements Component {
     }
 
     try {
-      // Load data for all groups in parallel
-      const promises = this.groups.map(async (group) => {
-        const data = await infoRegistry.getGroupData(group.id);
-        this.groupData.set(group.id, data);
+      // Load data for all groups in parallel with timeout
+      const loadPromises = this.groups.map(async (group) => {
+        try {
+          const data = await Promise.race([
+            infoRegistry.getGroupData(group.id),
+            new Promise<never>((_, reject) => setTimeout(() => reject(new Error('timeout')), 3000)),
+          ]);
+          this.groupData.set(group.id, data);
+        } catch {
+          // Use empty data on timeout/error
+          this.groupData.set(group.id, {});
+        }
       });
 
-      await Promise.all(promises);
+      await Promise.all(loadPromises);
     } catch (error) {
       this.error = error instanceof Error ? error.message : String(error);
     }
@@ -145,6 +156,7 @@ export class InfoOverlay implements Component {
    * Render the component.
    */
   render(width: number): string[] {
+    // While loading, show loading state
     if (this.loading) {
       return this.renderLoading(width);
     }
@@ -198,6 +210,11 @@ export class InfoOverlay implements Component {
     }
   }
 
+  /**
+   * Callback set by wrapper to trigger re-render.
+   */
+  requestRender?: () => void;
+
   /**
    * Refresh data for all groups (non-blocking).
    */
@@ -207,7 +224,14 @@ export class InfoOverlay implements Component {
       infoRegistry.invalidateCache(group.id);
       // Fetch fresh data (non-blocking)
       infoRegistry.getGroupData(group.id).then(data => {
+        const old = this.groupData.get(group.id);
+        const oldStr = JSON.stringify(old);
+        const newStr = JSON.stringify(data);
         this.groupData.set(group.id, data);
+        // Trigger re-render if data changed
+        if (oldStr !== newStr && this.requestRender) {
+          this.requestRender();
+        }
       }).catch(() => {
         // Ignore errors
       });