From a46c4b2322efc691596afa3beb465fa6586ce715 Mon Sep 17 00:00:00 2001
From: Juan Hoyos <juan.hoyos4@gmail.com>
Date: Tue, 30 Jul 2024 22:29:00 -0500
Subject: [PATCH] feat(obc): #39 allows to specify the action buttons in the
 ModelsList

---
 packages/obc/package.json                     |   2 +-
 .../components/tables/ModelsList/example.ts   |   6 +-
 .../tables/ModelsList/src/template.ts         | 104 +++++++++++-------
 3 files changed, 69 insertions(+), 43 deletions(-)

diff --git a/packages/obc/package.json b/packages/obc/package.json
index 801b335..f7f50f3 100644
--- a/packages/obc/package.json
+++ b/packages/obc/package.json
@@ -56,4 +56,4 @@
     "three": "0.160.1",
     "web-ifc": "0.0.56"
   }
-}
\ No newline at end of file
+}
diff --git a/packages/obc/src/components/tables/ModelsList/example.ts b/packages/obc/src/components/tables/ModelsList/example.ts
index 71221c6..ba5c9db 100644
--- a/packages/obc/src/components/tables/ModelsList/example.ts
+++ b/packages/obc/src/components/tables/ModelsList/example.ts
@@ -56,7 +56,11 @@ fragmentsManager.onFragmentsLoaded.add((model) => {
   Allright! Now that some basic events are setup, it's time to create a new fresh models list component:
   */
 
-const [modelsList] = CUI.tables.modelsList({ components });
+const [modelsList] = CUI.tables.modelsList({
+  components,
+  tags: { schema: true, viewDefinition: false },
+  actions: { download: false },
+});
 
 /* MD
   Now that we have a brand new models list created, we need to add it to the HTML page. For it, let's create simple BIM panel component where we include the models list and also a pre-made IFC load button 👇
diff --git a/packages/obc/src/components/tables/ModelsList/src/template.ts b/packages/obc/src/components/tables/ModelsList/src/template.ts
index ce64be6..ae5e196 100644
--- a/packages/obc/src/components/tables/ModelsList/src/template.ts
+++ b/packages/obc/src/components/tables/ModelsList/src/template.ts
@@ -5,15 +5,25 @@ import * as BUI from "@thatopen/ui";
 
 export interface ModelsListUIState {
   components: OBC.Components;
-  schemaTag?: boolean;
-  viewDefinitionTag?: boolean;
+  tags?: {
+    schema?: boolean;
+    viewDefinition?: boolean;
+  };
+  actions?: {
+    visibility?: boolean;
+    download?: boolean;
+    dispose?: boolean;
+  };
 }
 
 export const modelsListTemplate = (state: ModelsListUIState) => {
-  const { components } = state;
+  const { components, actions, tags } = state;
 
-  const schemaTag = state.schemaTag ?? true;
-  const viewDefinitionTag = state.viewDefinitionTag ?? true;
+  const disposeBtn = actions?.dispose ?? true;
+  const downloadBtn = actions?.download ?? true;
+  const visibilityBtn = actions?.visibility ?? true;
+  const schemaTag = tags?.schema ?? true;
+  const viewDefinitionTag = tags?.viewDefinition ?? true;
 
   const fragments = components.get(OBC.FragmentsManager);
 
@@ -66,41 +76,53 @@ export const modelsListTemplate = (state: ModelsListUIState) => {
         `;
       }
 
-      const onDeleteClick = () => fragments.disposeGroup(model);
+      let disposeTemplate: BUI.TemplateResult | undefined;
+      if (disposeBtn) {
+        const onDisposeClick = () => fragments.disposeGroup(model);
+        disposeTemplate = BUI.html`<bim-button @click=${onDisposeClick} icon="mdi:delete"></bim-button>`;
+      }
 
-      const onHideClick = (e: Event) => {
-        const hider = components.get(OBC.Hider);
-        const button = e.target as BUI.Button;
-        hider.set(button.hasAttribute("data-model-hidden"), modelIdMap);
-        button.toggleAttribute("data-model-hidden");
-        button.icon = button.hasAttribute("data-model-hidden")
-          ? "mdi:eye-off"
-          : "mdi:eye";
-      };
+      let visibilityTemplate: BUI.TemplateResult | undefined;
+      if (visibilityBtn) {
+        const onHideClick = (e: Event) => {
+          const hider = components.get(OBC.Hider);
+          const button = e.target as BUI.Button;
+          hider.set(button.hasAttribute("data-model-hidden"), modelIdMap);
+          button.toggleAttribute("data-model-hidden");
+          button.icon = button.hasAttribute("data-model-hidden")
+            ? "mdi:eye-off"
+            : "mdi:eye";
+        };
+        visibilityTemplate = BUI.html`<bim-button @click=${onHideClick} icon="mdi:eye"></bim-button>`;
+      }
 
-      const onSaveClick = () => {
-        const input = document.createElement("input");
-        input.type = "file";
-        input.accept = ".ifc";
-        input.multiple = false;
-        input.addEventListener("change", async () => {
-          if (!(input.files && input.files.length === 1)) return;
-          const originalFile = input.files[0];
-          const originalBuffer = await originalFile.arrayBuffer();
-          const propsManager = components.get(OBC.IfcPropertiesManager);
-          const modifiedBuffer = await propsManager.saveToIfc(
-            model,
-            new Uint8Array(originalBuffer),
-          );
-          const modifiedFile = new File([modifiedBuffer], originalFile.name);
-          const a = document.createElement("a");
-          a.href = URL.createObjectURL(modifiedFile);
-          a.download = modifiedFile.name;
-          a.click();
-          URL.revokeObjectURL(a.href);
-        });
-        input.click();
-      };
+      let downloadTemplate: BUI.TemplateResult | undefined;
+      if (downloadBtn) {
+        const onSaveClick = () => {
+          const input = document.createElement("input");
+          input.type = "file";
+          input.accept = ".ifc";
+          input.multiple = false;
+          input.addEventListener("change", async () => {
+            if (!(input.files && input.files.length === 1)) return;
+            const originalFile = input.files[0];
+            const originalBuffer = await originalFile.arrayBuffer();
+            const propsManager = components.get(OBC.IfcPropertiesManager);
+            const modifiedBuffer = await propsManager.saveToIfc(
+              model,
+              new Uint8Array(originalBuffer),
+            );
+            const modifiedFile = new File([modifiedBuffer], originalFile.name);
+            const a = document.createElement("a");
+            a.href = URL.createObjectURL(modifiedFile);
+            a.download = modifiedFile.name;
+            a.click();
+            URL.revokeObjectURL(a.href);
+          });
+          input.click();
+        };
+        downloadTemplate = BUI.html`<bim-button @click=${onSaveClick} icon="flowbite:download-solid"></bim-button>`;
+      }
 
       return BUI.html`
        <div style="display: flex; flex: 1; gap: var(--bim-ui_size-4xs); justify-content: space-between; overflow: auto;">
@@ -114,9 +136,9 @@ export const modelsListTemplate = (state: ModelsListUIState) => {
           </div>
         </div>
         <div style="display: flex; align-self: flex-start; flex-shrink: 0;">
-          <bim-button @click=${onSaveClick} icon="flowbite:download-solid"></bim-button>
-          <bim-button @click=${onHideClick} icon="mdi:eye"></bim-button>
-          <bim-button @click=${onDeleteClick} icon="mdi:delete"></bim-button>
+          ${downloadTemplate}
+          ${visibilityTemplate}
+          ${disposeTemplate}
         </div>
        </div>
       `;