rebase

awslabs · Nov 21, 2024 · e95ddb1 · e95ddb1
1 parent 4f3b06a
commit e95ddb1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/packages/threat-composer-app/src/utils/isMemoryRouterUsed/index.ts b/packages/threat-composer-app/src/utils/isMemoryRouterUsed/index.ts
@@ -13,12 +13,12 @@
   See the License for the specific language governing permissions and
   limitations under the License.
  ******************************************************************************************************************** */
-import { APP_MODE_IDE_EXTENSION } from '@aws/threat-composer';
+import { APP_MODE_IDE_EXTENSION, APP_MODE_BROWSER_EXTENSION } from '@aws/threat-composer';
 
 const appMode = process.env.REACT_APP_APP_MODE;
 
 const isMemoryRouterUsed = () => {
-  return appMode === APP_MODE_IDE_EXTENSION;
+  return appMode === APP_MODE_IDE_EXTENSION || appMode === APP_MODE_BROWSER_EXTENSION;
 };
 
 export default isMemoryRouterUsed;
diff --git a/packages/threat-composer/src/components/report/ThreatModel/index.tsx b/packages/threat-composer/src/components/report/ThreatModel/index.tsx
@@ -15,7 +15,7 @@
  ******************************************************************************************************************** */
 import { FC, useMemo } from 'react';
 import ThreatModelView, { ThreatModelViewProps } from './components/ThreatModelView';
-import { APP_MODE_IDE_EXTENSION } from '../../../configs/appMode';
+import { APP_MODE_IDE_EXTENSION, APP_MODE_BROWSER_EXTENSION } from '../../../configs/appMode';
 import { useGlobalSetupContext, useWorkspacesContext } from '../../../contexts';
 import { DataExchangeFormat, ViewNavigationEvent } from '../../../customTypes';
 import useImportExport from '../../../hooks/useExportImport';
@@ -49,7 +49,7 @@ const ThreatModel: FC<ThreatModelProps> = ({
     <ThreatModelView
       {...props}
       onPrintButtonClick={() => onPrintButtonClick?.(getWorkspaceData())}
-      showPrintDownloadButtons={appMode !== APP_MODE_IDE_EXTENSION}
+      showPrintDownloadButtons={appMode !== APP_MODE_IDE_EXTENSION && appMode !== APP_MODE_BROWSER_EXTENSION }
       composerMode={composerMode}
       data={getWorkspaceData()}
       downloadFileName={downloadFileName}