import { create } from 'zustand';
import { query } from '@/modules';
import { message } from 'antd';
type OrgStore = {
  showEdit: boolean;
  setShowEdit: (showEdit: boolean) => void;
  formData: any;
  setFormData: (formData: any) => void;
  loading: boolean;
  setLoading: (loading: boolean) => void;
  list: any[];
  getList: () => Promise<void>;
  updateData: (data: any) => Promise<void>;
  deleteData: (id: string) => Promise<void>;
  org: any;
  setOrg: (org: any) => void;
  users: { id: string; username: string; role?: string }[];
  orgId: string;
  setOrgId: (orgId: string) => void;
  getOrg: () => Promise<any>;
};
export const useOrgStore = create<OrgStore>((set, get) => {
  return {
    showEdit: false,
    setShowEdit: (showEdit) => set({ showEdit }),
    formData: {},
    setFormData: (formData) => set({ formData }),
    loading: false,
    setLoading: (loading) => set({ loading }),
    list: [],
    getList: async () => {
      set({ loading: true });

      const res = await query.post({
        path: 'org',
        key: 'list',
      });
      set({ loading: false });
      if (res.code === 200) {
        set({ list: res.data });
      } else {
        message.error(res.message || 'Request failed');
      }
    },
    updateData: async (data) => {
      const { getList } = get();
      const res = await query.post({
        path: 'org',
        key: 'update',
        data,
      });
      if (res.code === 200) {
        message.success('Success');
        set({ showEdit: false, formData: [] });
        getList();
      } else {
        message.error(res.message || 'Request failed');
      }
    },
    deleteData: async (id) => {
      const { getList } = get();
      const res = await query.post({
        path: 'org',
        key: 'delete',
        id,
      });
      if (res.code === 200) {
        getList();
        message.success('Success');
      } else {
        message.error(res.message || 'Request failed');
      }
    },
    org: {},
    setOrg: (org) => set({ org }),
    orgId: '',
    setOrgId: (orgId) => set({ orgId }),
    users: [],
    getOrg: async () => {
      const { orgId } = get();
      const loaded = message.loading('Loading...', 0);
      const res = await query.post({
        path: 'org',
        key: 'get',
        id: orgId,
      });
      loaded();
      if (res.code === 200) {
        const { org, users } = res.data || {};
        set({ org, users });
      } else {
        message.error(res.message || 'Request failed');
      }
    },
  };
});
