From 786fa02b90d64044aee23011a18f6e202856a362 Mon Sep 17 00:00:00 2001
From: Kyle Spearrin <kyle.spearrin@gmail.com>
Date: Fri, 9 Nov 2018 17:44:45 -0500
Subject: [PATCH] added collpase/expand functions to groupings

---
 src/angular/components/groupings.component.ts | 36 ++++++++++++++++++-
 src/services/constants.service.ts             |  2 ++
 2 files changed, 37 insertions(+), 1 deletion(-)

diff --git a/src/angular/components/groupings.component.ts b/src/angular/components/groupings.component.ts
index 23381634e1..9983823dfa 100644
--- a/src/angular/components/groupings.component.ts
+++ b/src/angular/components/groupings.component.ts
@@ -13,6 +13,10 @@ import { TreeNode } from '../../models/domain/treeNode';
 
 import { CollectionService } from '../../abstractions/collection.service';
 import { FolderService } from '../../abstractions/folder.service';
+import { StorageService } from '../../abstractions/storage.service';
+import { UserService } from '../../abstractions/user.service';
+
+import { ConstantsService } from '../../services/constants.service';
 
 export class GroupingsComponent {
     @Input() showFolders = true;
@@ -40,9 +44,22 @@ export class GroupingsComponent {
     selectedFolderId: string = null;
     selectedCollectionId: string = null;
 
-    constructor(protected collectionService: CollectionService, protected folderService: FolderService) { }
+    private collapsedGroupings: Set<string>;
+    private collapsedGroupingsKey: string;
+
+    constructor(protected collectionService: CollectionService, protected folderService: FolderService,
+        protected storageService: StorageService, protected userService: UserService) { }
 
     async load(setLoaded = true) {
+        const userId = await this.userService.getUserId();
+        this.collapsedGroupingsKey = ConstantsService.collapsedGroupingsKey + '_' + userId;
+        const collapsedGroupings = await this.storageService.get<string[]>(this.collapsedGroupingsKey);
+        if (collapsedGroupings == null) {
+            this.collapsedGroupings = new Set<string>();
+        } else {
+            this.collapsedGroupings = new Set(collapsedGroupings);
+        }
+
         await this.loadFolders();
         await this.loadCollections();
 
@@ -119,4 +136,21 @@ export class GroupingsComponent {
         this.selectedFolderId = null;
         this.selectedCollectionId = null;
     }
+
+    collapse(grouping: FolderView | CollectionView, idPrefix = '') {
+        if (grouping.id == null) {
+            return;
+        }
+        const id = idPrefix + grouping.id;
+        if (this.isCollapsed(grouping, idPrefix)) {
+            this.collapsedGroupings.delete(id);
+        } else {
+            this.collapsedGroupings.add(id);
+        }
+        this.storageService.save(this.collapsedGroupingsKey, this.collapsedGroupings);
+    }
+
+    isCollapsed(grouping: FolderView | CollectionView, idPrefix = '') {
+        return this.collapsedGroupings.has(idPrefix + grouping.id);
+    }
 }
diff --git a/src/services/constants.service.ts b/src/services/constants.service.ts
index 987f53ce48..cd0a260359 100644
--- a/src/services/constants.service.ts
+++ b/src/services/constants.service.ts
@@ -13,6 +13,7 @@ export class ConstantsService {
     static readonly installedVersionKey: string = 'installedVersion';
     static readonly localeKey: string = 'locale';
     static readonly themeKey: string = 'theme';
+    static readonly collapsedGroupingsKey: string = 'collapsedGroupings';
 
     readonly environmentUrlsKey: string = ConstantsService.environmentUrlsKey;
     readonly disableGaKey: string = ConstantsService.disableGaKey;
@@ -27,4 +28,5 @@ export class ConstantsService {
     readonly installedVersionKey: string = ConstantsService.installedVersionKey;
     readonly localeKey: string = ConstantsService.localeKey;
     readonly themeKey: string = ConstantsService.themeKey;
+    readonly collapsedGroupingsKey: string = ConstantsService.collapsedGroupingsKey;
 }