feat: manually mirror opencoze's code from bytedance

Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf

frontend/packages/data/knowledge/knowledge-modal-base/src/batch-frequency-modal/hooks.tsx

@@ -0,0 +1,146 @@
+/*
+ * 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 { useMemo, useState } from 'react';
+
+import { useRequest } from 'ahooks';
+import {
+  getUpdateIntervalOptions,
+  isFeishuOrLarkDocumentSource,
+  useDataModalWithCoze,
+} from '@coze-data/utils';
+import { DataNamespace, dataReporter } from '@coze-data/reporter';
+import { KnowledgeE2e } from '@coze-data/e2e';
+import { REPORT_EVENTS } from '@coze-arch/report-events';
+import { I18n } from '@coze-arch/i18n';
+import { DocumentSource, UpdateType } from '@coze-arch/bot-api/knowledge';
+import { KnowledgeApi } from '@coze-arch/bot-api';
+import { Select, Toast } from '@coze-arch/coze-design';
+
+import {
+  useBatchCheckboxDoc,
+  type IBatchCheckboxDocProps,
+} from '../batch-checkbox-doc';
+
+import styles from './index.module.less';
+
+export type TBatchFrequencyModalProps = IBatchCheckboxDocProps & {
+  onfinish: () => void;
+};
+
+export const useBatchFrequencyModal = (props: TBatchFrequencyModalProps) => {
+  const { documentList, onfinish } = props;
+  const [updateInterval, setUpdateInterval] = useState<number>(0);
+
+  const documents = useMemo(
+    () =>
+      documentList?.filter(
+        item =>
+          [
+            DocumentSource.Web,
+            DocumentSource.FrontCrawl,
+            // DocumentSource.ThirdParty,
+          ].includes(item?.source_type as DocumentSource) ||
+          isFeishuOrLarkDocumentSource(item?.source_type),
+      ),
+    [documentList],
+  );
+
+  const { node, checkedList, initCheckedList } = useBatchCheckboxDoc({
+    documentList: documents,
+    showTag: true,
+  });
+
+  const { loading, run } = useRequest(
+    async () => {
+      await KnowledgeApi.BatchUpdateDocument({
+        document_ids: checkedList,
+        update_rule: {
+          update_type: updateInterval ? UpdateType.Cover : UpdateType.NoUpdate,
+          update_interval: updateInterval,
+        },
+      });
+    },
+    {
+      onSuccess: () => {
+        Toast.success(I18n.t('Update_success'));
+        close();
+        onfinish();
+        initModalData();
+      },
+      onError: error => {
+        dataReporter.errorEvent(DataNamespace.KNOWLEDGE, {
+          eventName: REPORT_EVENTS.KnowledgeBatchUpdateDocument,
+          error,
+        });
+      },
+      manual: true,
+    },
+  );
+  const initModalData = () => {
+    setUpdateInterval(0);
+    initCheckedList();
+  };
+
+  const { modal, open, close } = useDataModalWithCoze({
+    className: styles['batch-update-frequency-modal'],
+    title: I18n.t('knowledge_optimize_014'),
+    centered: true,
+    cancelText: I18n.t('Cancel'),
+    okText: I18n.t('knowledge_optimize_007'),
+    okButtonProps: {
+      disabled: !checkedList.length,
+      loading,
+    },
+    onOk: () => {
+      run();
+    },
+    onCancel: () => {
+      close();
+      initModalData();
+    },
+  });
+  return {
+    node: modal(
+      <div className={styles['batch-update-frequency']}>
+        <div className={styles['batch-update-frequency-title']}>
+          {I18n.t('knowledge_optimize_015')}
+        </div>
+        <div className={styles['batch-update-frequency-content']}>
+          <div className={styles['batch-update-frequency-content-select']}>
+            <div
+              className={styles['batch-update-frequency-content-select-label']}
+            >
+              {I18n.t('datasets_frequencyModal_frequency')}
+              <span className={styles['frequency-label-required']}>*</span>
+            </div>
+            <Select
+              data-testid={KnowledgeE2e.SegmentDetailBatchFrequencyModalSelect}
+              style={{ width: '100%' }}
+              value={updateInterval}
+              onChange={v => setUpdateInterval(v as number)}
+              placeholder={I18n.t('datasets_frequencyModal_frequency')}
+              optionList={getUpdateIntervalOptions()}
+            ></Select>
+          </div>
+          {node}
+        </div>
+      </div>,
+    ),
+    open,
+    close,
+  };
+};