@stina/extension-api 0.25.0 → 0.26.0-alpha.ea677a6

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,140 +335,15 @@ 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(
-    schedulerCallbacks.map((callback) => callback(payload, execContext)),
+    schedulerCallbacks.map((callback) => Promise.resolve(callback(payload, execContext))),
   )
 
   // Log any errors
@@ -471,6 +352,18 @@ async function handleSchedulerFire(payload: SchedulerFirePayload): Promise<void>
       console.error(`Error in scheduler callback ${index}:`, result.reason)
     }
   })
+
+  // Report result back to host
+  const failed = results.find((r) => r.status === 'rejected') as PromiseRejectedResult | undefined
+  try {
+    await sendRequest<void>('scheduler.reportFireResult', {
+      jobId: payload.id,
+      success: !failed,
+      error: failed ? String(failed.reason) : undefined,
+    })
+  } catch {
+    // Best effort — don't crash if reporting fails
+  }
 }
 
 // ============================================================================
@@ -612,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)
 
@@ -667,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)
 
@@ -917,7 +786,7 @@ function buildContext(
       async cancel(jobId: string): Promise<void> {
         await sendRequest<void>('scheduler.cancel', { jobId })
       },
-      onFire(callback: (payload: SchedulerFirePayload, context: ExecutionContext) => void): Disposable {
+      onFire(callback: (payload: SchedulerFirePayload, context: ExecutionContext) => void | Promise<void>): Disposable {
         schedulerCallbacks.push(callback)
         return {
           dispose: () => {
@@ -954,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
@@ -1012,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),
       })
     }
 

package/src/schemas/components.schema.ts

@@ -251,6 +251,17 @@ export const SelectPropsSchema = z
   .passthrough()
   .describe('Select component')
 
+export const IconPickerPropsSchema = z
+  .object({
+    component: z.literal('IconPicker'),
+    label: z.string().optional().describe('Picker label'),
+    value: z.string().optional().describe('Currently selected icon name'),
+    onChangeAction: ExtensionActionRefSchema.describe('Action to call on change'),
+    style: ExtensionComponentStyleSchema.optional(),
+  })
+  .passthrough()
+  .describe('IconPicker component')
+
 export const VerticalStackPropsSchema = z
   .object({
     component: z.literal('VerticalStack'),
@@ -449,6 +460,7 @@ export type ButtonProps = z.infer<typeof ButtonPropsSchema>
 export type TextInputProps = z.infer<typeof TextInputPropsSchema>
 export type DateTimeInputProps = z.infer<typeof DateTimeInputPropsSchema>
 export type SelectProps = z.infer<typeof SelectPropsSchema>
+export type IconPickerProps = z.infer<typeof IconPickerPropsSchema>
 export type VerticalStackProps = z.infer<typeof VerticalStackPropsSchema>
 export type HorizontalStackProps = z.infer<typeof HorizontalStackPropsSchema>
 export type GridProps = z.infer<typeof GridPropsSchema>

package/src/schemas/index.ts

@@ -113,6 +113,7 @@ export {
   TextInputPropsSchema,
   DateTimeInputPropsSchema,
   SelectPropsSchema,
+  IconPickerPropsSchema,
   VerticalStackPropsSchema,
   HorizontalStackPropsSchema,
   GridPropsSchema,
@@ -145,6 +146,7 @@ export {
   type TextInputProps,
   type DateTimeInputProps,
   type SelectProps,
+  type IconPickerProps,
   type VerticalStackProps,
   type HorizontalStackProps,
   type GridProps,

package/src/types.components.ts

@@ -283,6 +283,14 @@ export interface SelectProps extends ExtensionComponentData {
   onChangeAction: ExtensionActionRef
 }
 
+/** The extension API properties for the IconPicker component. */
+export interface IconPickerProps extends ExtensionComponentData {
+  component: 'IconPicker'
+  label?: string
+  value?: string
+  onChangeAction: ExtensionActionRef
+}
+
 /** The extension API properties for the VerticalStack component. */
 export interface VerticalStackProps extends ExtensionComponentData {
   component: 'VerticalStack'

package/src/types.contributions.ts

@@ -373,6 +373,23 @@ export interface ToolDefinition {
   description: LocalizedString
   /** Parameter schema (JSON Schema) */
   parameters?: Record<string, unknown>
+  /**
+   * Confirmation configuration. If set, user must confirm before tool runs.
+   * If not set, tool runs without confirmation.
+   */
+  confirmation?: ToolConfirmationConfig
+}
+
+/**
+ * Configuration for tool confirmation.
+ */
+export interface ToolConfirmationConfig {
+  /**
+   * Default confirmation prompt to show the user.
+   * If not provided, a generic prompt like "Allow {toolName} to run?" is used.
+   * @example { en: "Allow sending email?", sv: "Tillåt att skicka e-post?" }
+   */
+  prompt?: LocalizedString
 }
 
 // ============================================================================

package/src/types.ts

@@ -48,6 +48,7 @@ export type {
   ProviderConfigValidation,
   // Tools
   ToolDefinition,
+  ToolConfirmationConfig,
   // Commands
   CommandDefinition,
   // Prompts