@tamagui/static 1.0.1-beta.20 → 1.0.1-beta.201

package/src/extractor/loadTamagui.ts

@@ -0,0 +1,478 @@
+import { readFileSync } from 'fs'
+/* eslint-disable no-console */
+import { basename, dirname, extname, join, relative, sep } from 'path'
+
+import generate from '@babel/generator'
+import traverse from '@babel/traverse'
+import * as t from '@babel/types'
+import { Color, colorLog } from '@tamagui/cli-color'
+import { getDefaultTamaguiConfig } from '@tamagui/config-default-node'
+import type { StaticConfigParsed, TamaguiInternalConfig } from '@tamagui/core-node'
+import { createTamagui } from '@tamagui/core-node'
+import esbuild from 'esbuild'
+import { ensureDir, existsSync, removeSync, writeFileSync } from 'fs-extra'
+
+import { SHOULD_DEBUG } from '../constants.js'
+import { getNameToPaths, registerRequire, unregisterRequire } from '../require.js'
+import { babelParse } from './babelParse'
+import { bundle } from './bundle'
+
+type NameToPaths = {
+  [key: string]: Set<string>
+}
+
+export type LoadedComponents = {
+  moduleName: string
+  nameToInfo: Record<
+    string,
+    {
+      staticConfig: StaticConfigParsed
+    }
+  >
+}
+
+export type TamaguiProjectInfo = {
+  components: LoadedComponents[]
+  tamaguiConfig: TamaguiInternalConfig
+  nameToPaths: NameToPaths
+}
+
+type Props = {
+  components: string[]
+  config?: string
+  forceExports?: boolean
+}
+
+const cache = {}
+
+// TODO needs a plugin for webpack / vite to run this once at startup and not again until changed...
+
+export async function loadTamagui(props: Props): Promise<TamaguiProjectInfo> {
+  const key = JSON.stringify(props)
+  if (cache[key]) {
+    if (cache[key] instanceof Promise) {
+      return await cache[key]
+    }
+    return cache[key]
+  }
+
+  let resolver: Function = () => {}
+  cache[key] = new Promise((res) => {
+    resolver = res
+  })
+
+  const tmpDir = join(process.cwd(), '.tamagui', 'tamagui-node')
+  const configOutPath = join(tmpDir, `tamagui.config.js`)
+  const baseComponents = props.components.filter((x) => x !== '@tamagui/core')
+  const componentOutPaths = baseComponents.map((componentModule) =>
+    join(
+      tmpDir,
+      `${componentModule
+        .split(sep)
+        .join('-')
+        .replace(/[^a-z0-9]+/gi, '')}-components.config.js`
+    )
+  )
+
+  const external = ['@tamagui/core', '@tamagui/core-node', 'react', 'react-dom', 'react-native-svg']
+  const configEntry = props.config ? join(process.cwd(), props.config) : ''
+
+  if (process.env.NODE_ENV === 'development' && process.env.DEBUG?.startsWith('tamagui')) {
+    console.log(`Building config entry`, configEntry)
+  }
+
+  // build them to node-compat versions
+  try {
+    await ensureDir(tmpDir)
+  } catch {
+    //
+  }
+
+  colorLog(
+    Color.FgYellow,
+    `
+Tamagui built config and components:`
+  )
+  colorLog(
+    Color.Dim,
+    `
+  - Config: .${sep}${relative(process.cwd(), configOutPath)}
+  - Components: .${sep}${relative(process.cwd(), componentOutPaths.join(', '))}
+`
+  )
+
+  await Promise.all([
+    props.config
+      ? bundle({
+          entryPoints: [configEntry],
+          external,
+          outfile: configOutPath,
+        })
+      : null,
+    ...baseComponents.map((componentModule, i) => {
+      return bundle({
+        entryPoints: [componentModule],
+        resolvePlatformSpecificEntries: true,
+        external,
+        outfile: componentOutPaths[i],
+      })
+    }),
+  ])
+
+  try {
+    registerRequire()
+    const out = require(configOutPath)
+    const config = out.default || out
+
+    if (!config) {
+      throw new Error(`No config: ${config}`)
+    }
+
+    let components = loadComponents({
+      ...props,
+      components: componentOutPaths,
+    })
+
+    if (!components) {
+      throw new Error(`No components found: ${componentOutPaths.join(', ')}`)
+    }
+
+    // map from built back to original module names
+    for (const component of components) {
+      component.moduleName = baseComponents[componentOutPaths.indexOf(component.moduleName)]
+      if (!component.moduleName) {
+        throw new Error(`Tamagui internal err`)
+      }
+    }
+
+    // always load core so we can optimize if directly importing
+    const coreComponents = loadComponents({
+      ...props,
+      components: ['@tamagui/core-node'],
+    })
+    if (coreComponents) {
+      coreComponents[0].moduleName = '@tamagui/core'
+      components = [...components, ...coreComponents]
+    }
+
+    if (process.env.NODE_ENV === 'development' && process.env.DEBUG?.startsWith('tamagui')) {
+      console.log('Loaded components', components)
+    }
+
+    cache[key] = {
+      components,
+      nameToPaths: {},
+      tamaguiConfig: config,
+    }
+
+    // init core-node
+    createTamagui(cache[key].tamaguiConfig)
+
+    resolver(cache[key])
+
+    return cache[key]
+  } finally {
+    unregisterRequire()
+  }
+}
+
+export function resolveWebOrNativeSpecificEntry(entry: string) {
+  const resolved = require.resolve(entry)
+  const ext = extname(resolved)
+  const fileName = basename(resolved).replace(ext, '')
+  const specificExt = process.env.TAMAGUI_TARGET === 'web' ? 'web' : 'native'
+  const specificFile = join(dirname(resolved), fileName + '.' + specificExt + ext)
+  if (existsSync(specificFile)) {
+    return specificFile
+  }
+  return entry
+}
+
+const esbuildOptions = {
+  loader: 'tsx',
+  target: 'es2018',
+  format: 'cjs',
+  jsx: 'transform',
+  platform: 'node',
+} as const
+
+// loads in-process using esbuild-register
+export function loadTamaguiSync(props: Props): TamaguiProjectInfo {
+  const key = JSON.stringify(props)
+  if (cache[key]) {
+    return cache[key]
+  }
+
+  const { unregister } = require('esbuild-register/dist/node').register(esbuildOptions)
+
+  try {
+    registerRequire()
+
+    // lets shim require and avoid importing react-native + react-native-web
+    // we just need to read the config around them
+    process.env.IS_STATIC = 'is_static'
+    const devValueOG = globalThis['__DEV__' as any]
+    globalThis['__DEV__' as any] = process.env.NODE_ENV === 'development'
+
+    try {
+      // config
+      let tamaguiConfig: TamaguiInternalConfig | null = null
+      if (props.config) {
+        const configPath = join(process.cwd(), props.config)
+        const exp = require(configPath)
+        tamaguiConfig = (exp['default'] || exp) as TamaguiInternalConfig
+        if (!tamaguiConfig || !tamaguiConfig.parsed) {
+          const confPath = require.resolve(configPath)
+          throw new Error(`Can't find valid config in ${confPath}:
+          
+  Be sure you "export default" the config.`)
+        }
+      }
+
+      // components
+      const components = loadComponents(props)
+      if (!components) {
+        throw new Error(`No components loaded`)
+      }
+      if (process.env.DEBUG === 'tamagui') {
+        console.log(`components`, components)
+      }
+
+      // undo shims
+      process.env.IS_STATIC = undefined
+      globalThis['__DEV__' as any] = devValueOG
+
+      // set up core-node
+      if (props.config && tamaguiConfig) {
+        createTamagui(tamaguiConfig as any)
+      }
+
+      cache[key] = {
+        components,
+        tamaguiConfig,
+        nameToPaths: getNameToPaths(),
+      }
+    } catch (err) {
+      if (err instanceof Error) {
+        console.warn(
+          `Error loading tamagui.config.ts (set DEBUG=tamagui to see full stack), running tamagui without custom config`
+        )
+        console.log(`\n\n    ${err.message}\n\n`)
+        if (SHOULD_DEBUG) {
+          console.log(err.stack)
+        }
+      } else {
+        console.error(`Error loading tamagui.config.ts`, err)
+      }
+      return {
+        components: [],
+        tamaguiConfig: getDefaultTamaguiConfig(),
+        nameToPaths: {},
+      }
+    }
+
+    return cache[key]
+  } finally {
+    unregister()
+    unregisterRequire()
+  }
+}
+
+function interopDefaultExport(mod: any) {
+  return mod?.default ?? mod
+}
+
+const cacheComponents: Record<string, LoadedComponents[]> = {}
+
+function transformAddExports(ast: t.File) {
+  const usedNames = new Set<string>()
+
+  // avoid clobbering
+  traverse(ast, {
+    ExportNamedDeclaration(nodePath) {
+      if (nodePath.node.specifiers) {
+        for (const spec of nodePath.node.specifiers) {
+          usedNames.add(t.isIdentifier(spec.exported) ? spec.exported.name : spec.exported.value)
+        }
+      }
+    },
+  })
+
+  traverse(ast, {
+    VariableDeclaration(nodePath) {
+      // top level only
+      if (!t.isProgram(nodePath.parent)) return
+      const decs = nodePath.node.declarations
+      if (decs.length > 1) return
+      const [dec] = decs
+      if (!t.isIdentifier(dec.id)) return
+      if (!dec.init) return
+      if (usedNames.has(dec.id.name)) return
+      usedNames.add(dec.id.name)
+      nodePath.replaceWith(
+        t.exportNamedDeclaration(t.variableDeclaration('let', [dec]), [
+          t.exportSpecifier(t.identifier(dec.id.name), t.identifier(dec.id.name)),
+        ])
+      )
+    },
+  })
+
+  return generate(ast as any, {
+    concise: false,
+    filename: 'test.tsx',
+    retainLines: false,
+    sourceMaps: false,
+  }).code
+}
+
+function loadComponents(props: Props): null | LoadedComponents[] {
+  const componentsModules = props.components
+  const key = componentsModules.join('')
+  if (cacheComponents[key]) {
+    return cacheComponents[key]
+  }
+  try {
+    const info: LoadedComponents[] = componentsModules.flatMap((name) => {
+      const extension = extname(name)
+      const isLocal = Boolean(extension)
+      // during props.config pass we are passing in pre-bundled stuff
+      const writeTmp = isLocal && !props.config
+      const fileContents = writeTmp ? readFileSync(name, 'utf-8') : ''
+      const loadModule = writeTmp
+        ? join(dirname(name), `.tamagui-dynamic-eval-${basename(name)}.tsx`)
+        : name
+      let writtenContents = fileContents
+      let didBabel = false
+
+      const esbuildit = (src: string, target?: 'modern') =>
+        esbuild.transformSync(src, {
+          ...esbuildOptions,
+          ...(target === 'modern' && {
+            target: 'es2022',
+            jsx: 'transform',
+            loader: 'tsx',
+            platform: 'neutral',
+            format: 'esm',
+          }),
+        }).code
+
+      function attemptLoad({ forceExports = false } = {}) {
+        // need to write to tsx to enable reading it properly (:/ esbuild-register)
+        if (writeTmp) {
+          writtenContents = forceExports
+            ? esbuildit(transformAddExports(babelParse(esbuildit(fileContents, 'modern'))))
+            : esbuildit(fileContents)
+
+          writeFileSync(loadModule, writtenContents)
+        }
+        return {
+          moduleName: name,
+          nameToInfo: getComponentStaticConfigByName(
+            name,
+            interopDefaultExport(require(loadModule))
+          ),
+        }
+      }
+
+      const dispose = () => {
+        writeTmp && removeSync(loadModule)
+      }
+
+      try {
+        const res = attemptLoad({
+          forceExports: true,
+        })
+        didBabel = true
+        return res
+      } catch (err) {
+        console.log('babel err', err, writtenContents)
+        // ok
+        writtenContents = fileContents
+        if (process.env.DEBUG?.startsWith('tamagui')) {
+          console.log(`Error parsing babel likely`, err)
+        }
+      } finally {
+        dispose()
+      }
+
+      try {
+        return attemptLoad({
+          forceExports: false,
+        })
+      } catch (err) {
+        if (!process.env.TAMAGUI_DISABLE_WARN_DYNAMIC_LOAD) {
+          console.log(`
+
+Tamagui attempted but failed to dynamically load components in:
+  ${name}
+
+This will leave some styled() tags unoptimized.
+Disable this file (or dynamic loading altogether):
+
+  disableExtractFoundComponents: ['${name}'] | true
+
+Quiet this warning with environment variable:
+      
+  TAMAGUI_DISABLE_WARN_DYNAMIC_LOAD=1
+
+`)
+          console.log(err)
+          console.log(
+            `At: ${loadModule}`,
+            `\ndidBabel: ${didBabel}`,
+            `\nIn:`,
+            writtenContents,
+            `\nwriteTmp: `,
+            writeTmp
+          )
+        }
+        return []
+      } finally {
+        dispose()
+      }
+    })
+    cacheComponents[key] = info
+    return info
+  } catch (err: any) {
+    console.log(`Tamagui error bundling components`, err.message, err.stack)
+    return null
+  }
+}
+
+function getComponentStaticConfigByName(name: string, exported: any) {
+  if (!exported || typeof exported !== 'object' || Array.isArray(exported)) {
+    throw new Error(`Invalid export from package ${name}: ${typeof exported}`)
+  }
+  const components: Record<string, { staticConfig: StaticConfigParsed }> = {}
+  try {
+    for (const key in exported) {
+      const found = getTamaguiComponent(key, exported[key])
+      if (found) {
+        // remove non-stringifyable
+        const { Component, ...sc } = found.staticConfig
+        components[key] = { staticConfig: sc }
+      }
+    }
+  } catch (err) {
+    console.error(`Tamagui failed getting components`)
+    if (err instanceof Error) {
+      console.error(err.message, err.stack)
+    } else {
+      console.error(err)
+    }
+  }
+  return components
+}
+
+function getTamaguiComponent(
+  name: string,
+  Component: any
+): undefined | { staticConfig: StaticConfigParsed } {
+  if (name[0].toUpperCase() !== name[0]) {
+    return
+  }
+  const staticConfig = Component?.staticConfig as StaticConfigParsed | undefined
+  if (staticConfig) {
+    return Component
+  }
+}

package/src/extractor/logLines.ts

@@ -0,0 +1,16 @@
+const prefix = '           '
+
+export const logLines = (str: string, singleLine = false) => {
+  if (singleLine) {
+    return prefix + str.split(' ').join(`\n${prefix}`)
+  }
+  let lines: string[] = ['']
+  const items = str.split(' ')
+  for (const item of items) {
+    if (item.length + lines[lines.length - 1].length > 85) {
+      lines.push('')
+    }
+    lines[lines.length - 1] += item + ' '
+  }
+  return lines.map((line, i) => prefix + (i == 0 ? '' : ' ') + line.trim()).join('\n')
+}

package/src/extractor/normalizeTernaries.ts

@@ -0,0 +1,63 @@
+import generate from '@babel/generator'
+import * as t from '@babel/types'
+import invariant from 'invariant'
+
+import { Ternary } from '../types.js'
+
+export function normalizeTernaries(ternaries: Ternary[]) {
+  invariant(
+    Array.isArray(ternaries),
+    'extractStaticTernaries expects param 1 to be an array of ternaries'
+  )
+
+  if (ternaries.length === 0) {
+    return []
+  }
+
+  const ternariesByKey: { [key: string]: Ternary } = {}
+
+  for (let idx = -1, len = ternaries.length; ++idx < len; ) {
+    const { test, consequent, alternate, remove, ...rest } = ternaries[idx]
+
+    let ternaryTest = test
+
+    // strip parens
+    if (t.isExpressionStatement(test)) {
+      ternaryTest = (test as any).expression
+    }
+
+    // convert `!thing` to `thing` with swapped consequent and alternate
+    let shouldSwap = false
+    if (t.isUnaryExpression(test) && test.operator === '!') {
+      ternaryTest = test.argument
+      shouldSwap = true
+    } else if (t.isBinaryExpression(test)) {
+      if (test.operator === '!==' || test.operator === '!=') {
+        ternaryTest = t.binaryExpression(test.operator, test.left, test.right)
+        shouldSwap = true
+      }
+    }
+
+    const key = generate(ternaryTest as any).code
+
+    if (!ternariesByKey[key]) {
+      ternariesByKey[key] = {
+        ...rest,
+        alternate: {},
+        consequent: {},
+        test: ternaryTest,
+        remove,
+      }
+    }
+    const altStyle = (shouldSwap ? consequent : alternate) ?? {}
+    const consStyle = (shouldSwap ? alternate : consequent) ?? {}
+    Object.assign(ternariesByKey[key].alternate!, altStyle)
+    Object.assign(ternariesByKey[key].consequent!, consStyle)
+  }
+
+  const ternaryExpression = Object.keys(ternariesByKey).map((key) => {
+    return ternariesByKey[key]
+  })
+
+  return ternaryExpression
+}

package/src/extractor/removeUnusedHooks.ts

@@ -0,0 +1,76 @@
+import { NodePath } from '@babel/traverse'
+import * as t from '@babel/types'
+
+const hooks = {
+  useMedia: true,
+  useTheme: true,
+}
+
+export function removeUnusedHooks(compFn: NodePath<any>, shouldPrintDebug: boolean | 'verbose') {
+  compFn.scope.crawl()
+  // check the top level statements
+  let bodyStatements = compFn?.get('body')
+  if (!bodyStatements) {
+    console.log('no body statemnts?', compFn)
+    return
+  }
+  if (!Array.isArray(bodyStatements)) {
+    if (bodyStatements.isFunctionExpression()) {
+      bodyStatements = bodyStatements.scope.path.get('body')
+    } else {
+      bodyStatements = bodyStatements.get('body')
+    }
+  }
+  if (!bodyStatements || !Array.isArray(bodyStatements)) {
+    return
+  }
+  const statements = bodyStatements as NodePath<any>[]
+  for (const statement of statements) {
+    if (!statement.isVariableDeclaration()) {
+      continue
+    }
+    const declarations = statement.get('declarations')
+    if (!Array.isArray(declarations)) {
+      continue
+    }
+    const isBindingReferenced = (name: string) => {
+      return !!statement.scope.getBinding(name)?.referenced
+    }
+    for (const declarator of declarations) {
+      const id = declarator.get('id')
+      const init = declarator.node.init
+      if (Array.isArray(id) || Array.isArray(init)) {
+        continue
+      }
+      const shouldRemove = (() => {
+        const isHook =
+          t.isCallExpression(init) && t.isIdentifier(init.callee) && hooks[init.callee.name]
+        if (!isHook) {
+          return false
+        }
+        if (t.isIdentifier(id.node)) {
+          // remove "const media = useMedia()"
+          const name = id.node.name
+          return !isBindingReferenced(name)
+        } else if (t.isObjectPattern(id.node)) {
+          // remove "const { sm } = useMedia()"
+          const propPaths = id.get('properties') as NodePath<any>[]
+          return propPaths.every((prop) => {
+            if (!prop.isObjectProperty()) return false
+            const value = prop.get('value')
+            if (Array.isArray(value) || !value.isIdentifier()) return false
+            const name = value.node.name
+            return !isBindingReferenced(name)
+          })
+        }
+        return false
+      })()
+      if (shouldRemove) {
+        declarator.remove()
+        if (shouldPrintDebug) {
+          console.log(`  [🪝] removed ${id.node['name'] ?? ''}`)
+        }
+      }
+    }
+  }
+}

package/src/extractor/timer.ts

@@ -0,0 +1,29 @@
+export const timer = () => {
+  const start = Date.now()
+  let last = start
+  return {
+    mark: (name: string, print = false) => {
+      if (print) {
+        const took = Date.now() - last
+        last = Date.now()
+        // eslint-disable-next-line no-console
+        console.log(`Time ${name}: ${took}ms`)
+        if (took > 10) {
+          // eslint-disable-next-line no-console
+          console.log('  long timer')
+        }
+      }
+    },
+    done: (print = false) => {
+      if (print) {
+        const total = Date.now() - start
+        // eslint-disable-next-line no-console
+        console.log(`Total time: ${total}ms`)
+        if (total > 50) {
+          // eslint-disable-next-line no-console
+          console.log('  long timer')
+        }
+      }
+    },
+  }
+}