@lobehub/chat 0.63.2 → 0.64.0

package/src/store/plugin/action.ts

@@ -14,6 +14,8 @@ import { pluginSelectors } from '@/store/plugin/selectors';
 import { LobeSessions } from '@/types/session';
 import { setNamespace } from '@/utils/storeDebug';
 
+import { DevPlugin, defaultDevPlugin } from './initialState';
+import { DevListDispatch, devPluginListReducer } from './reducers/devPluginList';
 import { PluginDispatch, pluginManifestReducer } from './reducers/manifest';
 import { PluginStore } from './store';
 
@@ -24,9 +26,12 @@ const t = setNamespace('plugin');
  */
 export interface PluginAction {
   checkLocalEnabledPlugins: (sessions: LobeSessions) => void;
+  dispatchDevPluginList: (payload: DevListDispatch) => void;
   dispatchPluginManifest: (payload: PluginDispatch) => void;
   fetchPluginManifest: (name: string) => Promise<void>;
+  saveToDevList: (value: DevPlugin) => void;
   updateManifestLoadingState: (key: string, value: boolean | undefined) => void;
+  updateNewDevPlugin: (value: Partial<DevPlugin>) => void;
   updatePluginSettings: <T>(id: string, settings: Partial<T>) => void;
   useFetchPluginList: () => SWRResponse<LobeChatPluginsMarketIndex>;
 }
@@ -55,13 +60,18 @@ export const createPluginSlice: StateCreator<
 
     set({ manifestPrepared: true }, false, t('checkLocalEnabledPlugins'));
   },
+  dispatchDevPluginList: (payload) => {
+    const { devPluginList } = get();
+
+    const nextList = devPluginListReducer(devPluginList, payload);
+    set({ devPluginList: nextList }, false, t('dispatchDevList', payload));
+  },
   dispatchPluginManifest: (payload) => {
     const { pluginManifestMap } = get();
     const nextManifest = pluginManifestReducer(pluginManifestMap, payload);
 
     set({ pluginManifestMap: nextManifest }, false, t('dispatchPluginManifest', payload));
   },
-
   fetchPluginManifest: async (name) => {
     const plugin = pluginSelectors.getPluginMetaById(name)(get());
     // 1. 校验文件
@@ -103,6 +113,10 @@ export const createPluginSlice: StateCreator<
     // 4. 存储 manifest 信息
     get().dispatchPluginManifest({ id: plugin.identifier, plugin: data, type: 'addManifest' });
   },
+  saveToDevList: (value) => {
+    get().dispatchDevPluginList({ plugin: value, type: 'addItem' });
+    set({ newDevPlugin: defaultDevPlugin }, false, t('saveToDevList'));
+  },
 
   updateManifestLoadingState: (key, value) => {
     set(
@@ -113,7 +127,13 @@ export const createPluginSlice: StateCreator<
       t('updateManifestLoadingState'),
     );
   },
-
+  updateNewDevPlugin: (newDevPlugin) => {
+    set(
+      { newDevPlugin: merge({}, get().newDevPlugin, newDevPlugin) },
+      false,
+      t('updateNewDevPlugin'),
+    );
+  },
   updatePluginSettings: (id, settings) => {
     set(
       produce((draft) => {
@@ -123,6 +143,7 @@ export const createPluginSlice: StateCreator<
       t('updatePluginSettings'),
     );
   },
+
   useFetchPluginList: () =>
     useSWR<LobeChatPluginsMarketIndex>('fetchPluginList', getPluginList, {
       onSuccess: (pluginMarketIndex) => {

package/src/store/plugin/initialState.ts

@@ -1,20 +1,36 @@
-import { LobeChatPluginMeta } from '@lobehub/chat-plugin-sdk';
+import { LobeChatPluginManifest, LobeChatPluginMeta } from '@lobehub/chat-plugin-sdk';
 
 import { PluginManifestMap } from '@/types/plugin';
 
 export type PluginManifestLoadingState = Record<string, boolean>;
 export type PluginsSettings = Record<string, any>;
 
+export interface DevPlugin extends LobeChatPluginMeta {
+  apiMode: 'openapi' | 'simple';
+  enableSettings: boolean;
+  manifestConfig?: LobeChatPluginManifest;
+  manifestMode: 'local' | 'url';
+}
+
 export interface PluginStoreState {
+  devPluginList: DevPlugin[];
   manifestPrepared: boolean;
+  newDevPlugin: Partial<DevPlugin>;
   pluginList: LobeChatPluginMeta[];
   pluginManifestLoading: PluginManifestLoadingState;
   pluginManifestMap: PluginManifestMap;
   pluginsSettings: PluginsSettings;
 }
+export const defaultDevPlugin: Partial<DevPlugin> = {
+  apiMode: 'simple',
+  enableSettings: false,
+  manifestMode: 'url',
+};
 
 export const initialState: PluginStoreState = {
+  devPluginList: [],
   manifestPrepared: false,
+  newDevPlugin: defaultDevPlugin,
   pluginList: [],
   pluginManifestLoading: {},
   pluginManifestMap: {},

package/src/store/plugin/reducers/devPluginList.ts

@@ -0,0 +1,56 @@
+import { produce } from 'immer';
+
+import { DevPlugin } from '../initialState';
+
+export type DevListState = DevPlugin[];
+
+export type AddPluginAction = {
+  plugin: DevPlugin;
+  type: 'addItem';
+};
+
+export type DeletePluginAction = {
+  id: string;
+  type: 'deleteItem';
+};
+
+export type UpdatePluginAction = {
+  id: string;
+  plugin: DevPlugin;
+  type: 'updateItem';
+};
+
+export type DevListDispatch = AddPluginAction | DeletePluginAction | UpdatePluginAction;
+
+export const devPluginListReducer = (
+  state: DevListState,
+  payload: DevListDispatch,
+): DevListState => {
+  switch (payload.type) {
+    case 'addItem': {
+      return [...state, payload.plugin];
+    }
+
+    case 'deleteItem': {
+      return produce(state, (draftState) => {
+        const deleteIndex = state.findIndex((plugin) => plugin.identifier === payload.id);
+        if (deleteIndex !== -1) {
+          draftState.splice(deleteIndex, 1);
+        }
+      });
+    }
+
+    case 'updateItem': {
+      return produce(state, (draftState) => {
+        const updateIndex = state.findIndex((plugin) => plugin.identifier === payload.id);
+        if (updateIndex !== -1) {
+          draftState[updateIndex] = payload.plugin;
+        }
+      });
+    }
+
+    default: {
+      return state;
+    }
+  }
+};

package/src/store/plugin/selectors.ts

@@ -23,8 +23,13 @@ const enabledSchema =
       );
   };
 
+const pluginList = (s: PluginStoreState) => [...s.pluginList, ...s.devPluginList];
+
 const getPluginMetaById = (id: string) => (s: PluginStoreState) =>
-  pluginHelpers.getPluginFormList(s.pluginList, id);
+  pluginHelpers.getPluginFormList(pluginList(s), id);
+
+const getDevPluginById = (id: string) => (s: PluginStoreState) =>
+  s.devPluginList.find((i) => i.identifier === id);
 
 const getPluginManifestById = (id: string) => (s: PluginStoreState) => s.pluginManifestMap[id];
 const getPluginSettingsById = (id: string) => (s: PluginStoreState) => s.pluginsSettings[id];
@@ -41,7 +46,7 @@ const getPluginManifestLoadingStatus = (id: string) => (s: PluginStoreState) =>
 };
 
 const displayPluginList = (s: PluginStoreState) =>
-  s.pluginList.map((p) => ({
+  pluginList(s).map((p) => ({
     author: p.author,
     avatar: p.meta?.avatar,
     createAt: p.createAt,
@@ -54,8 +59,10 @@ const displayPluginList = (s: PluginStoreState) =>
 export const pluginSelectors = {
   displayPluginList,
   enabledSchema,
+  getDevPluginById,
   getPluginManifestById,
   getPluginManifestLoadingStatus,
   getPluginMetaById,
   getPluginSettingsById,
+  pluginList,
 };

package/src/store/plugin/store.ts

@@ -19,7 +19,10 @@ const createStore: StateCreator<PluginStore, [['zustand/devtools', never]]> = (.
 
 //  ===============  persist 本地缓存中间件配置 ============ //
 
-type SessionPersist = Pick<PluginStore, 'pluginList' | 'pluginManifestMap' | 'pluginsSettings'>;
+type SessionPersist = Pick<
+  PluginStore,
+  'pluginList' | 'pluginManifestMap' | 'pluginsSettings' | 'devPluginList'
+>;
 
 const storeName = 'LOBE_PLUGIN';
 
@@ -27,6 +30,7 @@ const persistOptions: PersistOptions<PluginStore, SessionPersist> = {
   name: storeName,
 
   partialize: (s) => ({
+    devPluginList: s.devPluginList,
     pluginList: s.pluginList,
     pluginManifestMap: s.pluginManifestMap,
     pluginsSettings: s.pluginsSettings,