import {
  CurriedCreateTagsFunction,
  HyperScriptHelperNamespace,
  HyperScriptHelperRootTag,
  HyperScriptFunction,
} from './types'
import createTag from './createTag'
import {isNil} from 'ramda'

function createTags<T extends string, NodeData, Node>(
  h: HyperScriptFunction<NodeData, Node>
): CurriedCreateTagsFunction<T, NodeData, Node>
function createTags<T extends string, NodeData, Node>(
  h: HyperScriptFunction<NodeData, Node>,
  tagNames: T[]
): HyperScriptHelperNamespace<T, NodeData, Node>
function createTags<T extends string, NodeData, Node>(
  h: HyperScriptFunction<NodeData, Node>,
  tagNames: T[],
  rootTagName: T
): HyperScriptHelperRootTag<T, NodeData, Node>
function createTags<T extends string, NodeData, Node>(
  h: HyperScriptFunction<NodeData, Node>,
  tagNames?: T[],
  rootTagName?: T
): unknown {
  const _createTag = createTag(h)

  function curriedCreateTags(names: T[], rootName?: T): unknown {
    const root = !isNil(rootName) ? _createTag(rootName) : Object.create(null)

    return names.reduce((result, tagName) => {
      result[tagName] = _createTag(tagName)
      return result
    }, root)
  }

  if (!isNil(tagNames)) {
    return !isNil(rootTagName)
      ? curriedCreateTags(tagNames, rootTagName)
      : curriedCreateTags(tagNames)
  }

  return curriedCreateTags
}

export default createTags