lsd/packages/lsd-react/.storybook/withTheme.decorator.tsx

import { Global, css } from '@emotion/react'
import { useGlobals } from '@storybook/preview-api'
import { Decorator } from '@storybook/react'
import React, { useEffect } from 'react'
import { ThemeProvider } from '../src'
import { storybookThemes } from './themes'

export const withTheme: Decorator = (Story, context) => {
  const StoryComponent = Story as any as React.ComponentType
  const isDoc = context.viewMode === 'docs'

  const theme = storybookThemes.getTheme(context)
  const [globals, setGlobals] = useGlobals()

  useEffect(() => {
    const background = (context.parameters.backgrounds?.values ?? []).find(
      (value) => theme.name.startsWith(value.name),
    )?.value

    globals.backgrounds?.value !== background &&
      setGlobals({
        ...globals,
        backgrounds: {
          ...(globals.background ?? {}),
          value: background,
        },
      })
  }, [theme])

  return (
    <div>
      <ThemeProvider theme={theme} injectCssVars={true}>
        <StoryComponent />
        {isDoc && (
          <Global
            styles={css`
              .docs-story {
                ${theme.cssVars}

                background: rgb(var(--lsd-surface-primary));
              }
            `}
          />
        )}
      </ThemeProvider>
    </div>
  )
}