jan/web/hooks/useLoadTheme.ts

import { useCallback, useEffect } from 'react'

import { useTheme } from 'next-themes'

import { fs, joinPath } from '@janhq/core'

import { useAtom, useAtomValue, useSetAtom } from 'jotai'

import cssVars from '@/utils/jsonToCssVariables'

import { janDataFolderPathAtom } from '@/helpers/atoms/AppConfig.atom'
import {
  janThemesPathAtom,
  selectedThemeIdAtom,
  themeDataAtom,
  themesOptionsAtom,
} from '@/helpers/atoms/Setting.atom'

type NativeThemeProps = 'light' | 'dark'

export const useLoadTheme = async () => {
  const janDataFolderPath = useAtomValue(janDataFolderPathAtom)
  const setThemeOptions = useSetAtom(themesOptionsAtom)
  const setThemePath = useSetAtom(janThemesPathAtom)
  const [themeData, setThemeData] = useAtom(themeDataAtom)
  const [selectedIdTheme, setSelectedIdTheme] = useAtom(selectedThemeIdAtom)
  const { setTheme } = useTheme()

  const setNativeTheme = useCallback(
    (nativeTheme: NativeThemeProps) => {
      if (nativeTheme === 'dark') {
        window?.electronAPI.setNativeThemeDark()
        setTheme('dark')
        localStorage.setItem('nativeTheme', 'dark')
      } else {
        window?.electronAPI.setNativeThemeLight()
        setTheme('light')
        localStorage.setItem('nativeTheme', 'light')
      }
    },
    [setTheme]
  )

  const getThemes = useCallback(async () => {
    const folderPath = await joinPath([janDataFolderPath, 'themes'])
    const installedThemes = await fs.readdirSync(folderPath)

    const themesOptions: { name: string; value: string }[] = installedThemes
      .filter((x: string) => x !== '.DS_Store')
      .map(async (x: string) => {
        const y = await joinPath([`${folderPath}/${x}`, `theme.json`])
        const c: Theme = JSON.parse(await fs.readFileSync(y, 'utf-8'))
        return { name: c?.displayName, value: c.id }
      })
    Promise.all(themesOptions).then((results) => {
      setThemeOptions(results)
    })

    if (janDataFolderPath.length > 0) {
      if (!selectedIdTheme.length) return setSelectedIdTheme('joi-light')
      setThemePath(folderPath)
      const filePath = await joinPath([
        `${folderPath}/${selectedIdTheme}`,
        `theme.json`,
      ])
      const theme: Theme = JSON.parse(await fs.readFileSync(filePath, 'utf-8'))

      setThemeData(theme)
      setNativeTheme(theme.nativeTheme)
      const variables = cssVars(theme.variables)
      const headTag = document.getElementsByTagName('head')[0]
      const styleTag = document.createElement('style')
      styleTag.innerHTML = `:root {${variables}}`
      headTag.appendChild(styleTag)
    }
  }, [
    janDataFolderPath,
    selectedIdTheme,
    setNativeTheme,
    setSelectedIdTheme,
    setThemeData,
    setThemeOptions,
    setThemePath,
  ])

  useEffect(() => {
    getThemes()
    setNativeTheme(themeData?.nativeTheme as NativeThemeProps)
  }, [
    getThemes,
    selectedIdTheme,
    setNativeTheme,
    setSelectedIdTheme,
    themeData?.nativeTheme,
  ])
}