Merge pull request #62 from Lenny4/dev

show the size of all folders
Lenny4 · Jun 18, 2023 · 96d5873 · 96d5873
2 parents 3961c56 + 6a16043
commit 96d5873
Show file tree

Hide file tree

Showing 12 changed files with 270 additions and 73 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -10,6 +10,12 @@
 
 ### Removed
 
+## [0.2.0] - 2023-06-18
+
+### Added
+
+- Show the size of all folders in workspace
+
 ## [0.1.2] - 2023-06-14
 
 Note: need to install it manually from the github if on windows.

diff --git a/lib/class/folder_property.dart b/lib/class/folder_property.dart
@@ -0,0 +1,22 @@
+import 'package:json_annotation/json_annotation.dart';
+
+part 'folder_property.g.dart';
+
+@JsonSerializable()
+class FolderProperty {
+  int? size;
+  String path;
+  int? nbChildren;
+  List<FolderProperty> folderProperties;
+
+  FolderProperty(
+      {this.size,
+      required this.path,
+      this.nbChildren,
+      required this.folderProperties});
+
+  factory FolderProperty.fromJson(Map<String, dynamic> json) =>
+      _$FolderPropertyFromJson(json);
+
+  Map<String, dynamic> toJson() => _$FolderPropertyToJson(this);
+}
diff --git a/lib/class/folder_property.g.dart b/lib/class/folder_property.g.dart
diff --git a/lib/class/workspace.dart b/lib/class/workspace.dart
@@ -1,3 +1,4 @@
+import 'package:deepfacelab_client/class/folder_property.dart';
 import 'package:deepfacelab_client/class/locale_storage_question.dart';
 import 'package:json_annotation/json_annotation.dart';
 
@@ -8,9 +9,13 @@ class Workspace {
   String name;
   String path;
   List<LocaleStorageQuestion>? localeStorageQuestions;
+  FolderProperty? folderProperty;
 
   Workspace(
-      {required this.name, required this.path, this.localeStorageQuestions});
+      {required this.name,
+      required this.path,
+      this.localeStorageQuestions,
+      this.folderProperty});
 
   factory Workspace.fromJson(Map<String, dynamic> json) =>
       _$WorkspaceFromJson(json);

diff --git a/lib/class/workspace.g.dart b/lib/class/workspace.g.dart
diff --git a/lib/main.dart b/lib/main.dart
@@ -13,6 +13,7 @@ import 'package:deepfacelab_client/screens/window_command_screen.dart';
 import 'package:deepfacelab_client/screens/workspace_screen.dart';
 import 'package:deepfacelab_client/service/locale_storage_service.dart';
 import 'package:deepfacelab_client/widget/installation/has_requirements_widget.dart';
+import 'package:file_sizes/file_sizes.dart';
 import 'package:filesystem_picker/filesystem_picker.dart';
 import 'package:flutter/gestures.dart';
 import 'package:flutter/material.dart';
@@ -112,7 +113,8 @@ class Root extends HookWidget {
                 destination: NavigationRailDestination(
                   icon: const Icon(Icons.movie),
                   selectedIcon: const Icon(Icons.movie),
-                  label: Text(workspace.name),
+                  label: Text(
+                      '${workspace.name}\n${FileSize.getSize(workspace.folderProperty?.size ?? 0)}'),
                 ),
                 widget: WorkspaceScreen(initWorkspace: workspace)),
           );

diff --git a/lib/screens/workspace_screen.dart b/lib/screens/workspace_screen.dart
@@ -68,7 +68,7 @@ class WorkspaceScreen extends HookWidget {
                     if (initWorkspace != null) ...[
                       const Divider(),
                       FileManagerWidget(
-                        rootPath: initWorkspace!.path,
+                        workspace: initWorkspace,
                         controller: mainController.value,
                         updateFileMissing: updateFileMissingController,
                       ),

diff --git a/lib/service/file_manager_service.dart b/lib/service/file_manager_service.dart
@@ -0,0 +1,86 @@
+import 'dart:io';
+
+import 'package:collection/collection.dart';
+import 'package:deepfacelab_client/class/folder_property.dart';
+import 'package:deepfacelab_client/service/workspace_service.dart';
+
+import '../class/workspace.dart';
+
+class FileManagerService {
+  Future<FolderProperty> _updateFolderProperty(
+      {required FolderProperty folderProperty,
+      required List<FileSystemEntity> fileSystemEntities}) async {
+    folderProperty.nbChildren = fileSystemEntities.length;
+    var size = 0;
+    List<String> foldersFound = [];
+    for (var fileSystemEntity in fileSystemEntities) {
+      if (fileSystemEntity is File) {
+        size += await fileSystemEntity.length();
+      }
+      if (fileSystemEntity is Directory) {
+        var thisFileSystemEntities =
+            await Directory(fileSystemEntity.path).list().toList();
+        foldersFound.add(fileSystemEntity.path);
+        var thisFolderProperty = folderProperty.folderProperties
+            .firstWhereOrNull(
+                (FolderProperty f) => f.path == fileSystemEntity.path);
+        if (thisFolderProperty == null) {
+          thisFolderProperty =
+              FolderProperty(path: fileSystemEntity.path, folderProperties: []);
+          folderProperty.folderProperties.add(thisFolderProperty);
+        }
+        await _updateFolderProperty(
+            folderProperty: thisFolderProperty,
+            fileSystemEntities: thisFileSystemEntities);
+        size += thisFolderProperty.size!;
+      }
+    }
+    folderProperty.folderProperties = folderProperty.folderProperties
+        .where((f) => foldersFound.contains(f.path))
+        .toList();
+    folderProperty.size = size;
+    return folderProperty;
+  }
+
+  Future<FolderProperty> updateFolderProperty(
+      {required String path,
+      required Workspace workspace,
+      List<FileSystemEntity>? fileSystemEntities,
+      bool force = false}) async {
+    // region get thisFolderProperty
+    workspace.folderProperty ??=
+        FolderProperty(path: workspace.path, folderProperties: []);
+    FolderProperty? thisFolderProperty = workspace.folderProperty;
+    var pathArray = path
+        .replaceAll(workspace.path, '')
+        .split(Platform.pathSeparator)
+        .where((element) => element != '')
+        .toList();
+    var i = 0;
+    while (thisFolderProperty != null && thisFolderProperty.path != path) {
+      thisFolderProperty = thisFolderProperty.folderProperties.firstWhereOrNull(
+          (f) => f.path.endsWith(Platform.pathSeparator + pathArray[i]));
+      i++;
+    }
+    // endregion
+    if (thisFolderProperty == null) {
+      return await updateFolderProperty(
+          path: workspace.path, workspace: workspace, force: true);
+    }
+    fileSystemEntities ??= await Directory(path).list().toList();
+    if (!force && thisFolderProperty.nbChildren == fileSystemEntities.length) {
+      return thisFolderProperty;
+    }
+    if (thisFolderProperty.path != workspace.path) {
+      return await updateFolderProperty(
+          path: workspace.path, workspace: workspace, force: true);
+    }
+    thisFolderProperty = await _updateFolderProperty(
+        folderProperty: thisFolderProperty,
+        fileSystemEntities: fileSystemEntities);
+    workspace.folderProperty = thisFolderProperty;
+    WorkspaceService().createUpdateWorkspace(
+        oldWorkspace: workspace, newWorkspace: workspace);
+    return thisFolderProperty;
+  }
+}
diff --git a/lib/service/workspace_service.dart b/lib/service/workspace_service.dart
@@ -46,8 +46,7 @@ class WorkspaceService {
     if (index != null) {
       storage?.workspaces![index] = newWorkspace;
     }
-    storage?.workspaces =
-        storage.workspaces?.map((w) => Workspace.fromJson(w.toJson())).toList();
+    storage?.workspaces = [...?storage.workspaces];
     store.dispatch({'storage': storage});
   }