@lobehub/chat/src/store/discover/slices/plugin/action.ts

Lobe Chat - an open-source, high-performance chatbot framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.

import { CategoryItem, CategoryListQuery } from '@lobehub/market-sdk';
import useSWR, { type SWRResponse } from 'swr';
import type { StateCreator } from 'zustand/vanilla';

import { discoverService } from '@/services/discover';
import { DiscoverStore } from '@/store/discover';
import { globalHelpers } from '@/store/global/helpers';
import {
  DiscoverPluginDetail,
  IdentifiersResponse,
  PluginListResponse,
  PluginQueryParams,
} from '@/types/discover';

export interface PluginAction {
  usePluginCategories: (params: CategoryListQuery) => SWRResponse<CategoryItem[]>;
  usePluginDetail: (params: {
    identifier?: string;
    withManifest?: boolean;
  }) => SWRResponse<DiscoverPluginDetail | undefined>;
  usePluginIdentifiers: () => SWRResponse<IdentifiersResponse>;
  usePluginList: (params?: PluginQueryParams) => SWRResponse<PluginListResponse>;
}

export const createPluginSlice: StateCreator<
  DiscoverStore,
  [['zustand/devtools', never]],
  [],
  PluginAction
> = () => ({
  usePluginCategories: (params) => {
    const locale = globalHelpers.getCurrentLanguage();
    return useSWR(
      ['plugin-categories', locale, ...Object.values(params)].filter(Boolean).join('-'),
      async () => discoverService.getPluginCategories(params),
      {
        revalidateOnFocus: false,
      },
    );
  },

  usePluginDetail: ({ identifier, withManifest }) => {
    const locale = globalHelpers.getCurrentLanguage();
    return useSWR(
      !identifier
        ? null
        : ['plugin-details', locale, identifier, withManifest].filter(Boolean).join('-'),
      async () => discoverService.getPluginDetail({ identifier: identifier!, withManifest }),
      {
        revalidateOnFocus: false,
      },
    );
  },

  usePluginIdentifiers: () => {
    return useSWR('plugin-identifiers', async () => discoverService.getPluginIdentifiers(), {
      revalidateOnFocus: false,
    });
  },

  usePluginList: (params = {}) => {
    const locale = globalHelpers.getCurrentLanguage();
    return useSWR(
      ['plugin-list', locale, ...Object.values(params)].filter(Boolean).join('-'),
      async () =>
        discoverService.getPluginList({
          ...params,
          page: params.page ? Number(params.page) : 1,
          pageSize: params.pageSize ? Number(params.pageSize) : 21,
        }),
      {
        revalidateOnFocus: false,
      },
    );
  },
});