@testdriverai/agent 7.8.0-canary.10

package/mcp-server/src/session.ts

@@ -0,0 +1,194 @@
+/**
+ * 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;
+}
+
+// Generate unique session ID
+function generateSessionId(): string {
+  return `td-${Date.now()}-${Math.random().toString(36).substring(2, 9)}`;
+}
+
+/**
+ * Session Manager - handles session state for MCP server
+ */
+export class SessionManager {
+  private sessions: Map<string, SessionState> = new Map();
+  private currentSessionId: string | null = null;
+
+  /**
+   * Create a new session
+   */
+  createSession(options: {
+    sandboxId?: string;
+    os?: "linux" | "windows";
+    resolution?: string;
+    keepAlive?: number;
+    testFile?: string;
+  }): SessionState {
+    const sessionId = generateSessionId();
+    const keepAlive = options.keepAlive ?? 300000; // Default 5 minutes
+
+    const session: SessionState = {
+      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(): SessionState | null {
+    if (!this.currentSessionId) return null;
+    return this.sessions.get(this.currentSessionId) ?? null;
+  }
+
+  /**
+   * Get a session by ID
+   */
+  getSession(sessionId: string): SessionState | null {
+    return this.sessions.get(sessionId) ?? null;
+  }
+
+  /**
+   * Update session state
+   */
+  updateSession(
+    sessionId: string,
+    updates: Partial<SessionState>
+  ): SessionState | null {
+    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: string, sandboxId: string): SessionState | null {
+    return this.updateSession(sessionId, {
+      sandboxId,
+      status: "active",
+    });
+  }
+
+  /**
+   * Extend session expiry time by adding additional milliseconds
+   */
+  extendSession(sessionId: string, additionalMs: number): SessionState | null {
+    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: string): SessionState | null {
+    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: string): boolean {
+    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: string): number {
+    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: string): void {
+    this.updateSession(sessionId, { status: "expired" });
+    if (this.currentSessionId === sessionId) {
+      this.currentSessionId = null;
+    }
+  }
+
+  /**
+   * Delete a session
+   */
+  deleteSession(sessionId: string): boolean {
+    if (this.currentSessionId === sessionId) {
+      this.currentSessionId = null;
+    }
+    return this.sessions.delete(sessionId);
+  }
+
+  /**
+   * Get session summary for status reporting
+   */
+  getSessionSummary(sessionId: string): {
+    sessionId: string;
+    status: string;
+    sandboxId: string | null;
+    timeRemaining: number;
+  } | null {
+    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/tsconfig.json

@@ -0,0 +1,16 @@
+{
+  "compilerOptions": {
+    "target": "ES2022",
+    "module": "ESNext",
+    "moduleResolution": "bundler",
+    "esModuleInterop": true,
+    "strict": true,
+    "skipLibCheck": true,
+    "outDir": "dist",
+    "rootDir": "src",
+    "declaration": true,
+    "resolveJsonModule": true
+  },
+  "include": ["src/**/*"],
+  "exclude": ["node_modules", "dist", "src/mcp-app.ts", "src/mcp-app.css"]
+}

package/mcp-server/vite.config.ts

@@ -0,0 +1,23 @@
+import { defineConfig } from "vite";
+import { viteSingleFile } from "vite-plugin-singlefile";
+
+const INPUT = process.env.INPUT;
+if (!INPUT) {
+  throw new Error("INPUT environment variable is not set");
+}
+
+const isDevelopment = process.env.NODE_ENV === "development";
+
+export default defineConfig({
+  plugins: [viteSingleFile()],
+  build: {
+    sourcemap: isDevelopment ? "inline" : undefined,
+    cssMinify: !isDevelopment,
+    minify: !isDevelopment,
+    rollupOptions: {
+      input: INPUT,
+    },
+    outDir: "dist",
+    emptyOutDir: false,
+  },
+});

package/package.json

@@ -0,0 +1,158 @@
+{
+  "name": "@testdriverai/agent",
+  "version": "7.8.0-canary.10",
+  "description": "Next generation autonomous AI agent for end-to-end testing of web & desktop",
+  "main": "sdk.js",
+  "types": "sdk.d.ts",
+  "exports": {
+    ".": {
+      "types": "./sdk.d.ts",
+      "default": "./sdk.js"
+    },
+    "./core": {
+      "types": "./lib/core/index.d.ts",
+      "default": "./lib/core/index.js"
+    },
+    "./vitest": {
+      "types": "./interfaces/vitest-plugin.d.ts",
+      "default": "./interfaces/vitest-plugin.mjs"
+    },
+    "./vitest/plugin": {
+      "types": "./interfaces/vitest-plugin.d.ts",
+      "default": "./interfaces/vitest-plugin.mjs"
+    },
+    "./vitest/setup": "./lib/vitest/setup.mjs",
+    "./vitest/setup-aws": "./lib/vitest/setup-aws.mjs",
+    "./vitest/hooks": {
+      "types": "./lib/vitest/hooks.d.ts",
+      "default": "./lib/vitest/hooks.mjs"
+    },
+    "./presets": "./lib/presets/index.mjs"
+  },
+  "bin": {
+    "testdriverai": "bin/testdriverai.js",
+    "testdriverai-mcp": "mcp-server/dist/server.mjs"
+  },
+  "scripts": {
+    "start": "node bin/testdriverai.js",
+    "dev": "DEV=true node bin/testdriverai.js",
+    "debug": "DEV=true VERBOSE=true node bin/testdriverai.js",
+    "docs": "npm run docs:skills && cd docs && npx mint@latest dev --port 3002",
+    "docs:dev": "cd docs && npx mint dev --port 3002",
+    "docs:build": "npm run docs:skills && cd docs && npx mint@latest build",
+    "docs:links": "node docs/_scripts/link-replacer.js",
+    "docs:skills": "node docs/_scripts/generate-skills.js",
+    "docs:examples": "node docs/_scripts/generate-examples.js",
+    "docs:extract-urls": "node docs/_scripts/extract-example-urls.js",
+    "bundle": "node build.mjs",
+    "test": "mocha test/*",
+    "test:sdk": "vitest run",
+    "test:sdk:dev": "vitest run --project dev",
+    "test:sdk:staging": "vitest run --project staging",
+    "test:sdk:canary": "vitest run --project canary",
+    "test:sdk:stable": "vitest run --project stable",
+    "test:sdk:windows": "TD_OS=windows vitest run",
+    "test:sdk:mac": "TEST_PLATFORM=mac vitest run",
+    "test:sdk:linux": "TEST_PLATFORM=linux vitest run",
+    "test:sdk:watch": "vitest",
+    "test:sdk:ui": "vitest --ui",
+    "test:sdk:coverage": "vitest run --coverage",
+    "test:sdk:single": "vitest run --reporter=verbose",
+    "test:sdk:report": "open test-results/index.html",
+    "test:sdk:results": "node scripts/view-test-results.mjs",
+    "docs-lint": "cd docs && vale **/*.mdx",
+    "docs-upload": "node upload-docs-to-openai.js",
+    "runall": "dotenv testdriverai run --config testdriver.config.yaml --path './testdriver/*.testdriver.yaml'",
+    "test-assert": "node index.js run --path=testdriver/acceptance/assert.yaml",
+    "generate-report": "junit-viewer --results=out.xml --save=report.html",
+    "serve-report": "npx http-server . -p 8080 -o report.html",
+    "report": "npm run generate-report && npm run serve-report",
+    "generate:skills": "node scripts/generate-skills.js",
+    "build:mcp": "cd mcp-server && npm ci && npm run build",
+    "prepublishOnly": "npm run build:mcp"
+  },
+  "author": "",
+  "license": "ISC",
+  "dependencies": {
+    "@modelcontextprotocol/ext-apps": "^1.0.0",
+    "@modelcontextprotocol/sdk": "^1.24.0",
+    "@npmcli/redact": "^3.2.2",
+    "@oclif/core": "^4.5.0",
+    "@oclif/plugin-commands": "^4.1.28",
+    "@oclif/plugin-help": "^6.2.30",
+    "@oclif/plugin-not-found": "^3.2.59",
+    "@oclif/plugin-warn-if-update-available": "^3.1.43",
+    "@octokit/rest": "^20.1.1",
+    "@sentry/node": "^9.47.1",
+    "@stoplight/yaml-ast-parser": "^0.0.50",
+    "ably": "^2.6.0",
+    "ajv": "^8.17.1",
+    "arktype": "^2.1.19",
+    "axios": "^1.7.7",
+    "chalk": "^4.1.2",
+    "cli-progress": "^3.12.0",
+    "diff": "^8.0.2",
+    "dotenv": "^16.6.1",
+    "eventemitter2": "^6.4.9",
+    "jimp": "^0.22.12",
+    "js-yaml": "^4.1.0",
+    "junit-report-builder": "^5.1.1",
+    "markdown-parser": "0.0.8",
+    "marked": "^12.0.1",
+    "marked-terminal": "^7.0.0",
+    "marky": "^1.2.5",
+    "open": "^10.1.2",
+    "pixelmatch": "^7.1.0",
+    "remark-parse": "^11.0.0",
+    "sanitize-filename": "^1.6.3",
+    "semver": "^7.7.4",
+    "strip-ansi": "^6.0.1",
+    "terminal-image": "^4.1.0",
+    "tmp": "^0.2.3",
+    "winston": "^3.13.0",
+    "winston-transport": "^4.9.0",
+    "ws": "^8.18.1",
+    "zod": "^3.24.0"
+  },
+  "overrides": {
+    "glob": "^11.0.1",
+    "rimraf": "^5.0.10"
+  },
+  "peerDependencies": {
+    "vitest": ">=4.0.0"
+  },
+  "peerDependenciesMeta": {
+    "vitest": {
+      "optional": true
+    }
+  },
+  "devDependencies": {
+    "@eslint/js": "^9.10.0",
+    "@vitest/ui": "^4.0.15",
+    "chai": "^5.1.2",
+    "esbuild": "0.20.2",
+    "esbuild-plugin-fileloc": "^0.0.6",
+    "eslint": "^9.10.0",
+    "globals": "^15.9.0",
+    "http-server": "^14.1.1",
+    "junit-viewer": "^4.11.1",
+    "mint": "^4.2.67",
+    "mocha": "^10.8.2",
+    "node-addon-api": "^8.0.0",
+    "prettier": "3.3.3",
+    "vitest": "^4.0.18"
+  },
+  "optionalDependencies": {
+    "@esbuild/linux-x64": "^0.21.5"
+  },
+  "oclif": {
+    "bin": "testdriverai",
+    "dirname": "testdriverai",
+    "commands": "./interfaces/cli/commands",
+    "plugins": [
+      "@oclif/plugin-*"
+    ],
+    "topicSeparator": " ",
+    "default": "init"
+  }
+}