feat: add APIs to support plugin state in request (opensearch-project…

…#312)

* feat: add APIs to support plugin state in request

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

* feat: add APIs to support plugin state in request

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

---------

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

src/core/server/utils/index.ts

@@ -33,3 +33,4 @@ export * from './from_root';
 export * from './package_json';
 export * from './streams';
 export { getWorkspaceIdFromUrl, cleanWorkspaceId } from '../../utils';
+export { updateWorkspaceState, getWorkspaceState } from './workspace';

src/core/server/utils/workspace.test.ts

@@ -0,0 +1,19 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { httpServerMock } from '../mocks';
+import { getWorkspaceState, updateWorkspaceState } from './workspace';
+
+describe('updateWorkspaceState', () => {
+  it('update with payload', () => {
+    const requestMock = httpServerMock.createOpenSearchDashboardsRequest();
+    updateWorkspaceState(requestMock, {
+      id: 'foo',
+    });
+    expect(getWorkspaceState(requestMock)).toEqual({
+      id: 'foo',
+    });
+  });
+});

src/core/server/utils/workspace.ts

@@ -0,0 +1,43 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+/**
+ * This file is using {@link PluginsStates} to store workspace info into request.
+ * The best practice would be using {@link Server.register} to register plugins into the hapi server
+ * but OSD is wrappering the hapi server and the hapi server instance is hidden as internal implementation.
+ */
+import { PluginsStates } from '@hapi/hapi';
+import { OpenSearchDashboardsRequest, ensureRawRequest } from '../http/router';
+
+/**
+ * This function will be used as a proxy
+ * because `ensureRequest` is only importable from core module.
+ *
+ * @param workspaceId string
+ * @returns void
+ */
+export const updateWorkspaceState = (
+  request: OpenSearchDashboardsRequest,
+  payload: Partial<PluginsStates['workspace']>
+) => {
+  const rawRequest = ensureRawRequest(request);
+
+  if (!rawRequest.plugins) {
+    rawRequest.plugins = {};
+  }
+
+  if (!rawRequest.plugins.workspace) {
+    rawRequest.plugins.workspace = {};
+  }
+
+  rawRequest.plugins.workspace = {
+    ...rawRequest.plugins.workspace,
+    ...payload,
+  };
+};
+
+export const getWorkspaceState = (request: OpenSearchDashboardsRequest) => {
+  return ensureRawRequest(request).plugins?.workspace;
+};

src/legacy/server/osd_server.d.ts

@@ -60,6 +60,14 @@ declare module 'hapi' {
   }
 }
 
+declare module '@hapi/hapi' {
+  interface PluginsStates {
+    workspace?: {
+      id?: string;
+    };
+  }
+}
+
 type OsdMixinFunc = (osdServer: OsdServer, server: Server, config: any) => Promise<any> | void;
 
 export interface PluginsSetup {