feat: commit custom lengthUnits

Author: sonofmagic

packages/tailwindcss-patch/src/core/patches/index.ts

@@ -1 +1,2 @@
 export * from './exportContext'
+export * from './supportCustomUnits'

packages/tailwindcss-patch/src/core/patches/supportCustomUnits/index.ts

@@ -0,0 +1,79 @@
+import fs from 'node:fs'
+import path from 'node:path'
+import * as t from '@babel/types'
+import type { ArrayExpression, StringLiteral } from '@babel/types'
+import type { ILengthUnitsPatchDangerousOptions, ILengthUnitsPatchOptions } from './types'
+import { generate, parse, traverse } from '@/babel'
+
+function findAstNode(content: string, options: ILengthUnitsPatchOptions) {
+  const DOPTS = options.dangerousOptions as Required<ILengthUnitsPatchDangerousOptions>
+  const ast = parse(content)
+
+  let arrayRef: ArrayExpression | undefined
+  let changed = false
+  traverse(ast, {
+    Identifier(path) {
+      if (
+        path.node.name === DOPTS.variableName
+        && t.isVariableDeclarator(path.parent)
+        && t.isArrayExpression(path.parent.init)
+      ) {
+        arrayRef = path.parent.init
+        const set = new Set(path.parent.init.elements.map(x => (<StringLiteral>x).value))
+        for (let i = 0; i < options.units.length; i++) {
+          const unit = options.units[i]
+          if (!set.has(unit)) {
+            path.parent.init.elements = path.parent.init.elements.map((x) => {
+              if (t.isStringLiteral(x)) {
+                return {
+                  type: x?.type,
+                  value: x?.value,
+                }
+              }
+              return x
+            })
+            path.parent.init.elements.push({
+              type: 'StringLiteral',
+              value: unit,
+            })
+            changed = true
+          }
+        }
+      }
+    },
+  })
+  return {
+    arrayRef,
+    changed,
+  }
+}
+
+export function monkeyPatchForSupportingCustomUnit(rootDir: string, options: ILengthUnitsPatchOptions) {
+  const { dangerousOptions } = options
+  const DOPTS = dangerousOptions as Required<ILengthUnitsPatchDangerousOptions>
+  const dataTypesFilePath = path.resolve(rootDir, DOPTS.lengthUnitsFilePath)
+  const dataTypesFileContent = fs.readFileSync(dataTypesFilePath, {
+    encoding: 'utf8',
+  })
+  const { arrayRef, changed } = findAstNode(dataTypesFileContent, options)
+  if (arrayRef && changed) {
+    const { code } = generate(arrayRef, {
+      jsescOption: {
+        quotes: 'single',
+      },
+    })
+
+    if (arrayRef.start && arrayRef.end) {
+      const prev = dataTypesFileContent.slice(0, arrayRef.start)
+      const next = dataTypesFileContent.slice(arrayRef.end as number)
+      const newCode = prev + code + next
+      if (DOPTS.overwrite) {
+        fs.writeFileSync(DOPTS.destPath ?? dataTypesFilePath, newCode, {
+          encoding: 'utf8',
+        })
+        console.log('patch tailwindcss for custom length unit successfully!')
+      }
+    }
+    return code
+  }
+}

packages/tailwindcss-patch/src/core/patches/supportCustomUnits/types.ts

@@ -0,0 +1,15 @@
+export interface ILengthUnitsPatchDangerousOptions {
+  packageName?: string
+  gteVersion?: string
+  lengthUnitsFilePath?: string
+  variableName?: string
+  overwrite?: boolean
+  destPath?: string
+}
+
+export interface ILengthUnitsPatchOptions {
+  units: string[]
+  paths?: string[]
+  dangerousOptions?: ILengthUnitsPatchDangerousOptions
+  basedir?: string
+}

packages/tailwindcss-patch/src/core/runtime-patcher.ts

@@ -1,7 +1,7 @@
 import path from 'node:path'
 import { gte } from 'semver'
 import type { PackageJson } from 'pkg-types'
-import { monkeyPatchForExposingContextV2, monkeyPatchForExposingContextV3 } from './patches'
+import { monkeyPatchForExposingContextV2, monkeyPatchForExposingContextV3, monkeyPatchForSupportingCustomUnit } from './patches'
 
 import type { InternalPatchOptions } from '@/types'
 
@@ -12,6 +12,14 @@ export function internalPatch(pkgJsonPath: string | undefined, options: Internal
     const twDir = path.dirname(pkgJsonPath)
     if (gte(pkgJson.version!, '3.0.0')) {
       options.version = pkgJson.version
+      monkeyPatchForSupportingCustomUnit(twDir, {
+        units: ['rpx'],
+        dangerousOptions: {
+          lengthUnitsFilePath: 'lib/util/dataTypes.js',
+          variableName: 'lengthUnits',
+          overwrite: true,
+        },
+      })
       const result = monkeyPatchForExposingContextV3(twDir, options)
       return result
     }