opencatd-open/frontend/src/stores/key.ts

// src/stores/key.ts
import { defineStore } from 'pinia';
import { ref } from 'vue';
import type { AxiosResponse } from 'axios';
import request from '@/api/client';
import type { ApiKey, NewApiKeyPayload } from '@/types';

export const useKeyStore = defineStore('key', () => {
    const loading = ref(false);
    const error = ref<string | null>(null);
    const totalKeys = ref(0);
    const keys = ref<ApiKey[]>([]);
    const key = ref<ApiKey | null>(null);

    const fetchKeys = async (pageSize = 20, page = 1, active?: boolean | boolean[]) => {
        loading.value = true;
        error.value = null;
        try {
            const response = await request.get('/keys', {
                params: {
                    pageSize,
                    page,
                    active,
                },
            });

            keys.value = response.data.data?.keys ?? [];
            totalKeys.value = response.data.data?.total ?? 0;
        } catch (err: any) {
            error.value = err.response?.data?.error || '获取ApiKeys失败';
            throw error
        } finally {
            loading.value = false;
        }
    };

    const fetchKey = async (id: number | string) => {
        if (keys.value.length > 0) {
            const findkey = keys.value.find(item => item.id === id)
            if (findkey) {
                key.value = findkey;
                return;
            }
        }
        loading.value = true;
        error.value = null;
        try {
            const response = await request.get(`/keys/${id}`);
            key.value = response.data.data;
            if (key.value && (key.value.support_models?.length ?? 0) < 3) {
                key.value.support_models = key.value.support_models_array ? JSON.stringify(key.value.support_models) : ''
            }
            if (key.value && !key.value.support_models_array) {
                key.value.support_models_array = key.value.support_models ? JSON.parse(key.value.support_models) : []
            }

        } catch (err: any) {
            error.value = err.response?.data?.error || '获取ApiKey失败';
            throw error
        } finally {
            loading.value = false;
        }
    };

    const refreshKey = async (id: number | string) => {
        loading.value = true;
        error.value = null;
        try {
            const response = await request.get(`/keys/${id}`);
            key.value = response.data.data;
            if (key.value && (key.value.support_models?.length ?? 0) < 3) {
                key.value.support_models = key.value.support_models_array ? JSON.stringify(key.value.support_models) : ''
            }
            if (key.value && !key.value.support_models_array) {
                key.value.support_models_array = key.value.support_models ? JSON.parse(key.value.support_models) : []
            }
        } catch (err: any) {
            error.value = err.response?.data?.error || '获取ApiKey失败';
            throw error
        } finally {
            loading.value = false;
        }
    }

    const createKey = async (data: NewApiKeyPayload) => {
        loading.value = true;
        error.value = null;
        try {
            const response: AxiosResponse = await request.post('/keys', data);
            return response;
        } catch (err: any) {
            error.value = err.response?.data?.error || '创建ApiKey失败';
            throw error
        } finally {
            loading.value = false;
        }
    }

    const updateKey = async (keyInfo: ApiKey) => {
        loading.value = true;
        error.value = null;
        try {
            const response: AxiosResponse = await request.put(`/keys/${keyInfo.id}`, keyInfo);
            return response;
        } catch (err: any) {
            error.value = err.response?.data?.error || '更新ApiKey失败';
            throw error
        } finally {
            loading.value = false;
        }
    }

    const keyOption = async (option: string, ids: (number | string)[]) => {
        loading.value = true;
        error.value = null;
        try {
            const response: AxiosResponse = await request.post(`/keys/batch/${option}`, { ids });

            return response
        } catch (err: any) {
            error.value = err.response?.data?.error || '操作失败';
            throw err;
        } finally {
            loading.value = false;
        }
    }
    //todo 更新 批量操作

    return {
        loading, error,
        key, keys, totalKeys,
        fetchKeys,
        fetchKey,
        refreshKey,
        createKey,
        updateKey,
        keyOption,
    }
});