packages/babel-plugin-transform-solid-jsx/src/ssr/template.js - main - githubmirror/taro

packages/babel-plugin-transform-solid-jsx/src/ssr/template.js

74 строки2 KB

phy-lei

feat: [4.x]完成solid-componets-library 和 babel-plugin-transform-solid-jsx 两个包的开发 (#15539)

03 июн 2024, 18:40

03 июн 2024, 18:40c768863

import * as t from '@babel/types'

import { registerImportMethod } from '../shared/utils'

export function createTemplate(path, result) {
  if (!result.template) {
    return result.exprs[0]
  }

  let template, id

  if (!Array.isArray(result.template)) {
    template = t.stringLiteral(result.template)
  } else if (result.template.length === 1) {
    template = t.stringLiteral(result.template[0])
  } else {
    const strings = result.template.map(tmpl => t.stringLiteral(tmpl))
    template = t.arrayExpression(strings)
  }

  const templates =
    path.scope.getProgramParent().data.templates ||
    (path.scope.getProgramParent().data.templates = [])
  const found = templates.find(tmp => {
    if (t.isArrayExpression(tmp.template) && t.isArrayExpression(template)) {
      return tmp.template.elements.every(
        (el, i) => template.elements[i] && el.value === template.elements[i].value
      )
    }
    return tmp.template.value === template.value
  })
  if (!found) {
    id = path.scope.generateUidIdentifier('tmpl$')
    templates.push({
      id,
      template,
      renderer: 'ssr'
    })
  } else id = found.id

  if (result.wontEscape) {
    if (!Array.isArray(result.template) || result.template.length === 1) return id
    else if (
      Array.isArray(result.template) &&
      result.template.length === 2 &&
      result.templateValues[0].type === 'CallExpression' &&
      result.templateValues[0].callee.name === '_$ssrHydrationKey'
    ) {
      // remove unnecessary ssr call when only hydration key is used
      return t.binaryExpression(
        '+',
        t.binaryExpression(
          '+',
          t.memberExpression(id, t.numericLiteral(0), true),
          result.templateValues[0]
        ),
        t.memberExpression(id, t.numericLiteral(1), true)
      )
    }
  }
  return t.callExpression(
    registerImportMethod(path, 'ssr'),
    Array.isArray(result.template) && result.template.length > 1
      ? [id, ...result.templateValues]
      : [id]
  )
}

export function appendTemplates(path, templates) {
  const declarators = templates.map(template => {
    return t.variableDeclarator(template.id, template.template)
  })
  path.node.body.unshift(t.variableDeclaration('var', declarators))
}