feat(wdb): finish export package store wdb snapshot to asb, import pa…

…ckage should use it

src/core/composable_component/header/utils/data/ExportAssetType.re

@@ -21,6 +21,7 @@ type wdb = {
   name: string,
   path: string,
   bufferView: bufferViewIndex,
+  snapshot: imageIndex,
 };
 
 type image = {

src/core/composable_component/header/utils/export/BuildJsonDataUtils.re → src/core/composable_component/header/utils/export/HeaderBuildJsonDataUtils.re

@@ -250,7 +250,7 @@ let _buildMaterialData =
 
 let _buildWDBData =
     (
-      imageUint8ArrayMap,
+      (imageIndexMap, imageUint8ArrayMap),
       imageAlignedByteLength,
       imageBufferViewArr,
       (editorState, engineState),
@@ -264,7 +264,7 @@ let _buildWDBData =
            (engineState, wdbArr, arrayBufferArr, bufferViewArr, byteOffset),
            node,
          ) => {
-           let {name, wdbGameObject}: NodeAssetType.wdbNodeData =
+           let {name, wdbGameObject, imageDataIndex}: NodeAssetType.wdbNodeData =
              WDBNodeAssetService.getNodeData(node);
 
            let (engineState, wdbArrayBuffer) =
@@ -289,6 +289,11 @@ let _buildWDBData =
                       ),
                     bufferView:
                       imageBufferViewIndex + (bufferViewArr |> Js.Array.length),
+                    snapshot:
+                      imageIndexMap
+                      |> WonderCommonlib.ImmutableSparseMapService.unsafeGet(
+                           imageDataIndex,
+                         ),
                   }: ExportAssetType.wdb,
                 ),
              arrayBufferArr |> ArrayService.push(wdbArrayBuffer),
@@ -414,7 +419,7 @@ let buildJsonData = (imageUint8ArrayMap, (editorState, engineState)) => {
     bufferTotalAlignedByteLength,
   ) =
     _buildWDBData(
-      imageUint8ArrayMap,
+      (imageIndexMap, imageUint8ArrayMap),
       imageAlignedByteLength,
       imageBufferViewArr,
       (editorState, engineState),

src/core/composable_component/header/utils/export/HeaderExportASBUtils.re

@@ -105,13 +105,13 @@ let generateASB = (imageUint8ArrayMap, (editorState, engineState)) => {
     (imageUint8ArrayArr, wdbArrayBufferArr),
     bufferTotalAlignedByteLength,
   ) =
-    BuildJsonDataUtils.buildJsonData(
+    HeaderBuildJsonDataUtils.buildJsonData(
       imageUint8ArrayMap,
       (editorState, engineState),
     );
 
   let jsonUint8Array =
-    BuildJsonDataUtils.buildJsonUint8Array(
+    HeaderBuildJsonDataUtils.buildJsonUint8Array(
       bufferTotalAlignedByteLength,
       (
         Js.Array.concat(wdbBufferViewArr, imageBufferViewArr),

src/core/composable_component/header/utils/import/HeaderBuildAssetDataUtils.re

@@ -377,7 +377,11 @@ let buildMaterialData =
       (editorState, engineState),
     );
 
-  (basicMaterialMap, lightMaterialMap, (editorState, engineState));
+  (
+    imageDataIndexMap,
+    (basicMaterialMap, lightMaterialMap),
+    (editorState, engineState),
+  );
 };
 
 let _buildScriptEventFunctionEditorData =
@@ -582,6 +586,7 @@ let _mergeImageUint8ArrayDataMap =
 
 let buildWDBData =
     (
+      imageDataIndexMap,
       {wdbs, bufferViews}: ExportAssetType.assets,
       buffer,
       (editorState, engineState),
@@ -595,7 +600,7 @@ let buildWDBData =
   wdbs
   |> WonderBsMost.Most.from
   |> WonderBsMost.Most.concatMap(
-       ({name, bufferView, path}: ExportAssetType.wdb) => {
+       ({name, bufferView, path, snapshot}: ExportAssetType.wdb) => {
        let editorState = StateEditorService.getState();
        let engineState = StateEngineService.unsafeGetState();
 
@@ -609,9 +614,8 @@ let buildWDBData =
            path,
            (editorState, engineState),
          );
-
        HeaderImportASBWDBUtils.importWDB(
-         (name, arrayBuffer),
+         (imageDataIndexMap, snapshot, name, arrayBuffer),
          (assetNodeId, parentFolderNode),
          (editorState, engineState),
        )

src/core/composable_component/header/utils/import/HeaderImportASBUtils.re

@@ -64,8 +64,15 @@ let importASB = asb => {
        );
      })
   |> WonderBsMost.Most.flatMap(
-       ((basicMaterialMap, lightMaterialMap, (editorState, engineState))) =>
+       (
+         (
+           imageDataIndexMap,
+           (basicMaterialMap, lightMaterialMap),
+           (editorState, engineState),
+         ),
+       ) =>
        HeaderBuildAssetDataUtils.buildWDBData(
+         imageDataIndexMap,
          asbRecord,
          buffer,
          (editorState, engineState),

src/core/composable_component/header/utils/import/HeaderImportASBWDBUtils.re

@@ -1,12 +1,15 @@
 let importWDB =
     (
-      (name, wdbArrayBuffer),
+      (imageDataIndexMap, snapshot, name, wdbArrayBuffer),
       (wdbNodeId, parentFolderNode),
       (editorState, engineState),
     ) =>
   WDBAssetLogicService.importAssetWDB(
     (name, wdbArrayBuffer),
-    (wdbNodeId, parentFolderNode),
-    false,
+    (wdbNodeId, parentFolderNode, false),
+    WDBAssetLogicService.createWDBNodeUseImageDataMapSnapshot((
+      imageDataIndexMap,
+      snapshot,
+    )),
     (editorState, engineState),
   );

src/core/composable_component/mainEditor/composable_component/bottom_components/composable_component/project/composable_component/asset/composable_component/utils/AssetHeaderUtils.re

@@ -56,16 +56,15 @@ let _handleAssetWDBType =
   WDBAssetLogicService.importAssetWDB(
     (
       FileNameService.getBaseName(fileName)
-      ->(
-          OperateTreeAssetLogicService.getUniqueNodeName(
-            selectedFolderNodeInAssetTree,
-            engineState,
-          )
-        ),
+      |> OperateTreeAssetLogicService.getUniqueNodeName(
+           _,
+           selectedFolderNodeInAssetTree,
+           engineState,
+         ),
       wdbArrayBuffer,
     ),
-    (wdbNodeId, selectedFolderNodeInAssetTree),
-    true,
+    (wdbNodeId, selectedFolderNodeInAssetTree, true),
+    WDBAssetLogicService.createWDBNodeUseCreatedSnapshot,
     (editorState, engineState),
   )
   |> then_(
@@ -287,7 +286,9 @@ let fileLoad = ((uiState, dispatchFunc), createJsZipFunc, event) => {
          FileReader.makeSureCanLoadSameNameFileAgain(target);
 
          dispatchFunc(
-           AppStore.UpdateAction(Update([|UpdateStore.Inspector,UpdateStore.Project|])),
+           AppStore.UpdateAction(
+             Update([|UpdateStore.Inspector, UpdateStore.Project|]),
+           ),
          );
 
          resolve();

src/service/stateTuple/logic/asset/WDBAssetLogicService.re

@@ -33,11 +33,56 @@ let _createWDBNodeAndSnapshot =
      );
 };
 
+let createWDBNodeUseImageDataMapSnapshot =
+    (
+      (imageDataIndexMap, snapshot),
+      (wdbNodeId, name, gameObject, parentFolderNode),
+      editorState,
+    ) =>
+  editorState
+  |> OperateTreeAssetEditorService.insertNode(
+       NodeAssetService.getNodeId(~node=parentFolderNode),
+       WDBNodeAssetService.buildNode(
+         ~nodeId=wdbNodeId,
+         ~name,
+         ~wdbGameObject=gameObject,
+         ~imageDataIndex=
+           imageDataIndexMap
+           |> WonderCommonlib.ImmutableSparseMapService.unsafeGet(snapshot),
+       ),
+     );
+
+let createWDBNodeUseCreatedSnapshot =
+    ((wdbNodeId, name, gameObject, parentFolderNode), editorState) => {
+  let inspectorEngineState =
+    (editorState, StateInspectorEngineService.unsafeGetState())
+    |> AssetTreeInspectorUtils.disposeContainerGameObjectAllChildrenAndReallocateCPUMemory
+    |> WDBInspectorEngineUtils.createWDBIntoInspectorCanvas(
+         gameObject,
+         (StateEditorService.getState(), StateEngineService.unsafeGetState()),
+       )
+    |> StateLogicService.renderInspectorEngineStateAndReturnState;
+
+  let editorState =
+    editorState
+    |> _createWDBNodeAndSnapshot(
+         parentFolderNode,
+         (wdbNodeId, name, gameObject),
+       );
+
+  (editorState, inspectorEngineState)
+  |> AssetTreeInspectorUtils.disposeContainerGameObjectAllChildrenAndReallocateCPUMemory
+  |> AssetTreeInspectorUtils.setCameraDefaultDistance
+  |> StateInspectorEngineService.setState;
+
+  editorState;
+};
+
 let importAssetWDB =
     (
       (name, wdbArrayBuffer),
-      (wdbNodeId, parentFolderNode),
-      isLoadImage,
+      (wdbNodeId, parentFolderNode, isLoadImage),
+      createWDBNodeFunc,
       (editorState, engineState),
     ) => {
   let allGameObjectsRef = ref([||]);
@@ -61,32 +106,10 @@ let importAssetWDB =
            engineState,
          );
 
-       /* TODO need test */
-       let inspectorEngineState =
-         (editorState, StateInspectorEngineService.unsafeGetState())
-         |> AssetTreeInspectorUtils.disposeContainerGameObjectAllChildrenAndReallocateCPUMemory
-         |> WDBInspectorEngineUtils.createWDBIntoInspectorCanvas(
-              gameObject,
-              (
-                StateEditorService.getState(),
-                StateEngineService.unsafeGetState(),
-              ),
-            )
-         |> StateLogicService.renderInspectorEngineStateAndReturnState;
-
-       let editorState =
-         editorState
-         |> _createWDBNodeAndSnapshot(
-              parentFolderNode,
-              (wdbNodeId, name, gameObject),
-            );
-
-       (editorState, inspectorEngineState)
-       |> AssetTreeInspectorUtils.disposeContainerGameObjectAllChildrenAndReallocateCPUMemory
-       |> AssetTreeInspectorUtils.setCameraDefaultDistance
-       |> StateInspectorEngineService.setState;
-
-       editorState |> StateEditorService.setState |> ignore;
+       editorState
+       |> createWDBNodeFunc((wdbNodeId, name, gameObject, parentFolderNode))
+       |> StateEditorService.setState
+       |> ignore;
 
        engineState
        |> GameObjectEngineService.setAllGameObjectsIsRenderIfHasMeshRenderer(

test/integration/header/headerExportPackage_test.re

@@ -14,6 +14,14 @@ let _ =
   describe("header export package", () => {
     let sandbox = getSandboxDefaultVal();
 
+    let boxTexturedWDBArrayBuffer = ref(Obj.magic(1));
+    let sceneWDBArrayBuffer = ref(Obj.magic(1));
+
+    beforeAll(() => {
+      boxTexturedWDBArrayBuffer := WDBTool.convertGLBToWDB("BoxTextured");
+      sceneWDBArrayBuffer := WDBTool.generateSceneWDB();
+    });
+
     beforeEach(() => {
       sandbox := createSandbox();
 
@@ -145,7 +153,7 @@ let _ =
 
             MainEditorAssetHeaderOperateNodeTool.addMaterial();
 
-            let wpkArrayBuffer = ExportPackageTool.exportWPK();
+            ExportPackageTool.exportWPK() |> ignore;
 
             let editorState = StateEditorService.getState();
 
@@ -195,7 +203,7 @@ let _ =
               (),
             );
 
-            let wpkArrayBuffer = ExportPackageTool.exportWPK();
+            ExportPackageTool.exportWPK() |> ignore;
 
             let editorState = StateEditorService.getState();
 
@@ -212,10 +220,67 @@ let _ =
               ({base64, uint8Array}) =>
                 uint8Array
                 |> OptionService.unsafeGet
-                |> expect == BufferUtils.convertBase64ToUint8Array(imgCanvasFakeBase64Str)
+                |> expect
+                == BufferUtils.convertBase64ToUint8Array(
+                     imgCanvasFakeBase64Str,
+                   )
             );
           },
         );
       });
+
+      describe("test export all wdbs->snapshot", () =>
+        testPromise(
+          "upload one wdb w1;
+           export;
+
+           should convert w1->snapshot base64 to uint8Array;",
+          () => {
+            let (
+              addedMaterialNodeId,
+              newMaterialComponent,
+              imgCanvasFakeBase64Str,
+              (inspectorCanvasDom, imgCanvasDom),
+            ) =
+              MainEditorLightMaterialForAssetTool.prepareInspectorMaterialSphereAndImgCanvas(
+                ~sandbox,
+                (),
+              );
+
+            MainEditorAssetUploadTool.loadOneWDB(
+              ~arrayBuffer=boxTexturedWDBArrayBuffer^,
+              (),
+            )
+            |> then_(uploadedWDBNodeId => {
+                 ExportPackageTool.exportWPK() |> ignore;
+
+                 let editorState = StateEditorService.getState();
+                 let engineState = StateEngineService.unsafeGetState();
+
+                 let {imageDataIndex}: wdbNodeData =
+                   editorState
+                   |> OperateTreeAssetEditorService.unsafeFindNodeById(
+                        uploadedWDBNodeId,
+                      )
+                   |> WDBNodeAssetService.getNodeData;
+
+                 editorState
+                 |> ImageDataMapAssetEditorService.unsafeGetData(
+                      imageDataIndex,
+                    )
+                 |> (
+                   ({base64, uint8Array}) =>
+                     uint8Array
+                     |> OptionService.unsafeGet
+                     |> expect
+                     == BufferUtils.convertBase64ToUint8Array(
+                          imgCanvasFakeBase64Str,
+                        )
+                     |> resolve
+                 );
+               });
+          },
+        )
+      );
     });
   });