@testdriverai/agent 7.8.0-test.38

package/mcp-server/dist/session.d.ts

@@ -0,0 +1,85 @@
+/**
+ * Session state management for TestDriver MCP Server
+ * Tracks sandbox sessions for MCP server
+ */
+export interface SessionState {
+    sessionId: string;
+    sandboxId: string | null;
+    status: "initializing" | "active" | "expired" | "error";
+    os: "linux" | "windows";
+    resolution: string;
+    createdAt: number;
+    expiresAt: number;
+    keepAlive: number;
+    testFile: string | null;
+    lastScreenshot: string | null;
+    errorMessage?: string;
+}
+/**
+ * Session Manager - handles session state for MCP server
+ */
+export declare class SessionManager {
+    private sessions;
+    private currentSessionId;
+    /**
+     * Create a new session
+     */
+    createSession(options: {
+        sandboxId?: string;
+        os?: "linux" | "windows";
+        resolution?: string;
+        keepAlive?: number;
+        testFile?: string;
+    }): SessionState;
+    /**
+     * Get the current active session
+     */
+    getCurrentSession(): SessionState | null;
+    /**
+     * Get a session by ID
+     */
+    getSession(sessionId: string): SessionState | null;
+    /**
+     * Update session state
+     */
+    updateSession(sessionId: string, updates: Partial<SessionState>): SessionState | null;
+    /**
+     * Mark session as active with sandbox ID
+     */
+    activateSession(sessionId: string, sandboxId: string): SessionState | null;
+    /**
+     * Extend session expiry time by adding additional milliseconds
+     */
+    extendSession(sessionId: string, additionalMs: number): SessionState | null;
+    /**
+     * Refresh session expiry to reset the keepAlive timer from now
+     * Called on each command to keep session alive during active use
+     */
+    refreshSession(sessionId: string): SessionState | null;
+    /**
+     * Check if session is still valid
+     */
+    isSessionValid(sessionId: string): boolean;
+    /**
+     * Get time remaining until session expires
+     */
+    getTimeRemaining(sessionId: string): number;
+    /**
+     * Mark session as expired/ended
+     */
+    endSession(sessionId: string): void;
+    /**
+     * Delete a session
+     */
+    deleteSession(sessionId: string): boolean;
+    /**
+     * Get session summary for status reporting
+     */
+    getSessionSummary(sessionId: string): {
+        sessionId: string;
+        status: string;
+        sandboxId: string | null;
+        timeRemaining: number;
+    } | null;
+}
+export declare const sessionManager: SessionManager;

package/mcp-server/dist/session.js

@@ -0,0 +1,152 @@
+/**
+ * Session state management for TestDriver MCP Server
+ * Tracks sandbox sessions for MCP server
+ */
+// Generate unique session ID
+function generateSessionId() {
+    return `td-${Date.now()}-${Math.random().toString(36).substring(2, 9)}`;
+}
+/**
+ * Session Manager - handles session state for MCP server
+ */
+export class SessionManager {
+    sessions = new Map();
+    currentSessionId = null;
+    /**
+     * Create a new session
+     */
+    createSession(options) {
+        const sessionId = generateSessionId();
+        const keepAlive = options.keepAlive ?? 300000; // Default 5 minutes
+        const session = {
+            sessionId,
+            sandboxId: options.sandboxId ?? null,
+            status: "initializing",
+            os: options.os ?? "linux",
+            resolution: options.resolution ?? "1366x768",
+            createdAt: Date.now(),
+            expiresAt: Date.now() + keepAlive,
+            keepAlive,
+            testFile: options.testFile ?? null,
+            lastScreenshot: null,
+        };
+        this.sessions.set(sessionId, session);
+        this.currentSessionId = sessionId;
+        return session;
+    }
+    /**
+     * Get the current active session
+     */
+    getCurrentSession() {
+        if (!this.currentSessionId)
+            return null;
+        return this.sessions.get(this.currentSessionId) ?? null;
+    }
+    /**
+     * Get a session by ID
+     */
+    getSession(sessionId) {
+        return this.sessions.get(sessionId) ?? null;
+    }
+    /**
+     * Update session state
+     */
+    updateSession(sessionId, updates) {
+        const session = this.sessions.get(sessionId);
+        if (!session)
+            return null;
+        const updated = { ...session, ...updates };
+        this.sessions.set(sessionId, updated);
+        return updated;
+    }
+    /**
+     * Mark session as active with sandbox ID
+     */
+    activateSession(sessionId, sandboxId) {
+        return this.updateSession(sessionId, {
+            sandboxId,
+            status: "active",
+        });
+    }
+    /**
+     * Extend session expiry time by adding additional milliseconds
+     */
+    extendSession(sessionId, additionalMs) {
+        const session = this.sessions.get(sessionId);
+        if (!session)
+            return null;
+        return this.updateSession(sessionId, {
+            expiresAt: session.expiresAt + additionalMs,
+        });
+    }
+    /**
+     * Refresh session expiry to reset the keepAlive timer from now
+     * Called on each command to keep session alive during active use
+     */
+    refreshSession(sessionId) {
+        const session = this.sessions.get(sessionId);
+        if (!session)
+            return null;
+        return this.updateSession(sessionId, {
+            expiresAt: Date.now() + session.keepAlive,
+        });
+    }
+    /**
+     * Check if session is still valid
+     */
+    isSessionValid(sessionId) {
+        const session = this.sessions.get(sessionId);
+        if (!session)
+            return false;
+        if (session.status === "expired" || session.status === "error")
+            return false;
+        if (Date.now() > session.expiresAt) {
+            this.updateSession(sessionId, { status: "expired" });
+            return false;
+        }
+        return true;
+    }
+    /**
+     * Get time remaining until session expires
+     */
+    getTimeRemaining(sessionId) {
+        const session = this.sessions.get(sessionId);
+        if (!session)
+            return 0;
+        return Math.max(0, session.expiresAt - Date.now());
+    }
+    /**
+     * Mark session as expired/ended
+     */
+    endSession(sessionId) {
+        this.updateSession(sessionId, { status: "expired" });
+        if (this.currentSessionId === sessionId) {
+            this.currentSessionId = null;
+        }
+    }
+    /**
+     * Delete a session
+     */
+    deleteSession(sessionId) {
+        if (this.currentSessionId === sessionId) {
+            this.currentSessionId = null;
+        }
+        return this.sessions.delete(sessionId);
+    }
+    /**
+     * Get session summary for status reporting
+     */
+    getSessionSummary(sessionId) {
+        const session = this.sessions.get(sessionId);
+        if (!session)
+            return null;
+        return {
+            sessionId: session.sessionId,
+            status: session.status,
+            sandboxId: session.sandboxId,
+            timeRemaining: this.getTimeRemaining(sessionId),
+        };
+    }
+}
+// Singleton instance
+export const sessionManager = new SessionManager();

package/mcp-server/mcp-app.html

@@ -0,0 +1,28 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+  <meta charset="UTF-8">
+  <meta name="viewport" content="width=device-width, initial-scale=1.0">
+  <meta name="color-scheme" content="light dark">
+  <title>TestDriver Screenshot</title>
+</head>
+<body>
+  <main class="main">
+    <div class="screenshot-wrapper">
+      <div class="loading-overlay" id="loading-overlay">
+        <div class="loading-spinner"></div>
+        <span class="loading-text">Waiting for screenshot...</span>
+      </div>
+      <div class="screenshot-container" id="screenshot-container" style="display: none;">
+        <img id="screenshot" alt="Screenshot" />
+        <div id="overlays"></div>
+      </div>
+    </div>
+    <div class="status-bar">
+      <span id="action-status"></span>
+      <span id="session-info"></span>
+    </div>
+  </main>
+  <script type="module" src="/src/mcp-app.ts"></script>
+</body>
+</html>

package/mcp-server/mcp-config.example.json

@@ -0,0 +1,19 @@
+{
+  "inputs": [
+    {
+      "type": "promptString",
+      "id": "testdriver-api-key",
+      "description": "TestDriver API Key From https://console.testdriver.ai/team",
+      "password": true
+    }
+  ],
+  "servers": {
+    "testdriver": {
+      "command": "npx",
+      "args": ["-p", "testdriverai", "testdriverai-mcp"],
+      "env": {
+        "TD_API_KEY": "${input:testdriver-api-key}"
+      }
+    }
+  }
+}