From 312138f9810a11db9e7f56d3c90bcc11833bdfa1 Mon Sep 17 00:00:00 2001
From: Hayden <48267247+hayden-fr@users.noreply.github.com>
Date: Sat, 22 Feb 2025 18:30:29 +0800
Subject: [PATCH] fix: auto open root folder (#151)

---
 src/hooks/explorer.ts | 10 ++++++++--
 src/hooks/model.ts    |  3 +++
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/src/hooks/explorer.ts b/src/hooks/explorer.ts
index 6087f95..bb8e53e 100644
--- a/src/hooks/explorer.ts
+++ b/src/hooks/explorer.ts
@@ -1,7 +1,7 @@
 import { genModelFullName, useModels } from 'hooks/model'
 import { cloneDeep, filter, find } from 'lodash'
 import { BaseModel, Model, SelectOptions } from 'types/typings'
-import { computed, ref } from 'vue'
+import { computed, ref, watch } from 'vue'
 
 export interface FolderPathItem {
   name: string
@@ -26,7 +26,7 @@ export type TreeItemNode = ModelTreeNode & {
 }
 
 export const useModelExplorer = () => {
-  const { data, folders, ...modelRest } = useModels()
+  const { data, folders, initialized, ...modelRest } = useModels()
 
   const folderPaths = ref<FolderPathItem[]>([])
 
@@ -144,6 +144,12 @@ export const useModelExplorer = () => {
     folderPaths.value = folderItems
   }
 
+  watch(initialized, (val) => {
+    if (val) {
+      openFolder(dataTreeList.value[0])
+    }
+  })
+
   return {
     folders,
     folderPaths,
diff --git a/src/hooks/model.ts b/src/hooks/model.ts
index 6e74a2b..ad40314 100644
--- a/src/hooks/model.ts
+++ b/src/hooks/model.ts
@@ -50,10 +50,12 @@ export const useModels = defineStore('models', (store) => {
   const loading = useLoading()
 
   const folders = ref<ModelFolder>({})
+  const initialized = ref(false)
 
   const refreshFolders = async () => {
     return request('/models').then((resData) => {
       folders.value = resData
+      initialized.value = true
     })
   }
 
@@ -233,6 +235,7 @@ export const useModels = defineStore('models', (store) => {
   }
 
   return {
+    initialized: initialized,
     folders: folders,
     data: models,
     refresh: refreshAllModels,