feat: manually mirror opencoze's code from bytedance
Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf
This commit is contained in:
@@ -0,0 +1,171 @@
|
||||
/*
|
||||
* 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 { useState } from 'react';
|
||||
|
||||
import { useProjectId, useSpaceId } from '@coze-project-ide/framework';
|
||||
import { usePrimarySidebarStore } from '@coze-project-ide/biz-components';
|
||||
import { FilterKnowledgeType } from '@coze-data/knowledge-modal-base';
|
||||
import { useKnowledgeListModalContent } from '@coze-data/knowledge-modal-adapter';
|
||||
import { useSelectDatabaseModal } from '@coze-data/database-v2';
|
||||
import { I18n } from '@coze-arch/i18n';
|
||||
import { UITabsModal } from '@coze-arch/bot-semi';
|
||||
import { ResType, ResourceCopyScene } from '@coze-arch/bot-api/plugin_develop';
|
||||
|
||||
import SiderCategory from '../components/sider-category';
|
||||
import { useResourceOperation } from './use-resource-operation';
|
||||
|
||||
type TabType = 'knowledge' | 'database';
|
||||
|
||||
export const ImportDataModal = (props: { onClose?: () => void }) => {
|
||||
const { onClose } = props;
|
||||
const [activeKey, setActiveKey] = useState<TabType>('knowledge');
|
||||
|
||||
const projectID = useProjectId();
|
||||
const spaceID = useSpaceId();
|
||||
|
||||
const refetch = usePrimarySidebarStore(state => state.refetch);
|
||||
|
||||
const resourceOperation = useResourceOperation({ projectId: projectID });
|
||||
|
||||
const { renderContent, renderSearch } = useKnowledgeListModalContent({
|
||||
showFilters: ['scope-type', 'search-type'],
|
||||
datasetList: [],
|
||||
onDatasetListChange: async list => {
|
||||
if (list[0].dataset_id && list[0].name) {
|
||||
await resourceOperation({
|
||||
scene: ResourceCopyScene.CopyResourceFromLibrary,
|
||||
resource: {
|
||||
id: list[0].dataset_id,
|
||||
res_id: list[0].dataset_id,
|
||||
name: list[0].name,
|
||||
res_type: ResType.Knowledge,
|
||||
},
|
||||
});
|
||||
refetch();
|
||||
onClose?.();
|
||||
}
|
||||
},
|
||||
canCreate: false,
|
||||
defaultType: FilterKnowledgeType.ALL,
|
||||
});
|
||||
|
||||
const {
|
||||
renderContent: renderDatabaseContent,
|
||||
renderFilter: renderDatabaseFilter,
|
||||
renderInput: renderDatabaseInput,
|
||||
} = useSelectDatabaseModal({
|
||||
visible: true,
|
||||
onClose: () => void 0,
|
||||
onAddDatabase: async id => {
|
||||
await resourceOperation({
|
||||
scene: ResourceCopyScene.CopyResourceFromLibrary,
|
||||
resource: {
|
||||
id,
|
||||
res_id: id,
|
||||
name: '',
|
||||
res_type: ResType.Database,
|
||||
},
|
||||
});
|
||||
refetch();
|
||||
onClose?.();
|
||||
},
|
||||
onClickDatabase: () => void 0,
|
||||
enterFrom: 'project',
|
||||
spaceId: spaceID,
|
||||
});
|
||||
|
||||
return (
|
||||
<UITabsModal
|
||||
keepDOM={false}
|
||||
visible
|
||||
onCancel={onClose}
|
||||
tabs={{
|
||||
tabsProps: {
|
||||
lazyRender: true,
|
||||
activeKey,
|
||||
onChange: (key: string) => setActiveKey(key as TabType),
|
||||
},
|
||||
tabPanes: [
|
||||
{
|
||||
tabPaneProps: {
|
||||
tab: I18n.t('resource_type_knowledge'),
|
||||
itemKey: 'knowledge',
|
||||
},
|
||||
content: (
|
||||
<div className="w-full h-full flex">
|
||||
<div className="w-[218px] pt-[16px] px-[12px] shrink-0 flex flex-col gap-[12px]">
|
||||
{renderSearch()}
|
||||
<SiderCategory
|
||||
label={I18n.t('project_resource_modal_library_resources', {
|
||||
resource: I18n.t('resource_type_knowledge'),
|
||||
})}
|
||||
selected
|
||||
></SiderCategory>
|
||||
</div>
|
||||
<div className="grow-[1] bg-[white] pt-[16px] px-[12px]">
|
||||
{renderContent()}
|
||||
</div>
|
||||
</div>
|
||||
),
|
||||
},
|
||||
{
|
||||
tabPaneProps: {
|
||||
tab: I18n.t('resource_type_database'),
|
||||
itemKey: 'database',
|
||||
},
|
||||
content: (
|
||||
<div className="w-full h-full flex">
|
||||
<div className="w-[218px] pt-[16px] px-[12px] shrink-0 flex flex-col gap-[12px]">
|
||||
{renderDatabaseInput()}
|
||||
<SiderCategory
|
||||
label={I18n.t('project_resource_modal_library_resources', {
|
||||
resource: I18n.t('resource_type_database'),
|
||||
})}
|
||||
selected
|
||||
></SiderCategory>
|
||||
</div>
|
||||
<div className="grow-[1] bg-[white] pt-[16px] px-[12px] flex flex-col">
|
||||
{renderDatabaseFilter()}
|
||||
{renderDatabaseContent()}
|
||||
</div>
|
||||
</div>
|
||||
),
|
||||
},
|
||||
],
|
||||
}}
|
||||
/>
|
||||
);
|
||||
};
|
||||
|
||||
const useImportData = () => {
|
||||
const [visible, setVisible] = useState(false);
|
||||
|
||||
const open = () => {
|
||||
setVisible(true);
|
||||
};
|
||||
|
||||
const close = () => {
|
||||
setVisible(false);
|
||||
};
|
||||
return {
|
||||
modal: visible ? <ImportDataModal onClose={close} /> : null,
|
||||
open,
|
||||
close,
|
||||
};
|
||||
};
|
||||
|
||||
export default useImportData;
|
||||
Reference in New Issue
Block a user