import _ from 'lodash'
import nameClass from '../util/nameClass'

export default function () {
  return function ({ addUtilities, theme, variants }) {
    const generators = [
      (size, modifier) => ({
        [`${nameClass('space-y', modifier)} > :not(template) ~ :not(template)`]: {
          '--space-y-reverse': '0',
          'margin-top': `calc(${size === '0' ? '0px' : size} * calc(1 - var(--space-y-reverse)))`,
          'margin-bottom': `calc(${size === '0' ? '0px' : size} * var(--space-y-reverse))`,
        },
        [`${nameClass('space-x', modifier)} > :not(template) ~ :not(template)`]: {
          '--space-x-reverse': '0',
          'margin-right': `calc(${size === '0' ? '0px' : size} * var(--space-x-reverse))`,
          'margin-left': `calc(${size === '0' ? '0px' : size} * calc(1 - var(--space-x-reverse)))`,
        },
      }),
    ]

    const utilities = _.flatMap(generators, generator => {
      return [
        ..._.flatMap(theme('space'), generator),
        {
          '.space-y-reverse > :not(template) ~ :not(template)': {
            '--space-y-reverse': '1',
          },
          '.space-x-reverse > :not(template) ~ :not(template)': {
            '--space-x-reverse': '1',
          },
        },
      ]
    })

    addUtilities(utilities, variants('space'))
  }
}