From 8ffa71859b6c08b79f643d7c99595eeb632e7ff0 Mon Sep 17 00:00:00 2001
From: samuelea <batonenbois@gmail.com>
Date: Tue, 17 Oct 2023 14:52:33 -0400
Subject: [PATCH] theming with theme object

---
 postcss.config.cjs => backpostcss.config.cjs  |   0
 .../ThemeProvider/ThemeProvider.stories.tsx   |   3 +-
 .../ThemeProvider/ThemeProvider.tsx           |  69 ++++---
 src/components/ThemeProvider/index.ts         |   1 +
 src/css/vars.css.ts                           |  10 +-
 src/tokens/color.ts                           | 171 ++++++++----------
 src/tokens/index.ts                           |  35 ++--
 7 files changed, 140 insertions(+), 149 deletions(-)
 rename postcss.config.cjs => backpostcss.config.cjs (100%)

diff --git a/postcss.config.cjs b/backpostcss.config.cjs
similarity index 100%
rename from postcss.config.cjs
rename to backpostcss.config.cjs
diff --git a/src/components/ThemeProvider/ThemeProvider.stories.tsx b/src/components/ThemeProvider/ThemeProvider.stories.tsx
index 150e4b46a..f9e47bd0d 100644
--- a/src/components/ThemeProvider/ThemeProvider.stories.tsx
+++ b/src/components/ThemeProvider/ThemeProvider.stories.tsx
@@ -11,8 +11,9 @@ export default {
 
 export const Default = () => {
   const { theme, setTheme } = useTheme()
+
   const toggleTheme = () => {
-    setTheme(theme === 'light' ? 'dark' : 'light')
+    setTheme(theme)
   }
 
   return <Button label="Toggle theme" onClick={toggleTheme} />
diff --git a/src/components/ThemeProvider/ThemeProvider.tsx b/src/components/ThemeProvider/ThemeProvider.tsx
index 39afab4aa..e445f691b 100644
--- a/src/components/ThemeProvider/ThemeProvider.tsx
+++ b/src/components/ThemeProvider/ThemeProvider.tsx
@@ -7,12 +7,36 @@ import {
   useState,
 } from 'react'
 
-const THEMES = ['dark', 'light'] as const
+export interface Color {
+  r: number
+  g: number
+  b: number
+}
 
-export type Theme = (typeof THEMES)[number]
+export interface Theme {
+  foreground: Color
+  background: Color
+  backgroundBackdrop: Color,
+  backgroundRaised: Color,
+  statusPositive: Color
+  statusNegative: Color
+  statusWarning: Color
+  statusInfo: Color
+  primaryButton: string
+}
+
+const DEFAULT_THEME: Theme = {
+  foreground: { r: 255, g: 255, b: 255 },
+  background: { r: 0, g: 0, b: 0 },
+  backgroundBackdrop: { r: 34, g: 34, b: 34 },
+  backgroundRaised: { r: 54, g: 54, b: 54 },
+  statusPositive: { r: 31, g: 194, b: 102 },
+  statusNegative: { r: 194, g: 80, b: 31 },
+  statusWarning: { r: 244, g: 176, b: 62 },
+  statusInfo: { r: 0, g: 118, b: 204 },
+  primaryButton: 'linear-gradient(89.69deg, #4411E1 0.27%, #7537F9 99.73%)'
+}
 
-const DEFAULT_THEME = 'dark'
-const THEME_ATTR = 'data-theme'
 const STORAGE_KEY = '@sequence.theme'
 
 interface ThemeContextValue {
@@ -25,18 +49,13 @@ interface ThemeProviderProps {
 }
 
 const getTheme = (): Theme => {
-  const persistedTheme = localStorage.getItem(STORAGE_KEY) as Theme
+  const persistedThemeRaw = localStorage.getItem(STORAGE_KEY)
 
-  if (THEMES.includes(persistedTheme)) {
+  if (persistedThemeRaw) {
+    const persistedTheme = JSON.parse(persistedThemeRaw) as Theme
     return persistedTheme
   }
 
-  // else if (matchMedia(`(prefers-color-scheme: light)`).matches) {
-  //   return 'light'
-  // } else if (matchMedia(`(prefers-color-scheme: dark)`).matches) {
-  //   return 'dark'
-  // }
-
   return DEFAULT_THEME
 }
 
@@ -45,6 +64,7 @@ const ThemeContext = createContext<ThemeContextValue | null>(null)
 export const ThemeProvider = (props: PropsWithChildren<ThemeProviderProps>) => {
   const [theme, setTheme] = useState<Theme>(props.theme || getTheme())
 
+  
   useEffect(() => {
     // Add is-apple class
     ;/Mac/.test(window.navigator.userAgent) &&
@@ -53,32 +73,23 @@ export const ThemeProvider = (props: PropsWithChildren<ThemeProviderProps>) => {
 
   // Allow prop theme override
   useEffect(() => {
-    if (props.theme && THEMES.includes(props.theme)) {
+    if (props.theme) {
       setTheme(props.theme)
     }
   }, [props.theme])
 
-  // Set the data-theme attribtute on the document root element
-  useEffect(() => {
-    const root = document.querySelector(':root')
-
-    if (root) {
-      root.setAttribute(THEME_ATTR, theme)
-    }
-  }, [theme])
 
   // Create the context value
   const value: ThemeContextValue = useMemo(() => {
     return {
       theme,
-      setTheme: (mode: Theme) => {
-        if (THEMES.includes(mode)) {
-          // Save to local storage
-          localStorage.setItem(STORAGE_KEY, mode)
-
-          // Set the theme state which will cause a re-render
-          setTheme(mode)
-        }
+      setTheme: (newTheme: Theme) => {
+        const themeString = JSON.stringify(newTheme)
+        // Save to local storage
+        localStorage.setItem(STORAGE_KEY, themeString)
+
+        // Set the theme state which will cause a re-render
+        setTheme(newTheme)
       },
     }
   }, [theme])
diff --git a/src/components/ThemeProvider/index.ts b/src/components/ThemeProvider/index.ts
index 0b666958b..63a70806e 100644
--- a/src/components/ThemeProvider/index.ts
+++ b/src/components/ThemeProvider/index.ts
@@ -1 +1,2 @@
 export { ThemeProvider, useTheme } from './ThemeProvider'
+export type { Theme, Color } from './ThemeProvider'
\ No newline at end of file
diff --git a/src/css/vars.css.ts b/src/css/vars.css.ts
index a6eb37285..78d40fa64 100644
--- a/src/css/vars.css.ts
+++ b/src/css/vars.css.ts
@@ -3,10 +3,12 @@ import {
   createGlobalThemeContract,
 } from '@vanilla-extract/css'
 
+import { Theme } from '~/components/ThemeProvider' 
 import { capitalize } from '~/helpers'
-import { ColorScheme, tokens } from '~/tokens'
+import { getColors, getTokens, NetworkColors } from '~/tokens'
 
 import { mapVarName } from './utils'
+import { getColorSchemes } from '~/tokens/color'
 
 type MapTokens<P extends string, T> = {
   [K in keyof T & string as `${P}${Capitalize<K>}`]: string
@@ -21,8 +23,6 @@ const mapTokens = <P extends string, T extends {}>(
   }, {}) as MapTokens<P, T>
 }
 
-type NetworkColors = typeof tokens.colors.network
-
 const mapNetworkColors = <
   T extends NetworkColors,
   K extends keyof NetworkColors,
@@ -43,8 +43,8 @@ export const baseVars = createGlobalThemeContract(baseTokens, mapVarName)
 
 createGlobalTheme(':root', baseVars, baseTokens)
 
-const makeColorScheme = (mode: ColorScheme = 'light') => {
-  const colorSchemeTokens = colors.colorSchemes[mode]
+const makeColorScheme = (theme: Theme) => {
+  const colorSchemeTokens = getColorSchemes(theme)
 
   return {
     colors: {
diff --git a/src/tokens/color.ts b/src/tokens/color.ts
index c4b4a2049..1b430a5f1 100644
--- a/src/tokens/color.ts
+++ b/src/tokens/color.ts
@@ -1,13 +1,15 @@
+import { Theme, Color } from '~/components/ThemeProvider'
+
+export type NetworkColors = typeof networkColors
+
 export interface ColorTokens {
   base: BaseColors
   context: ContextColors
-  network: typeof networkColors
+  network: NetworkColors
   gradient: Gradients
-  colorSchemes: ColorSchemes
+  colorSchemes: ColorSchemeTokens
 }
 
-export type ColorScheme = 'dark' | 'light'
-
 interface ColorSchemeTokens {
   background: BackgroundColors
   border: BorderColors
@@ -17,8 +19,6 @@ interface ColorSchemeTokens {
   // context: ContextColors // ContextColors are global and not color scheme specific
 }
 
-type ColorSchemes<T = ColorSchemeTokens> = { [key in ColorScheme]: T }
-
 interface BaseColors {
   black: string
   white: string
@@ -69,68 +69,46 @@ interface Gradients {
   secondary: string
 }
 
-const backgroundColors: ColorSchemes<BackgroundColors> = {
-  dark: {
-    primary: 'rgba(0, 0, 0, 1)',
-    secondary: 'rgba(255, 255, 255, 0.1)',
-    contrast: 'rgba(0, 0, 0, 0.5)',
-    muted: 'rgba(255, 255, 255, 0.05)',
-    control: 'rgba(255, 255, 255, 0.25)',
-    inverse: 'rgba(255, 255, 255, 1)',
-    backdrop: 'rgba(34, 34, 34, 0.9)',
-    overlay: 'rgba(0, 0, 0, 0.7)',
-    raised: 'rgba(54, 54, 54, 0.7)',
-  },
-  light: {
-    primary: 'rgba(244, 244, 244, 1)',
-    secondary: 'rgba(0, 0, 0, 0.1)',
-    contrast: 'rgba(244, 244, 244, 0.5)',
-    muted: 'rgba(0, 0, 0, 0.05)',
-    control: 'rgba(0, 0, 0, 0.25)',
-    inverse: 'rgba(0, 0, 0, 1)',
-    backdrop: 'rgba(221, 221, 221, 0.9)',
-    overlay: 'rgba(244, 244, 244, 0.7)',
-    raised: 'rgba(192, 192, 192, 0.7)',
-  },
+const makeRGBA = (color: Color, alpha: number) => {
+  return `rgba(${color.r},${color.g},${color.b},${alpha})`
 }
 
-const borderColors: ColorSchemes<BorderColors> = {
-  dark: {
-    normal: 'rgba(255, 255, 255, 0.25)',
-    focus: 'rgba(255, 255, 255, 0.5)',
-  },
-  light: {
-    normal: 'rgba(0, 0, 0, 0.25)',
-    focus: 'rgba(0, 0, 0, 0.5)',
-  },
+const getBackgroundColors = (theme: Theme): BackgroundColors => {
+  return ({
+    primary: makeRGBA(theme.background, 1),
+    secondary: makeRGBA(theme.foreground, 1),
+    contrast: makeRGBA(theme.background, 0.5),
+    muted: makeRGBA(theme.foreground, 0.05),
+    control: makeRGBA(theme.foreground, 0.25),
+    inverse: makeRGBA(theme.foreground, 1),
+    backdrop: makeRGBA(theme.backgroundBackdrop, 0.9),
+    overlay:  makeRGBA(theme.background, 0.7),
+    raised: makeRGBA(theme.backgroundRaised, 0.7),
+  })
 }
 
-const buttonColors: ColorSchemes<ButtonColors> = {
-  dark: {
-    glass: 'rgba(255, 255, 255, 0.15)',
-    emphasis: 'rgba(0, 0, 0, 0.5)',
-    inverse: 'rgba(255, 255, 255, 0.8)',
-  },
-  light: {
-    glass: 'rgba(0, 0, 0, 0.15)',
-    emphasis: 'rgba(255, 255, 255, 0.5)',
-    inverse: 'rgba(0, 0, 0, 0.8)',
-  },
+const getBorderColors = (theme: Theme): BorderColors => {
+  return ({
+    normal: makeRGBA(theme.foreground, 0.25),
+    focus: makeRGBA(theme.foreground, 0.5),
+  })
 }
 
-const textColors: ColorSchemes<TextColors> = {
-  dark: {
-    '100': 'rgba(255, 255, 255, 1)',
-    '80': 'rgba(255, 255, 255, 0.8)',
-    '50': 'rgba(255, 255, 255, 0.5)',
-    inverse100: 'rgba(0, 0, 0, 1)',
-  },
-  light: {
-    '100': 'rgba(0, 0, 0, 1)',
-    '80': 'rgba(0, 0, 0, 0.8)',
-    '50': 'rgba(0, 0, 0, 0.5)',
-    inverse100: 'rgba(255, 255, 255, 1)',
-  },
+const getButtonColors = (theme: Theme): ButtonColors => {
+  return ({
+    glass: makeRGBA(theme.foreground, 0.15),
+    emphasis: makeRGBA(theme.background, 0.5),
+    inverse: makeRGBA(theme.foreground, 0.8),
+  })
+}
+
+const getTextColors = (theme: Theme): TextColors => {
+  return ({
+    '100': makeRGBA(theme.foreground, 1),
+    '80': makeRGBA(theme.foreground, 0.8),
+    '50': makeRGBA(theme.foreground, 0.5),
+    inverse100: makeRGBA(theme.background, 1),
+  })
 }
 
 // ContextColors are global and not color scheme specific
@@ -141,16 +119,17 @@ const contextColors: ContextColors = {
   warning: '#F4B03E',
 }
 
-// Gradients are global and not color scheme specific
-const gradients: Gradients = {
-  backdrop: `linear-gradient(
-    243.18deg, 
-    rgba(86, 52, 189, 0.85) 0%, 
-    rgba(49, 41, 223, 0.85) 63.54%, 
-    rgba(7, 98, 149, 0.85) 100%
-  )`,
-  primary: `linear-gradient(89.69deg, #4411E1 0.27%, #7537F9 99.73%)`,
-  secondary: `linear-gradient(32.51deg, #951990 -15.23%, #3A35B1 48.55%, #20A8B0 100%)`,
+const getGradients = (theme: Theme): Gradients => {
+  return ({
+    backdrop: `linear-gradient(
+      243.18deg, 
+      rgba(86, 52, 189, 0.85) 0%, 
+      rgba(49, 41, 223, 0.85) 63.54%, 
+      rgba(7, 98, 149, 0.85) 100%
+    )`,
+    primary: theme.primaryButton,
+    secondary: `linear-gradient(32.51deg, #951990 -15.23%, #3A35B1 48.55%, #20A8B0 100%)`,
+  })
 }
 
 const networkColors = {
@@ -180,30 +159,26 @@ const networkColors = {
   },
 }
 
-const colorSchemes: ColorSchemes = {
-  dark: {
-    background: backgroundColors.dark,
-    border: borderColors.dark,
-    button: buttonColors.dark,
-    text: textColors.dark,
-  },
-  light: {
-    background: backgroundColors.light,
-    border: borderColors.light,
-    button: buttonColors.light,
-    text: textColors.light,
-  },
-}
-
-export const colors: ColorTokens = {
-  base: {
-    black: '#000000',
-    white: '#ffffff',
-    inherit: 'inherit',
-    transparent: 'transparent',
-  },
-  context: contextColors,
-  gradient: gradients,
-  network: networkColors,
-  colorSchemes,
+export const getColorSchemes = (theme: Theme): ColorSchemeTokens => {
+  return ({
+    background: getBackgroundColors(theme),
+    border: getBorderColors(theme),
+    button: getButtonColors(theme),
+    text: getTextColors(theme),
+  })
+}
+
+export const getColors = (theme: Theme): ColorTokens => {
+  return ({
+    base: {
+      black: '#000000',
+      white: '#ffffff',
+      inherit: 'inherit',
+      transparent: 'transparent',
+    },
+    context: contextColors,
+    gradient: getGradients(theme),
+    network: networkColors,
+    colorSchemes: getColorSchemes(theme),
+  })
 }
diff --git a/src/tokens/index.ts b/src/tokens/index.ts
index 83b51e6ff..1c8466bb6 100644
--- a/src/tokens/index.ts
+++ b/src/tokens/index.ts
@@ -1,5 +1,6 @@
+import { Theme } from '~/components/ThemeProvider'
 import { borderWidths, radii } from './border'
-import { colors } from './color'
+import { getColors } from './color'
 import { blur } from './effect'
 import { opacity } from './opacity'
 import { space } from './space'
@@ -11,21 +12,23 @@ import {
   lineHeights,
 } from './typography'
 
-export const tokens = {
-  blur,
-  borderWidths,
-  colors,
-  fonts,
-  fontSizes,
-  fontWeights,
-  letterSpacings,
-  lineHeights,
-  opacity,
-  radii,
-  space,
+export const getTokens = (theme: Theme) => {
+  return ({
+    blur,
+    borderWidths,
+    colors: getColors(theme),
+    fonts,
+    fontSizes,
+    fontWeights,
+    letterSpacings,
+    lineHeights,
+    opacity,
+    radii,
+    space,
+  })
 }
 
-export { colors } from './color'
+export { getColors } from './color'
 
-export type { ColorScheme } from './color'
-export type Tokens = typeof tokens
+export type { NetworkColors } from './color' 
+export type Tokens = ReturnType<typeof getTokens>