import type { InferenceClient } from "@karakeep/shared/inference";
import { buildTextPrompt } from "@karakeep/shared/prompts";

import { inferTags } from "./inferenceClient";
import type { Bookmark } from "./types";

export async function extractBookmarkContent(
  bookmark: Bookmark,
): Promise<string> {
  if (bookmark.content.type === "link") {
    const parts = [];

    if (bookmark.content.url) {
      parts.push(`URL: ${bookmark.content.url}`);
    }

    if (bookmark.title) {
      parts.push(`Title: ${bookmark.title}`);
    }

    if (bookmark.content.description) {
      parts.push(`Description: ${bookmark.content.description}`);
    }

    if (bookmark.content.htmlContent) {
      parts.push(`Content: ${bookmark.content.htmlContent}`);
    }

    return parts.join("\n");
  }

  if (bookmark.content.type === "text" && bookmark.content.text) {
    return bookmark.content.text;
  }

  return "";
}

export async function runTaggingForModel(
  bookmark: Bookmark,
  inferenceClient: InferenceClient,
  lang: string = "english",
  contextLength: number = 8000,
): Promise<string[]> {
  const content = await extractBookmarkContent(bookmark);

  if (!content) {
    return [];
  }

  try {
    // Use the shared prompt builder with empty custom prompts and default tag style
    const prompt = await buildTextPrompt(
      lang,
      [], // No custom prompts for comparison tool
      content,
      contextLength,
      "as-generated", // Use tags as generated by the model
    );

    const tags = await inferTags(inferenceClient, prompt);
    return tags;
  } catch (error) {
    throw new Error(
      `Failed to generate tags: ${error instanceof Error ? error.message : String(error)}`,
    );
  }
}