import IconsResolver from "unplugin-icons/resolver";
import Components from "unplugin-vue-components/vite";
import { ArcoResolver } from "unplugin-vue-components/resolvers";
import type { PluginOption } from "vite";

export function setupAutoImport(viteEnv: Env.ImportMeta) {
  // 从环境变量中提取图标相关的前缀
  const { VITE_ICON_PREFIX, VITE_ICON_LOCAL_PREFIX } = viteEnv;

  // 从本地图标前缀中提取图标集合名称
  const collectionName = VITE_ICON_LOCAL_PREFIX.replace(`${VITE_ICON_PREFIX}-`, "");
  const plugins: PluginOption[] = [
    Components({
      // 自动导入的组件类型声明文件
      dts: "src/types/auto-import.d.ts",
      types: [
        {
          from: "vue-router",
          names: ["RouterLink", "RouterView"]
        }
      ],

      // 配置图标解析器
      resolvers: [
        IconsResolver({
          customCollections: [collectionName], // 本地图标集合
          componentPrefix: VITE_ICON_PREFIX // 图标组件的前缀
        }),
        ArcoResolver()
      ]
    })
  ];
  return plugins;
}