pollination-react-io 1.28.6 → 1.30.0

package/build/FilePreview/FilePreview.types.d.ts

@@ -9,4 +9,6 @@ export declare type FilePreviewProps = {
     inputProps?: HTMLProps<HTMLInputElement>;
     onPreviewChange?: (...any: any) => void;
     client?: APIClient;
+    enableFilePicker?: boolean;
+    localPath?: string;
 };

package/build/hooks/usePollinationPanel.d.ts

@@ -2,4 +2,5 @@ import { PanelMessageOut } from './types';
 export declare const usePollinationPanel: () => {
     getFileBase64: (key?: any, accept?: any) => PanelMessageOut;
     saveFileBase64: (key: string, file: File | Blob, subFolder: string, fileName: string) => Promise<PanelMessageOut>;
+    fromFileToBase64: (filePath: string) => PanelMessageOut;
 };

package/build/index.esm.js

@@ -37948,9 +37948,15 @@ var usePollinationPanel = function () {
             return;
         return panel.GetArtifactAsBase64(key, accept);
     }, [panel]);
+    var fromFileToBase64 = useCallback(function (filePath) {
+        if (!panel)
+            return;
+        return panel.FromFileToBase64(filePath);
+    }, [panel]);
     return {
         getFileBase64: getFileBase64,
-        saveFileBase64: saveFileBase64
+        saveFileBase64: saveFileBase64,
+        fromFileToBase64: fromFileToBase64
     };
 };
 
@@ -40029,8 +40035,13 @@ var useGetGeometry = function () {
     var host = getHost();
     var _a = useState(undefined), geometry = _a[0], setGeometry = _a[1];
     var geometryRef = useRef(undefined);
+    var runPreview = function (identifier) {
+        if (typeof window.parent.chrome.webview.hostObjects.sync.geometry.Mark === 'function') {
+            window.parent.chrome.webview.hostObjects.sync.geometry.Mark(identifier);
+        }
+    };
     var setReturnVal = useCallback(function (_a, force) {
-        var data = _a.data;
+        var data = _a.data; _a.identifier;
         var bytes = new TextEncoder().encode(data);
         if (!force && geometryRef.current) {
             if (!lodash_isequal(bytes, geometryRef.current)) {
@@ -40069,6 +40080,7 @@ var useGetGeometry = function () {
         };
         res = window.parent.chrome.webview.hostObjects.sync.geometry.Get(key, selection);
         if (res) {
+            runPreview(res.identifier);
             setReturnVal(res, force);
         }
     }, [setReturnVal]);
@@ -47164,8 +47176,11 @@ var jszip_min = createCommonjsModule(function (module, exports) {
 var shortDebounce = useDebounce(200);
 var ICON_SIZE = 18;
 var FilePreview = function (_a) {
-    var projectOwner = _a.projectOwner, projectName = _a.projectName, jobId = _a.jobId, value = _a.value, client = _a.client, inputProps = _a.inputProps, onPreviewChange = _a.onPreviewChange // Use it for streamlit setComponentValue
+    var projectOwner = _a.projectOwner, projectName = _a.projectName, jobId = _a.jobId, value = _a.value, // cloud controller
+    client = _a.client, inputProps = _a.inputProps, onPreviewChange = _a.onPreviewChange, // Use it for streamlit setComponentValue
+    enableFilePicker = _a.enableFilePicker, localPath = _a.localPath // local controller
     ;
+    var fromFileToBase64 = usePollinationPanel().fromFileToBase64;
     var getExtension = function (fileName) {
         if (!fileName)
             return;
@@ -47174,13 +47189,14 @@ var FilePreview = function (_a) {
             return arr.at(arr.length - 1);
         }
     };
+    var host = getHost();
     var _b = useState(value), selArtifact = _b[0], setSelArtifact = _b[1];
+    var _c = useState(localPath), selFilePath = _c[0], setSelFilePath = _c[1];
     var downloadArtifact = useArtifacts(projectOwner, projectName, client).downloadArtifact;
-    var _c = useState(), preview = _c[0], setPreview = _c[1];
-    var _d = useState(), currBlob = _d[0], setCurrBlob = _d[1];
-    var _e = useState(getExtension(value === null || value === void 0 ? void 0 : value.file_name)), currFileExtension = _e[0], setCurrFileExtension = _e[1];
+    var _d = useState(), preview = _d[0], setPreview = _d[1];
+    var _e = useState(), currBlob = _e[0], setCurrBlob = _e[1];
+    var _f = useState(getExtension((value === null || value === void 0 ? void 0 : value.file_name) || getExtension(localPath))), currFileExtension = _f[0], setCurrFileExtension = _f[1];
     var allowedFileTypes = ['application/zip', 'text/plain'];
-    var host = getHost();
     var parser = {
         'text/plain': function (blob) {
             var reader = new FileReader();
@@ -47220,13 +47236,70 @@ var FilePreview = function (_a) {
             }); });
         }
     };
-    // Internal onChange
+    // Internal onChange cloud
     var onChange = useCallback(function (selArtifact) {
-        // console.log(selArtifact)
         setSelArtifact(selArtifact);
         !selArtifact && setPreview(undefined);
         !selArtifact && setCurrFileExtension(undefined);
     }, []);
+    // Internal onChange local
+    var onChangeLocal = useCallback(function (path) {
+        setSelFilePath(path);
+        !selFilePath && setPreview(undefined);
+        !selFilePath && setCurrFileExtension(undefined);
+    }, []);
+    var base64AndFetch = function (path) {
+        if (!path)
+            return;
+        var data = fromFileToBase64(path);
+        var base64Data = data === null || data === void 0 ? void 0 : data.data;
+        if (!base64Data)
+            return;
+        // bas64 mapping
+        var t;
+        switch (getExtension(path)) {
+            case 'zip':
+                t = 'application/zip';
+                break;
+            case 'vsf':
+                t = 'text/plain';
+        }
+        fetchAndSet("data:".concat(t, ";base64,").concat(base64Data), path);
+    };
+    var fetchAndSet = function (url, path) {
+        if (path === void 0) { path = undefined; }
+        fetch(url)
+            .then(function (response) { return response.blob()
+            .then(function (blob) {
+            var t = blob.type;
+            if (!allowedFileTypes.includes(t)) {
+                console.error("Cloud Artifact ".concat(t, " not supported."));
+                setPreview(undefined);
+                setCurrBlob(undefined);
+                setCurrFileExtension(undefined);
+                return;
+            }
+            switch (t) {
+                case 'application/zip':
+                    parser['application/zip'](blob);
+                    break;
+                case 'text/plain':
+                    parser['text/plain'](blob);
+                    break;
+                default:
+                    setPreview(undefined);
+                    setCurrBlob(undefined);
+                    setCurrFileExtension(undefined);
+                    break;
+            }
+            setCurrBlob(blob);
+            if (path) {
+                setSelFilePath(path);
+            }
+            var ext = getExtension((selArtifact === null || selArtifact === void 0 ? void 0 : selArtifact.file_name) || path);
+            setCurrFileExtension(ext);
+        }); });
+    };
     var download = function (blob, fileName) {
         var file = new File([blob], fileName);
         var a = document.createElement('a');
@@ -47239,54 +47312,31 @@ var FilePreview = function (_a) {
         a.click();
     };
     useEffect(function () {
+        if (enableFilePicker) {
+            base64AndFetch(selFilePath);
+            return;
+        }
         downloadArtifact(selArtifact === null || selArtifact === void 0 ? void 0 : selArtifact.key, jobId)
-            .then(function (url) {
-            fetch(url)
-                .then(function (response) { return response.blob()
-                .then(function (blob) {
-                var t = blob.type;
-                if (!allowedFileTypes.includes(t)) {
-                    console.error("Cloud Artifact ".concat(t, " not supported."));
-                    setPreview(undefined);
-                    setCurrBlob(undefined);
-                    setCurrFileExtension(undefined);
-                    return;
-                }
-                switch (t) {
-                    case 'application/zip':
-                        parser['application/zip'](blob);
-                        break;
-                    case 'text/plain':
-                        parser['text/plain'](blob);
-                        break;
-                    default:
-                        setPreview(undefined);
-                        setCurrBlob(undefined);
-                        setCurrFileExtension(undefined);
-                        break;
-                }
-                setCurrBlob(blob);
-                var ext = getExtension(selArtifact === null || selArtifact === void 0 ? void 0 : selArtifact.file_name);
-                setCurrFileExtension(ext);
-            }); });
-        });
-    }, [selArtifact, projectOwner, projectName]);
+            .then(fetchAndSet);
+    }, [selArtifact, selFilePath, projectOwner, projectName]);
     shortDebounce(function () {
         onPreviewChange && onPreviewChange(preview);
     }, [onPreviewChange, preview]);
-    // TODO: Add buttons latest part (e.g. download)
     return (React__default.createElement(React__default.Fragment, null,
-        !value && React__default.createElement(SelectCloudArtifacts, { projectOwner: projectOwner, projectName: projectName, value: selArtifact, hidden: false, client: client, jobId: jobId, fileNameMatch: typeof inputProps.accept === 'string'
+        !value && !enableFilePicker && React__default.createElement(SelectCloudArtifacts, { projectOwner: projectOwner, projectName: projectName, value: selArtifact, hidden: false, client: client, jobId: jobId, fileNameMatch: typeof inputProps.accept === 'string'
                 ? ".*(".concat(inputProps.accept.replace(/\./g, '').split(', ').join('$|'), "$)")
                 : undefined, onChange: onChange }),
-        selArtifact &&
+        !localPath && enableFilePicker && React__default.createElement(SelectLocalArtifactNative, { onChange: onChangeLocal, value: selFilePath, buttonLabel: "Select ".concat(inputProps.accept.split(',').length > 1 ?
+                'any of' : 'an', " ").concat(inputProps.accept, " ").concat(inputProps.accept.split(',').length > 1 ? 'file types' : 'file', "."), fileMatchPattern: "File types ".concat(inputProps.accept, "|*").concat(inputProps.accept.split(', ').join(';*')) }),
+        (selArtifact || selFilePath) && // 
             React__default.createElement(React__default.Fragment, null,
+                (value || localPath) ? undefined : React__default.createElement("br", null),
                 (host !== 'web' && currFileExtension === 'vsf'
                     && typeof preview === 'string') &&
-                    React__default.createElement(SendResults, { results: JSON.parse(preview), defaultKey: selArtifact.file_name, buttonLabel: 'Result Preview' }),
+                    React__default.createElement(SendResults, { results: JSON.parse(preview), defaultKey: (selArtifact === null || selArtifact === void 0 ? void 0 : selArtifact.file_name) || (selFilePath === null || selFilePath === void 0 ? void 0 : selFilePath.replace(/^.*[\\\/]/, '')), buttonLabel: 'Result Preview' }),
                 (host !== 'web' && currFileExtension === 'zip') &&
                     preview && typeof preview === 'string' &&
-                    React__default.createElement(SendResults, { results: JSON.parse(preview), defaultKey: selArtifact.file_name, buttonLabel: 'Result Preview' }),
+                    React__default.createElement(SendResults, { results: JSON.parse(preview), defaultKey: (selArtifact === null || selArtifact === void 0 ? void 0 : selArtifact.file_name) || (selFilePath === null || selFilePath === void 0 ? void 0 : selFilePath.replace(/^.*[\\\/]/, '')), buttonLabel: 'Result Preview' }),
                 (host === 'web') &&
                     React__default.createElement(Button, { type: 'button', onClick: function () { return download(currBlob, selArtifact.file_name); }, disabled: !currBlob, style: {
                             width: 'fit-content',