@stina/extension-api 0.26.0-alpha.48430e8 → 0.26.0-alpha.ea677a6

package/src/runtime/index.ts

@@ -0,0 +1,8 @@
+/**
+ * Runtime module re-exports.
+ * These are extracted from the main runtime.ts for modularity.
+ */
+
+export { buildExtensionStorageAPI, buildUserStorageAPI } from './storageApi.js'
+export { buildExtensionSecretsAPI, buildUserSecretsAPI } from './secretsApi.js'
+export { createExecutionContext } from './executionContext.js'

package/src/runtime/secretsApi.ts

@@ -0,0 +1,48 @@
+/**
+ * Secrets API builders for extension and user-scoped secrets.
+ */
+
+import type { SecretsAPI } from '../types.js'
+import type { RequestMessage } from '../messages.js'
+
+type SendRequest = <T>(method: RequestMessage['method'], payload: unknown) => Promise<T>
+
+/**
+ * Build extension-scoped secrets API (shared across all users).
+ */
+export function buildExtensionSecretsAPI(sendRequest: SendRequest): SecretsAPI {
+  return {
+    async set(key: string, value: string): Promise<void> {
+      return sendRequest<void>('secrets.set', { key, value })
+    },
+    async get(key: string): Promise<string | undefined> {
+      return sendRequest<string | undefined>('secrets.get', { key })
+    },
+    async delete(key: string): Promise<boolean> {
+      return sendRequest<boolean>('secrets.delete', { key })
+    },
+    async list(): Promise<string[]> {
+      return sendRequest<string[]>('secrets.list', {})
+    },
+  }
+}
+
+/**
+ * Build user-scoped secrets API (isolated per user).
+ */
+export function buildUserSecretsAPI(sendRequest: SendRequest, userId: string): SecretsAPI {
+  return {
+    async set(key: string, value: string): Promise<void> {
+      return sendRequest<void>('secrets.setForUser', { userId, key, value })
+    },
+    async get(key: string): Promise<string | undefined> {
+      return sendRequest<string | undefined>('secrets.getForUser', { userId, key })
+    },
+    async delete(key: string): Promise<boolean> {
+      return sendRequest<boolean>('secrets.deleteForUser', { userId, key })
+    },
+    async list(): Promise<string[]> {
+      return sendRequest<string[]>('secrets.listForUser', { userId })
+    },
+  }
+}

package/src/runtime/storageApi.ts

@@ -0,0 +1,67 @@
+/**
+ * Storage API builders for extension and user-scoped storage.
+ */
+
+import type { StorageAPI, Query, QueryOptions } from '../types.js'
+import type { RequestMessage } from '../messages.js'
+
+type SendRequest = <T>(method: RequestMessage['method'], payload: unknown) => Promise<T>
+
+/**
+ * Create a storage API that delegates to the host via request messages.
+ * Used for both extension-scoped and user-scoped storage by varying the method prefix.
+ */
+function buildStorageAPI(
+  sendRequest: SendRequest,
+  methodSuffix: '' | 'ForUser',
+  userId?: string
+): StorageAPI {
+  const extraPayload = userId ? { userId } : {}
+
+  return {
+    async put<T extends object>(collection: string, id: string, data: T): Promise<void> {
+      return sendRequest<void>(`storage.put${methodSuffix}`, { ...extraPayload, collection, id, data })
+    },
+    async get<T>(collection: string, id: string): Promise<T | undefined> {
+      return sendRequest<T | undefined>(`storage.get${methodSuffix}`, { ...extraPayload, collection, id })
+    },
+    async delete(collection: string, id: string): Promise<boolean> {
+      return sendRequest<boolean>(`storage.delete${methodSuffix}`, { ...extraPayload, collection, id })
+    },
+    async find<T>(collection: string, query?: Query, options?: QueryOptions): Promise<T[]> {
+      return sendRequest<T[]>(`storage.find${methodSuffix}`, { ...extraPayload, collection, query, options })
+    },
+    async findOne<T>(collection: string, query: Query): Promise<T | undefined> {
+      return sendRequest<T | undefined>(`storage.findOne${methodSuffix}`, { ...extraPayload, collection, query })
+    },
+    async count(collection: string, query?: Query): Promise<number> {
+      return sendRequest<number>(`storage.count${methodSuffix}`, { ...extraPayload, collection, query })
+    },
+    async putMany<T extends object>(collection: string, docs: Array<{ id: string; data: T }>): Promise<void> {
+      return sendRequest<void>(`storage.putMany${methodSuffix}`, { ...extraPayload, collection, docs })
+    },
+    async deleteMany(collection: string, query: Query): Promise<number> {
+      return sendRequest<number>(`storage.deleteMany${methodSuffix}`, { ...extraPayload, collection, query })
+    },
+    async dropCollection(collection: string): Promise<void> {
+      return sendRequest<void>(`storage.dropCollection${methodSuffix}`, { ...extraPayload, collection })
+    },
+    async listCollections(): Promise<string[]> {
+      return sendRequest<string[]>(`storage.listCollections${methodSuffix}`, { ...extraPayload })
+    },
+  }
+}
+
+/**
+ * Build extension-scoped storage API (shared across all users).
+ */
+export function buildExtensionStorageAPI(sendRequest: SendRequest): StorageAPI {
+  return buildStorageAPI(sendRequest, '')
+}
+
+/**
+ * Build user-scoped storage API (isolated per user).
+ */
+export function buildUserStorageAPI(sendRequest: SendRequest, userId: string): StorageAPI {
+  return buildStorageAPI(sendRequest, 'ForUser', userId)
+}

package/src/runtime.ts

@@ -25,8 +25,6 @@ import type {
   ChatInstructionMessage,
   StorageAPI,
   SecretsAPI,
-  Query,
-  QueryOptions,
   LogAPI,
   AIProvider,
   Tool,
@@ -51,6 +49,14 @@ import type {
 
 import { generateMessageId } from './messages.js'
 
+import {
+  buildExtensionStorageAPI,
+  buildUserStorageAPI,
+  buildExtensionSecretsAPI,
+  buildUserSecretsAPI,
+  createExecutionContext,
+} from './runtime/index.js'
+
 // ============================================================================
 // Environment Detection and Message Port
 // ============================================================================
@@ -329,136 +335,11 @@ function handleSettingsChanged(key: string, value: unknown): void {
   }
 }
 
-/**
- * Build extension-scoped storage API
- */
-function buildExtensionStorageAPI(): StorageAPI {
-  return {
-    async put<T extends object>(collection: string, id: string, data: T): Promise<void> {
-      return sendRequest<void>('storage.put', { collection, id, data })
-    },
-    async get<T>(collection: string, id: string): Promise<T | undefined> {
-      return sendRequest<T | undefined>('storage.get', { collection, id })
-    },
-    async delete(collection: string, id: string): Promise<boolean> {
-      return sendRequest<boolean>('storage.delete', { collection, id })
-    },
-    async find<T>(collection: string, query?: Query, options?: QueryOptions): Promise<T[]> {
-      return sendRequest<T[]>('storage.find', { collection, query, options })
-    },
-    async findOne<T>(collection: string, query: Query): Promise<T | undefined> {
-      return sendRequest<T | undefined>('storage.findOne', { collection, query })
-    },
-    async count(collection: string, query?: Query): Promise<number> {
-      return sendRequest<number>('storage.count', { collection, query })
-    },
-    async putMany<T extends object>(collection: string, docs: Array<{ id: string; data: T }>): Promise<void> {
-      return sendRequest<void>('storage.putMany', { collection, docs })
-    },
-    async deleteMany(collection: string, query: Query): Promise<number> {
-      return sendRequest<number>('storage.deleteMany', { collection, query })
-    },
-    async dropCollection(collection: string): Promise<void> {
-      return sendRequest<void>('storage.dropCollection', { collection })
-    },
-    async listCollections(): Promise<string[]> {
-      return sendRequest<string[]>('storage.listCollections', {})
-    },
-  }
-}
-
-/**
- * Build user-scoped storage API
- */
-function buildUserStorageAPI(userId: string): StorageAPI {
-  return {
-    async put<T extends object>(collection: string, id: string, data: T): Promise<void> {
-      return sendRequest<void>('storage.putForUser', { userId, collection, id, data })
-    },
-    async get<T>(collection: string, id: string): Promise<T | undefined> {
-      return sendRequest<T | undefined>('storage.getForUser', { userId, collection, id })
-    },
-    async delete(collection: string, id: string): Promise<boolean> {
-      return sendRequest<boolean>('storage.deleteForUser', { userId, collection, id })
-    },
-    async find<T>(collection: string, query?: Query, options?: QueryOptions): Promise<T[]> {
-      return sendRequest<T[]>('storage.findForUser', { userId, collection, query, options })
-    },
-    async findOne<T>(collection: string, query: Query): Promise<T | undefined> {
-      return sendRequest<T | undefined>('storage.findOneForUser', { userId, collection, query })
-    },
-    async count(collection: string, query?: Query): Promise<number> {
-      return sendRequest<number>('storage.countForUser', { userId, collection, query })
-    },
-    async putMany<T extends object>(collection: string, docs: Array<{ id: string; data: T }>): Promise<void> {
-      return sendRequest<void>('storage.putManyForUser', { userId, collection, docs })
-    },
-    async deleteMany(collection: string, query: Query): Promise<number> {
-      return sendRequest<number>('storage.deleteManyForUser', { userId, collection, query })
-    },
-    async dropCollection(collection: string): Promise<void> {
-      return sendRequest<void>('storage.dropCollectionForUser', { userId, collection })
-    },
-    async listCollections(): Promise<string[]> {
-      return sendRequest<string[]>('storage.listCollectionsForUser', { userId })
-    },
-  }
-}
-
-/**
- * Build extension-scoped secrets API
- */
-function buildExtensionSecretsAPI(): SecretsAPI {
-  return {
-    async set(key: string, value: string): Promise<void> {
-      return sendRequest<void>('secrets.set', { key, value })
-    },
-    async get(key: string): Promise<string | undefined> {
-      return sendRequest<string | undefined>('secrets.get', { key })
-    },
-    async delete(key: string): Promise<boolean> {
-      return sendRequest<boolean>('secrets.delete', { key })
-    },
-    async list(): Promise<string[]> {
-      return sendRequest<string[]>('secrets.list', {})
-    },
-  }
-}
-
-/**
- * Build user-scoped secrets API
- */
-function buildUserSecretsAPI(userId: string): SecretsAPI {
-  return {
-    async set(key: string, value: string): Promise<void> {
-      return sendRequest<void>('secrets.setForUser', { userId, key, value })
-    },
-    async get(key: string): Promise<string | undefined> {
-      return sendRequest<string | undefined>('secrets.getForUser', { userId, key })
-    },
-    async delete(key: string): Promise<boolean> {
-      return sendRequest<boolean>('secrets.deleteForUser', { userId, key })
-    },
-    async list(): Promise<string[]> {
-      return sendRequest<string[]>('secrets.listForUser', { userId })
-    },
-  }
-}
+// Storage and Secrets APIs are now in runtime/storageApi.ts and runtime/secretsApi.ts
+// ExecutionContext builder is in runtime/executionContext.ts
 
 async function handleSchedulerFire(payload: SchedulerFirePayload): Promise<void> {
-  // Create request-scoped execution context with storage and secrets
-  const execContext: ExecutionContext = {
-    userId: payload.userId,
-    extension: {
-      id: extensionContext!.extension.id,
-      version: extensionContext!.extension.version,
-      storagePath: extensionContext!.extension.storagePath,
-    },
-    storage: buildExtensionStorageAPI(),
-    userStorage: payload.userId ? buildUserStorageAPI(payload.userId) : buildExtensionStorageAPI(),
-    secrets: buildExtensionSecretsAPI(),
-    userSecrets: payload.userId ? buildUserSecretsAPI(payload.userId) : buildExtensionSecretsAPI(),
-  }
+  const execContext = createExecutionContext(sendRequest, extensionContext!, payload.userId)
 
   // Run callbacks concurrently to avoid blocking
   const results = await Promise.allSettled(
@@ -624,19 +505,7 @@ async function handleToolExecuteRequest(
   }
 
   try {
-    // Create request-scoped execution context with storage and secrets
-    const execContext: ExecutionContext = {
-      userId: payload.userId,
-      extension: {
-        id: extensionContext!.extension.id,
-        version: extensionContext!.extension.version,
-        storagePath: extensionContext!.extension.storagePath,
-      },
-      storage: buildExtensionStorageAPI(),
-      userStorage: payload.userId ? buildUserStorageAPI(payload.userId) : buildExtensionStorageAPI(),
-      secrets: buildExtensionSecretsAPI(),
-      userSecrets: payload.userId ? buildUserSecretsAPI(payload.userId) : buildExtensionSecretsAPI(),
-    }
+    const execContext = createExecutionContext(sendRequest, extensionContext!, payload.userId)
 
     const result = await tool.execute(payload.params, execContext)
 
@@ -679,19 +548,7 @@ async function handleActionExecuteRequest(
   }
 
   try {
-    // Create request-scoped execution context with storage and secrets
-    const execContext: ExecutionContext = {
-      userId: payload.userId,
-      extension: {
-        id: extensionContext!.extension.id,
-        version: extensionContext!.extension.version,
-        storagePath: extensionContext!.extension.storagePath,
-      },
-      storage: buildExtensionStorageAPI(),
-      userStorage: payload.userId ? buildUserStorageAPI(payload.userId) : buildExtensionStorageAPI(),
-      secrets: buildExtensionSecretsAPI(),
-      userSecrets: payload.userId ? buildUserSecretsAPI(payload.userId) : buildExtensionSecretsAPI(),
-    }
+    const execContext = createExecutionContext(sendRequest, extensionContext!, payload.userId)
 
     const result = await action.execute(payload.params, execContext)
 
@@ -966,12 +823,12 @@ function buildContext(
 
   // Add storage API if permitted (new collection-based storage)
   if (hasPermission('storage.collections')) {
-    ;(context as { storage: StorageAPI }).storage = buildExtensionStorageAPI()
+    ;(context as { storage: StorageAPI }).storage = buildExtensionStorageAPI(sendRequest)
   }
 
   // Add secrets API if permitted
   if (hasPermission('secrets.manage')) {
-    ;(context as { secrets: SecretsAPI }).secrets = buildExtensionSecretsAPI()
+    ;(context as { secrets: SecretsAPI }).secrets = buildExtensionSecretsAPI(sendRequest)
   }
 
   // Add background workers API if permitted
@@ -1024,10 +881,10 @@ function buildContext(
           error: (message, data) =>
             postMessage({ type: 'log', payload: { level: 'error', message: `[${taskId}] ${message}`, data } }),
         }),
-        createStorageAPI: () => buildExtensionStorageAPI(),
-        createUserStorageAPI: (userId) => buildUserStorageAPI(userId),
-        createSecretsAPI: () => buildExtensionSecretsAPI(),
-        createUserSecretsAPI: (userId) => buildUserSecretsAPI(userId),
+        createStorageAPI: () => buildExtensionStorageAPI(sendRequest),
+        createUserStorageAPI: (userId) => buildUserStorageAPI(sendRequest, userId),
+        createSecretsAPI: () => buildExtensionSecretsAPI(sendRequest),
+        createUserSecretsAPI: (userId) => buildUserSecretsAPI(sendRequest, userId),
       })
     }