import {
  createWebSearchProviderContractFields,
  mergeScopedSearchConfig,
  resolveProviderWebSearchPluginConfig,
  type WebSearchProviderPlugin,
  type WebSearchProviderToolDefinition,
} from "openclaw/plugin-sdk/provider-web-search-config-contract";
import { resolvePerplexityRuntimeTransport } from "./perplexity-web-search-provider.shared.js";

const PERPLEXITY_CREDENTIAL_PATH = "plugins.entries.perplexity.config.webSearch.apiKey";

function isRecord(value: unknown): value is Record<string, unknown> {
  return typeof value === "object" && value !== null && !Array.isArray(value);
}

function createPerplexityParameters(transport?: string): Record<string, unknown> {
  const properties: Record<string, unknown> = {
    query: { type: "string", description: "Search query string." },
    count: {
      type: "number",
      description: "Number of results to return (1-10).",
      minimum: 1,
      maximum: 10,
    },
    freshness: {
      type: "string",
      description: "Filter by time: 'day' (24h), 'week', 'month', or 'year'.",
    },
  };

  if (transport !== "chat_completions") {
    properties.country = {
      type: "string",
      description: "Native Perplexity Search API only. 2-letter country code.",
    };
    properties.language = {
      type: "string",
      description: "Native Perplexity Search API only. ISO 639-1 language code.",
    };
    properties.date_after = {
      type: "string",
      description:
        "Native Perplexity Search API only. Only results published after this date (YYYY-MM-DD).",
    };
    properties.date_before = {
      type: "string",
      description:
        "Native Perplexity Search API only. Only results published before this date (YYYY-MM-DD).",
    };
    properties.domain_filter = {
      type: "array",
      items: { type: "string" },
      description: "Native Perplexity Search API only. Domain filter (max 20).",
    };
    properties.max_tokens = {
      type: "number",
      description: "Native Perplexity Search API only. Total content budget across all results.",
      minimum: 1,
      maximum: 1000000,
    };
    properties.max_tokens_per_page = {
      type: "number",
      description: "Native Perplexity Search API only. Max tokens extracted per page.",
      minimum: 1,
    };
  }

  return {
    type: "object",
    properties,
    required: ["query"],
  };
}

function hasPerplexityLegacyOverride(searchConfig?: Record<string, unknown>): boolean {
  const perplexity = isRecord(searchConfig?.perplexity) ? searchConfig.perplexity : undefined;
  return (
    (typeof perplexity?.baseUrl === "string" && perplexity.baseUrl.trim().length > 0) ||
    (typeof perplexity?.model === "string" && perplexity.model.trim().length > 0)
  );
}

function createPerplexityToolDefinition(
  searchConfig?: Record<string, unknown>,
  runtimeTransport?: string,
): WebSearchProviderToolDefinition {
  const schemaTransport =
    runtimeTransport ??
    (hasPerplexityLegacyOverride(searchConfig) ? "chat_completions" : undefined);

  return {
    description:
      schemaTransport === "chat_completions"
        ? "Search the web using Perplexity Sonar via Perplexity/OpenRouter chat completions. Returns AI-synthesized answers with citations from web-grounded search."
        : "Search the web using Perplexity. Runtime routing decides between native Search API and Sonar chat-completions compatibility. Structured filters are available on the native Search API path.",
    parameters: createPerplexityParameters(schemaTransport),
    execute: async (args) => {
      const { executePerplexitySearch } =
        await import("./perplexity-web-search-provider.runtime.js");
      return await executePerplexitySearch(args, searchConfig);
    },
  };
}

export function createPerplexityWebSearchProvider(): WebSearchProviderPlugin {
  return {
    id: "perplexity",
    label: "Perplexity Search",
    hint: "Requires Perplexity API key or OpenRouter API key · structured results",
    onboardingScopes: ["text-inference"],
    credentialLabel: "Perplexity API key",
    envVars: ["PERPLEXITY_API_KEY", "OPENROUTER_API_KEY"],
    placeholder: "pplx-...",
    signupUrl: "https://www.perplexity.ai/settings/api",
    docsUrl: "https://docs.openclaw.ai/perplexity",
    autoDetectOrder: 50,
    credentialPath: PERPLEXITY_CREDENTIAL_PATH,
    ...createWebSearchProviderContractFields({
      credentialPath: PERPLEXITY_CREDENTIAL_PATH,
      searchCredential: { type: "scoped", scopeId: "perplexity" },
      configuredCredential: { pluginId: "perplexity" },
    }),
    resolveRuntimeMetadata: (ctx) => ({
      perplexityTransport: resolvePerplexityRuntimeTransport({
        searchConfig: mergeScopedSearchConfig(
          ctx.searchConfig,
          "perplexity",
          resolveProviderWebSearchPluginConfig(ctx.config, "perplexity"),
        ),
        resolvedKey: ctx.resolvedCredential?.value,
        keySource: ctx.resolvedCredential?.source ?? "missing",
        fallbackEnvVar: ctx.resolvedCredential?.fallbackEnvVar,
      }),
    }),
    createTool: (ctx) =>
      createPerplexityToolDefinition(
        mergeScopedSearchConfig(
          ctx.searchConfig,
          "perplexity",
          resolveProviderWebSearchPluginConfig(ctx.config, "perplexity"),
        ),
        ctx.runtimeMetadata?.perplexityTransport,
      ),
  };
}