pro/src/store/department.ts

import { atomWithMutation, atomWithQuery, queryClientAtom } from 'jotai-tanstack-query'
import systemServ from '@/service/system.ts'
import { IApiResult, IPage } from '@/types'
import { IDepartment } from '@/types/department'
import { atom, createStore } from 'jotai'
import { t } from 'i18next'
import { message } from 'antd'


const store = createStore()

export const departPageAtom = atom<IPage>({})

export const defaultDepart = {
  id: 0,
  parent_id: 0,
  name: '',
  manager_user_id: 0,
  phone: '',
  sort: 0,
} as IDepartment

export const batchIdsAtom = atom<number[]>([])

export const selectedDepartAtom = atom<IDepartment>({} as IDepartment)

export const departTreeAtom = atomWithQuery(() => {

  return {
    queryKey: [ 'departTree' ],
    queryFn: async () => {
      return await systemServ.dept.tree()
    },
    select: (res) => {
      return res.data.tree ?? []
    }
  }
})

export const saveOrUpdateDepartAtom = atomWithMutation<IApiResult, IDepartment>((get) => {

  return {
    mutationKey: [ 'saveOrUpdateDepart' ],
    mutationFn: async (data: IDepartment) => {
      if (data.id) {
        return await systemServ.dept.update(data)
      }
      return await systemServ.dept.add(data)
    },
    onSuccess: (res) => {
      const isAdd = !!res.data?.id
      message.success(t(isAdd ? 'message.saveSuccess' : 'message.editSuccess', '保存成功'))

      if (isAdd) {
        store.set(selectedDepartAtom, prev => ({
          ...prev,
          id: res.data.id
        }))
      }

      get(queryClientAtom).invalidateQueries({ queryKey: [ 'departTree' ] }).then()
    }
  }

})


export const deleteDepartAtom = atomWithMutation<IApiResult, number[]>((get) => {

  return {
    mutationKey: [ 'deleteDepart', get(batchIdsAtom) ],
    mutationFn: async (ids: number[]) => {
      return await systemServ.dept.batchDelete(ids)
    },
    onSuccess: () => {
      message.success(t('message.deleteSuccess', '删除成功'))
      store.set(batchIdsAtom, [])
      get(queryClientAtom).invalidateQueries({ queryKey: [ 'departTree' ] }).then()

    }
  }

})