import type { Message } from "./GlobalChatbox.types";

export const createId = () =>
  `${Date.now()}-${Math.random().toString(36).slice(2, 8)}`;
export const PRESET_PROMPTS = [
  "分析当前管网中的水力瓶颈管道，并给出改造建议。",
  "供水服务分区分析。",
  "帮我分析当前管网压力异常点，并按风险等级排序。",
  "帮我生成一份今日运行简报，包含问题、原因和建议。",
  "查询关键 SCADA 点位最近 24 小时的异常波动。",
  "排查当前管网爆管风险，并说明优先处置建议。",
];

export const stripMarkdown = (md: string): string =>
  md
    .replace(/```[\s\S]*?```/g, "")
    .replace(/`([^`]+)`/g, "$1")
    .replace(/!\[.*?\]\(.*?\)/g, "")
    .replace(/\[([^\]]+)\]\(.*?\)/g, "$1")
    .replace(/#{1,6}\s+/g, "")
    .replace(/\*\*\*(.+?)\*\*\*/g, "$1")
    .replace(/\*\*(.+?)\*\*/g, "$1")
    .replace(/\*(.+?)\*/g, "$1")
    .replace(/~~(.+?)~~/g, "$1")
    .replace(/>\s+/g, "")
    .replace(/[-*+]\s+/g, "")
    .replace(/\d+\.\s+/g, "")
    .replace(/\n{2,}/g, "\n")
    .replace(/<[^>]+>/g, "")
    .trim();

export const cloneMessage = (message: Message): Message => ({
  ...message,
  progress: message.progress ? [...message.progress] : undefined,
  artifacts: message.artifacts ? [...message.artifacts] : undefined,
});

export const cloneMessages = (messages: Message[]) => messages.map(cloneMessage);