pi-context-map 0.4.1 → 0.4.4

package/extensions/index.ts

@@ -194,10 +194,14 @@ export default async function piContextMap(pi: ExtensionAPI): Promise<void> {
 		liveServer.stop();
 	});
 
-	if (serverUrl) {
-		console.log(`[pi-context-map] Live server running at ${serverUrl}`);
-		console.log(
-			`[pi-context-map] Run /context-map to generate a report, or /context-map stop to terminate.`,
-		);
-	}
+	// Kill server when process exits
+	process.on("exit", () => liveServer.stop());
+	process.on("SIGINT", () => {
+		liveServer.stop();
+		process.exit(0);
+	});
+	process.on("SIGTERM", () => {
+		liveServer.stop();
+		process.exit(0);
+	});
 }

package/extensions/live-server.ts

@@ -51,8 +51,9 @@ export class LiveReportServer {
 	 * Start the server. Returns a Promise that resolves to the URL, or null on failure.
 	 */
 	public start(): Promise<string | null> {
+		// Kill any pre-existing server to ensure only one instance runs
 		if (this.server) {
-			return Promise.resolve(this.url);
+			this.stop();
 		}
 
 		return new Promise((resolve) => {
@@ -69,7 +70,6 @@ export class LiveReportServer {
 					const addr = this.server?.address() as AddressInfo | null;
 					if (addr) {
 						this.port = addr.port;
-						console.log(`[pi-context-map] Live server: ${this.url}`);
 						resolve(this.url);
 					} else {
 						resolve(null);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-context-map",
-  "version": "0.4.1",
+  "version": "0.4.4",
   "description": "Professional context profiler for Pi that visualizes the session context window, token distribution, and integrates with Nexus packages for actionable insights.",
   "keywords": [
     "pi-package",