import { SnippetTypes } from '../utils/prop-types'
import { HuiThemesPalette } from '../themes'

export type SnippetStyles = {
  color: string
  border: string
  bgColor: string
}

export const getStyles = (
  type: SnippetTypes,
  palette: HuiThemesPalette,
  fill?: boolean
): SnippetStyles => {
  const styles: { [key in SnippetTypes]: SnippetStyles } = {
    default: {
      color: palette.foreground,
      border: palette.border,
      bgColor: palette.background
    },
    success: {
      color: palette.success,
      border: palette.success,
      bgColor: palette.background
    },
    warning: {
      color: palette.warning,
      border: palette.warning,
      bgColor: palette.background
    },
    error: {
      color: palette.error,
      border: palette.error,
      bgColor: palette.background
    },
    secondary: {
      color: palette.secondary,
      border: palette.secondary,
      bgColor: palette.background
    },
    lite: {
      color: palette.foreground,
      border: palette.border,
      bgColor: palette.accents_1
    },
    dark: {
      color: palette.background,
      border: palette.foreground,
      bgColor: palette.foreground
    }
  }

  const filledTypes: Array<SnippetTypes> = ['success', 'warning', 'error', 'secondary']
  const style = styles[type]
  const shouldFilled = filledTypes.includes(type)
  if (!fill || !shouldFilled) return style

  return {
    ...style,
    color: style.bgColor,
    bgColor: style.color
  }
}