import {z} from 'zod'

export type ObjectIDFilter = string | {_type: 'EqualTo', value: string} | {_type: 'NotEqual', value: string} | {_type: 'IsNull'} | {_type: 'NotNull'} | {_type: 'EqualToOrNull', value: string} | {_type: 'Includes', value: Array<string>};

export type DateFilter = string | {_type: 'EqualTo', value: string} | {_type: 'NotEqual', value: string} | {_type: 'IsNull'} | {_type: 'NotNull'} | {_type: 'EqualToOrNull', value: string} | {_type: 'GreaterThan', value: string} | {_type: 'LessThan', value: string} | {_type: 'GreaterThanOrEqual', value: string} | {_type: 'LessThanOrEqual', value: string} | {_type: 'InBetween', startValue: string, endValue: string};

export type ShortTextFilter = string | {_type: 'EqualTo', value: string} | {_type: 'NotEqual', value: string} | {_type: 'IsNull'} | {_type: 'NotNull'} | {_type: 'EqualToOrNull', value: string} | {_type: 'Search', value: string} | {_type: 'Includes', value: Array<string>};

export type NumberFilter = number | {_type: 'EqualTo', value: number} | {_type: 'NotEqual', value: number} | {_type: 'IsNull'} | {_type: 'NotNull'} | {_type: 'EqualToOrNull', value: number} | {_type: 'GreaterThan', value: number} | {_type: 'LessThan', value: number} | {_type: 'GreaterThanOrEqual', value: number} | {_type: 'LessThanOrEqual', value: number} | {_type: 'InBetween', startValue: number, endValue: number};

export type JSONFilter = {_type: 'IsNull'} | {_type: 'NotNull'} | {[key: string]: string | number | boolean | {_type: 'EqualTo', value: string | number | boolean} | {_type: 'NotEqual', value: string | number | boolean} | {_type: 'IsNull'} | {_type: 'NotNull'} | {_type: 'EqualToOrNull', value: string | number | boolean} | {_type: 'Search', value: string} | {_type: 'Includes', value: Array<string | number | boolean>} | {_type: 'GreaterThan', value: string | number | boolean} | {_type: 'LessThan', value: string | number | boolean} | {_type: 'GreaterThanOrEqual', value: string | number | boolean} | {_type: 'LessThanOrEqual', value: string | number | boolean} | {_type: 'InBetween', startValue: string | number | boolean, endValue: string | number | boolean}};

export type BooleanFilter = boolean | {_type: 'EqualTo', value: boolean} | {_type: 'NotEqual', value: boolean} | {_type: 'IsNull'} | {_type: 'NotNull'} | {_type: 'EqualToOrNull', value: boolean};

export type LongTextFilter = string | {_type: 'EqualTo', value: string} | {_type: 'NotEqual', value: string} | {_type: 'IsNull'} | {_type: 'NotNull'} | {_type: 'EqualToOrNull', value: string} | {_type: 'Search', value: string} | {_type: 'Includes', value: Array<string>};

export type AIRunQuerySchema = {_id?: ObjectIDFilter | undefined, createdAt?: DateFilter | undefined, updatedAt?: DateFilter | undefined, deletedAt?: DateFilter | undefined, projectId?: ObjectIDFilter | undefined, runType?: ShortTextFilter | undefined, codeFixTaskType?: ShortTextFilter | undefined, taskNumber?: NumberFilter | undefined, status?: ShortTextFilter | undefined, codeFixRecommendation?: ShortTextFilter | undefined, userId?: ObjectIDFilter | undefined, conversationId?: ObjectIDFilter | undefined, triggeredByIncidentId?: ObjectIDFilter | undefined, triggeredByAlertId?: ObjectIDFilter | undefined, followUpForAlertId?: ObjectIDFilter | undefined, ruleWaitMinutes?: NumberFilter | undefined, llmProviderId?: ObjectIDFilter | undefined, triggeredByTelemetryExceptionId?: ObjectIDFilter | undefined, triggeredByAiInsightId?: ObjectIDFilter | undefined, triggeredBySecurityCaseId?: ObjectIDFilter | undefined, triggeredByAutoRemediationSuggestionId?: ObjectIDFilter | undefined, monitorId?: ObjectIDFilter | undefined, aiAgentId?: ObjectIDFilter | undefined, attemptCount?: NumberFilter | undefined, handedBackCount?: NumberFilter | undefined, startedAt?: DateFilter | undefined, completedAt?: DateFilter | undefined, lastHeartbeatAt?: DateFilter | undefined, llmCallCount?: NumberFilter | undefined, toolCallCount?: NumberFilter | undefined, totalTokens?: NumberFilter | undefined, totalCostInUSDCents?: NumberFilter | undefined, egressManifest?: JSONFilter | undefined, codeRepositoryId?: ObjectIDFilter | undefined, codeTaskId?: ObjectIDFilter | undefined, needsUserInput?: BooleanFilter | undefined, errorMessage?: LongTextFilter | undefined, analysisTldr?: LongTextFilter | undefined, humanVerdict?: ShortTextFilter | undefined, humanVerdictAt?: DateFilter | undefined, humanVerdictByUserId?: ObjectIDFilter | undefined, autoGrade?: ShortTextFilter | undefined, autoGradeAt?: DateFilter | undefined};

export type AIRunSelectSchema = {_id?: boolean | undefined, createdAt?: boolean | undefined, updatedAt?: boolean | undefined, deletedAt?: boolean | undefined, projectId?: boolean | undefined, runType?: boolean | undefined, codeFixTaskType?: boolean | undefined, taskNumber?: boolean | undefined, status?: boolean | undefined, codeFixRecommendation?: boolean | undefined, userId?: boolean | undefined, conversationId?: boolean | undefined, triggeredByIncidentId?: boolean | undefined, triggeredByAlertId?: boolean | undefined, followUpForAlertId?: boolean | undefined, ruleWaitMinutes?: boolean | undefined, llmProviderId?: boolean | undefined, triggeredByTelemetryExceptionId?: boolean | undefined, triggeredByAiInsightId?: boolean | undefined, triggeredBySecurityCaseId?: boolean | undefined, triggeredByAutoRemediationSuggestionId?: boolean | undefined, monitorId?: boolean | undefined, aiAgentId?: boolean | undefined, attemptCount?: boolean | undefined, handedBackCount?: boolean | undefined, startedAt?: boolean | undefined, completedAt?: boolean | undefined, lastHeartbeatAt?: boolean | undefined, llmCallCount?: boolean | undefined, toolCallCount?: boolean | undefined, totalTokens?: boolean | undefined, totalCostInUSDCents?: boolean | undefined, egressManifest?: boolean | undefined, codeRepositoryId?: boolean | undefined, codeTaskId?: boolean | undefined, needsUserInput?: boolean | undefined, errorMessage?: boolean | undefined, analysisTldr?: boolean | undefined, humanVerdict?: boolean | undefined, humanVerdictAt?: boolean | undefined, humanVerdictByUserId?: boolean | undefined, autoGrade?: boolean | undefined, autoGradeAt?: boolean | undefined};

export type SortOrder = 'ASC' | 'DESC';

export type AIRunSortSchema = {_id?: SortOrder | undefined, createdAt?: SortOrder | undefined, updatedAt?: SortOrder | undefined, deletedAt?: SortOrder | undefined, projectId?: SortOrder | undefined, runType?: SortOrder | undefined, codeFixTaskType?: SortOrder | undefined, taskNumber?: SortOrder | undefined, status?: SortOrder | undefined, codeFixRecommendation?: SortOrder | undefined, userId?: SortOrder | undefined, conversationId?: SortOrder | undefined, triggeredByIncidentId?: SortOrder | undefined, triggeredByAlertId?: SortOrder | undefined, followUpForAlertId?: SortOrder | undefined, ruleWaitMinutes?: SortOrder | undefined, llmProviderId?: SortOrder | undefined, triggeredByTelemetryExceptionId?: SortOrder | undefined, triggeredByAiInsightId?: SortOrder | undefined, triggeredBySecurityCaseId?: SortOrder | undefined, triggeredByAutoRemediationSuggestionId?: SortOrder | undefined, monitorId?: SortOrder | undefined, aiAgentId?: SortOrder | undefined, attemptCount?: SortOrder | undefined, handedBackCount?: SortOrder | undefined, startedAt?: SortOrder | undefined, completedAt?: SortOrder | undefined, lastHeartbeatAt?: SortOrder | undefined, llmCallCount?: SortOrder | undefined, toolCallCount?: SortOrder | undefined, totalTokens?: SortOrder | undefined, totalCostInUSDCents?: SortOrder | undefined, codeRepositoryId?: SortOrder | undefined, codeTaskId?: SortOrder | undefined, needsUserInput?: SortOrder | undefined, errorMessage?: SortOrder | undefined, analysisTldr?: SortOrder | undefined, humanVerdict?: SortOrder | undefined, humanVerdictAt?: SortOrder | undefined, humanVerdictByUserId?: SortOrder | undefined, autoGrade?: SortOrder | undefined, autoGradeAt?: SortOrder | undefined};

export type CreateApiAiRunGetListArgs = {body: {query?: AIRunQuerySchema | undefined, select?: AIRunSelectSchema | undefined, sort?: AIRunSortSchema | undefined, limit?: number | undefined, skip?: number | undefined}};

export type CreateApiAiRunGetListOptions = {baseUrl?: string | undefined, ApiKey?: string | undefined, fetch?: typeof fetch | undefined};

export const aIRun = z.object({_id: z.string().optional(), createdAt: z.object({}).optional(), updatedAt: z.object({}).optional(), deletedAt: z.object({}).optional(), version: z.number().optional(), projectId: z.string(), runType: z.enum(["Chat", "Investigation", "CodeFix", "RemediationPlan", "RemediationExecution"]), codeFixTaskType: z.enum(["FixException", "WriteRegressionTest", "ImproveExceptionHandling", "ImproveInstrumentation", "FixPerformance", "ImproveLogging", "ImproveTracing", "FixFromIncident", "GitHubIssueFix", "GitHubPullRequestRevision", "GitHubPullRequestReview", "GitHubPullRequestProductionReview", "CodeRequest"]).optional(), taskNumber: z.number().optional(), status: z.enum(["Queued", "Running", "WaitingForApproval", "Completed", "NoFixFound", "Error", "Cancelled", "Stale"]), codeFixRecommendation: z.enum(["Pending", "Recommended", "NotRecommended"]), userId: z.string().optional(), conversationId: z.string().optional(), triggeredByIncidentId: z.string().optional(), triggeredByAlertId: z.string().optional(), followUpForAlertId: z.string().optional(), ruleWaitMinutes: z.number().optional(), llmProviderId: z.string().optional(), triggeredByTelemetryExceptionId: z.string().optional(), triggeredByAiInsightId: z.string().optional(), triggeredBySecurityCaseId: z.string().optional(), triggeredByAutoRemediationSuggestionId: z.string().optional(), monitorId: z.string().optional(), aiAgentId: z.string().optional(), attemptCount: z.number(), handedBackCount: z.number(), startedAt: z.object({}).optional(), completedAt: z.object({}).optional(), lastHeartbeatAt: z.object({}).optional(), llmCallCount: z.number().optional(), toolCallCount: z.number().optional(), totalTokens: z.number().optional(), totalCostInUSDCents: z.number().optional(), egressManifest: z.object({}).optional(), codeRepositoryId: z.string().optional(), codeTaskId: z.string().optional(), needsUserInput: z.boolean().optional(), errorMessage: z.string().optional(), analysisTldr: z.string().optional(), humanVerdict: z.enum(["Confirmed", "Rejected"]).optional(), humanVerdictAt: z.object({}).optional(), humanVerdictByUserId: z.string().optional(), autoGrade: z.enum(["Match", "Partial", "Mismatch"]).optional(), autoGradeAt: z.object({}).optional()});

export const createApiAiRunGetListResponse = z.object({data: z.array(aIRun).optional(), count: z.number().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
};

/**
 * List AI Run
 * 
 * POST /ai-run/get-list
 */
export const createApiAiRunGetList = async (args: CreateApiAiRunGetListArgs, options?: CreateApiAiRunGetListOptions) => {
const {baseUrl = "https://oneuptime.com/api", fetch: fetchFn = fetch, ApiKey} = options ?? {}

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

if (ApiKey !== undefined) headers.set("APIKey", ApiKey)

const url = `${baseUrl.replace(/\/+$/, '')}/ai-run/get-list${query.size > 0 ? `?${query}` : ''}`

const res = await fetchFn(url, {method: "POST", headers, body: JSON.stringify(args.body)})

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

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