refactoring types

Signed-off-by: Ramakrishna Chilaka <[email protected]>
opensearch-project · Mar 25, 2024 · 078bd72 · 078bd72
1 parent 2c37acf
commit 078bd72
Show file tree

Hide file tree

Showing 13 changed files with 47 additions and 29 deletions.
diff --git a/public/pages/Aliases/containers/Aliases/Aliases.tsx b/public/pages/Aliases/containers/Aliases/Aliases.tsx
@@ -115,6 +115,7 @@ const defaultFilter = {
 
 class Aliases extends MDSEnabledComponent<AliasesProps, AliasesState> {
   static contextType = CoreServicesContext;
+
   constructor(props: AliasesProps) {
     super(props);
     const {
@@ -482,7 +483,7 @@ class Aliases extends MDSEnabledComponent<AliasesProps, AliasesState> {
   }
 }
 
-export default function AliasContainer(props: Omit<AliasesProps, "commonService">) {
+export default function AliasContainer(props: Omit<AliasesProps, "commonService" | keyof DataSourceMenuProperties>) {
   const context = useContext(ServicesContext);
   const dataSourceMenuProps = useContext(DataSourceMenuContext);
   return <Aliases {...props} commonService={context?.commonService as CommonService} {...dataSourceMenuProps} />;

diff --git a/public/pages/ComposableTemplates/containers/ComposableTemplates/ComposableTemplates.tsx b/public/pages/ComposableTemplates/containers/ComposableTemplates/ComposableTemplates.tsx
@@ -461,7 +461,7 @@ class ComposableTemplates extends MDSEnabledComponent<ComposableTemplatesProps,
 }
 
 export default function ComposableTemplatesContainer(
-  props: Omit<ComposableTemplatesProps, "commonService" | "loading" | "componentMapTemplate">
+  props: Omit<ComposableTemplatesProps, "commonService" | "loading" | "componentMapTemplate" | keyof DataSourceMenuProperties>
 ) {
   const context = useContext(ServicesContext);
   const dataSourceMenuProps = useContext(DataSourceMenuContext);

diff --git a/...CreateComposableTemplate/containers/CreateComposableTemplate/CreateComposableTemplate.tsx b/...CreateComposableTemplate/containers/CreateComposableTemplate/CreateComposableTemplate.tsx
@@ -75,7 +75,7 @@ class CreateComposableTemplate extends Component<CreateComposableTemplateProps>
   }
 }
 
-export default function (props: CreateComposableTemplateProps) {
+export default function (props: Omit<CreateComposableTemplateProps, keyof DataSourceMenuProperties>) {
   const dataSourceMenuProps = useContext(DataSourceMenuContext);
   useUpdateUrlWithDataSourceProperties();
   return <CreateComposableTemplate {...props} {...dataSourceMenuProps} />;

diff --git a/public/pages/CreateDataStream/containers/CreateDataStream/CreateDataStream.tsx b/public/pages/CreateDataStream/containers/CreateDataStream/CreateDataStream.tsx
@@ -64,7 +64,7 @@ class CreateDataStream extends Component<CreateDataStreamProps> {
   }
 }
 
-export default function (props: CreateDataStreamProps) {
+export default function (props: Omit<CreateDataStreamProps, keyof DataSourceMenuProperties>) {
   const dataSourceMenuProps = useContext(DataSourceMenuContext);
   useUpdateUrlWithDataSourceProperties();
   return <CreateDataStream {...props} {...dataSourceMenuProps} />;

diff --git a/public/pages/CreateIndex/containers/CreateIndex/CreateIndex.tsx b/public/pages/CreateIndex/containers/CreateIndex/CreateIndex.tsx
@@ -9,15 +9,15 @@ import { RouteComponentProps } from "react-router-dom";
 import IndexForm from "../IndexForm";
 import { BREADCRUMBS, IndicesUpdateMode, ROUTES } from "../../../../utils/constants";
 import { CoreServicesContext } from "../../../../components/core_services";
-import { CommonService } from "../../../../services/index";
 import { DataSourceMenuContext, DataSourceMenuProperties } from "../../../../services/DataSourceMenuContext";
 import { useUpdateUrlWithDataSourceProperties } from "../../../../components/MDSEnabledComponent";
 
-interface CreateIndexProps extends RouteComponentProps<{ index?: string; mode?: IndicesUpdateMode }>, DataSourceMenuProperties {
+interface CreateIndexPropsBase extends RouteComponentProps<{ index?: string; mode?: IndicesUpdateMode }> {
   isEdit?: boolean;
-  commonService: CommonService;
 }
 
+interface CreateIndexProps extends CreateIndexPropsBase, DataSourceMenuProperties {}
+
 export class CreateIndex extends Component<CreateIndexProps> {
   static contextType = CoreServicesContext;
 
@@ -63,7 +63,7 @@ export class CreateIndex extends Component<CreateIndexProps> {
   }
 }
 
-export default function (props: CreateIndexProps) {
+export default function (props: CreateIndexPropsBase) {
   const dataSourceMenuProperties = useContext(DataSourceMenuContext);
   useUpdateUrlWithDataSourceProperties();
   return <CreateIndex {...props} {...dataSourceMenuProperties} />;

diff --git a/public/pages/CreateIndex/containers/IndexForm/index.tsx b/public/pages/CreateIndex/containers/IndexForm/index.tsx
@@ -46,7 +46,9 @@ export const getAliasActionsByDiffArray = (
   }, [] as IAliasAction[]);
 };
 
-export interface IndexFormProps extends Pick<IndexDetailProps, "readonly" | "sourceIndices">, DataSourceMenuProperties {
+export interface IndexFormProps
+  extends Pick<IndexDetailProps, "readonly" | "sourceIndices">,
+    Pick<DataSourceMenuProperties, "dataSourceId"> {
   index?: string;
   value?: Partial<IndexItemRemote>;
   mode?: IndicesUpdateMode;

diff --git a/public/pages/CreateIndexTemplate/containers/CreateIndexTemplate/CreateIndexTemplate.tsx b/public/pages/CreateIndexTemplate/containers/CreateIndexTemplate/CreateIndexTemplate.tsx
@@ -12,7 +12,12 @@ import { CoreServicesContext } from "../../../../components/core_services";
 import { DataSourceMenuContext, DataSourceMenuProperties } from "../../../../services/DataSourceMenuContext";
 import { useUpdateUrlWithDataSourceProperties } from "../../../../components/MDSEnabledComponent";
 
-interface CreateIndexTemplateProps extends RouteComponentProps<{ template?: string; mode?: string }>, DataSourceMenuProperties {}
+interface CreateIndexTemplateProps
+  extends RouteComponentProps<{
+      template?: string;
+      mode?: string;
+    }>,
+    DataSourceMenuProperties {}
 
 class CreateIndexTemplate extends Component<CreateIndexTemplateProps> {
   static contextType = CoreServicesContext;
@@ -74,7 +79,7 @@ class CreateIndexTemplate extends Component<CreateIndexTemplateProps> {
   }
 }
 
-export default function (props: CreateIndexTemplateProps) {
+export default function (props: Omit<CreateIndexTemplateProps, keyof DataSourceMenuProperties>) {
   const dataSourceMenuProps = useContext(DataSourceMenuContext);
   useUpdateUrlWithDataSourceProperties();
   return <CreateIndexTemplate {...props} {...dataSourceMenuProps} />;

diff --git a/public/pages/DataStreams/containers/DataStreams/DataStreams.tsx b/public/pages/DataStreams/containers/DataStreams/DataStreams.tsx
@@ -428,7 +428,7 @@ class DataStreams extends MDSEnabledComponent<DataStreamsProps, DataStreamsState
   }
 }
 
-export default function DataStreamsContainer(props: Omit<DataStreamsProps, "commonService">) {
+export default function DataStreamsContainer(props: Omit<DataStreamsProps, "commonService" | keyof DataSourceMenuProperties>) {
   const context = useContext(ServicesContext);
   const dataSourceMenuProperties = useContext(DataSourceMenuContext);
   return <DataStreams {...props} commonService={context?.commonService as CommonService} {...dataSourceMenuProperties} />;

diff --git a/public/pages/Indices/containers/Indices/Indices.tsx b/public/pages/Indices/containers/Indices/Indices.tsx
@@ -62,6 +62,7 @@ interface IndicesState extends DataSourceMenuProperties {
 
 export class Indices extends MDSEnabledComponent<IndicesProps, IndicesState> {
   static contextType = CoreServicesContext;
+
   constructor(props: IndicesProps) {
     super(props);
     const { from, size, search, sortField, sortDirection, showDataStreams } = getURLQueryParams(this.props.location);
@@ -101,14 +102,14 @@ export class Indices extends MDSEnabledComponent<IndicesProps, IndicesState> {
   }
 
   async componentDidUpdate(prevProps: IndicesProps, prevState: IndicesState) {
-    const prevQuery = Indices.getQueryObjectFromState(prevState);
-    const currQuery = Indices.getQueryObjectFromState(this.state);
+    const prevQuery = this.getQueryObjectFromState(prevState);
+    const currQuery = this.getQueryObjectFromState(this.state);
     if (!_.isEqual(prevQuery, currQuery)) {
       await this.getIndices();
     }
   }
 
-  static getQueryObjectFromState({
+  getQueryObjectFromState({
     from,
     size,
     search,
@@ -117,14 +118,19 @@ export class Indices extends MDSEnabledComponent<IndicesProps, IndicesState> {
     showDataStreams,
     dataSourceId,
   }: IndicesState): IndicesQueryParams {
-    return { from, size, search, sortField, sortDirection, showDataStreams, dataSourceId };
+    const queryObj = { from, size, search, sortField, sortDirection, showDataStreams };
+    if (!this.props.multiDataSourceEnabled) {
+      // don't send dataSourceId, if MDS is not enabled
+      return queryObj;
+    }
+    return { ...queryObj, dataSourceId };
   }
 
   getIndices = async (): Promise<void> => {
     this.setState({ loadingIndices: true });
     try {
       const { indexService, history } = this.props;
-      const queryObject = Indices.getQueryObjectFromState(this.state);
+      const queryObject = this.getQueryObjectFromState(this.state);
       const queryParamsString = queryString.stringify({ ...queryObject, dataSourceLabel: this.state.dataSourceLabel });
       history.replace({ ...this.props.location, search: queryParamsString });
 
@@ -313,7 +319,7 @@ export class Indices extends MDSEnabledComponent<IndicesProps, IndicesState> {
   }
 }
 
-export default function (props: IndicesProps) {
+export default function (props: Omit<IndicesProps, keyof DataSourceMenuProperties>) {
   const dataSourceMenuProps = useContext(DataSourceMenuContext);
   return <Indices {...props} {...dataSourceMenuProps} />;
 }
diff --git a/public/pages/Indices/models/interfaces.ts b/public/pages/Indices/models/interfaces.ts
@@ -19,5 +19,5 @@ export interface IndicesQueryParams {
   sortField: keyof ManagedCatIndex;
   sortDirection: Direction;
   showDataStreams: boolean;
-  dataSourceId: string;
+  dataSourceId?: string;
 }
diff --git a/public/pages/Main/Main.tsx b/public/pages/Main/Main.tsx
@@ -201,8 +201,8 @@ export default class Main extends Component<MainProps, MainState> {
     };
 
     if (this.props.multiDataSourceEnabled && this.isDataSourceEnabledForPath(pathname)) {
-      services.indexService = new IndexService(http, this.state.dataSourceId);
-      services.commonService = new CommonService(http, this.state.dataSourceId);
+      services.indexService = new IndexService(http, this.state.dataSourceId, this.props.multiDataSourceEnabled);
+      services.commonService = new CommonService(http, this.state.dataSourceId, this.props.multiDataSourceEnabled);
     }
     return services;
   }
@@ -311,7 +311,7 @@ export default class Main extends Component<MainProps, MainState> {
     const { landingPage } = this.props;
 
     const ROUTE_STYLE = { padding: "25px 25px" };
-    const DataSourceMenu = this.props.dataSourceManagement.ui.DataSourceMenu;
+    const DataSourceMenu = this.props.dataSourceManagement?.ui?.DataSourceMenu;
 
     return (
       <CoreServicesConsumer>
@@ -626,7 +626,7 @@ export default class Main extends Component<MainProps, MainState> {
                                 path={`${ROUTES.CREATE_INDEX}/:index/:mode`}
                                 render={(props: RouteComponentProps) => (
                                   <div style={ROUTE_STYLE}>
-                                    <CreateIndex {...props} commonService={services.commonService} />
+                                    <CreateIndex {...props} />
                                   </div>
                                 )}
                               />
@@ -642,7 +642,7 @@ export default class Main extends Component<MainProps, MainState> {
                                 path={ROUTES.CREATE_INDEX}
                                 render={(props: RouteComponentProps) => (
                                   <div style={ROUTE_STYLE}>
-                                    <CreateIndex {...props} commonService={services.commonService} />
+                                    <CreateIndex {...props} />
                                   </div>
                                 )}
                               />

diff --git a/public/pages/Templates/containers/Templates/Templates.tsx b/public/pages/Templates/containers/Templates/Templates.tsx
@@ -417,7 +417,7 @@ class Templates extends MDSEnabledComponent<TemplatesProps, TemplatesState> {
   }
 }
 
-export default function TemplatesContainer(props: Omit<TemplatesProps, "commonService">) {
+export default function TemplatesContainer(props: Omit<TemplatesProps, "commonService" | keyof DataSourceMenuProperties>) {
   const context = useContext(ServicesContext);
   const dataSourceMenuProps = useContext(DataSourceMenuContext);
   return <Templates {...props} commonService={context?.commonService as CommonService} {...dataSourceMenuProps} />;

diff --git a/public/services/MDSEnabledClientService.ts b/public/services/MDSEnabledClientService.ts
@@ -3,15 +3,19 @@ import { HttpFetchQuery, HttpSetup } from "opensearch-dashboards/public";
 export abstract class MDSEnabledClientService {
   httpClient: HttpSetup;
   dataSourceId: string;
+  mdsEnabled: boolean;
 
-  constructor(httpClient: HttpSetup, dataSourceId: string = "") {
+  constructor(httpClient: HttpSetup, dataSourceId: string = "", mdsEnabled: boolean = false) {
     this.httpClient = httpClient;
     this.dataSourceId = dataSourceId;
+    this.mdsEnabled = mdsEnabled;
   }
 
-  patchQueryObjectWithDataSourceId(queryObject: HttpFetchQuery) {
-    queryObject = queryObject || {};
-    queryObject.dataSourceId = this.dataSourceId;
+  patchQueryObjectWithDataSourceId(queryObject: HttpFetchQuery): HttpFetchQuery {
+    if (this.mdsEnabled) {
+      queryObject = queryObject || {};
+      queryObject.dataSourceId = this.dataSourceId;
+    }
     return queryObject;
   }
 }