TERES_web_frontend/src/hooks/setting-hooks.ts

import { useUserData } from "./useUserData";
import { useEffect, useState, useCallback } from "react";
import logger from "@/utils/logger";
import type { IUserInfo, ISystemStatus } from "@/interfaces/database/user-setting";
import userService from "@/services/user_service";
import { rsaPsw } from "../utils/encryption";
import type { IFactory, IMyLlmModel } from "@/interfaces/database/llm";
import type { LLMFactory } from "@/constants/llm";

/**
 * 个人中心设置
 */
export function useProfileSetting() {
  const { fetchUserInfo, userInfo } = useUserData();

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

  const updateUserInfo = async (newUserInfo: Partial<IUserInfo>) => {
    try {
      await userService.updateSetting(newUserInfo);
    } catch (error) {
      logger.error('更新用户信息失败:', error);
      throw error;
    }
  };

  const changeUserPassword = async (data: { password: string; new_password: string }) => {
    try {
      const newPassword = rsaPsw(data.new_password);
      const oldPassword = rsaPsw(data.password);
      const res = await userService.updatePassword({
        password: oldPassword,
        new_password: newPassword,
      });
    } catch (error) {
      throw error;
    }
  };

  return {
    userInfo,
    updateUserInfo,
    changeUserPassword,
  };
}

/**
 * LLM 模型设置
 */
export function useLlmModelSetting() {
  const [llmFactory, setLlmFactory] = useState<IFactory[]>([]);
  const [myLlm, setMyLlm] = useState<Record<LLMFactory, IMyLlmModel>>();

  const fetchLlmFactory = async () => {
    try {
      const res = await userService.llm_factories_list();
      const arr = res.data.data || [];
      setLlmFactory(arr);
    } catch (error) {
      logger.error('获取模型工厂失败:', error);
      throw error;
    }
  }

  const fetchMyLlm = async () => {
    try {
      const res = await userService.my_llm();
      const llm_dic = res.data.data || {};
      setMyLlm(llm_dic);
    } catch (error) {
      logger.error('获取我的模型失败:', error);
      throw error;
    }
  }
  useEffect(() => {
    fetchLlmFactory();
    fetchMyLlm();
  }, []);

  const refreshLlmModel = async () => {
    await fetchMyLlm();
    // await fetchLlmFactory();
    logger.info('刷新我的模型成功');
  }

  return {
    llmFactory,
    myLlm,
    refreshLlmModel,
  }
}

/**
 * 系统状态设置
 */
export function useSystemStatus() {
  const [systemStatus, setSystemStatus] = useState<ISystemStatus | null>(null);
  const [loading, setLoading] = useState(false);
  const [error, setError] = useState<string | null>(null);

  const fetchSystemStatus = useCallback(async () => {
    try {
      setLoading(true);
      setError(null);
      const res = await userService.system_status();
      if (res.data.code === 0) {
        setSystemStatus(res.data.data);
      } else {
        throw new Error(res.data.message || '获取系统状态失败');
      }
    } catch (error: any) {
      const errorMessage = error.response?.data?.message || error.message || '获取系统状态失败';
      setError(errorMessage);
      logger.error('获取系统状态失败:', error);
    } finally {
      setLoading(false);
    }
  }, []);

  return {
    systemStatus,
    loading,
    error,
    fetchSystemStatus,
  };
}

/**
 * 团队设置
 */
export function useTeamSetting() {
  const { userInfo, tenantInfo, tenantList, refreshUserData } = useUserData();
  const [tenantUsers, setTenantUsers] = useState<any[]>([]);
  const [loading, setLoading] = useState(false);
  const [error, setError] = useState<string | null>(null);

  // 获取租户用户列表
  const fetchTenantUsers = useCallback(async () => {
    if (!tenantInfo?.tenant_id) return;
    
    try {
      const response = await userService.listTenantUser(tenantInfo.tenant_id);
      if (response.data.code === 0) {
        setTenantUsers(response.data.data || []);
      }
    } catch (error) {
      logger.error('获取租户用户失败:', error);
    }
  }, [tenantInfo?.tenant_id]);

  useEffect(() => {
    if (tenantInfo?.tenant_id) {
      fetchTenantUsers();
    }
  }, [fetchTenantUsers]);

  const inviteUser = async (email: string) => {
    if (!email.trim() || !tenantInfo?.tenant_id) return;

    setLoading(true);
    setError(null);

    try {
      const response = await userService.addTenantUser(tenantInfo.tenant_id, email.trim());
      if (response.data.code === 0) {
        await fetchTenantUsers();
        await refreshUserData();
        return { success: true };
      } else {
        const errorMsg = response.data.message || '邀请失败';
        setError(errorMsg);
        return { success: false, error: errorMsg };
      }
    } catch (error: any) {
      const errorMsg = error.message || '邀请失败';
      setError(errorMsg);
      return { success: false, error: errorMsg };
    } finally {
      setLoading(false);
    }
  };

  const deleteUser = async (userId: string) => {
    if (!tenantInfo?.tenant_id) return;

    try {
      const response = await userService.deleteTenantUser({
        tenantId: tenantInfo.tenant_id,
        userId
      });
      if (response.data.code === 0) {
        await fetchTenantUsers();
        return { success: true };
      }
      return { success: false };
    } catch (error) {
      logger.error('删除用户失败:', error);
      return { success: false };
    }
  };

  const agreeTenant = async (tenantId: string) => {
    try {
      const response = await userService.agreeTenant(tenantId);
      if (response.data.code === 0) {
        await refreshUserData();
        return { success: true };
      }
      return { success: false };
    } catch (error) {
      logger.error('同意加入失败:', error);
      return { success: false };
    }
  };

  const quitTenant = async (tenantId: string) => {
    if (!userInfo?.id) return;

    try {
      const response = await userService.deleteTenantUser({
        tenantId,
        userId: userInfo.id
      });
      if (response.data.code === 0) {
        await refreshUserData();
        return { success: true };
      }
      return { success: false };
    } catch (error) {
      logger.error('退出租户失败:', error);
      return { success: false };
    }
  };

  return {
    userInfo,
    tenantInfo,
    tenantList,
    tenantUsers,
    loading,
    error,
    inviteUser,
    deleteUser,
    agreeTenant,
    quitTenant,
    refreshUserData,
  };
}