From ed7fd63fc37d4f7590b71ba8180da01a41bbe3f9 Mon Sep 17 00:00:00 2001
From: suzhou <suzhou@amazon.com>
Date: Fri, 16 Jun 2023 09:16:49 +0800
Subject: [PATCH] Add currentWorkspace$ (#15)

* feat: add currentWorkspace$

Signed-off-by: SuZhoue-Joe <suzhou@amazon.com>

* fix: type error

Signed-off-by: SuZhoue-Joe <suzhou@amazon.com>

* feat: add emit on currentWorkspace$

Signed-off-by: SuZhoue-Joe <suzhou@amazon.com>

---------

Signed-off-by: SuZhoue-Joe <suzhou@amazon.com>
---
 .../fatal_errors/fatal_errors_service.mock.ts |  6 ++-
 .../public/workspace/workspaces_client.ts     | 46 ++++++++++++++-----
 2 files changed, 39 insertions(+), 13 deletions(-)

diff --git a/src/core/public/fatal_errors/fatal_errors_service.mock.ts b/src/core/public/fatal_errors/fatal_errors_service.mock.ts
index 5079fc8f4b6a..e495d66ae568 100644
--- a/src/core/public/fatal_errors/fatal_errors_service.mock.ts
+++ b/src/core/public/fatal_errors/fatal_errors_service.mock.ts
@@ -30,7 +30,7 @@
 
 import type { PublicMethodsOf } from '@osd/utility-types';
 import { FatalErrorsService, FatalErrorsSetup } from './fatal_errors_service';
-import { BehaviorSubject, Subject } from 'rxjs';
+import { BehaviorSubject } from 'rxjs';
 import { WorkspaceAttribute } from '../workspace';
 
 const createSetupContractMock = () => {
@@ -62,12 +62,14 @@ export const fatalErrorsServiceMock = {
 };
 
 const currentWorkspaceId$ = new BehaviorSubject<string>('');
-const workspaceList$ = new Subject<WorkspaceAttribute[]>();
+const workspaceList$ = new BehaviorSubject<WorkspaceAttribute[]>([]);
+const currentWorkspace$ = new BehaviorSubject<WorkspaceAttribute | null>(null);
 
 const createWorkspacesSetupContractMock = () => ({
   client: {
     currentWorkspaceId$,
     workspaceList$,
+    currentWorkspace$,
     stop: jest.fn(),
     enterWorkspace: jest.fn(),
     exitWorkspace: jest.fn(),
diff --git a/src/core/public/workspace/workspaces_client.ts b/src/core/public/workspace/workspaces_client.ts
index 91d83dd1639f..f37fd89ae249 100644
--- a/src/core/public/workspace/workspaces_client.ts
+++ b/src/core/public/workspace/workspaces_client.ts
@@ -3,7 +3,8 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 import type { PublicContract } from '@osd/utility-types';
-import { BehaviorSubject } from 'rxjs';
+import { BehaviorSubject, combineLatest } from 'rxjs';
+import { isEqual } from 'lodash';
 import { HttpFetchError, HttpFetchOptions, HttpSetup } from '../http';
 import { WorkspaceAttribute, WorkspaceFindOptions } from '.';
 import { WORKSPACES_API_BASE_URL, WORKSPACE_ERROR_REASON_MAP } from './consts';
@@ -41,26 +42,34 @@ export class WorkspacesClient {
   private http: HttpSetup;
   public currentWorkspaceId$ = new BehaviorSubject<string>('');
   public workspaceList$ = new BehaviorSubject<WorkspaceAttribute[]>([]);
+  public currentWorkspace$ = new BehaviorSubject<WorkspaceAttribute | null>(null);
   constructor(http: HttpSetup) {
     this.http = http;
-    /**
-     * Add logic to check if current workspace id is still valid
-     * If not, remove the current workspace id and notify other subscribers
-     */
-    this.workspaceList$.subscribe(async (workspaceList) => {
-      const currentWorkspaceId = this.currentWorkspaceId$.getValue();
-      if (currentWorkspaceId) {
-        const findItem = workspaceList.find((item) => item.id === currentWorkspaceId);
-        if (!findItem) {
+
+    combineLatest([this.workspaceList$, this.currentWorkspaceId$]).subscribe(
+      ([workspaceList, currentWorkspaceId]) => {
+        const currentWorkspace = this.findWorkspace([workspaceList, currentWorkspaceId]);
+
+        /**
+         * Do a simple idempotent verification here
+         */
+        if (!isEqual(currentWorkspace, this.currentWorkspace$.getValue())) {
+          this.currentWorkspace$.next(currentWorkspace);
+        }
+
+        if (currentWorkspaceId && !currentWorkspace?.id) {
           /**
            * Current workspace is staled
            */
           this.currentWorkspaceId$.error({
             reason: WORKSPACE_ERROR_REASON_MAP.WORKSPACE_STALED,
           });
+          this.currentWorkspace$.error({
+            reason: WORKSPACE_ERROR_REASON_MAP.WORKSPACE_STALED,
+          });
         }
       }
-    });
+    );
 
     /**
      * Initialize workspace list
@@ -68,6 +77,21 @@ export class WorkspacesClient {
     this.updateWorkspaceListAndNotify();
   }
 
+  private findWorkspace(payload: [WorkspaceAttribute[], string]): WorkspaceAttribute | null {
+    const [workspaceList, currentWorkspaceId] = payload;
+    if (!currentWorkspaceId || !workspaceList || !workspaceList.length) {
+      return null;
+    }
+
+    const findItem = workspaceList.find((item) => item?.id === currentWorkspaceId);
+
+    if (!findItem) {
+      return null;
+    }
+
+    return findItem;
+  }
+
   /**
    * Add a non-throw-error fetch method for internal use.
    */