chore: replace with useRuntimeVisualRefresh from component toolkit

cloudscape-design · Sep 19, 2023 · 619925b · 619925b
1 parent 6d19d70
commit 619925b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 123 deletions.
diff --git a/src/internal/hooks/use-visual-mode/__tests__/use-visual-mode.test.tsx b/src/internal/hooks/use-visual-mode/__tests__/use-visual-mode.test.tsx
diff --git a/src/internal/hooks/use-visual-mode/index.ts b/src/internal/hooks/use-visual-mode/index.ts
@@ -2,46 +2,6 @@
 // SPDX-License-Identifier: Apache-2.0
 
 import { ALWAYS_VISUAL_REFRESH } from '../../environment';
-import { isDevelopment } from '../../is-development';
-import { warnOnce } from '@cloudscape-design/component-toolkit/internal';
+import { useRuntimeVisualRefresh } from '@cloudscape-design/component-toolkit/internal';
 
-const awsuiVisualRefreshFlag = Symbol.for('awsui-visual-refresh-flag');
-interface ExtendedWindow extends Window {
-  [awsuiVisualRefreshFlag]?: () => boolean;
-}
-declare const window: ExtendedWindow;
-
-export const useVisualRefresh = ALWAYS_VISUAL_REFRESH ? () => true : useVisualRefreshDynamic;
-
-// We expect VR is to be set only once and before the application is rendered.
-let visualRefreshState: undefined | boolean = undefined;
-
-// for testing
-export function clearVisualRefreshState() {
-  visualRefreshState = undefined;
-}
-
-function detectVisualRefresh() {
-  return typeof document !== 'undefined' && !!document.querySelector('.awsui-visual-refresh');
-}
-
-export function useVisualRefreshDynamic() {
-  if (visualRefreshState === undefined) {
-    visualRefreshState = detectVisualRefresh();
-    if (!visualRefreshState && typeof window !== 'undefined' && window[awsuiVisualRefreshFlag]?.()) {
-      document.body.classList.add('awsui-visual-refresh');
-      visualRefreshState = true;
-    }
-  }
-  if (isDevelopment) {
-    const newVisualRefreshState = detectVisualRefresh();
-    if (newVisualRefreshState !== visualRefreshState) {
-      warnOnce(
-        'Visual Refresh',
-        'Dynamic visual refresh change detected. This is not supported. ' +
-          'Make sure `awsui-visual-refresh` is attached to the `<body>` element before initial React render'
-      );
-    }
-  }
-  return visualRefreshState;
-}
+export const useVisualRefresh = ALWAYS_VISUAL_REFRESH ? () => true : useRuntimeVisualRefresh;