ConceptPortal-public/rsconcept/frontend/src/context/ThemeContext.tsx

'use client';

import clsx from 'clsx';
import { createContext, useCallback, useContext, useLayoutEffect, useMemo, useState } from 'react';

import ConceptTooltip from '@/components/Common/ConceptTooltip';
import useLocalStorage from '@/hooks/useLocalStorage';
import { animationDuration } from '@/utils/animations';
import { darkT, IColorTheme, lightT } from '@/utils/color';
import { globalIDs } from '@/utils/constants';

interface IThemeContext {
  viewportHeight: string
  mainHeight: string
  
  colors: IColorTheme
  
  darkMode: boolean
  toggleDarkMode: () => void

  noNavigationAnimation: boolean
  noNavigation: boolean
  toggleNoNavigation: () => void

  noFooter: boolean
  setNoFooter: (value: boolean) => void

  showScroll: boolean
  setShowScroll: (value: boolean) => void
}

const ThemeContext = createContext<IThemeContext | null>(null);
export const useConceptTheme = () => {
  const context = useContext(ThemeContext);
  if (!context) {
    throw new Error('useConceptTheme has to be used within <ThemeState.Provider>');
  }
  return context;
}

interface ThemeStateProps {
  children: React.ReactNode
}

export const ThemeState = ({ children }: ThemeStateProps) => {
  const [darkMode, setDarkMode] = useLocalStorage('darkMode', false);
  const [colors, setColors] = useState<IColorTheme>(lightT);
  const [noNavigation, setNoNavigation] = useState(false);
  const [noNavigationAnimation, setNoNavigationAnimation] = useState(false);
  const [noFooter, setNoFooter] = useState(false);
  const [showScroll, setShowScroll] = useState(false);

  function setDarkClass(isDark: boolean) {
    const root = window.document.documentElement;
    if (isDark) {
      root.classList.add('dark');
    } else {
      root.classList.remove('dark');
    }
    root.setAttribute('data-color-scheme', !isDark ? 'light' : 'dark');
  }

  useLayoutEffect(() => {
    setDarkClass(darkMode);
  }, [darkMode]);

  useLayoutEffect(() => {
    setColors(darkMode ? darkT : lightT)
  }, [darkMode, setColors]);

  const toggleNoNavigation = useCallback(
  () => {
    if (noNavigation) {
      setNoNavigationAnimation(false);
      setNoNavigation(false);
    } else {
      setNoNavigationAnimation(true);
      setTimeout(() => setNoNavigation(true), animationDuration.navigationToggle);
    }
  }, [noNavigation]);

  const mainHeight = useMemo(
  () => {
    return !noNavigation ?
      'calc(100vh - 7rem - 2px)'
    : '100vh';
  }, [noNavigation]);

  const viewportHeight = useMemo(
  () => {
    return !noNavigation ?
      'calc(100vh - 3rem - 2px)'
    : '100vh';
  }, [noNavigation]);

  return (
  <ThemeContext.Provider value={{
    darkMode, colors,
    noNavigationAnimation, noNavigation, noFooter, showScroll,
    toggleDarkMode: () => setDarkMode(prev => !prev),
    toggleNoNavigation: toggleNoNavigation,
    setNoFooter, setShowScroll,
    viewportHeight, mainHeight
  }}>
    <>
      <ConceptTooltip float
        id={`${globalIDs.tooltip}`}
        layer='z-topmost'
        place='right-start'
        className={clsx(
          'mt-3 translate-y-1/2',
          'max-w-[20rem]'
        )}
      />
      {children}
    </>
  </ThemeContext.Provider>);
}