123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266
import { MINI } from './constants.ts'
const MODE_SLASH = 0const MODE_TEXT = 1const MODE_WHITESPACE = 2const MODE_TAGNAME = 3const MODE_COMMENT = 4const MODE_PROP_SET = 5const MODE_PROP_APPEND = 6
const CHILD_APPEND = 0const CHILD_RECURSE = 2const TAG_SET = 3const PROPS_ASSIGN = 4const PROP_SET = MODE_PROP_SETconst PROP_APPEND = MODE_PROP_APPEND
// Turn a result of a build(...) call into a tree that is more// convenient to analyze and transform (e.g. Babel plugins).// For example:// 	treeify(//		build'<div href="1${a}" ...${b}><${x} /></div>`,//		[X, Y, Z]//	)// returns:// 	{// 		tag: 'div',//		props: [ { href: ["1", X] },	Y ],// 		children: [ { tag: Z, props: [], children: [] } ]// 	}export const treeify = (built: any, fields: any) => {  const _treeify = (built: any): any => {    let tag = ''    let currentProps: any = null    const props = []    const children: any[] = []
    for (let i = 1; i < built.length; i++) {      const type = built[i++]      const value = built[i] ? fields[built[i++] - 1] : built[++i]
      if (type === TAG_SET) {        tag = value      } else if (type === PROPS_ASSIGN) {        props.push(value)        currentProps = null      } else if (type === PROP_SET) {        if (!currentProps) {          currentProps = Object.create(null)          props.push(currentProps)        }        currentProps[built[++i]] = [value]      } else if (type === PROP_APPEND) {        currentProps[built[++i]].push(value)      } else if (type === CHILD_RECURSE) {        children.push(_treeify(value))      } else if (type === CHILD_APPEND) {        children.push(value)      }    }
    return { tag, props, children }  }  const { children } = _treeify(built)  return children.length > 1 ? children : children[0]}
export const evaluate = (h: any, built: any, fields: any, args: any): any => {  let tmp
  // `build()` used the first element of the operation list as  // temporary workspace. Now that `build()` is done we can use  // that space to track whether the current element is "dynamic"  // (i.e. it or any of its descendants depend on dynamic values).  built[0] = 0
  for (let i = 1; i < built.length; i++) {    const type = built[i++]
    // Set `built[0]`'s appropriate bits if this element depends on a dynamic value.    const value = built[i] ? ((built[0] |= type ? 1 : 2), fields[built[i++]]) : built[++i]
    if (type === TAG_SET) {      args[0] = value    } else if (type === PROPS_ASSIGN) {      args[1] = Object.assign(args[1] || {}, value)    } else if (type === PROP_SET) {      ;(args[1] = args[1] || {})[built[++i]] = value    } else if (type === PROP_APPEND) {      args[1][built[++i]] += `${value}`    } else if (type) {      // type === CHILD_RECURSE      // Set the operation list (including the staticness bits) as      // `this` for the `h` call.      tmp = h.apply(value, evaluate(h, value, fields, ['', null]))      args.push(tmp)
      if (value[0]) {        // Set the 2nd lowest bit it the child element is dynamic.        built[0] |= 2      } else {        // Rewrite the operation list in-place if the child element is static.        // The currently evaluated piece `CHILD_RECURSE, 0, [...]` becomes        // `CHILD_APPEND, 0, tmp`.        // Essentially the operation list gets optimized for potential future        // re-evaluations.        built[i - 2] = CHILD_APPEND        built[i] = tmp      }    } else {      // type === CHILD_APPEND      args.push(value)    }  }
  return args}
export const build = function (statics: TemplateStringsArray, ...rest: any[]) {  const fields = [statics, ...rest]  // @ts-ignore  const h = this
  let mode: any = MODE_TEXT  let buffer = ''  let quote = ''  let current: any = [0]  let char: any  let propName: any
  const commit = (field?: any) => {    if (mode === MODE_TEXT && (field || (buffer = buffer.replace(/^\s*\n\s*|\s*\n\s*$/g, '')))) {      if (MINI) {        current.push(field ? fields[field] : buffer)      } else {        current.push(CHILD_APPEND, field, buffer)      }    } else if (mode === MODE_TAGNAME && (field || buffer)) {      if (MINI) {        current[1] = field ? fields[field] : buffer      } else {        current.push(TAG_SET, field, buffer)      }      mode = MODE_WHITESPACE    } else if (mode === MODE_WHITESPACE && buffer === '...' && field) {      if (MINI) {        current[2] = Object.assign(current[2] || {}, fields[field])      } else {        current.push(PROPS_ASSIGN, field, 0)      }    } else if (mode === MODE_WHITESPACE && buffer && !field) {      if (MINI) {        ;(current[2] = current[2] || {})[buffer] = true      } else {        current.push(PROP_SET, 0, true, buffer)      }    } else if (mode >= MODE_PROP_SET) {      if (MINI) {        if (mode === MODE_PROP_SET) {          ;(current[2] = current[2] || {})[propName] = field            ? buffer              ? buffer + fields[field]              : fields[field]            : buffer          mode = MODE_PROP_APPEND        } else if (field || buffer) {          current[2][propName] += field ? buffer + fields[field] : buffer        }      } else {        if (buffer || (!field && mode === MODE_PROP_SET)) {          current.push(mode, 0, buffer, propName)          mode = MODE_PROP_APPEND        }        if (field) {          current.push(mode, field, 0, propName)          mode = MODE_PROP_APPEND        }      }    }
    buffer = ''  }
  for (let i = 0; i < statics.length; i++) {    if (i) {      if (mode === MODE_TEXT) {        commit()      }      commit(i)    }
    for (let j = 0; j < statics[i].length; j++) {      char = statics[i][j]
      if (mode === MODE_TEXT) {        if (char === '<') {          // commit buffer          commit()          if (MINI) {            current = [current, '', null]          } else {            current = [current]          }          mode = MODE_TAGNAME        } else {          buffer += char        }      } else if (mode === MODE_COMMENT) {        // Ignore everything until the last three characters are '-', '-' and '>'        if (buffer === '--' && char === '>') {          mode = MODE_TEXT          buffer = ''        } else {          buffer = char + buffer[0]        }      } else if (quote) {        if (char === quote) {          quote = ''        } else {          buffer += char        }      } else if (char === '"' || char === "'") {        quote = char      } else if (char === '>') {        commit()        mode = MODE_TEXT      } else if (!mode) {        // Ignore everything until the tag ends      } else if (char === '=') {        mode = MODE_PROP_SET        propName = buffer        buffer = ''      } else if (char === '/' && (mode < MODE_PROP_SET || statics[i][j + 1] === '>')) {        commit()        if (mode === MODE_TAGNAME) {          current = current[0]        }        mode = current        if (MINI) {          ;(current = current[0]).push(h(...mode.slice(1)))        } else {          ;(current = current[0]).push(CHILD_RECURSE, 0, mode)        }        mode = MODE_SLASH      } else if (char === ' ' || char === '\t' || char === '\n' || char === '\r') {        // <a disabled>        commit()        mode = MODE_WHITESPACE      } else {        buffer += char      }
      if (mode === MODE_TAGNAME && buffer === '!--') {        mode = MODE_COMMENT        current = current[0]      }    }  }  commit()
  if (MINI) {    return current.length > 2 ? current.slice(1) : current[1]  }  return current}