import { fetchOllamaWeb } from "./client.js";
import { getMissingApiKeyMessage, type OllamaSearchConfig } from "./config.js";
import { formatFetchResultWithMetadata, type FetchTruncationMetadata } from "./format.js";
import { normalizeWebFetchResponse, type NormalizedFetchResponse } from "./normalize.js";
import {
  registerFetchRetrieval as registerFetchRetrievalDefault,
  type FetchRetrievalRecord,
  type RegisterFetchRetrievalOptions,
} from "./retrieval.js";

export interface RunOllamaWebFetchOptions {
  config: OllamaSearchConfig;
  signal?: AbortSignal;
  fetchImpl?: typeof fetch;
  registerFetchRetrieval?: (payload: NormalizedFetchResponse, options?: RegisterFetchRetrievalOptions) => FetchRetrievalRecord;
}

export interface FetchResultDetails extends FetchRetrievalRecord {
  truncated: boolean;
  maxOutputChars: number;
  targets: FetchTruncationMetadata["targets"];
}

export interface RunOllamaWebFetchResult {
  formatted: string;
  normalized: FetchResultDetails;
}

export async function runOllamaWebFetch(url: string, options: RunOllamaWebFetchOptions): Promise<RunOllamaWebFetchResult> {
  const trimmedUrl = url.trim();
  if (!trimmedUrl) {
    throw new Error("Fetch URL must not be empty.");
  }

  if (!options.config.apiKey) {
    throw new Error(getMissingApiKeyMessage());
  }

  const raw = await fetchOllamaWeb({
    endpoint: options.config.fetchEndpoint,
    apiKey: options.config.apiKey,
    url: trimmedUrl,
    signal: options.signal,
    fetchImpl: options.fetchImpl,
  });

  const registerFetchRetrieval = options.registerFetchRetrieval ?? registerFetchRetrievalDefault;
  const normalizedPayload = normalizeWebFetchResponse(raw);
  const normalized = registerFetchRetrieval(normalizedPayload, { sourceUrl: trimmedUrl });
  const formattedResult = formatFetchResultWithMetadata(normalized, {
    maxOutputChars: options.config.maxOutputChars,
    fullContentRef: normalized.fullContentRef,
    recommendationThreshold: options.config.maxOutputChars,
  });

  return {
    formatted: formattedResult.text,
    normalized: {
      ...normalized,
      truncated: formattedResult.truncation.truncated,
      maxOutputChars: formattedResult.truncation.maxOutputChars,
      targets: formattedResult.truncation.targets,
    },
  };
}