feat: add typeAssert util back

Author: sxzz

package.json

@@ -135,6 +135,7 @@
     "rolldown-plugin-require-cjs": "catalog:dev",
     "typescript": "catalog:dev",
     "unocss": "catalog:docs",
+    "unplugin-ast": "catalog:dev",
     "unplugin-lightningcss": "catalog:peer",
     "unplugin-unused": "catalog:peer",
     "vite": "catalog:docs",

pnpm-lock.yaml

@@ -31,6 +31,7 @@ catalogs:
     rolldown-plugin-dts-snapshot: ^0.3.2
     rolldown-plugin-require-cjs: ^0.3.3
     typescript: ~5.9.3
+    unplugin-ast: ^0.15.4
     vitest: ^4.0.16
   docs:
     '@shikijs/vitepress-twoslash': ^3.20.0

pnpm-workspace.yaml

@@ -3,7 +3,7 @@ import path from 'node:path'
 import { RE_DTS } from 'rolldown-plugin-dts/filename'
 import { detectIndentation } from '../../utils/format.ts'
 import { stripExtname } from '../../utils/fs.ts'
-import { matchPattern, slash } from '../../utils/general.ts'
+import { matchPattern, slash, typeAssert } from '../../utils/general.ts'
 import type { NormalizedFormat, ResolvedConfig } from '../../config/types.ts'
 import type {
   ChunksByFormat,
@@ -59,9 +59,10 @@ export async function writeExports(
   options: ResolvedConfig,
   chunks: ChunksByFormat,
 ): Promise<void> {
-  const pkg = options.pkg!
-  const exports = options.exports as ExportsOptions
+  typeAssert(options.pkg)
+  typeAssert(options.exports)
 
+  const { pkg, exports } = options
   const { publishExports, ...generated } = await generateExports(
     pkg,
     chunks,

src/features/pkg/exports.ts

@@ -100,5 +100,10 @@ export function promiseWithResolvers<T>(): {
   const promise = new Promise<T>((res) => {
     resolve = res
   })
-  return { promise, resolve: resolve! }
+  typeAssert(resolve!)
+  return { promise, resolve }
 }
+
+export function typeAssert<T>(
+  value: T,
+): asserts value is Exclude<T, false | null | undefined> {}

src/utils/general.ts

@@ -1,5 +1,8 @@
 import { DtsSnapshot } from 'rolldown-plugin-dts-snapshot'
 import { RequireCJS } from 'rolldown-plugin-require-cjs'
+import { isCallOf } from 'unplugin-ast/ast-kit'
+import AST from 'unplugin-ast/rolldown'
+import { RemoveNode } from 'unplugin-ast/transformers'
 import { defineConfig } from './src/config.ts'
 
 export default defineConfig([
@@ -32,7 +35,14 @@ export default defineConfig([
         return exports
       },
     },
-    plugins: [RequireCJS(), DtsSnapshot()],
+    plugins: [
+      RequireCJS(),
+      DtsSnapshot(),
+      AST({
+        exclude: ['**/*.d.ts'],
+        transformer: [RemoveNode((node) => isCallOf(node, 'typeAssert'))],
+      }),
+    ],
     onSuccess() {
       console.info('🙏 Build succeeded!')
     },