105 lines
2.8 KiB
TypeScript
105 lines
2.8 KiB
TypeScript
/*
|
|
* Copyright 2025 coze-dev Authors
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
import { create } from 'zustand';
|
|
import { produce } from 'immer';
|
|
import {
|
|
type BizCtx,
|
|
type MockSet,
|
|
type MockSetBinding,
|
|
} from '@coze-arch/bot-api/debugger_api';
|
|
|
|
import { isCurrent } from '../utils';
|
|
import { type BasicMockSetInfo } from '../interface';
|
|
|
|
export interface EnabledMockSetInfo {
|
|
mockSetBinding: MockSetBinding;
|
|
mockSetDetail?: MockSet;
|
|
}
|
|
|
|
interface MockInfoStoreState {
|
|
bizCtx: BizCtx;
|
|
enabledMockSetInfo: Array<EnabledMockSetInfo>;
|
|
isPolling: boolean;
|
|
isLoading: boolean;
|
|
currentMockComp: Array<BasicMockSetInfo>;
|
|
timer?: NodeJS.Timeout;
|
|
restartTimer?: NodeJS.Timeout;
|
|
}
|
|
|
|
interface MockInfoStoreAction {
|
|
setPolling: (polling: boolean) => void;
|
|
setLoading: (loading: boolean) => void;
|
|
setCurrentBizCtx: (bizCtx: BizCtx) => void;
|
|
setEnabledMockSetInfo: (mockSetList?: Array<EnabledMockSetInfo>) => void;
|
|
removeMockComp: (mockComp: BasicMockSetInfo) => number;
|
|
addMockComp: (mockComp: BasicMockSetInfo) => number;
|
|
setTimer: (timer?: NodeJS.Timeout) => void;
|
|
setRestartTimer: (timer?: NodeJS.Timeout) => void;
|
|
}
|
|
|
|
export const useMockInfoStore = create<
|
|
MockInfoStoreState & MockInfoStoreAction
|
|
>((set, get) => ({
|
|
bizCtx: {},
|
|
enabledMockSetInfo: [],
|
|
isPolling: false,
|
|
isLoading: false,
|
|
currentMockComp: [],
|
|
setPolling: polling => {
|
|
set({ isPolling: polling });
|
|
},
|
|
setLoading: loading => {
|
|
set({ isLoading: loading });
|
|
},
|
|
setCurrentBizCtx: bizCtx => {
|
|
set({ bizCtx });
|
|
},
|
|
setEnabledMockSetInfo: enabledMockSetInfo => {
|
|
set({ enabledMockSetInfo });
|
|
},
|
|
addMockComp: mockSetInfo => {
|
|
set(
|
|
produce<MockInfoStoreState>(s => {
|
|
const index = s.currentMockComp.findIndex(item =>
|
|
isCurrent(item, mockSetInfo),
|
|
);
|
|
index <= -1 && s.currentMockComp.push(mockSetInfo);
|
|
}),
|
|
);
|
|
return get().currentMockComp.length;
|
|
},
|
|
removeMockComp: mockSetInfo => {
|
|
set(
|
|
produce<MockInfoStoreState>(s => {
|
|
const index = s.currentMockComp.findIndex(item =>
|
|
isCurrent(item, mockSetInfo),
|
|
);
|
|
if (index > -1) {
|
|
s.currentMockComp.splice(index, 1);
|
|
}
|
|
}),
|
|
);
|
|
return get().currentMockComp.length;
|
|
},
|
|
setTimer: timer => {
|
|
set({ timer });
|
|
},
|
|
setRestartTimer: timer => {
|
|
set({ timer });
|
|
},
|
|
}));
|