/**
 * Basic usage example for AgentBay Browser functionality in TypeScript.
 * This example demonstrates the core browser operations without external dependencies.
 */

import {
  AgentBay,
  CreateSessionParamsClass,
  ActOptions,
  ObserveOptions,
  ExtractOptions,
} from "wuying-agentbay-sdk";
import { z } from "zod";

const PageInfoSchema = z.object({
  title: z.string().describe("page title"),
  url: z.string().describe("page url"),
});
async function main() {
  // Get API key from environment variable
  const apiKey = process.env.AGENTBAY_API_KEY;
  if (!apiKey) {
    console.log("Error: AGENTBAY_API_KEY environment variable not set");
    return;
  }

  try {
    // Initialize AgentBay client
    console.log("Initializing AgentBay client...");
    const agentBay = new AgentBay({ apiKey });

    // Create a session with browser image
    console.log("Creating a new session...");
    // const params :CreateSessionParams = {
    //   imageId:'browser_latest',
    // }
    const params = new CreateSessionParamsClass({
      imageId:'browser_latest',
    })
    const sessionResult = await agentBay.create(params);

    if (!sessionResult.success) {
      console.log("Failed to create session");
      return;
    }

    const session = sessionResult.session;
    console.log(`Session created with ID: ${session.sessionId}`);

    // Initialize browser
    console.log("Initializing browser...");
    // Note: persistentPath is currently ignored by the SDK implementation
    const initialized = await session.browser.initializeAsync({});
    if (!initialized) {
      console.log("Failed to initialize browser");
      return;
    }

    console.log("Browser initialized successfully");

    // Get browser endpoint for external connections (like Playwright)
    const endpointUrl = await session.browser.getEndpointUrl();
    console.log("Browser endpoint URL:", endpointUrl);

    // Example: Mock page object (in real usage, you'd get this from Playwright)
    const mockPage = {
      url: () => "https://example.com",
      title: () => "Example Domain",
    };

    // Example 1: Perform an action
    console.log("\n--- Example 1: Performing an action ---");
    try {
      const actOptions: ActOptions = {
        action: "Click the 'More information...' link",
        timeout: 5,
      };

      const actResult = await session.browser.operator.act(actOptions, mockPage);
      console.log("Action result:", {
        success: actResult.success,
        message: actResult.message,
      });
    } catch (error) {
      console.log("Action failed:", error);
    }

    // Example 2: Observe elements on the page
    console.log("\n--- Example 2: Observing page elements ---");
    try {
      const observeOptions: ObserveOptions = {
        instruction: "Find all links and buttons on the page",
      };

      const [observeSuccess, observations] =
        await session.browser.operator.observe(observeOptions, mockPage);
      console.log("Observe success:", observeSuccess);
      console.log("Number of observations:", observations.length);

      observations.forEach((obs: any, index: number) => {
        console.log(`Observation ${index + 1}:`, {
          selector: obs.selector,
          description: obs.description,
          method: obs.method,
        });
      });
    } catch (error) {
      console.log("Observation failed:", error);
    }

    // Example 3: Extract structured data from the page
    console.log("\n--- Example 3: Extracting structured data ---");
    try {
      const extractOptions: ExtractOptions<typeof PageInfoSchema> = {
        instruction: "Extract the page title and URL",
        schema: PageInfoSchema,
        use_text_extract: false,
      };

      const [extractSuccess, extracted] = await session.browser.operator.extract(
        extractOptions,
        mockPage
      );
      console.log("Extract success:", extractSuccess);

      if (extractSuccess && extracted) {
        console.log("Extracted data:", {
          title: extracted.title,
          url: extracted.url,
        });
      }
    } catch (error) {
      console.log("Extraction failed:", error);
    }

    // Clean up
    console.log("\n--- Cleanup ---");
    console.log("Browser session completed successfully");
  } catch (error) {
    console.error("Error in main function:", error);
  }
}

if (require.main === module) {
  main().catch(console.error);
}