import { DownloadIcon, FileTextIcon, LoaderIcon, FileTypeIcon } from "lucide-react";
import {
  useCallback,
  useEffect,
  useMemo,
  useState,
  type HTMLAttributes,
} from "react";
import { toast } from "sonner";
import { Streamdown } from "streamdown";

import {
  Artifact,
  ArtifactAction,
  ArtifactActions,
  ArtifactContent,
  ArtifactHeader,
  ArtifactTitle,
} from "@/components/ai-elements/artifact";
import {
  DropdownMenu,
  DropdownMenuContent,
  DropdownMenuItem,
  DropdownMenuTrigger,
} from "@/components/ui/dropdown-menu";
import { DropdownSelector } from "@/components/ui/dropdown-selector";
import { ToggleGroup, ToggleGroupItem } from "@/components/ui/toggle-group";
import { CodeEditor } from "@/components/workspace/code-editor";
import { useArtifactContent } from "@/core/artifacts/hooks";
import { urlOfArtifact } from "@/core/artifacts/utils";
import { useI18n } from "@/core/i18n/hooks";
import { POST_MESSAGE_TYPES, sendToParent } from "@/core/iframe-messages";
import { installSkill } from "@/core/skills/api";
import { streamdownPlugins } from "@/core/streamdown";
import { checkCodeFile, getFileName } from "@/core/utils/files";
import { useMarkdownDownload } from "@/core/utils/markdown-download";
import { cn, copyToClipboard, truncateMiddle } from "@/lib/utils";

import { CitationLink } from "../citations/citation-link";

import { useArtifacts } from "./context";

export function ArtifactFileDetail({
  className,
  filepath: filepathFromProps,
  threadId,
}: {
  className?: string;
  filepath: string;
  threadId: string;
}) {
  const { t } = useI18n();
  const { artifacts, setOpen, select, fullscreen, setFullscreen } =
    useArtifacts();
  const isWriteFile = useMemo(() => {
    return filepathFromProps.startsWith("write-file:");
  }, [filepathFromProps]);
  const filepath = useMemo(() => {
    if (isWriteFile) {
      const url = new URL(filepathFromProps);
      return decodeURIComponent(url.pathname);
    }
    return filepathFromProps;
  }, [filepathFromProps, isWriteFile]);
  const isSkillFile = useMemo(() => {
    return filepath.endsWith(".skill");
  }, [filepath]);
  const { isCodeFile, language } = useMemo(() => {
    if (isWriteFile) {
      let language = checkCodeFile(filepath).language;
      language ??= "text";
      return { isCodeFile: true, language };
    }
    // Treat .skill files as markdown (they contain SKILL.md)
    if (isSkillFile) {
      return { isCodeFile: true, language: "markdown" };
    }
    return checkCodeFile(filepath);
  }, [filepath, isWriteFile, isSkillFile]);
  const previewable = useMemo(() => {
    return (language === "html" && !isWriteFile) || language === "markdown";
  }, [isWriteFile, language]);
  const { content } = useArtifactContent({
    threadId,
    filepath: filepathFromProps,
    enabled: isCodeFile && !isWriteFile,
  });

  const displayContent = content ?? "";

  const artifactOptions = useMemo(() => {
    return (artifacts ?? []).map((artifactPath) => ({
      value: artifactPath,
      label: getFileName(artifactPath),
    }));
  }, [artifacts]);

  const [viewMode, setViewMode] = useState<"code" | "preview">("code");
  const [isInstalling, setIsInstalling] = useState(false);
  const [zoom, setZoom] = useState(80);

  // 获取文件名（不含路径）
  const fileName = useMemo(() => getFileName(filepath), [filepath]);

  // 是否可以转换为docx/pdf（仅markdown文件支持）
  const canConvertToDocxPdf = language === "markdown";

  // 使用 Markdown 下载 hook
  const { isDownloading, downloadAsDocx, downloadAsPdf } = useMarkdownDownload({
    onError: (error, format) => {
      console.error(`Failed to download as ${format}:`, error);
      toast.error(`Failed to download as ${format.toUpperCase()}`);
    },
  });

  // 下载为 DOCX
  const handleDownloadDocx = useCallback(() => {
    if (content) {
      void downloadAsDocx(content, fileName);
    }
  }, [content, fileName, downloadAsDocx]);

  // 下载为 PDF
  const handleDownloadPdf = useCallback(() => {
    if (content) {
      void downloadAsPdf(content, fileName);
    }
  }, [content, fileName, downloadAsPdf]);

  // 全屏切换处理
  const handleFullscreenToggle = useCallback(() => {
    const newFullscreen = !fullscreen;
    setFullscreen(newFullscreen);
    sendToParent({
      type: POST_MESSAGE_TYPES.FULLSCREEN,
      fullscreen: newFullscreen,
    });
  }, [fullscreen, setFullscreen]);

  useEffect(() => {
    if (previewable) {
      setViewMode("preview");
    } else {
      setViewMode("code");
    }
  }, [previewable]);

  const handleInstallSkill = useCallback(async () => {
    if (isInstalling) return;

    setIsInstalling(true);
    try {
      const result = await installSkill({
        thread_id: threadId,
        path: filepath,
      });
      if (result.success) {
        toast.success(result.message);
      } else {
        toast.error(result.message ?? "Failed to install skill");
      }
    } catch (error) {
      console.error("Failed to install skill:", error);
      toast.error("Failed to install skill");
    } finally {
      setIsInstalling(false);
    }
  }, [threadId, filepath, isInstalling]);

  return (
    // 给滚动遮挡头部定位relative
    <Artifact className={cn("relative",className)}>
      <ArtifactHeader>
        <div className="flex items-center justify-start gap-2">
          {previewable && (
            <ToggleGroup
              type="single"
              variant={null}
              size="default"
              className="h-[28px]"
              value={viewMode}
              onValueChange={(value) => {
                if (value) {
                  setViewMode(value as "code" | "preview");
                }
              }}
            >
              <ToggleGroupItem value="code">
                <svg
                  width="18"
                  height="18"
                  viewBox="0 0 18 18"
                  fill="none"
                  xmlns="http://www.w3.org/2000/svg"
                >
                  <path
                    d="M5 6L2 9L5 12"
                    stroke="#150033"
                    strokeLinecap="round"
                    strokeLinejoin="round"
                  />
                  <path
                    d="M11 3L7 15"
                    stroke="#150033"
                    strokeLinecap="round"
                    strokeLinejoin="round"
                  />
                  <path
                    d="M13 6L16 9L13 12"
                    stroke="#150033"
                    strokeLinecap="round"
                    strokeLinejoin="round"
                  />
                </svg>
              </ToggleGroupItem>
              <ToggleGroupItem value="preview">
                <svg
                  xmlns="http://www.w3.org/2000/svg"
                  width="16"
                  height="10"
                  viewBox="0 0 16 10"
                  fill="none"
                >
                  <path
                    d="M8 0.5C10.4943 0.5 12.8473 1.84466 14.792 4.21973C15.1644 4.67466 15.1644 5.32534 14.792 5.78027C12.8473 8.15534 10.4943 9.5 8 9.5C5.50561 9.49989 3.15269 8.15543 1.20801 5.78027C0.835561 5.32534 0.835562 4.67466 1.20801 4.21973C3.15269 1.84457 5.50561 0.500106 8 0.5Z"
                    stroke="#666666"
                  />
                  <circle cx="8" cy="5" r="1.5" stroke="#666666" />
                </svg>
              </ToggleGroupItem>
            </ToggleGroup>
          )}
          {/* 放大缩小选择器 */}
          <ArtifactZoomSelector value={zoom} onChange={setZoom} />
        </div>
        <div className="flex min-w-0 grow items-center justify-center">
          <ArtifactTitle>
            {isWriteFile ? (
              <div className=" w-full text-center overflow-hidden text-ellipsis whitespace-nowrap px-2">{truncateMiddle(getFileName(filepath), 50)}</div>
            ) : (
              <DropdownSelector
                value={filepath}
                options={artifactOptions}
                onChange={select}
              />
            )}
          </ArtifactTitle>
        </div>
        <div className="flex items-center justify-end overflow-hidden">
          <ArtifactActions>
            {isCodeFile && (
              <ArtifactAction
                label={t.clipboard.copyToClipboard}
                disabled={!content}
                onClick={async () => {
                  try {
                    await copyToClipboard(displayContent ?? "");
                    toast.success(t.clipboard.copiedToClipboard);
                  } catch (error) {
                    toast.error("Failed to copy to clipboard");
                    console.error(error);
                  }
                }}
                tooltip={t.clipboard.copyToClipboard}
              >
                <svg
                  width="18"
                  height="18"
                  viewBox="0 0 18 18"
                  fill="none"
                  xmlns="http://www.w3.org/2000/svg"
                >
                  <path
                    d="M6 2H13C14.1046 2 15 2.89543 15 4V13"
                    stroke="#666666"
                    strokeLinecap="round"
                    strokeLinejoin="round"
                  />
                  <rect
                    x="2.5"
                    y="4.5"
                    width="10"
                    height="11"
                    rx="1.5"
                    stroke="#666666"
                  />
                </svg>
              </ArtifactAction>
            )}
            {!isWriteFile && (
              <DropdownMenu>
                <DropdownMenuTrigger asChild>
                  <ArtifactAction
                    label={t.common.download}
                    tooltip={t.common.download}
                  >
                    {isDownloading ? (
                      <LoaderIcon className="size-4 animate-spin" />
                    ) : (
                      <svg
                        width="18"
                        height="18"
                        viewBox="0 0 18 18"
                        fill="none"
                        xmlns="http://www.w3.org/2000/svg"
                      >
                        <path
                          d="M16 9V14C16 15.1046 15.1046 16 14 16H4C2.89543 16 2 15.1046 2 14V9"
                          stroke="#666666"
                          strokeLinecap="round"
                        />
                        <path
                          d="M9 2V13M9 13L5 9M9 13L13 9"
                          stroke="#666666"
                          strokeLinecap="round"
                          strokeLinejoin="round"
                        />
                      </svg>
                    )}
                  </ArtifactAction>
                </DropdownMenuTrigger>
                <DropdownMenuContent align="end" className="min-w-[160px]">
                  <DropdownMenuItem asChild>
                    <a
                      href={urlOfArtifact({
                        filepath,
                        threadId,
                        download: true,
                      })}
                      target="_blank"
                      className="w-full cursor-pointer"
                    >
                      <DownloadIcon className="size-4" />
                      {t.common.downloadOriginal}
                    </a>
                  </DropdownMenuItem>
                  {/* DOCX 和 PDF 导出选项仅对 Markdown 文件显示。 */}
                  {canConvertToDocxPdf && (
                    <>
                      <DropdownMenuItem
                        onClick={handleDownloadDocx}
                        disabled={isDownloading !== null || !content}
                        className="cursor-pointer"
                      >
                        <FileTextIcon className="size-4" />
                        {isDownloading === "docx" ? t.common.loading : t.common.downloadAsDocx}
                      </DropdownMenuItem>
                      <DropdownMenuItem
                        onClick={handleDownloadPdf}
                        disabled={isDownloading !== null || !content}
                        className="cursor-pointer"
                      >
                        <FileTypeIcon className="size-4" />
                        {isDownloading === "pdf" ? t.common.loading : t.common.downloadAsPdf}
                      </DropdownMenuItem>
                    </>
                  )}
                </DropdownMenuContent>
              </DropdownMenu>
            )}
            {/* 全屏按钮 */}
            <ArtifactAction
              label={
                fullscreen ? t.common.closeFullScreen : t.common.fullScreen
              }
              onClick={handleFullscreenToggle}
              tooltip={
                fullscreen ? t.common.closeFullScreen : t.common.fullScreen
              }
            >
              {fullscreen ? (
                <svg
                  width="18"
                  height="18"
                  viewBox="0 0 18 18"
                  fill="none"
                  xmlns="http://www.w3.org/2000/svg"
                >
                  <path
                    d="M6 2V4C6 5.10457 5.10457 6 4 6H2"
                    stroke="currentColor"
                    strokeLinecap="round"
                    strokeLinejoin="round"
                  />
                  <path
                    d="M6 16V14C6 12.8954 5.10457 12 4 12H2"
                    stroke="currentColor"
                    strokeLinecap="round"
                    strokeLinejoin="round"
                  />
                  <path
                    d="M12 2V4C12 5.10457 12.8954 6 14 6H16"
                    stroke="currentColor"
                    strokeLinecap="round"
                    strokeLinejoin="round"
                  />
                  <path
                    d="M12 16V14C12 12.8954 12.8954 12 14 12H16"
                    stroke="currentColor"
                    strokeLinecap="round"
                    strokeLinejoin="round"
                  />
                </svg>
              ) : (
                <svg
                  width="18"
                  height="18"
                  viewBox="0 0 18 18"
                  fill="none"
                  xmlns="http://www.w3.org/2000/svg"
                >
                  <path
                    d="M5.75 15.5H4.5C3.39543 15.5 2.5 14.6046 2.5 13.5V12.25M2.5 5.75V4.5C2.5 3.39543 3.39543 2.5 4.5 2.5H5.75M12.25 2.5H13.5C14.6046 2.5 15.5 3.39543 15.5 4.5V5.75M15.5 12.25V13.5C15.5 14.6046 14.6046 15.5 13.5 15.5H12.25"
                    stroke="currentColor"
                    strokeLinecap="round"
                    strokeLinejoin="round"
                  />
                </svg>
              )}
            </ArtifactAction>
            {!fullscreen && (
              <ArtifactAction
                label={t.common.close}
                onClick={() => setOpen(false)}
                tooltip={t.common.close}
              >
                <svg
                  width="18"
                  height="18"
                  viewBox="0 0 18 18"
                  fill="none"
                  xmlns="http://www.w3.org/2000/svg"
                >
                  <path
                    d="M4 14L14 4M4 4L14 14"
                    stroke="#666666"
                    strokeLinecap="round"
                  />
                </svg>
              </ArtifactAction>
            )}
          </ArtifactActions>
        </div>
      </ArtifactHeader>
      <ArtifactContent className=" rounded-b-[10px] bg-white p-0">
        {/* 遮挡多余的滚动顶部 */}
        <div className="absolute w-[calc(100%-40px)] bg-white z-20 h-5 rounded-t-[10px] top-[57px]"></div>
        {previewable &&
          viewMode === "preview" &&
          (language === "markdown" || language === "html") && (
            <ArtifactFilePreview
              content={displayContent}
              language={language ?? "text"}
              zoom={zoom}
            />
          )}
        {isCodeFile && viewMode === "code" && (
          <CodeEditor
            className="size-full py-[20px] resize-none rounded-none border-none"
            value={displayContent ?? ""}
            zoom={zoom}
            readonly
          />
        )}
        {!isCodeFile && (
          <iframe
            className="size-full"
            src={urlOfArtifact({ filepath, threadId })}
          />
        )}
      </ArtifactContent>
    </Artifact>
  );
}

export function ArtifactFilePreview({
  content,
  language,
  zoom = 100,
}: {
  content: string;
  language: string;
  zoom?: number;
}) {
  const zoomScale = zoom / 100;

  if (language === "markdown") {
    return (
      <div
        className={cn("size-full p-[20px]")}
        style={{ "--zoom-scale": zoomScale } as React.CSSProperties}
      >
        <Streamdown
          className="size-full"
          {...streamdownPlugins}
          components={{ a: CitationLink }}
        >
          {content ?? ""}
        </Streamdown>
      </div>
    );
  }
  if (language === "html") {
    return (
      <iframe
        className="size-full"
        title="Artifact preview"
        srcDoc={content}
        sandbox="allow-scripts allow-forms"
        style={{ zoom: zoomScale }}
      />
    );
  }
  return null;
}

// 缩放比例选项
const ZOOM_LEVELS = [50, 60, 70, 80, 90, 100, 110, 120, 130, 150, 175, 200];

export type ArtifactZoomSelectorProps = Omit<
  HTMLAttributes<HTMLDivElement>,
  "onChange"
> & {
  value?: number;
  onChange?: (value: number) => void;
};

export const ArtifactZoomSelector = ({
  value = 100,
  onChange,
  className,
  ...props
}: ArtifactZoomSelectorProps) => {
  const handleZoomIn = () => {
    const currentIndex = ZOOM_LEVELS.indexOf(value);
    const nextValue = ZOOM_LEVELS[currentIndex + 1];
    if (currentIndex < ZOOM_LEVELS.length - 1 && nextValue !== undefined) {
      onChange?.(nextValue);
    }
  };

  const handleZoomOut = () => {
    const currentIndex = ZOOM_LEVELS.indexOf(value);
    const prevValue = ZOOM_LEVELS[currentIndex - 1];
    if (currentIndex > 0 && prevValue !== undefined) {
      onChange?.(prevValue);
    }
  };

  const canZoomIn = ZOOM_LEVELS.indexOf(value) < ZOOM_LEVELS.length - 1;
  const canZoomOut = ZOOM_LEVELS.indexOf(value) > 0;

  return (
    <div
      className={cn(
        "inline-flex h-[28px] items-center gap-1 rounded-[10px] bg-white backdrop-blur-sm",
        "dark:border-gray-700/50 dark:bg-gray-800/90",
        className,
      )}
      {...props}
    >
      <button
        type="button"
        onClick={handleZoomIn}
        disabled={!canZoomIn}
        className={cn(
          "flex h-full w-10 items-center justify-center rounded py-1 transition-colors",
          "text-gray-400 hover:bg-gray-100 hover:text-gray-600",
          "disabled:cursor-not-allowed disabled:opacity-40 disabled:hover:bg-transparent",
          "dark:text-gray-500 dark:hover:bg-gray-700 dark:hover:text-gray-300",
        )}
        aria-label="放大"
      >
        <svg
          xmlns="http://www.w3.org/2000/svg"
          width="16"
          height="16"
          viewBox="0 0 16 16"
          fill="none"
        >
          <circle cx="7.55558" cy="7.55534" r="6.16667" stroke="#666666" />
          <path
            d="M13.8688 15.4646C14.064 15.6598 14.3806 15.6598 14.5759 15.4646C14.7711 15.2693 14.7711 14.9527 14.5759 14.7574L14.2223 15.111L13.8688 15.4646ZM14.2223 15.111L14.5759 14.7574L11.9092 12.0908L11.5557 12.4443L11.2021 12.7979L13.8688 15.4646L14.2223 15.111Z"
            fill="#666666"
          />
          <path
            d="M5.33325 7.5H9.7777M7.55547 5V10"
            stroke="#666666"
            strokeLinecap="round"
            strokeLinejoin="round"
          />
        </svg>
      </button>
      <span
        className={cn(
          "min-w-[36px] text-center text-xs font-medium text-gray-600",
          "dark:text-gray-300",
        )}
      >
        {value}%
      </span>
      <button
        type="button"
        onClick={handleZoomOut}
        disabled={!canZoomOut}
        className={cn(
          "flex h-full w-10 items-center justify-center rounded transition-colors",
          "text-gray-400 hover:bg-gray-100 hover:text-gray-600",
          "disabled:cursor-not-allowed disabled:opacity-40 disabled:hover:bg-transparent",
          "dark:text-gray-500 dark:hover:bg-gray-700 dark:hover:text-gray-300",
        )}
        aria-label="缩小"
      >
        <svg
          xmlns="http://www.w3.org/2000/svg"
          width="16"
          height="16"
          viewBox="0 0 16 16"
          fill="none"
        >
          <circle cx="7.55558" cy="7.55534" r="6.16667" stroke="#666666" />
          <path
            d="M13.8688 15.4646C14.064 15.6598 14.3806 15.6598 14.5759 15.4646C14.7711 15.2693 14.7711 14.9527 14.5759 14.7574L14.2223 15.111L13.8688 15.4646ZM14.2223 15.111L14.5759 14.7574L11.9092 12.0908L11.5557 12.4443L11.2021 12.7979L13.8688 15.4646L14.2223 15.111Z"
            fill="#666666"
          />
          <path
            d="M4.99927 7.5H9.99927"
            stroke="#666666"
            strokeLinecap="round"
            strokeLinejoin="round"
          />
        </svg>
      </button>
    </div>
  );
};