NapCatQQ/packages/napcat-webui-frontend/src/pages/dashboard/plugin.tsx

import { Button } from '@heroui/button';
import { useEffect, useState } from 'react';
import toast from 'react-hot-toast';
import { IoMdRefresh } from 'react-icons/io';
import { useDisclosure } from '@heroui/modal';

import PageLoading from '@/components/page_loading';
import PluginDisplayCard from '@/components/display_card/plugin_card';
import PluginManager, { PluginItem } from '@/controllers/plugin_manager';
import useDialog from '@/hooks/use-dialog';
import PluginConfigModal from '@/pages/dashboard/plugin_config_modal';

export default function PluginPage () {
  const [plugins, setPlugins] = useState<PluginItem[]>([]);
  const [loading, setLoading] = useState(false);
  const [pluginManagerNotFound, setPluginManagerNotFound] = useState(false);
  const dialog = useDialog();

  const { isOpen, onOpen, onOpenChange } = useDisclosure();
  const [currentPluginName, setCurrentPluginName] = useState<string>('');

  const loadPlugins = async () => {
    setLoading(true);
    setPluginManagerNotFound(false);
    try {
      const listResult = await PluginManager.getPluginList();

      if (listResult.pluginManagerNotFound) {
        setPluginManagerNotFound(true);
        setPlugins([]);
      } else {
        setPlugins(listResult.plugins);
      }
    } catch (e: any) {
      toast.error(e.message);
    } finally {
      setLoading(false);
    }
  };

  useEffect(() => {
    loadPlugins();
  }, []);

  const handleReload = async (name: string) => {
    const loadingToast = toast.loading('重载中...');
    try {
      await PluginManager.reloadPlugin(name);
      toast.success('重载成功', { id: loadingToast });
      loadPlugins();
    } catch (e: any) {
      toast.error(e.message, { id: loadingToast });
    }
  };

  const handleToggle = async (plugin: PluginItem) => {
    const isEnable = plugin.status !== 'active';
    const actionText = isEnable ? '启用' : '禁用';
    const loadingToast = toast.loading(`${actionText}中...`);
    try {
      await PluginManager.setPluginStatus(plugin.name, isEnable, plugin.filename);
      toast.success(`${actionText}成功`, { id: loadingToast });
      loadPlugins();
    } catch (e: any) {
      toast.error(e.message, { id: loadingToast });
    }
  };

  const handleUninstall = async (plugin: PluginItem) => {
    return new Promise<void>((resolve, reject) => {
      dialog.confirm({
        title: '卸载插件',
        content: (
          <div className="flex flex-col gap-2">
            <p>确定要卸载插件「{plugin.name}」吗? 此操作不可恢复。</p>
            <p className="text-small text-default-500">如果插件创建了数据文件，是否一并删除？</p>
          </div>
        ),
        // This 'dialog' utility might not support returning a value from UI interacting.
        // We might need to implement a custom confirmation flow if we want a checkbox.
        // Alternatively, use two buttons? "Uninstall & Clean", "Uninstall Only"?
        // Standard dialog usually has Confirm/Cancel.
        // Let's stick to a simpler "Uninstall" and then maybe a second prompt? Or just clean data?
        // User requested: "Uninstall prompts whether to clean data".
        // Let's use `window.confirm` for the second step or assume `dialog.confirm` is flexible enough?
        // I will implement a two-step confirmation or try to modify the dialog hook if visible (not visible here).
        // Let's use a standard `window.confirm` for the data cleanup question if the custom dialog doesn't support complex return.
        // Better: Inside onConfirm, ask again?
        onConfirm: async () => {
          // Ask for data cleanup
          // Since we are in an async callback, we can use another dialog or confirm.
          // Native confirm is ugly but works reliably for logic:
          const cleanData = window.confirm(`是否同时清理插件「${plugin.name}」的数据文件？\n点击“确定”清理数据，点击“取消”仅卸载插件。`);

          const loadingToast = toast.loading('卸载中...');
          try {
            await PluginManager.uninstallPlugin(plugin.name, plugin.filename, cleanData);
            toast.success('卸载成功', { id: loadingToast });
            loadPlugins();
            resolve();
          } catch (e: any) {
            toast.error(e.message, { id: loadingToast });
            reject(e);
          }
        },
        onCancel: () => {
          resolve();
        }
      });
    });
  };

  const handleConfig = (plugin: PluginItem) => {
    setCurrentPluginName(plugin.name); // Use Loaded Name for config lookup
    onOpen();
  };

  return (
    <>
      <title>插件管理 - NapCat WebUI</title>
      <div className='p-2 md:p-4 relative'>
        <PageLoading loading={loading} />
        <PluginConfigModal
          isOpen={isOpen}
          onOpenChange={onOpenChange}
          pluginName={currentPluginName}
        />

        <div className='flex mb-6 items-center gap-4'>
          <h1 className="text-2xl font-bold">插件管理</h1>
          <Button
            isIconOnly
            className="bg-default-100/50 hover:bg-default-200/50 text-default-700 backdrop-blur-md"
            radius='full'
            onPress={loadPlugins}
          >
            <IoMdRefresh size={24} />
          </Button>
        </div>

        {pluginManagerNotFound ? (
          <div className="flex flex-col items-center justify-center min-h-[400px] text-center">
            <div className="text-6xl mb-4">📦</div>
            <h2 className="text-xl font-semibold text-default-700 dark:text-white/90 mb-2">
              无插件加载
            </h2>
            <p className="text-default-500 dark:text-white/60 max-w-md">
              插件管理器未加载，请检查 plugins 目录是否存在
            </p>
          </div>
        ) : plugins.length === 0 ? (
          <div className="text-default-400">暂时没有安装插件</div>
        ) : (
          <div className='grid grid-cols-1 md:grid-cols-2 lg:grid-cols-3 xl:grid-cols-4 2xl:grid-cols-5 justify-start items-stretch gap-x-2 gap-y-4'>
            {plugins.map(plugin => (
              <PluginDisplayCard
                key={plugin.name}
                data={plugin}
                onReload={() => handleReload(plugin.name)}
                onToggleStatus={() => handleToggle(plugin)}
                onUninstall={() => handleUninstall(plugin)}
                onConfig={() => {
                  if (plugin.hasConfig) {
                    handleConfig(plugin);
                  } else {
                    toast.error('此插件没有配置哦');
                  }
                }}
                hasConfig={true}
              />
            ))}
          </div>
        )}
      </div>
    </>
  );
}