feat: open model folder

containers · Jan 29, 2024 · 2a8682c · 2a8682c
1 parent a431851
commit 2a8682c
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 8 deletions.
diff --git a/packages/backend/src/managers/modelsManager.spec.ts b/packages/backend/src/managers/modelsManager.spec.ts
@@ -97,14 +97,14 @@ test('getLocalModelsFromDisk should get models in local directory', () => {
       file: 'model-id-1-model',
       size: 32000,
       creation: now,
-      path: path.resolve(dirent[0].path, dirent[0].name, 'model-id-1-model'),
+      path: path.resolve(dirent[0].path, dirent[0].name),
     },
     {
       id: 'model-id-2',
       file: 'model-id-2-model',
       size: 32000,
       creation: now,
-      path: path.resolve(dirent[1].path, dirent[1].name, 'model-id-2-model'),
+      path: path.resolve(dirent[1].path, dirent[1].name),
     },
   ]);
 });
@@ -165,7 +165,7 @@ test('loadLocalModels should post a message with the message on disk and on cata
           file: 'model-id-1-model',
           id: 'model-id-1',
           size: 32000,
-          path: path.resolve(dirent[0].path, dirent[0].name, 'model-id-1-model'),
+          path: path.resolve(dirent[0].path, dirent[0].name),
         },
         id: 'model-id-1',
       },
@@ -219,7 +219,7 @@ test('deleteLocalModel deletes the model folder', async () => {
           file: 'model-id-1-model',
           id: 'model-id-1',
           size: 32000,
-          path: path.resolve(dirent[0].path, dirent[0].name, 'model-id-1-model'),
+          path: path.resolve(dirent[0].path, dirent[0].name),
         },
         id: 'model-id-1',
         state: 'deleting',
@@ -279,7 +279,7 @@ test('deleteLocalModel fails to delete the model folder', async () => {
           file: 'model-id-1-model',
           id: 'model-id-1',
           size: 32000,
-          path: path.resolve(dirent[0].path, dirent[0].name, 'model-id-1-model'),
+          path: path.resolve(dirent[0].path, dirent[0].name),
         },
         id: 'model-id-1',
         state: 'deleting',
@@ -296,7 +296,7 @@ test('deleteLocalModel fails to delete the model folder', async () => {
           file: 'model-id-1-model',
           id: 'model-id-1',
           size: 32000,
-          path: path.resolve(dirent[0].path, dirent[0].name, 'model-id-1-model'),
+          path: path.resolve(dirent[0].path, dirent[0].name),
         },
         id: 'model-id-1',
       },

diff --git a/packages/backend/src/managers/modelsManager.ts b/packages/backend/src/managers/modelsManager.ts
@@ -77,7 +77,7 @@ export class ModelsManager {
       result.set(d.name, {
         id: d.name,
         file: modelFile,
-        path: fullPath,
+        path: path.resolve(d.path, d.name),
         size: info.size,
         creation: info.mtime,
       });

diff --git a/packages/frontend/src/lib/table/model/ModelColumnActions.svelte b/packages/frontend/src/lib/table/model/ModelColumnActions.svelte
@@ -1,10 +1,11 @@
 <script lang="ts">
 import type { ModelInfo } from "@shared/src/models/IModelInfo";
 import { faTrash } from "@fortawesome/free-solid-svg-icons";
+import { faFolderOpen } from "@fortawesome/free-solid-svg-icons";
 import ListItemButtonIcon from "../../button/ListItemButtonIcon.svelte";
 import { studioClient } from "/@/utils/client";
 import Modal from "../../Modal.svelte";
-  import Button from "../../button/Button.svelte";
+import Button from "../../button/Button.svelte";
 export let object: ModelInfo;
 
 let deleteConfirmVisible: boolean = false;
@@ -17,8 +18,20 @@ async function goDeleteModel() {
   await studioClient.deleteLocalModel(object.id);
   deleteConfirmVisible= false;
 }
+
+function openModelFolder() {
+  console.log(object.file);
+  if (object && object.file) {
+    studioClient.openURL('file://'+object.file.path);
+  }
+}
 </script>
 
+<ListItemButtonIcon
+  icon={faFolderOpen}
+  onClick={() => openModelFolder()}
+  title="Open Model Folder"
+/>
 <ListItemButtonIcon
   icon={faTrash}
   onClick={() => deleteModel()}