import type { SourcegraphGraphQLAPIClient } from '../sourcegraph-api/graphql'
import { isError } from '../utils'

import type { Attribution, Guardrails } from '.'
import { ClientConfigSingleton } from '../sourcegraph-api/graphql/client'

// 10s timeout is enough to serve most attribution requests.
// It's a better user experience for chat attribution to wait
// a few seconds more and get attribution result.
const defaultTimeoutSeconds = 10

/**
 * This defines the user controllable configuration. Note: enablement is
 * controlled serverside.
 */
export interface GuardrailsClientConfig {
    experimentalGuardrailsTimeoutSeconds: number | undefined
}

export class SourcegraphGuardrailsClient implements Guardrails {
    constructor(
        private client: SourcegraphGraphQLAPIClient,
        private config: GuardrailsClientConfig
    ) {}

    public onConfigurationChange(newConfig: GuardrailsClientConfig): void {
        this.config = newConfig
    }

    public async searchAttribution(snippet: string): Promise<Attribution | Error> {
        // Short-circuit attribution search if turned off in site config.
        const clientConfig = await ClientConfigSingleton.getInstance().getConfig()
        if (!clientConfig?.attributionEnabled) {
            return new Error('Attribution search is turned off.')
        }

        const timeout =
            (this.config.experimentalGuardrailsTimeoutSeconds ?? defaultTimeoutSeconds) * 1000

        const result = await this.client.searchAttribution(snippet, timeout)

        if (isError(result)) {
            return result
        }

        return {
            limitHit: result.limitHit,
            repositories: result.nodes.map(repo => ({ name: repo.repositoryName })),
        }
    }
}