heicode/cc-haha/desktop/src/stores/resourceStore.ts

// desktop/src/stores/resourceStore.ts
//
// Zustand store for ResourceBinding / ResourceGrant. Talks to the local
// /api/heicode-resources/* proxy.

import { create } from 'zustand'
import {
  heicodeResourcesApi,
  type CreateBindingInput,
  type CreateGrantInput,
  type ResourceBinding,
  type ResourceGrant,
} from '../api/heicodeResources'

type ResourceState = {
  // Bindings
  bindings: ResourceBinding[]
  bindingsLoading: boolean
  bindingsFetched: boolean
  bindingsError: string | null

  // Grants
  grants: ResourceGrant[]
  grantsLoading: boolean
  grantsFetched: boolean
  grantsError: string | null

  // Mutation state (shared — only one create/delete in flight at a time)
  isMutating: boolean
  mutationError: string | null

  // Actions
  fetchBindings: () => Promise<void>
  fetchGrants: () => Promise<void>
  createBinding: (input: CreateBindingInput) => Promise<ResourceBinding | null>
  deleteBinding: (id: string) => Promise<boolean>
  createGrant: (input: CreateGrantInput) => Promise<ResourceGrant | null>
  revokeGrant: (id: string) => Promise<boolean>
  clearBindingsError: () => void
  clearGrantsError: () => void
  clearMutationError: () => void
}

function extractErrorMessage(err: unknown, fallback: string): string {
  if (err instanceof Error) return err.message
  return typeof err === 'string' ? err : fallback
}

export const useResourceStore = create<ResourceState>((set, get) => ({
  bindings: [],
  bindingsLoading: false,
  bindingsFetched: false,
  bindingsError: null,

  grants: [],
  grantsLoading: false,
  grantsFetched: false,
  grantsError: null,

  isMutating: false,
  mutationError: null,

  fetchBindings: async () => {
    set({ bindingsLoading: true, bindingsError: null })
    try {
      const res = await heicodeResourcesApi.listBindings()
      set({
        bindings: res.success ? (res.data?.items ?? []) : [],
        bindingsLoading: false,
        bindingsFetched: true,
        bindingsError: res.success ? null : (res.message || '加载资源绑定失败'),
      })
    } catch (err) {
      set({
        bindingsLoading: false,
        bindingsFetched: true,
        bindingsError: extractErrorMessage(err, '加载资源绑定失败'),
      })
    }
  },

  fetchGrants: async () => {
    set({ grantsLoading: true, grantsError: null })
    try {
      const res = await heicodeResourcesApi.listGrants()
      set({
        grants: res.success ? (res.data?.items ?? []) : [],
        grantsLoading: false,
        grantsFetched: true,
        grantsError: res.success ? null : (res.message || '加载资源授权失败'),
      })
    } catch (err) {
      set({
        grantsLoading: false,
        grantsFetched: true,
        grantsError: extractErrorMessage(err, '加载资源授权失败'),
      })
    }
  },

  createBinding: async (input) => {
    set({ isMutating: true, mutationError: null })
    try {
      const res = await heicodeResourcesApi.createBinding(input)
      if (!res.success || !res.data) {
        set({ isMutating: false, mutationError: res.message || '创建资源绑定失败' })
        return null
      }
      // Optimistic add (server is source of truth, but UX wins).
      set((s) => ({
        isMutating: false,
        bindings: [res.data!, ...s.bindings],
      }))
      return res.data
    } catch (err) {
      set({ isMutating: false, mutationError: extractErrorMessage(err, '创建资源绑定失败') })
      return null
    }
  },

  deleteBinding: async (id) => {
    set({ isMutating: true, mutationError: null })
    try {
      const res = await heicodeResourcesApi.deleteBinding(id)
      if (!res.success) {
        set({ isMutating: false, mutationError: res.message || '删除失败' })
        return false
      }
      // Server soft-deletes (status=revoked). Refetch is safest.
      await get().fetchBindings()
      set({ isMutating: false })
      return true
    } catch (err) {
      set({ isMutating: false, mutationError: extractErrorMessage(err, '删除失败') })
      return false
    }
  },

  createGrant: async (input) => {
    set({ isMutating: true, mutationError: null })
    try {
      const res = await heicodeResourcesApi.createGrant(input)
      if (!res.success || !res.data) {
        set({ isMutating: false, mutationError: res.message || '创建授权失败' })
        return null
      }
      set((s) => ({
        isMutating: false,
        grants: [res.data!, ...s.grants],
      }))
      return res.data
    } catch (err) {
      set({ isMutating: false, mutationError: extractErrorMessage(err, '创建授权失败') })
      return null
    }
  },

  revokeGrant: async (id) => {
    set({ isMutating: true, mutationError: null })
    try {
      const res = await heicodeResourcesApi.revokeGrant(id)
      if (!res.success) {
        set({ isMutating: false, mutationError: res.message || '撤销失败' })
        return false
      }
      await get().fetchGrants()
      set({ isMutating: false })
      return true
    } catch (err) {
      set({ isMutating: false, mutationError: extractErrorMessage(err, '撤销失败') })
      return false
    }
  },

  clearBindingsError: () => set({ bindingsError: null }),
  clearGrantsError: () => set({ grantsError: null }),
  clearMutationError: () => set({ mutationError: null }),
}))