@corva/ui 3.78.0-18 → 3.78.0-19

package/mcp-server/server.mjs

@@ -246,9 +246,9 @@ const validateDsn = dsn => {
   return url.href;
 };
 
-const MCP_SERVER_VERSION = '1.9.0';
+const MCP_SERVER_VERSION = '1.9.1';
 
-var version = "3.78.0-18";
+var version = "3.78.0-19";
 
 const CORVA_UI_VERSION = version;
 
@@ -1010,6 +1010,77 @@ const resolveProjectIdentity = async (server, logger) => {
   }
 };
 
+const createToolResponse = text => ({
+  content: [{
+    type: 'text',
+    text
+  }]
+});
+
+const createErrorResponse = message => ({
+  content: [{
+    type: 'text',
+    text: message
+  }],
+  isError: true
+});
+
+const withTimeout = async (promise, ms) => {
+  let timeoutId;
+  const timeout = new Promise(resolve => {
+    timeoutId = setTimeout(() => resolve(undefined), ms);
+  });
+  try {
+    return await Promise.race([promise, timeout]);
+  } finally {
+    clearTimeout(timeoutId);
+  }
+};
+
+/**
+ * A field of a listing is ONE line: every whitespace run becomes a space, the ends are trimmed,
+ * every other character survives. Not the first paragraph (drops words) and not indentation
+ * (CommonMark honours up to three spaces, so a fence, heading or blank line inside a description
+ * would still open a block of its own).
+ */
+const singleLine = text => text.replace(/\s+/g, ' ').trim();
+
+// The SDK gives roots/list 60 s, and a session that ends inside the wait would never be counted.
+const IDENTITY_FLUSH_TIMEOUT_MS = 2000;
+/**
+ * Records the session counter and the initialize span exactly once, with the project identity when
+ * it resolves within IDENTITY_FLUSH_TIMEOUT_MS. The deadline starts at the call, so call it only
+ * once telemetry has settled; an identity that resolves later is ignored here.
+ */
+const recordSessionStart = async ({
+  telemetry,
+  initData,
+  identity
+}) => {
+  if (!telemetry.isEnabled()) {
+    return;
+  }
+  const projectIdentity = await withTimeout(identity.catch(() => null), IDENTITY_FLUSH_TIMEOUT_MS);
+  const {
+    timestamp,
+    clientName,
+    clientVersion,
+    clientCapabilities
+  } = initData;
+  telemetry.metrics.recordSession(clientName, clientVersion, projectIdentity);
+  telemetry.recordInitialize({
+    timestamp,
+    clientName,
+    clientVersion,
+    clientCapabilities,
+    sessionId: telemetry.sessionId,
+    protocolVersion: MCP_PROTOCOL_VERSION,
+    networkTransport: MCP_NETWORK_TRANSPORT,
+    appKey: projectIdentity?.appKey ?? undefined,
+    packageName: projectIdentity?.packageName ?? undefined
+  });
+};
+
 // This file is auto-generated by `yarn mcp:generate-prompts`. Do not edit by hand.
 // Source: ./figma-to-code.prompt.md
 
@@ -21548,41 +21619,6 @@ const formatZIndex = zIndex => {
   return table(['Name', 'CSS Variable', 'Value'], rows);
 };
 
-const createToolResponse = text => ({
-  content: [{
-    type: 'text',
-    text
-  }]
-});
-
-const createErrorResponse = message => ({
-  content: [{
-    type: 'text',
-    text: message
-  }],
-  isError: true
-});
-
-const withTimeout = async (promise, ms) => {
-  let timeoutId;
-  const timeout = new Promise(resolve => {
-    timeoutId = setTimeout(() => resolve(undefined), ms);
-  });
-  try {
-    return await Promise.race([promise, timeout]);
-  } finally {
-    clearTimeout(timeoutId);
-  }
-};
-
-/**
- * A field of a listing is ONE line: every whitespace run becomes a space, the ends are trimmed,
- * every other character survives. Not the first paragraph (drops words) and not indentation
- * (CommonMark honours up to three spaces, so a fence, heading or blank line inside a description
- * would still open a block of its own).
- */
-const singleLine = text => text.replace(/\s+/g, ' ').trim();
-
 const parseQuery = queryLower => {
   const rawWords = queryLower.trim().split(/\s+/);
   const queryWords = rawWords.filter(w => w.length > 1);
@@ -23385,9 +23421,9 @@ class CorvaUiMcpServer {
   telemetryDisabledReason = null;
   telemetryInitPromise = null;
   shutdownRequested = false;
-  pendingInitData = null;
   projectIdentity = null;
   identityPromise = null;
+  sessionStartPromise = null;
   constructor() {
     const baseLogger = createLogger({
       prefix: '[CorvaUI MCP]'
@@ -23411,50 +23447,42 @@ class CorvaUiMcpServer {
   }
   setupInitializedHandler() {
     this.server.server.oninitialized = () => {
+      // The SDK calls this for every notifications/initialized, and a repeat must not count a
+      // second session.
+      if (this.sessionStartPromise) {
+        return;
+      }
       const timestamp = Date.now();
       const clientInfo = this.server.server.getClientVersion();
       const clientCapabilities = this.server.server.getClientCapabilities();
-      this.pendingInitData = {
-        timestamp,
-        clientName: clientInfo?.name,
-        clientVersion: clientInfo?.version,
-        clientCapabilities: clientCapabilities ? JSON.stringify(clientCapabilities) : undefined
-      };
       if (clientInfo?.name) {
         this.mcpLogger.info(`Client connected: ${clientInfo.name}${clientInfo.version ? ` v${clientInfo.version}` : ''}`);
       }
-      this.identityPromise = resolveProjectIdentity(this.server.server, this.mcpLogger).then(identity => {
-        this.projectIdentity = identity;
+      const identity = resolveProjectIdentity(this.server.server, this.mcpLogger).then(resolved => {
+        this.projectIdentity = resolved;
+        return resolved;
       }).catch(error => {
         this.mcpLogger.error('Failed to resolve project identity', error);
-      }).finally(() => {
-        this.flushPendingInitData();
+        return null;
+      });
+      this.identityPromise = identity.then(() => undefined);
+
+      // run() connects only after telemetry has settled, so initialized is the later of the two
+      // and the identity deadline starts here.
+      this.sessionStartPromise = recordSessionStart({
+        telemetry: this.telemetry,
+        initData: {
+          timestamp,
+          clientName: clientInfo?.name,
+          clientVersion: clientInfo?.version,
+          clientCapabilities: clientCapabilities ? JSON.stringify(clientCapabilities) : undefined
+        },
+        identity
+      }).catch(error => {
+        this.mcpLogger.error('Failed to record session start', error);
       });
     };
   }
-  flushPendingInitData() {
-    if (!this.pendingInitData || !this.telemetry.isEnabled()) {
-      return;
-    }
-    const {
-      timestamp,
-      clientName,
-      clientVersion,
-      clientCapabilities
-    } = this.pendingInitData;
-    this.telemetry.metrics.recordSession(clientName, clientVersion);
-    const attrs = {
-      timestamp,
-      clientName,
-      clientVersion,
-      clientCapabilities,
-      sessionId: this.telemetry.sessionId,
-      protocolVersion: MCP_PROTOCOL_VERSION,
-      networkTransport: MCP_NETWORK_TRANSPORT
-    };
-    this.telemetry.recordInitialize(attrs);
-    this.pendingInitData = null;
-  }
   executeToolWithObservability(toolName, args, handler, formatResult, options) {
     this.requestCount += 1;
     const startTime = Date.now();
@@ -23663,6 +23691,9 @@ class CorvaUiMcpServer {
     if (this.identityPromise) {
       await withTimeout(this.identityPromise, TELEMETRY_INIT_TIMEOUT_MS);
     }
+    if (this.sessionStartPromise) {
+      await withTimeout(this.sessionStartPromise, IDENTITY_FLUSH_TIMEOUT_MS);
+    }
 
     // No transport means shutdown came during the telemetry wait: the client never connected, so
     // there is no session to end.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@corva/ui",
-  "version": "3.78.0-18",
+  "version": "3.78.0-19",
   "license": "UNLICENSED",
   "description": "Shared components/utils for Corva ui projects",
   "repository": {