import {z} from 'zod'

export type SandboxMCPServer = 'jira' | 'github' | 'salesforce' | 'hubspot' | 'notion' | 'airtable' | 'linear' | 'asana' | 'google_sheets' | 'google_drive' | 'google_docs' | 'gmail' | 'google_calendar' | 'google_forms' | 'clickup' | 'close' | 'monday' | 'motion' | 'onedrive' | 'microsoft_teams' | 'outlook_mail' | 'cal.com' | 'quickbooks' | 'moneybird' | 'dropbox' | 'shopify' | 'woocommerce' | 'outlook_calendar' | 'resend' | 'wordpress' | 'mem0' | 'supabase' | 'slack' | 'confluence' | 'discord' | 'snowflake' | 'postgres' | 'mongodb' | 'youtube' | 'googleworkspaceatlas' | 'huggingface' | 'mock_notion' | 'arxiv_latex' | 'calculator' | 'clinicaltrialsgov' | 'fetch' | 'met_museum' | 'open_library' | 'osm' | 'pubmed' | 'us_weather' | 'duckduckgo' | 'whois' | 'wikipedia' | 'scholarly' | 'howtocook' | 'yahoo_finance' | '12306' | 'youtube_transcript' | 'weather' | 'twelvedata' | 'national_parks' | 'lara_translate' | 'e2b' | 'context7' | 'alchemy' | 'weights_and_biases' | 'oxylabs' | 'google_maps' | 'brave_search' | 'exa' | 'serper';

export type GetApiSandboxServerNameSandboxIdArgs = {server_name: SandboxMCPServer, sandbox_id: string};

export type GetApiSandboxServerNameSandboxIdOptions = {baseUrl?: string | undefined, HTTPBearer?: string | undefined, fetch?: typeof fetch | undefined};

export const sandboxMCPServer = z.enum(["jira", "github", "salesforce", "hubspot", "notion", "airtable", "linear", "asana", "google_sheets", "google_drive", "google_docs", "gmail", "google_calendar", "google_forms", "clickup", "close", "monday", "motion", "onedrive", "microsoft_teams", "outlook_mail", "cal.com", "quickbooks", "moneybird", "dropbox", "shopify", "woocommerce", "outlook_calendar", "resend", "wordpress", "mem0", "supabase", "slack", "confluence", "discord", "snowflake", "postgres", "mongodb", "youtube", "googleworkspaceatlas", "huggingface", "mock_notion", "arxiv_latex", "calculator", "clinicaltrialsgov", "fetch", "met_museum", "open_library", "osm", "pubmed", "us_weather", "duckduckgo", "whois", "wikipedia", "scholarly", "howtocook", "yahoo_finance", "12306", "youtube_transcript", "weather", "twelvedata", "national_parks", "lara_translate", "e2b", "context7", "alchemy", "weights_and_biases", "oxylabs", "google_maps", "brave_search", "exa", "serper"]);

export const sandboxStatus = z.enum(["idle", "occupied", "error"]);

export const sandboxInfo = z.object({sandbox_id: z.string(), server_url: z.string().nullable().optional(), server_name: sandboxMCPServer, status: sandboxStatus, benchmark: z.string().nullable().optional(), updated_at: z.string().nullable().optional(), metadata: z.record(z.string(), z.unknown()).nullable().optional(), auth_data: z.record(z.string(), z.unknown()).nullable().optional(), tags: z.array(z.string()).nullable().optional()});

export const validateResponse = async <Output>(
  schema: {
    readonly '~standard': {
      readonly validate: (value: unknown) => ({ readonly value: Output; readonly issues?: undefined } | { readonly issues: ReadonlyArray<{ readonly message: string; readonly path?: ReadonlyArray<PropertyKey | { readonly key: PropertyKey }> | undefined }> }) | Promise<{ readonly value: Output; readonly issues?: undefined } | { readonly issues: ReadonlyArray<{ readonly message: string; readonly path?: ReadonlyArray<PropertyKey | { readonly key: PropertyKey }> | undefined }> }>
    }
  },
  value: unknown
): Promise<Output> => {
  const result = await schema['~standard'].validate(value)

  if (result.issues !== undefined) {
    const detail = result.issues
      .map(({ message, path }) =>
        path?.length
          ? `${path.map((segment) => String(typeof segment === 'object' ? segment.key : segment)).join('.')}: ${message}`
          : message
      )
      .join('; ')

    throw new Error(`Response does not match the schema: ${detail}`, { cause: { issues: result.issues } })
  }

  return result.value
};

/**
 * Get sandbox details
 * 
 * GET /sandbox/{server_name}/{sandbox_id}
 */
export const getApiSandboxServerNameSandboxId = async (args: GetApiSandboxServerNameSandboxIdArgs, options?: GetApiSandboxServerNameSandboxIdOptions) => {
const {baseUrl = "https://api.klavis.ai", fetch: fetchFn = fetch, HTTPBearer} = options ?? {}

const query = new URLSearchParams()
const headers = new Headers()
headers.set('accept', "application/json")

if (HTTPBearer !== undefined) headers.set('authorization', `Bearer ${HTTPBearer}`)

const url = `${baseUrl.replace(/\/+$/, '')}/sandbox/${encodeURIComponent(String(args.server_name))}/${encodeURIComponent(String(args.sandbox_id))}${query.size > 0 ? `?${query}` : ''}`

const res = await fetchFn(url, {method: "GET", headers})

if (!res.ok) throw new Error(`Request failed with status ${res.status}`, {cause: {status: res.status, body: await res.text()}})

return validateResponse(sandboxInfo, await res.json())
};