@department-of-veterans-affairs/mobile-component-library/src/utils/hooks/useColorScheme.ts

VA Design System Mobile Component Library

import {
  ColorSchemeName,
  useColorScheme as RNUseColorScheme,
} from 'react-native'

/** Handles return of color scheme based on platform */
export function useColorScheme(): ColorSchemeName {
  // If not web Storybook, set with RN useColorScheme hook
  if (!process.env.STORYBOOK_WEB) {
    return RNUseColorScheme()
  } else {
    try {
      const webStorybookColorScheme =
        require('./useWebStorybookColorScheme').useWebStorybookColorScheme // eslint-disable-line
      return webStorybookColorScheme()
    } catch (error) {
      return null
    }
  }
}