@zjlab-fe/data-hub-ui 0.29.0 → 0.31.0

package/es/components/file-uploader/context/use-upload-engine.js

@@ -5,7 +5,7 @@ import { UploadStatus } from '../constants.js';
 import { createUploadError, UPLOAD_ERROR_CODE, parseError } from '../utils/error-handler.js';
 import { handlerRegistry } from './handler-registry.js';
 import { actionRegistry } from './action-registry.js';
-import { resetInterruptedChunks, getChunksForResume } from '../utils/upload-controllers.js';
+import { resetInterruptedChunks, getChunksForResume, getFailedChunks, resetFailedChunks, resetAllChunks } from '../utils/upload-controllers.js';
 import { fileStateStore } from './file-state-store.js';
 import { ensureHandlers } from '../utils/ensure-handlers.js';
 
@@ -25,6 +25,7 @@ function getHandlerContext() {
             pause: () => { },
             resume: () => __awaiter(this, void 0, void 0, function* () { }),
             cancel: () => { },
+            retry: () => __awaiter(this, void 0, void 0, function* () { return ({ success: false, retryType: 'smart' }); }),
             getFileById: () => undefined,
         };
     }
@@ -291,135 +292,111 @@ function useUploadEngine() {
      * @param forceFullRetry - Skip smart retry and do full retry immediately
      * @returns Object with success status and retry type used
      */
-    // const retryUpload = useCallback(
-    //   async (
-    //     fileId: string,
-    //     forceFullRetry: boolean = false,
-    //   ): Promise<{ success: boolean; retryType: 'smart' | 'full' }> => {
-    //     const file = stateRef.current.files.find((f) => f.id === fileId);
-    //     if (!file || !file.file) {
-    //       const error = createUploadError(UPLOAD_ERROR_CODE.MISSING_FILE, 'File is missing, cannot retry');
-    //       console.error(error.message);
-    //       return { success: false, retryType: 'smart' };
-    //     }
-    //     const h = ensureHandlers(fileId, file);
-    //     if (!h) {
-    //       return { success: false, retryType: 'smart' };
-    //     }
-    //     // Increment retry count
-    //     const retryCount = (file.retryCount ?? 0) + 1;
-    //     const failedChunksResult = getFailedChunks(file);
-    //     // If forcing full retry or no failed chunks, do full retry
-    //     if (forceFullRetry || !failedChunksResult.hasFailed) {
-    //       return performFullRetry(file, h, retryCount);
-    //     }
-    //     // ─────────────────────────────────────────────────────────────────────────
-    //     // SMART RETRY: Retry only failed chunks
-    //     // ─────────────────────────────────────────────────────────────────────────
-    //     try {
-    //       // Reset only failed chunks to pending
-    //       const updatedChunks = resetFailedChunks(file.chunks);
-    //       dispatch({
-    //         type: 'UPDATE_FILE',
-    //         id: fileId,
-    //         updates: {
-    //           status: UploadStatus.UPLOADING,
-    //           chunks: updatedChunks,
-    //           error: undefined,
-    //           errorCode: undefined,
-    //           retryCount,
-    //         },
-    //       });
-    //       // Get fresh URLs for all chunks (server needs full chunk count)
-    //       const urlResult = await h.onGetUploadUrls(file.fileName, file.chunks.length);
-    //       if (!urlResult.status) {
-    //         // URL request failed - try full retry as fallback
-    //         console.warn('Smart retry URL fetch failed, falling back to full retry');
-    //         return performFullRetry(file, h, retryCount);
-    //       }
-    //       // Filter URLs to only failed chunks
-    //       const failedUrls = filterUrlsForPendingChunks(
-    //         urlResult.uploadInfo.uploadUrls,
-    //         failedChunksResult.failedPartNums,
-    //       );
-    //       // Re-register handlers for background upload support
-    //       handlerRegistry.register(fileId, h, 'retryUpload:smart');
-    //       // Clear stale queue items and re-add with failed chunks only
-    //       const qm = getQueueManager();
-    //       qm.clearFileFromQueue(fileId);
-    //       const fileWithUrls = {
-    //         ...file,
-    //         uploadId: urlResult.uploadInfo.uploadId,
-    //         chunks: updatedChunks,
-    //         retryCount,
-    //       };
-    //       qm.addFile(fileWithUrls, failedUrls);
-    //       h.onRetry?.(file, 'smart');
-    //       return { success: true, retryType: 'smart' };
-    //     } catch (error) {
-    //       // Smart retry failed - fallback to full retry
-    //       const uploadError = parseError(error, 'Smart retry failed');
-    //       console.warn('Smart retry failed, falling back to full retry:', uploadError.message);
-    //       return performFullRetry(file, h, retryCount);
-    //     }
-    //     // ─────────────────────────────────────────────────────────────────────────
-    //     // FULL RETRY: Reset all chunks and start from beginning
-    //     // ─────────────────────────────────────────────────────────────────────────
-    //     async function performFullRetry(
-    //       uploadFile: UploadFile,
-    //       handlers: UploadHandlers,
-    //       retryAttempt: number,
-    //     ): Promise<{ success: boolean; retryType: 'full' }> {
-    //       try {
-    //         // Reset all chunks to pending
-    //         const resetChunks = resetAllChunks(uploadFile.chunks);
-    //         dispatch({
-    //           type: 'UPDATE_FILE',
-    //           id: fileId,
-    //           updates: {
-    //             status: UploadStatus.PENDING,
-    //             progress: 0,
-    //             error: undefined,
-    //             errorCode: undefined,
-    //             uploadId: '',
-    //             chunks: resetChunks,
-    //             retryCount: retryAttempt,
-    //           },
-    //         });
-    //         // Use startUpload for full retry
-    //         const resetFile: UploadFile = {
-    //           ...uploadFile,
-    //           status: UploadStatus.PENDING,
-    //           progress: 0,
-    //           error: undefined,
-    //           errorCode: undefined,
-    //           uploadId: '',
-    //           chunks: resetChunks,
-    //           retryCount: retryAttempt,
-    //         };
-    //         await startUpload(resetFile);
-    //         handlers.onRetry?.(uploadFile, 'full');
-    //         return { success: true, retryType: 'full' };
-    //       } catch (error) {
-    //         const uploadError = parseError(error, 'Retry failed');
-    //         console.error('Full retry failed:', uploadError.message);
-    //         dispatch({
-    //           type: 'UPDATE_FILE',
-    //           id: fileId,
-    //           updates: {
-    //             status: UploadStatus.FAILED,
-    //             error: uploadError.message,
-    //             errorCode: uploadError.code,
-    //             retryCount: retryAttempt,
-    //           },
-    //         });
-    //         handlers.onError?.(uploadFile, uploadError.originalError ?? new Error(uploadError.message));
-    //         return { success: false, retryType: 'full' };
-    //       }
-    //     }
-    //   },
-    //   [dispatch, getQueueManager, startUpload, ensureHandlers],
-    // );
+    const retryUpload = useCallback((fileId_1, ...args_1) => __awaiter(this, [fileId_1, ...args_1], void 0, function* (fileId, forceFullRetry = false) {
+        var _a, _b;
+        const file = fileStateStore.getFile(fileId);
+        if (!file || !file.file) {
+            const error = createUploadError(UPLOAD_ERROR_CODE.MISSING_FILE, 'File is missing, cannot retry');
+            console.error(error.message);
+            return { success: false, retryType: 'smart' };
+        }
+        if (file.status !== UploadStatus.FAILED) {
+            return { success: false, retryType: 'smart' };
+        }
+        const h = ensureHandlers(file, dispatch);
+        if (!h) {
+            return { success: false, retryType: 'smart' };
+        }
+        const retryCount = ((_a = file.retryCount) !== null && _a !== void 0 ? _a : 0) + 1;
+        const failedChunksResult = getFailedChunks(file);
+        if (forceFullRetry || !failedChunksResult.hasFailed) {
+            return performFullRetry(file, h, retryCount);
+        }
+        // ─────────────────────────────────────────────────────────────────────────
+        // SMART RETRY: Retry only failed chunks
+        // ─────────────────────────────────────────────────────────────────────────
+        try {
+            const updatedChunks = resetFailedChunks(file.chunks);
+            const urlResult = yield h.onGetUploadUrls(file.fileName, file.chunks.length, file);
+            if (!urlResult.success) {
+                console.warn('Smart retry URL fetch failed, falling back to full retry');
+                return performFullRetry(file, h, retryCount);
+            }
+            const { uploadId, uploadUrls } = urlResult.uploadInfo;
+            const chunksWithUrls = applyUploadUrlsToChunks(updatedChunks, uploadUrls);
+            dispatch({
+                type: 'UPDATE_FILE',
+                id: fileId,
+                updates: {
+                    status: UploadStatus.UPLOADING,
+                    uploadId,
+                    chunks: chunksWithUrls,
+                    error: undefined,
+                    errorCode: undefined,
+                    retryCount,
+                },
+            });
+            handlerRegistry.register(fileId, h, 'retryUpload:smart');
+            const qm = getQueueManager();
+            qm.clearFileFromQueue(fileId);
+            const fileWithUrls = Object.assign(Object.assign({}, file), { uploadId, chunks: chunksWithUrls, status: UploadStatus.UPLOADING, retryCount, error: undefined, errorCode: undefined });
+            qm.addFile(fileWithUrls);
+            const ctx = getHandlerContext();
+            (_b = h.onRetry) === null || _b === void 0 ? void 0 : _b.call(h, file, 'smart', ctx);
+            return { success: true, retryType: 'smart' };
+        }
+        catch (error) {
+            const uploadError = parseError(error, 'Smart retry failed');
+            console.warn('Smart retry failed, falling back to full retry:', uploadError.message);
+            return performFullRetry(file, h, retryCount);
+        }
+        // ─────────────────────────────────────────────────────────────────────────
+        // FULL RETRY: Reset all chunks and start from beginning
+        // ─────────────────────────────────────────────────────────────────────────
+        function performFullRetry(uploadFile, handlers, retryAttempt) {
+            return __awaiter(this, void 0, void 0, function* () {
+                var _a, _b, _c;
+                try {
+                    const resetChunks = resetAllChunks(uploadFile.chunks);
+                    dispatch({
+                        type: 'UPDATE_FILE',
+                        id: fileId,
+                        updates: {
+                            status: UploadStatus.PENDING,
+                            progress: 0,
+                            error: undefined,
+                            errorCode: undefined,
+                            uploadId: '',
+                            chunks: resetChunks,
+                            retryCount: retryAttempt,
+                        },
+                    });
+                    const resetFile = Object.assign(Object.assign({}, uploadFile), { status: UploadStatus.PENDING, progress: 0, error: undefined, errorCode: undefined, uploadId: '', chunks: resetChunks, retryCount: retryAttempt });
+                    const ctx = getHandlerContext();
+                    (_a = handlers.onRetry) === null || _a === void 0 ? void 0 : _a.call(handlers, uploadFile, 'full', ctx);
+                    yield startUpload(resetFile);
+                    return { success: true, retryType: 'full' };
+                }
+                catch (error) {
+                    const uploadError = parseError(error, 'Retry failed');
+                    console.error('Full retry failed:', uploadError.message);
+                    dispatch({
+                        type: 'UPDATE_FILE',
+                        id: fileId,
+                        updates: {
+                            status: UploadStatus.FAILED,
+                            error: uploadError.message,
+                            errorCode: uploadError.code,
+                            retryCount: retryAttempt,
+                        },
+                    });
+                    const ctx = getHandlerContext();
+                    (_b = handlers.onError) === null || _b === void 0 ? void 0 : _b.call(handlers, uploadFile, (_c = uploadError.originalError) !== null && _c !== void 0 ? _c : new Error(uploadError.message), ctx);
+                    return { success: false, retryType: 'full' };
+                }
+            });
+        }
+    }), [dispatch, getQueueManager, startUpload]);
     // ─────────────────────────────────────────────────────────────────────────────
     // PUBLIC API
     // ─────────────────────────────────────────────────────────────────────────────
@@ -433,7 +410,7 @@ function useUploadEngine() {
         /** Cancel a file upload */
         cancelUpload,
         /** Retry a failed upload (smart retry with full retry fallback) */
-        // retryUpload,
+        retryUpload,
     };
 }
 

package/es/components/file-uploader/hooks/use-upload.js

@@ -60,7 +60,7 @@ import { useUploadEngine } from '../context/use-upload-engine.js';
  */
 function useUpload() {
     const { state, dispatch, config } = useUploadContext();
-    const { startUpload, pauseUpload, resumeUpload, cancelUpload } = useUploadEngine();
+    const { startUpload, pauseUpload, resumeUpload, cancelUpload, retryUpload } = useUploadEngine();
     // ─────────────────────────────────────────────────────────────────────────────
     // FILE OPERATIONS
     // ─────────────────────────────────────────────────────────────────────────────
@@ -172,6 +172,18 @@ function useUpload() {
      * @param id - File ID to cancel
      */
     const cancel = useCallback((id) => cancelUpload(id), [cancelUpload]);
+    /**
+     * Retry a failed upload (smart retry with full retry fallback).
+     *
+     * @param id - File ID to retry
+     */
+    const retry = useCallback((id) => __awaiter(this, void 0, void 0, function* () {
+        const file = state.files.find((f) => f.id === id);
+        if (file && file.status === UploadStatus.FAILED) {
+            return retryUpload(id);
+        }
+        return { success: false, retryType: 'smart' };
+    }), [state.files, retryUpload]);
     // ─────────────────────────────────────────────────────────────────────────────
     // MUTEX FLAGS - Prevent concurrent batch operations
     // ─────────────────────────────────────────────────────────────────────────────
@@ -280,6 +292,7 @@ function useUpload() {
             pause,
             resume,
             cancel,
+            retry,
             getFileById,
         });
     });
@@ -297,6 +310,7 @@ function useUpload() {
         pause,
         resume,
         cancel,
+        retry,
         /** Get a specific file by ID */
         getFileById,
         /** Files waiting to be uploaded */

package/es/components/file-uploader/utils/upload-controllers.js

@@ -39,5 +39,47 @@ function resetInterruptedChunks(chunks) {
         return chunk;
     });
 }
+/**
+ * Get chunks that failed and need to be retried.
+ * Returns only FAILED chunks, preserves COMPLETED chunks.
+ */
+function getFailedChunks(file) {
+    const failedPartNums = [];
+    let completedCount = 0;
+    for (const chunk of file.chunks) {
+        if (chunk.status === ChunkStatus.COMPLETED) {
+            completedCount++;
+        }
+        else if (chunk.status === ChunkStatus.FAILED && chunk.partNum !== undefined) {
+            failedPartNums.push(chunk.partNum);
+        }
+    }
+    return {
+        failedPartNums,
+        failedCount: failedPartNums.length,
+        completedCount,
+        hasFailed: failedPartNums.length > 0,
+    };
+}
+/**
+ * Reset only failed chunks to pending state.
+ * Preserves completed chunks for efficient retry.
+ * Returns updated chunks array.
+ */
+function resetFailedChunks(chunks) {
+    return chunks.map((chunk) => {
+        if (chunk.status === ChunkStatus.FAILED) {
+            return Object.assign(Object.assign({}, chunk), { status: ChunkStatus.PENDING, progress: 0, error: undefined });
+        }
+        return chunk;
+    });
+}
+/**
+ * Reset all chunks to pending state (for full retry).
+ * Used when smart retry fails or server state is lost.
+ */
+function resetAllChunks(chunks) {
+    return chunks.map((chunk) => (Object.assign(Object.assign({}, chunk), { status: ChunkStatus.PENDING, progress: 0, error: undefined })));
+}
 
-export { getChunksForResume, resetInterruptedChunks };
+export { getChunksForResume, getFailedChunks, resetAllChunks, resetFailedChunks, resetInterruptedChunks };

package/es/components/limit-credit-modal/index.js

@@ -0,0 +1,36 @@
+import { jsx, jsxs } from 'react/jsx-runtime';
+import { Modal, Space, Button } from 'antd';
+import { isPublicNetwork } from '@zjlab-fe/util';
+import { createRoot } from 'react-dom/client';
+
+const LimitCreditModal = (props) => {
+    const isPublic = isPublicNetwork();
+    const handleOk = () => {
+        if (isPublic) {
+            window.open('https://alidocs.dingtalk.com/notable/share/form/v018oLl952Q1vXkzlap_dv19yqvsgs3oebp3pcjys_1qX0QQ0?source=link');
+        }
+        else {
+            window.open('https://alidocs.dingtalk.com/notable/share/form/v011X3lE5jKezRB4lJb_dv19yqvsgs3oebp3pcjys_1qX0QQ0?source=link');
+        }
+    };
+    return (jsx(Modal, Object.assign({}, props, { 
+        // classNames={{
+        //   content: 'modal-content-layout',
+        //   header: 'modal-title-font modal-title-layout',
+        //   body: 'modal-content-font modal-body-layout',
+        //   footer: 'modal-footer-layout',
+        // }}
+        title: jsx("div", { style: { textAlign: 'center', width: '100%' }, children: "\u63D0\u793A" }), footer: jsx("div", { style: { display: 'flex', justifyContent: 'center' }, children: jsxs(Space, { children: [jsx(Button, { onClick: props.onCancel, children: "\u53D6\u6D88" }), jsx(Button, { type: "primary", onClick: handleOk, children: "\u7533\u8BF7" })] }) }), children: jsx("div", { className: "modal-content-font", style: { textAlign: 'center' }, children: "\u989D\u5EA6\u5DF2\u7528\u5C3D\uFF0C\u8BF7\u70B9\u51FB\u7533\u8BF7\u3002" }) })));
+};
+LimitCreditModal.open = () => {
+    const div = document.createElement('div');
+    document.body.appendChild(div);
+    const root = createRoot(div);
+    const destroy = () => {
+        root.unmount();
+        div.remove();
+    };
+    root.render(jsx(LimitCreditModal, { open: true, onCancel: destroy }));
+};
+
+export { LimitCreditModal as default };

package/es/components/test-component/index.js

@@ -0,0 +1,9 @@
+import { jsxs, jsx } from 'react/jsx-runtime';
+import useStyle from './index.scss.js';
+
+useStyle();
+const TestComponent = (props) => {
+    return (jsxs("div", { className: "data-hub-ui-test-component-container", children: ["TestComponent Component", jsx("div", { children: "This is template, please delete it after you create your component." })] }));
+};
+
+export { TestComponent as default };

package/es/components/test-component/index.scss.js

@@ -0,0 +1,7 @@
+function useStyle() {
+  var style = document.createElement('style');
+  style.textContent = ``;
+  document.head.appendChild(style);
+}
+
+export { useStyle as default };

package/es/index.js

@@ -9,6 +9,7 @@ export { Uploader as FileUploader } from './components/file-uploader/components/
 export { UploaderDropZone } from './components/file-uploader/components/uploader-drop-zone.js';
 export { UploaderFileListLayout } from './components/file-uploader/components/uploader-file-list.js';
 export { UploaderFileItem } from './components/file-uploader/components/uploader-file-item.js';
+export { UploaderProgressBar } from './components/file-uploader/components/uploader-progress-bar.js';
 export { useUpload } from './components/file-uploader/hooks/use-upload.js';
 export { UploadStatus } from './components/file-uploader/constants.js';
 export { default as FilePreview } from './components/file-preview/index.js';
@@ -36,5 +37,6 @@ export { default as ModelCard } from './components/model-card/index.js';
 export { default as SDKModal } from './components/SDK-modal/index.js';
 export { default as TagView } from './components/tag-view/index.js';
 export { default as TagGroupFilter } from './components/tag-group-filter/index.js';
+export { default as LimitCreditModal } from './components/limit-credit-modal/index.js';
 export { default as DatasetBatchAction, EDatasetBatchType } from './components/dataset-batch-action/index.js';
 export { default as UploadDrawerUploadStoreProvider } from './components/uploadDrawer/UploadStoreProvider.js';