/**
 * Copyright (c) Microsoft Corporation.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import { Client } from '@modelcontextprotocol/sdk/client/index.js';
import { contextFactory } from '../browserContextFactory.js';
import { BrowserServerBackend } from '../browserServerBackend.js';
import { Context as BrowserContext } from '../context.js';
import { runTask } from '../loop/loop.js';
import { OpenAIDelegate } from '../loop/loopOpenAI.js';
import { ClaudeDelegate } from '../loop/loopClaude.js';
import { InProcessTransport } from '../mcp/inProcessTransport.js';
import * as mcpServer from '../mcp/server.js';

import type { LLMDelegate } from '../loop/loop.js';
import type { FullConfig } from '../config.js';

export class Context {
  readonly config: FullConfig;
  private _client: Client;
  private _delegate: LLMDelegate;

  constructor(config: FullConfig, client: Client) {
    this.config = config;
    this._client = client;
    if (process.env.OPENAI_API_KEY)
      this._delegate = new OpenAIDelegate();
    else if (process.env.ANTHROPIC_API_KEY)
      this._delegate = new ClaudeDelegate();
    else
      throw new Error('No LLM API key found. Please set OPENAI_API_KEY or ANTHROPIC_API_KEY environment variable.');
  }

  static async create(config: FullConfig) {
    const client = new Client({ name: 'Playwright Proxy', version: '1.0.0' });
    const browserContextFactory = contextFactory(config.browser);
    const server = mcpServer.createServer(new BrowserServerBackend(config, browserContextFactory));
    await client.connect(new InProcessTransport(server));
    await client.ping();
    return new Context(config, client);
  }

  async runTask(task: string, oneShot: boolean = false): Promise<mcpServer.ToolResponse> {
    const result = await runTask(this._delegate, this._client!, task, oneShot);
    return {
      content: [{ type: 'text', text: result }],
    };
  }

  async close() {
    await BrowserContext.disposeAll();
  }
}