96 lines
2.5 KiB
TypeScript
96 lines
2.5 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 { nanoid } from 'nanoid';
|
|
import mitt from 'mitt';
|
|
import { type PluginRegistryEntry } from '@coze-common/chat-area';
|
|
|
|
import {
|
|
type EventCenter,
|
|
type GrabPluginBizContext,
|
|
type PublicEventCenter,
|
|
type EventCallbacks,
|
|
} from './types/plugin-biz-context';
|
|
import { createSelectionStore } from './stores/selection';
|
|
import { createQuoteStore, subscribeQuoteUpdate } from './stores/quote';
|
|
import { createPreferenceStore } from './stores/preference';
|
|
import { ChatAreaGrabPlugin } from './plugin';
|
|
|
|
interface Preference {
|
|
enableGrab: boolean;
|
|
}
|
|
|
|
export type Scene = 'store' | 'other';
|
|
|
|
type CreateGrabPluginParams = {
|
|
preference: Preference;
|
|
scene?: Scene;
|
|
} & EventCallbacks;
|
|
|
|
export const publicEventCenter = mitt<PublicEventCenter>();
|
|
|
|
export const createGrabPlugin = (params: CreateGrabPluginParams) => {
|
|
const { preference, onQuote, onQuoteChange, scene } = params;
|
|
|
|
const grabPluginId = nanoid();
|
|
|
|
const grabPlugin: PluginRegistryEntry<GrabPluginBizContext> = {
|
|
createPluginBizContext: () => {
|
|
const eventCallbacks = {
|
|
onQuote,
|
|
onQuoteChange,
|
|
};
|
|
|
|
const storeSet = {
|
|
useSelectionStore: createSelectionStore('plugin'),
|
|
useQuoteStore: createQuoteStore('plugin'),
|
|
usePreferenceStore: createPreferenceStore('plugin'),
|
|
};
|
|
|
|
const eventCenter = mitt<EventCenter>();
|
|
|
|
// Default injection preference
|
|
storeSet.usePreferenceStore
|
|
.getState()
|
|
.updateEnableGrab(preference.enableGrab);
|
|
|
|
const unsubscribeQuoteStore = subscribeQuoteUpdate(
|
|
{
|
|
useQuoteStore: storeSet.useQuoteStore,
|
|
},
|
|
eventCallbacks,
|
|
);
|
|
|
|
const ctx = {
|
|
grabPluginId,
|
|
storeSet,
|
|
eventCallbacks,
|
|
unsubscribe: () => {
|
|
unsubscribeQuoteStore();
|
|
},
|
|
eventCenter,
|
|
publicEventCenter,
|
|
scene,
|
|
};
|
|
|
|
return ctx;
|
|
},
|
|
Plugin: ChatAreaGrabPlugin,
|
|
};
|
|
|
|
return { grabPlugin, grabPluginId };
|
|
};
|