feat(coding): add Pi Data Service product tools

shared/coding-conversation-product-tool-protocol.ts

@@ -1,11 +1,18 @@
 import type {
   AgentBrowserDetailsV1,
   ChangedFileDetailsV1,
+  DataServiceToolDetailsV1,
   GameAssetsDetailsV1,
   KnownToolDetails,
   RuntimeContextDetailsV1,
   TaskStateDetailsV1,
 } from './coding-conversation-contracts';
+import {
+  DATA_SERVICE_PI_TOOL_NAMES,
+  type DataServiceErrorContext,
+  type DataServiceToolData,
+  type DataServicePiToolName,
+} from './data-service';
 
 const PRODUCT_TOOL_NAMES = new Set([
   'agent_browser',
@@ -14,8 +21,21 @@ const PRODUCT_TOOL_NAMES = new Set([
   'task_state',
   'changed_file',
   'runtime_context',
+  ...DATA_SERVICE_PI_TOOL_NAMES,
 ]);
 
+const DATA_SERVICE_CONTEXT_KEYS = new Set([
+  'resource',
+  'limit',
+  'current',
+  'attempted',
+  'actual',
+  'allowed',
+  'current_revision',
+  'retry_after_seconds',
+]);
+const DATA_SERVICE_CONTEXT_RESOURCES = new Set(['instances', 'collections', 'documents', 'bytes']);
+
 function record(value: unknown): Record<string, unknown> | null {
   return value !== null && typeof value === 'object' && !Array.isArray(value)
     ? value as Record<string, unknown>
@@ -133,6 +153,63 @@ function runtimeContextDetails(value: Record<string, unknown>): RuntimeContextDe
   return { schema: 'runtime-context.v1', skills, commands };
 }
 
+function dataServiceContext(value: unknown): DataServiceErrorContext | undefined | null {
+  if (value === undefined) return undefined;
+  const source = record(value);
+  if (!source || Object.keys(source).length > 8) return null;
+  const context: DataServiceErrorContext = {};
+  for (const [key, item] of Object.entries(source)) {
+    if (!DATA_SERVICE_CONTEXT_KEYS.has(key)) return null;
+    if (key === 'resource') {
+      if (typeof item !== 'string' || !DATA_SERVICE_CONTEXT_RESOURCES.has(item)) return null;
+      context[key] = item;
+    } else if (Number.isSafeInteger(item) && (item as number) >= 0) {
+      context[key] = item as number;
+    } else {
+      return null;
+    }
+  }
+  return Object.keys(context).length > 0 ? context : undefined;
+}
+
+function dataServiceDetails(value: Record<string, unknown>): DataServiceToolDetailsV1 | null {
+  if (!DATA_SERVICE_PI_TOOL_NAMES.includes(value.operation as DataServicePiToolName)) return null;
+  if (typeof value.success !== 'boolean'
+    || !Number.isSafeInteger(value.status)
+    || (value.status as number) < 100 || (value.status as number) > 599
+    || (value.code !== null && text(value.code, 128) === null)
+    || (value.error !== null && text(value.error, 2_000) === null)
+    || typeof value.retryable !== 'boolean'
+    || !Object.prototype.hasOwnProperty.call(value, 'data')) return null;
+  if (value.success) {
+    if ((value.status as number) < 200 || (value.status as number) > 299
+      || value.code !== null || value.error !== null || value.retryable !== false) return null;
+  } else if ((value.status as number) < 400 || value.code === null || value.error === null) {
+    return null;
+  }
+  const retryAfter = value.retry_after_seconds;
+  if (retryAfter !== undefined
+    && (!Number.isSafeInteger(retryAfter) || (retryAfter as number) < 0 || (retryAfter as number) > 86_400)) {
+    return null;
+  }
+  const context = dataServiceContext(value.context);
+  if (context === null) return null;
+  const data = value.data === null ? null : record(value.data);
+  if (value.data !== null && !data) return null;
+  return {
+    schema: 'data-service.v1',
+    operation: value.operation as DataServicePiToolName,
+    success: value.success,
+    status: value.status as number,
+    code: value.code as string | null,
+    error: value.error as string | null,
+    retryable: value.retryable,
+    ...(retryAfter === undefined ? {} : { retry_after_seconds: retryAfter as number }),
+    ...(context === undefined ? {} : { context }),
+    data: data as DataServiceToolData | null,
+  };
+}
+
 export function productToolDetails(value: unknown): Exclude<KnownToolDetails, { schema: 'subagent.v1' | 'write-lease.v1' }> | null {
   const details = record(value);
   if (!details) return null;
@@ -144,6 +221,7 @@ export function productToolDetails(value: unknown): Exclude<KnownToolDetails, {
   if (details.schema === 'agent-browser.v1') return browserDetails(details);
   if (details.schema === 'game-assets.v1') return gameAssetDetails(details);
   if (details.schema === 'runtime-context.v1') return runtimeContextDetails(details);
+  if (details.schema === 'data-service.v1') return dataServiceDetails(details);
   return null;
 }