@pi-unipi/info-screen 0.1.3 → 0.1.4

package/core-groups.ts

@@ -180,6 +180,7 @@ const loadTimes: Array<{ name: string; type: string; ms: number }> = [];
 let totalLoadTimeMs = 0;
 let loadTrackingStarted = false;
 let loadTrackingStartMs = 0;
+const moduleStartTimes = new Map<string, number>();
 
 /** Start load time tracking */
 export function startLoadTracking(): void {
@@ -189,10 +190,28 @@ export function startLoadTracking(): void {
   }
 }
 
+/** Record when a module starts loading */
+export function recordModuleStart(name: string): void {
+  moduleStartTimes.set(name, Date.now());
+}
+
 /** Record a load time */
-export function recordLoadTime(name: string, type: string, ms: number): void {
-  loadTimes.push({ name, type, ms });
-  totalLoadTimeMs += ms;
+export function recordLoadTime(name: string, type: string, ms?: number): void {
+  // If no ms provided, calculate from start time
+  if (ms === undefined || ms === 0) {
+    const startTime = moduleStartTimes.get(name);
+    if (startTime) {
+      ms = Date.now() - startTime;
+    } else {
+      ms = 0;
+    }
+  }
+  // Avoid duplicates
+  const existing = loadTimes.find(t => t.name === name && t.type === type);
+  if (!existing) {
+    loadTimes.push({ name, type, ms });
+    totalLoadTimeMs += ms;
+  }
 }
 
 /** Finish load tracking */
@@ -209,7 +228,7 @@ export function getLoadTimes(): Array<{ name: string; type: string; ms: number }
 
 /** Get total load time */
 export function getTotalLoadTime(): number {
-  return totalLoadTimeMs;
+  return totalLoadTimeMs > 0 ? totalLoadTimeMs : (loadTrackingStarted ? Date.now() - loadTrackingStartMs : 0);
 }
 
 /**
@@ -510,16 +529,28 @@ export function registerCoreGroups(): void {
         return source === "sdk";
       });
 
-      // Build tool list as comma-separated values
+      // Build tool list as comma-separated values with wrapping
       const toolNames = tools.map((t) => `${t.name}`);
-      const toolListStr = toolNames.join(", ");
+      // Split into chunks of ~60 chars for wrapping
+      const chunks: string[] = [];
+      let current = "";
+      for (const name of toolNames) {
+        if (current && (current.length + name.length + 2) > 60) {
+          chunks.push(current);
+          current = name;
+        } else {
+          current = current ? `${current}, ${name}` : name;
+        }
+      }
+      if (current) chunks.push(current);
 
       return {
         total: { value: String(tools.length) },
         builtin: { value: String(builtin.length) },
         registered: { value: String(extension.length + sdk.length) },
         list: {
-          value: toolListStr.length > 0 ? toolListStr : "none",
+          value: chunks.length > 0 ? chunks[0] : "none",
+          detail: chunks.length > 1 ? chunks.slice(1).join("\n") : undefined,
         },
       };
     },
@@ -585,16 +616,28 @@ export function registerCoreGroups(): void {
       const global = skills.filter((s) => s.source === "global");
       const project = skills.filter((s) => s.source === "project");
 
-      // Build skill list as comma-separated values
+      // Build skill list as comma-separated values with wrapping
       const skillNames = skills.map((s) => `${s.name} (${s.source})`);
-      const skillListStr = skillNames.join(", ");
+      // Split into chunks of ~60 chars for wrapping
+      const chunks: string[] = [];
+      let current = "";
+      for (const name of skillNames) {
+        if (current && (current.length + name.length + 2) > 60) {
+          chunks.push(current);
+          current = name;
+        } else {
+          current = current ? `${current}, ${name}` : name;
+        }
+      }
+      if (current) chunks.push(current);
 
       return {
         count: { value: String(skills.length) },
         global: { value: String(global.length) },
         project: { value: String(project.length) },
         list: {
-          value: skillListStr.length > 0 ? skillListStr : "none",
+          value: chunks.length > 0 ? chunks[0] : "none",
+          detail: chunks.length > 1 ? chunks.slice(1).join("\n") : undefined,
         },
       };
     },

package/index.ts

@@ -12,10 +12,10 @@
 import type { ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
 import { UNIPI_EVENTS, MODULES, UNIPI_PREFIX, emitEvent, getPackageVersion } from "@pi-unipi/core";
 import { infoRegistry } from "./registry.js";
-import { registerCoreGroups, trackModule, trackTool, setPiApi, registerSkillDir, startLoadTracking, recordLoadTime, finishLoadTracking } from "./core-groups.js";
+import { registerCoreGroups, trackModule, trackTool, setPiApi, registerSkillDir, startLoadTracking, recordLoadTime, finishLoadTracking, recordModuleStart } from "./core-groups.js";
 
 /** Re-export infoRegistry, registerSkillDir, and load tracking for external use */
-export { infoRegistry, registerSkillDir, startLoadTracking, recordLoadTime, finishLoadTracking };
+export { infoRegistry, registerSkillDir, startLoadTracking, recordLoadTime, finishLoadTracking, recordModuleStart };
 import { getInfoSettings } from "./config.js";
 import { InfoOverlay } from "./tui/info-overlay.js";
 import { SettingsOverlay } from "./settings/settings-tui.js";
@@ -62,7 +62,7 @@ export default function (pi: ExtensionAPI) {
     if (event.name && event.name !== MODULES.INFO_SCREEN) {
       // Track the module
       trackModule(event.name, event.version || "unknown");
-      recordLoadTime(event.name, "module", event.loadTimeMs || 0);
+      recordLoadTime(event.name, "module", event.loadTimeMs);
 
       // Track tools from this module
       if (event.tools && Array.isArray(event.tools)) {

package/package.json

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

package/tui/info-overlay.ts

@@ -153,12 +153,12 @@ export class InfoOverlay implements Component {
       return this.renderError(width);
     }
 
-    // Check for new groups (but don't re-trigger loading)
+    // Check for new groups and re-fetch data
     const allGroups = infoRegistry.getAllGroups();
     const groupIds = allGroups.map(g => g.id).join(",");
     const currentIds = this.groups.map(g => g.id).join(",");
     
-    if (groupIds !== currentIds) {
+    if (groupIds !== currentIds || this.groups.length !== allGroups.length) {
       this.groups = allGroups;
       // Apply saved order
       const settings = getInfoSettings();
@@ -170,10 +170,11 @@ export class InfoOverlay implements Component {
           return (ai === -1 ? 999 : ai) - (bi === -1 ? 999 : bi);
         });
       }
-      // Load data for any new groups (non-blocking)
-      this.loadDataForNewGroups(allGroups);
     }
 
+    // Always re-fetch data for all groups to catch late updates
+    this.refreshAllData();
+
     if (this.groups.length === 0) {
       return this.renderEmpty(width);
     }
@@ -197,6 +198,22 @@ export class InfoOverlay implements Component {
     }
   }
 
+  /**
+   * Refresh data for all groups (non-blocking).
+   */
+  private refreshAllData(): void {
+    for (const group of this.groups) {
+      // Invalidate cache to get fresh data
+      infoRegistry.invalidateCache(group.id);
+      // Fetch fresh data (non-blocking)
+      infoRegistry.getGroupData(group.id).then(data => {
+        this.groupData.set(group.id, data);
+      }).catch(() => {
+        // Ignore errors
+      });
+    }
+  }
+
   /**
    * Render loading state.
    */