@pi-oxide/extension-js 0.2.4 → 0.3.0

package/package.json

@@ -1,7 +1,7 @@
 {
 	"name": "@pi-oxide/extension-js",
 	"type": "module",
-	"version": "0.2.4",
+	"version": "0.3.0",
 	"license": "LicenseRef-PiccoloNotebook-Fair-BYOK-1.0",
 	"main": "index.js",
 	"types": "index.d.ts",

package/runner.d.ts

@@ -9,8 +9,9 @@ declare global {
 type HostHandler<T = unknown, R = unknown> = (params: T) => Promise<R>;
 export declare function registerHostHandler<T, R>(action: string, handler: (params: T) => Promise<R>): void;
 export declare function registerHostHandlers(handlers: Record<string, HostHandler>): void;
+export declare function isValidMainThreadAction(action: string): boolean;
 export declare function normalizeParams(action: string, params: unknown): unknown;
-export declare function executeMainThreadCommand(command: Command): Promise<AsyncResponse>;
+export declare function executeMainThreadCommand(command: Command, relaySignal?: AbortSignal): Promise<AsyncResponse>;
 export declare function getActiveTabId(): number | null;
 export declare function initExtensionListeners(): void;
 export declare function removeExtensionListeners(): void;

package/schemas.d.ts

@@ -113,10 +113,70 @@ export declare const PageWaitParamsSchema: z.ZodObject<{
 }, {
     duration?: number | bigint | undefined;
 }>;
-export declare const PageClickParamsSchema: z.ZodRecord<z.ZodString, z.ZodUnknown>;
-export declare const PageFillParamsSchema: z.ZodRecord<z.ZodString, z.ZodUnknown>;
-export declare const PageTypeParamsSchema: z.ZodRecord<z.ZodString, z.ZodUnknown>;
-export declare const PageAppendParamsSchema: z.ZodRecord<z.ZodString, z.ZodUnknown>;
+export declare const PageClickParamsSchema: z.ZodEffects<z.ZodObject<{
+    refId: z.ZodOptional<z.ZodString>;
+    label: z.ZodOptional<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    refId?: string | undefined;
+    label?: string | undefined;
+}, {
+    refId?: string | undefined;
+    label?: string | undefined;
+}>, {
+    refId?: string | undefined;
+    label?: string | undefined;
+}, {
+    refId?: string | undefined;
+    label?: string | undefined;
+}>;
+export declare const PageFillParamsSchema: z.ZodEffects<z.ZodObject<{
+    refId: z.ZodOptional<z.ZodString>;
+    label: z.ZodOptional<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    refId?: string | undefined;
+    label?: string | undefined;
+}, {
+    refId?: string | undefined;
+    label?: string | undefined;
+}>, {
+    refId?: string | undefined;
+    label?: string | undefined;
+}, {
+    refId?: string | undefined;
+    label?: string | undefined;
+}>;
+export declare const PageTypeParamsSchema: z.ZodEffects<z.ZodObject<{
+    refId: z.ZodOptional<z.ZodString>;
+    label: z.ZodOptional<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    refId?: string | undefined;
+    label?: string | undefined;
+}, {
+    refId?: string | undefined;
+    label?: string | undefined;
+}>, {
+    refId?: string | undefined;
+    label?: string | undefined;
+}, {
+    refId?: string | undefined;
+    label?: string | undefined;
+}>;
+export declare const PageAppendParamsSchema: z.ZodEffects<z.ZodObject<{
+    refId: z.ZodOptional<z.ZodString>;
+    label: z.ZodOptional<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    refId?: string | undefined;
+    label?: string | undefined;
+}, {
+    refId?: string | undefined;
+    label?: string | undefined;
+}>, {
+    refId?: string | undefined;
+    label?: string | undefined;
+}, {
+    refId?: string | undefined;
+    label?: string | undefined;
+}>;
 export declare const PagePressParamsSchema: z.ZodObject<{
     key: z.ZodString;
 }, "strip", z.ZodTypeAny, {
@@ -124,9 +184,27 @@ export declare const PagePressParamsSchema: z.ZodObject<{
 }, {
     key: string;
 }>;
-export declare const PageSelectParamsSchema: z.ZodRecord<z.ZodString, z.ZodUnknown>;
-export declare const PageCheckParamsSchema: z.ZodRecord<z.ZodString, z.ZodUnknown>;
-export declare const PageHoverParamsSchema: z.ZodRecord<z.ZodString, z.ZodUnknown>;
+export declare const PageSelectParamsSchema: z.ZodObject<{
+    refId: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    refId: string;
+}, {
+    refId: string;
+}>;
+export declare const PageCheckParamsSchema: z.ZodObject<{
+    refId: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    refId: string;
+}, {
+    refId: string;
+}>;
+export declare const PageHoverParamsSchema: z.ZodObject<{
+    refId: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    refId: string;
+}, {
+    refId: string;
+}>;
 export declare const PageUnhoverParamsSchema: z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>;
 export declare const PageScrollParamsSchema: z.ZodObject<{
     direction: z.ZodDefault<z.ZodString>;
@@ -138,8 +216,20 @@ export declare const PageScrollParamsSchema: z.ZodObject<{
     direction?: string | undefined;
     amount?: number | undefined;
 }>;
-export declare const PageScrollToParamsSchema: z.ZodRecord<z.ZodString, z.ZodUnknown>;
-export declare const PageDblClickParamsSchema: z.ZodRecord<z.ZodString, z.ZodUnknown>;
+export declare const PageScrollToParamsSchema: z.ZodObject<{
+    refId: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    refId: string;
+}, {
+    refId: string;
+}>;
+export declare const PageDblClickParamsSchema: z.ZodObject<{
+    refId: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    refId: string;
+}, {
+    refId: string;
+}>;
 export declare const PageFindParamsSchema: z.ZodObject<{
     selector: z.ZodString;
 }, "strip", z.ZodTypeAny, {
@@ -171,17 +261,116 @@ export declare const TabCreateParamsSchema: z.ZodRecord<z.ZodString, z.ZodUnknow
 export declare const TabActivateParamsSchema: z.ZodUnion<[z.ZodNumber, z.ZodArray<z.ZodUnknown, "many">, z.ZodRecord<z.ZodString, z.ZodUnknown>]>;
 export declare const TabCloseParamsSchema: z.ZodUnion<[z.ZodNumber, z.ZodArray<z.ZodUnknown, "many">, z.ZodRecord<z.ZodString, z.ZodUnknown>]>;
 export declare const TabExecuteScriptParamsSchema: z.ZodRecord<z.ZodString, z.ZodUnknown>;
-export declare const TabClickParamsSchema: z.ZodRecord<z.ZodString, z.ZodUnknown>;
-export declare const TabFillParamsSchema: z.ZodRecord<z.ZodString, z.ZodUnknown>;
-export declare const TabScrollToParamsSchema: z.ZodRecord<z.ZodString, z.ZodUnknown>;
-export declare const TabTypeParamsSchema: z.ZodRecord<z.ZodString, z.ZodUnknown>;
-export declare const TabPressParamsSchema: z.ZodRecord<z.ZodString, z.ZodUnknown>;
-export declare const TabSelectParamsSchema: z.ZodRecord<z.ZodString, z.ZodUnknown>;
-export declare const TabCheckParamsSchema: z.ZodRecord<z.ZodString, z.ZodUnknown>;
-export declare const TabHoverParamsSchema: z.ZodRecord<z.ZodString, z.ZodUnknown>;
-export declare const TabUnhoverParamsSchema: z.ZodRecord<z.ZodString, z.ZodUnknown>;
-export declare const TabScrollParamsSchema: z.ZodRecord<z.ZodString, z.ZodUnknown>;
-export declare const TabDblClickParamsSchema: z.ZodRecord<z.ZodString, z.ZodUnknown>;
+export declare const TabClickParamsSchema: z.ZodObject<{
+    refId: z.ZodString;
+    tabId: z.ZodOptional<z.ZodUnion<[z.ZodNumber, z.ZodBigInt]>>;
+}, "strip", z.ZodTypeAny, {
+    refId: string;
+    tabId?: number | bigint | undefined;
+}, {
+    refId: string;
+    tabId?: number | bigint | undefined;
+}>;
+export declare const TabFillParamsSchema: z.ZodObject<{
+    refId: z.ZodString;
+    tabId: z.ZodOptional<z.ZodUnion<[z.ZodNumber, z.ZodBigInt]>>;
+}, "strip", z.ZodTypeAny, {
+    refId: string;
+    tabId?: number | bigint | undefined;
+}, {
+    refId: string;
+    tabId?: number | bigint | undefined;
+}>;
+export declare const TabScrollToParamsSchema: z.ZodObject<{
+    refId: z.ZodString;
+    tabId: z.ZodOptional<z.ZodUnion<[z.ZodNumber, z.ZodBigInt]>>;
+}, "strip", z.ZodTypeAny, {
+    refId: string;
+    tabId?: number | bigint | undefined;
+}, {
+    refId: string;
+    tabId?: number | bigint | undefined;
+}>;
+export declare const TabTypeParamsSchema: z.ZodObject<{
+    refId: z.ZodString;
+    tabId: z.ZodOptional<z.ZodUnion<[z.ZodNumber, z.ZodBigInt]>>;
+}, "strip", z.ZodTypeAny, {
+    refId: string;
+    tabId?: number | bigint | undefined;
+}, {
+    refId: string;
+    tabId?: number | bigint | undefined;
+}>;
+export declare const TabPressParamsSchema: z.ZodObject<{
+    key: z.ZodString;
+    tabId: z.ZodOptional<z.ZodUnion<[z.ZodNumber, z.ZodBigInt]>>;
+}, "strip", z.ZodTypeAny, {
+    key: string;
+    tabId?: number | bigint | undefined;
+}, {
+    key: string;
+    tabId?: number | bigint | undefined;
+}>;
+export declare const TabSelectParamsSchema: z.ZodObject<{
+    refId: z.ZodString;
+    tabId: z.ZodOptional<z.ZodUnion<[z.ZodNumber, z.ZodBigInt]>>;
+}, "strip", z.ZodTypeAny, {
+    refId: string;
+    tabId?: number | bigint | undefined;
+}, {
+    refId: string;
+    tabId?: number | bigint | undefined;
+}>;
+export declare const TabCheckParamsSchema: z.ZodObject<{
+    refId: z.ZodString;
+    tabId: z.ZodOptional<z.ZodUnion<[z.ZodNumber, z.ZodBigInt]>>;
+}, "strip", z.ZodTypeAny, {
+    refId: string;
+    tabId?: number | bigint | undefined;
+}, {
+    refId: string;
+    tabId?: number | bigint | undefined;
+}>;
+export declare const TabHoverParamsSchema: z.ZodObject<{
+    refId: z.ZodString;
+    tabId: z.ZodOptional<z.ZodUnion<[z.ZodNumber, z.ZodBigInt]>>;
+}, "strip", z.ZodTypeAny, {
+    refId: string;
+    tabId?: number | bigint | undefined;
+}, {
+    refId: string;
+    tabId?: number | bigint | undefined;
+}>;
+export declare const TabUnhoverParamsSchema: z.ZodObject<{
+    tabId: z.ZodOptional<z.ZodUnion<[z.ZodNumber, z.ZodBigInt]>>;
+}, "strip", z.ZodTypeAny, {
+    tabId?: number | bigint | undefined;
+}, {
+    tabId?: number | bigint | undefined;
+}>;
+export declare const TabScrollParamsSchema: z.ZodObject<{
+    direction: z.ZodDefault<z.ZodString>;
+    amount: z.ZodDefault<z.ZodNumber>;
+    tabId: z.ZodOptional<z.ZodUnion<[z.ZodNumber, z.ZodBigInt]>>;
+}, "strip", z.ZodTypeAny, {
+    direction: string;
+    amount: number;
+    tabId?: number | bigint | undefined;
+}, {
+    direction?: string | undefined;
+    amount?: number | undefined;
+    tabId?: number | bigint | undefined;
+}>;
+export declare const TabDblClickParamsSchema: z.ZodObject<{
+    refId: z.ZodString;
+    tabId: z.ZodOptional<z.ZodUnion<[z.ZodNumber, z.ZodBigInt]>>;
+}, "strip", z.ZodTypeAny, {
+    refId: string;
+    tabId?: number | bigint | undefined;
+}, {
+    refId: string;
+    tabId?: number | bigint | undefined;
+}>;
 export declare const TabEvaluateParamsSchema: z.ZodRecord<z.ZodString, z.ZodUnknown>;
 export declare const TabBackParamsSchema: z.ZodRecord<z.ZodString, z.ZodUnknown>;
 export declare const TabWaitForLoadParamsSchema: z.ZodRecord<z.ZodString, z.ZodUnknown>;

package/tool-registry.d.ts

@@ -42,17 +42,74 @@ export interface ToolDefinition<P, R> {
     description: string;
     params: z.ZodSchema<P>;
     returns: z.ZodSchema<R>;
-    handler: (params: P) => Promise<R>;
+    handler: (params: P, callId?: number, runId?: string, signal?: AbortSignal) => Promise<R>;
     paramTypes: ToolDocParam[];
     returnType?: string;
     returnDoc: string;
     errorCode: string;
     errorCategory?: string;
 }
+export type ExecutionContextId = string;
+export type CallContext = {
+    action: string;
+    callId?: number;
+    runId?: string;
+    signal?: AbortSignal;
+};
+export type JsCallSpec<P, R> = {
+    action: string;
+    namespace: string;
+    name: string;
+    description: string;
+    params: z.ZodSchema<P>;
+    returns: z.ZodSchema<R>;
+    fields?: string[];
+    aliases?: Array<{
+        namespace: string;
+        name: string;
+        fields?: string[];
+    }>;
+    owner: ExecutionContextId;
+    handler: (params: P, ctx: CallContext) => Promise<R>;
+    errorCode: string;
+    errorCategory?: string;
+    paramTypes?: ToolDocParam[];
+    returnType?: string;
+    returnDoc?: string;
+};
+export type SerializableJsCallManifestEntry = {
+    action: string;
+    namespace: string;
+    name: string;
+    publicName: string;
+    description: string;
+    fields: string[] | null;
+    aliases: Array<{
+        namespace: string;
+        name: string;
+        fields: string[] | null;
+    }> | null;
+    owner: ExecutionContextId;
+    paramsDoc: ToolDocParam[];
+    returnsDoc: {
+        type: string;
+        description: string;
+    };
+    errorCode: string;
+    errorCategory?: string;
+};
 export declare function setRunnerAbortController(controller: AbortController | null): void;
+export declare function getRunnerSignal(): AbortSignal | undefined;
 export declare function throwIfAborted(): void;
-export declare function registerTool<P, R>(tool: ToolDefinition<P, R>): void;
+export declare function registerJsCall<P, R>(spec: JsCallSpec<P, R>): void;
 export declare function getTool(action: string): ToolDefinition<unknown, unknown> | undefined;
 export declare function clearRegistry(): void;
-export declare function dispatchTool(action: string, params: unknown): Promise<AsyncResponse>;
+export declare function freezeJsRegistry(): void;
+export declare function clearJsRegistry(): void;
+export declare function getSerializableJsManifest(): SerializableJsCallManifestEntry[];
+/** Convert a serializable manifest entry to the shape expected by WASM registerJsCall/registerJsCallBatch. */
+export declare function manifestEntryToWasm(entry: SerializableJsCallManifestEntry): Record<string, unknown>;
+/** Rust/WASM often passes BTreeMap params as a JS Map; Zod object schemas need plain objects. */
+export declare function coerceWasmParams(params: unknown): unknown;
+export declare function dispatchTool(action: string, params: unknown, callId?: number, runId?: string, signal?: AbortSignal): Promise<AsyncResponse>;
 export declare function listTools(): ToolDoc[];