src/state/shell/color-mode.tsx


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

import React from 'react'
import {isWeb} from '#/platform/detection'
import * as persisted from '#/state/persisted'

type StateContext = persisted.Schema['colorMode']
type SetContext = (v: persisted.Schema['colorMode']) => void

const stateContext = React.createContext<StateContext>('system')
const setContext = React.createContext<SetContext>(
  (_: persisted.Schema['colorMode']) => {},
)

export function Provider({children}: React.PropsWithChildren<{}>) {
  const [state, setState] = React.useState(persisted.get('colorMode'))

  const setStateWrapped = React.useCallback(
    (colorMode: persisted.Schema['colorMode']) => {
      setState(colorMode)
      persisted.write('colorMode', colorMode)
      updateDocument(colorMode)
    },
    [setState],
  )

  React.useEffect(() => {
    return persisted.onUpdate(() => {
      setState(persisted.get('colorMode'))
      updateDocument(persisted.get('colorMode'))
    })
  }, [setStateWrapped])

  return (
    <stateContext.Provider value={state}>
      <setContext.Provider value={setStateWrapped}>
        {children}
      </setContext.Provider>
    </stateContext.Provider>
  )
}

export function useColorMode() {
  return React.useContext(stateContext)
}

export function useSetColorMode() {
  return React.useContext(setContext)
}

function updateDocument(colorMode: string) {
  if (isWeb && typeof window !== 'undefined') {
    const html = window.document.documentElement
    // remove any other color mode classes
    html.className = html.className.replace(/colorMode--\w+/g, '')
    html.classList.add(`colorMode--${colorMode}`)
  }
}