export interface AgentToolImageMarkdownItem {
  originalUrl: string;
  thumbnailUrl?: string | null;
  imageSize?: string;
}

export function prependImageModelBadge(
  text: string,
  modelDisplayName?: string | null,
): string {
  const normalizedText = String(text || "").trim();
  if (!normalizedText) return normalizedText;

  const normalizedModelName = String(modelDisplayName || "").trim();
  if (!normalizedModelName) return normalizedText;

  if (
    /^\s*[◈◇◆🔹⬡✦•·]\s*[^\n]+/im.test(normalizedText) ||
    /^\s*(?:Routed to|Using model|使用模型|模型)\s*[:：]/im.test(normalizedText)
  ) {
    return normalizedText;
  }

  return `◈ ${normalizedModelName}\n\n${normalizedText}`;
}

export function buildAgentToolImageMarkdown(params: AgentToolImageMarkdownItem): string {
  const originalUrl = String(params.originalUrl || "").trim();
  const thumbnailUrl = String(params.thumbnailUrl || originalUrl).trim() || originalUrl;
  const imageSize = String(params.imageSize || "2K").trim() || "2K";
  return `![thumb|||${originalUrl}|||${imageSize}](${thumbnailUrl})`;
}

export function buildAgentToolImagesMarkdown(items: AgentToolImageMarkdownItem[]) {
  return items
    .map((item) =>
      buildAgentToolImageMarkdown({
        originalUrl: item.originalUrl,
        thumbnailUrl: item.thumbnailUrl,
        imageSize: item.imageSize,
      }),
    )
    .join("\n");
}

export function buildAgentToolVideoMarkdown(videoUrl: string) {
  const normalizedUrl = String(videoUrl || "").trim();
  if (!normalizedUrl) return "";
  return [
    `<video src="${normalizedUrl}" controls style="max-width: 100%; border-radius: 8px;"></video>`,
    `[下载视频](${normalizedUrl})`,
  ].join("\n\n");
}