@thinkrun/mcp 0.3.5

package/dist/src/server.d.ts

@@ -0,0 +1,135 @@
+/**
+ * ThinkRun MCP Server — exposes browser automation as MCP tools.
+ *
+ * Security: Selectors are validated via cssSelector() zod refinement to reject
+ * script injection attempts. The server-side also applies JSON.stringify()
+ * escaping when embedding selectors in error messages (defense-in-depth).
+ */
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import type { ThinkRunClient } from './client.js';
+/** Mutable ref for the current default session (used by session_use tool). */
+export interface DefaultSessionRef {
+    current: string | undefined;
+}
+/** Mutable ref for the current client (used by set_mode tool). */
+export interface ClientRef {
+    current: ThinkRunClient;
+}
+/** Result from a set_mode handler — includes the new client and the resolved mode. */
+export interface SetModeResult {
+    client: ThinkRunClient;
+    /** The actual mode resolved (e.g. 'auto' resolves to 'local' or 'cloud'). */
+    resolvedMode: 'local' | 'cloud';
+}
+/**
+ * Callback invoked when the set_mode tool is called.
+ * Should create and return a new client for the requested mode.
+ * Throw an error to reject the mode switch (e.g. missing API key).
+ */
+export type SetModeHandler = (mode: 'local' | 'cloud' | 'auto') => Promise<SetModeResult>;
+export interface CreateServerOptions {
+    /**
+     * Default session ID to use when the AI agent doesn't provide one.
+     * In extension mode, the server-side spawner knows the session ID —
+     * this removes the need for the AI to call session_create first.
+     */
+    defaultSessionId?: string;
+    /**
+     * When provided, the server uses this ref for the default session and
+     * session_use can update it. Use the same ref when creating the server
+     * so that session_use updates the default for subsequent tool calls.
+     */
+    defaultSessionRef?: DefaultSessionRef;
+    /**
+     * When provided, enables the set_mode tool for runtime client switching.
+     * The clientRef.current is swapped by the set_mode handler.
+     */
+    clientRef?: ClientRef;
+    /**
+     * Called by set_mode to create a new client for the requested mode.
+     * Required when clientRef is provided.
+     */
+    onSetMode?: SetModeHandler;
+}
+/** Tool result shape returned by MCP tool handlers. */
+export type ToolResult = {
+    content: Array<{
+        type: 'text';
+        text: string;
+    }>;
+    isError: boolean;
+};
+export type LocalSessionLifecycleOperation = 'tab_attach' | 'tab_switch' | 'tab_focus' | 'window_new' | 'session_release';
+export interface ApplyLifecycleDefaultSessionArgs {
+    currentDefaultSessionId?: string;
+    operation: LocalSessionLifecycleOperation;
+    resultSessionId?: string;
+    releasedSessionId?: string;
+}
+export declare const MAX_GET_HTML_CHARS = 200000;
+export declare const MAX_SLEEP_MS = 30000;
+/**
+ * Apply the PRD's local-session mutation policy without requiring tool registration.
+ * This is exported so tests can lock the behavior down before the MCP tools are added.
+ */
+export declare function applyLifecycleDefaultSession(args: ApplyLifecycleDefaultSessionArgs): string | undefined;
+/**
+ * clear_logs tool handler logic. Exported for unit tests so we can assert
+ * on content/isError without going through the MCP transport.
+ * Catches rejections from client.clearLogs (e.g. network or unexpected API errors)
+ * and returns a structured ToolResult so the MCP tool always responds with content/isError.
+ */
+export declare function handleClearLogs(client: ThinkRunClient, args: {
+    sessionId?: string;
+}, sidResolver: (args: {
+    sessionId?: string;
+}) => string): Promise<ToolResult>;
+export declare function handleTabAttach(client: ThinkRunClient, args: {
+    tabId: number;
+}, defaultSessionRef: DefaultSessionRef): Promise<ToolResult>;
+export declare function handleTabSwitch(client: ThinkRunClient, args: {
+    tabId: number;
+}, defaultSessionRef: DefaultSessionRef): Promise<ToolResult>;
+export declare function handleWindowNew(client: ThinkRunClient, args: {
+    url?: string;
+}, defaultSessionRef: DefaultSessionRef): Promise<ToolResult>;
+export declare function handleTabFocus(client: ThinkRunClient, args: {
+    sessionId?: string;
+}, sidResolver: (args: {
+    sessionId?: string;
+}) => string, defaultSessionRef: DefaultSessionRef): Promise<ToolResult>;
+export declare function handleSessionRelease(client: ThinkRunClient, args: {
+    sessionId?: string;
+}, defaultSessionRef: DefaultSessionRef): Promise<ToolResult>;
+export declare function handleLocalActionRun(client: ThinkRunClient, args: {
+    instruction: string;
+    maxIterations?: number;
+}): Promise<ToolResult>;
+export declare function handleLocalActionStatus(client: ThinkRunClient, args: {
+    actionId: string;
+}): Promise<ToolResult>;
+export declare function handleLocalActionCancel(client: ThinkRunClient, args: {
+    actionId: string;
+}): Promise<ToolResult>;
+export declare function handleLocalDiagnostics(client: ThinkRunClient): Promise<ToolResult>;
+export declare function handleLocalResetConnection(client: ThinkRunClient): Promise<ToolResult>;
+export declare function handleGetUrl(client: ThinkRunClient, args: {
+    sessionId?: string;
+}, sidResolver: (args: {
+    sessionId?: string;
+}) => string): Promise<ToolResult>;
+export declare function handleGetTitle(client: ThinkRunClient, args: {
+    sessionId?: string;
+}, sidResolver: (args: {
+    sessionId?: string;
+}) => string): Promise<ToolResult>;
+export declare function handleGetHtml(client: ThinkRunClient, args: {
+    sessionId?: string;
+}, sidResolver: (args: {
+    sessionId?: string;
+}) => string): Promise<ToolResult>;
+export declare function handleSleep(args: {
+    durationMs: number;
+}): Promise<ToolResult>;
+export declare function createServer(client: ThinkRunClient, options?: CreateServerOptions): McpServer;
+//# sourceMappingURL=server.d.ts.map

package/dist/src/server.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../src/server.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAEpE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AA0DlD,8EAA8E;AAC9E,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,MAAM,GAAG,SAAS,CAAC;CAC7B;AAED,kEAAkE;AAClE,MAAM,WAAW,SAAS;IACxB,OAAO,EAAE,cAAc,CAAC;CACzB;AAED,sFAAsF;AACtF,MAAM,WAAW,aAAa;IAC5B,MAAM,EAAE,cAAc,CAAC;IACvB,6EAA6E;IAC7E,YAAY,EAAE,OAAO,GAAG,OAAO,CAAC;CACjC;AAED;;;;GAIG;AACH,MAAM,MAAM,cAAc,GAAG,CAAC,IAAI,EAAE,OAAO,GAAG,OAAO,GAAG,MAAM,KAAK,OAAO,CAAC,aAAa,CAAC,CAAC;AAE1F,MAAM,WAAW,mBAAmB;IAClC;;;;OAIG;IACH,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B;;;;OAIG;IACH,iBAAiB,CAAC,EAAE,iBAAiB,CAAC;IACtC;;;OAGG;IACH,SAAS,CAAC,EAAE,SAAS,CAAC;IACtB;;;OAGG;IACH,SAAS,CAAC,EAAE,cAAc,CAAC;CAC5B;AAED,uDAAuD;AACvD,MAAM,MAAM,UAAU,GAAG;IACvB,OAAO,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC/C,OAAO,EAAE,OAAO,CAAC;CAClB,CAAC;AAEF,MAAM,MAAM,8BAA8B,GACtC,YAAY,GACZ,YAAY,GACZ,WAAW,GACX,YAAY,GACZ,iBAAiB,CAAC;AAEtB,MAAM,WAAW,gCAAgC;IAC/C,uBAAuB,CAAC,EAAE,MAAM,CAAC;IACjC,SAAS,EAAE,8BAA8B,CAAC;IAC1C,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B;AAED,eAAO,MAAM,kBAAkB,SAAU,CAAC;AAC1C,eAAO,MAAM,YAAY,QAAS,CAAC;AAiGnC;;;GAGG;AACH,wBAAgB,4BAA4B,CAC1C,IAAI,EAAE,gCAAgC,GACrC,MAAM,GAAG,SAAS,CAepB;AAED;;;;;GAKG;AACH,wBAAsB,eAAe,CACnC,MAAM,EAAE,cAAc,EACtB,IAAI,EAAE;IAAE,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,EAC5B,WAAW,EAAE,CAAC,IAAI,EAAE;IAAE,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,KAAK,MAAM,GACpD,OAAO,CAAC,UAAU,CAAC,CAiBrB;AAED,wBAAsB,eAAe,CACnC,MAAM,EAAE,cAAc,EACtB,IAAI,EAAE;IAAE,KAAK,EAAE,MAAM,CAAA;CAAE,EACvB,iBAAiB,EAAE,iBAAiB,GACnC,OAAO,CAAC,UAAU,CAAC,CAerB;AAED,wBAAsB,eAAe,CACnC,MAAM,EAAE,cAAc,EACtB,IAAI,EAAE;IAAE,KAAK,EAAE,MAAM,CAAA;CAAE,EACvB,iBAAiB,EAAE,iBAAiB,GACnC,OAAO,CAAC,UAAU,CAAC,CAcrB;AAED,wBAAsB,eAAe,CACnC,MAAM,EAAE,cAAc,EACtB,IAAI,EAAE;IAAE,GAAG,CAAC,EAAE,MAAM,CAAA;CAAE,EACtB,iBAAiB,EAAE,iBAAiB,GACnC,OAAO,CAAC,UAAU,CAAC,CAerB;AAiBD,wBAAsB,cAAc,CAClC,MAAM,EAAE,cAAc,EACtB,IAAI,EAAE;IAAE,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,EAC5B,WAAW,EAAE,CAAC,IAAI,EAAE;IAAE,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,KAAK,MAAM,EACrD,iBAAiB,EAAE,iBAAiB,GACnC,OAAO,CAAC,UAAU,CAAC,CAyBrB;AAED,wBAAsB,oBAAoB,CACxC,MAAM,EAAE,cAAc,EACtB,IAAI,EAAE;IAAE,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,EAC5B,iBAAiB,EAAE,iBAAiB,GACnC,OAAO,CAAC,UAAU,CAAC,CAkBrB;AAED,wBAAsB,oBAAoB,CACxC,MAAM,EAAE,cAAc,EACtB,IAAI,EAAE;IAAE,WAAW,EAAE,MAAM,CAAC;IAAC,aAAa,CAAC,EAAE,MAAM,CAAA;CAAE,GACpD,OAAO,CAAC,UAAU,CAAC,CAarB;AAED,wBAAsB,uBAAuB,CAC3C,MAAM,EAAE,cAAc,EACtB,IAAI,EAAE;IAAE,QAAQ,EAAE,MAAM,CAAA;CAAE,GACzB,OAAO,CAAC,UAAU,CAAC,CAUrB;AAED,wBAAsB,uBAAuB,CAC3C,MAAM,EAAE,cAAc,EACtB,IAAI,EAAE;IAAE,QAAQ,EAAE,MAAM,CAAA;CAAE,GACzB,OAAO,CAAC,UAAU,CAAC,CAUrB;AAED,wBAAsB,sBAAsB,CAAC,MAAM,EAAE,cAAc,GAAG,OAAO,CAAC,UAAU,CAAC,CAexF;AAED,wBAAsB,0BAA0B,CAAC,MAAM,EAAE,cAAc,GAAG,OAAO,CAAC,UAAU,CAAC,CAe5F;AAED,wBAAsB,YAAY,CAChC,MAAM,EAAE,cAAc,EACtB,IAAI,EAAE;IAAE,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,EAC5B,WAAW,EAAE,CAAC,IAAI,EAAE;IAAE,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,KAAK,MAAM,GACpD,OAAO,CAAC,UAAU,CAAC,CAYrB;AAED,wBAAsB,cAAc,CAClC,MAAM,EAAE,cAAc,EACtB,IAAI,EAAE;IAAE,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,EAC5B,WAAW,EAAE,CAAC,IAAI,EAAE;IAAE,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,KAAK,MAAM,GACpD,OAAO,CAAC,UAAU,CAAC,CAYrB;AAED,wBAAsB,aAAa,CACjC,MAAM,EAAE,cAAc,EACtB,IAAI,EAAE;IAAE,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,EAC5B,WAAW,EAAE,CAAC,IAAI,EAAE;IAAE,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,KAAK,MAAM,GACpD,OAAO,CAAC,UAAU,CAAC,CA2BrB;AAED,wBAAsB,WAAW,CAAC,IAAI,EAAE;IAAE,UAAU,EAAE,MAAM,CAAA;CAAE,GAAG,OAAO,CAAC,UAAU,CAAC,CAqBnF;AAID,wBAAgB,YAAY,CAAC,MAAM,EAAE,cAAc,EAAE,OAAO,CAAC,EAAE,mBAAmB,GAAG,SAAS,CA8kC7F"}