Skip to content

Bun 提供了一个通用插件 API,可用于扩展运行时和打包器。

插件拦截导入并执行自定义加载逻辑:读取文件、转译代码等。它们可用于添加对其他文件类型的支持,如 .scss.yaml。在 Bun 打包器的上下文中,插件可用于实现框架级功能,如 CSS 提取、宏和客户端 - 服务器代码共存。

生命周期钩子

插件可以注册回调在打包生命周期的各个点运行:

  • onStart():打包器开始打包时运行一次
  • onResolve():在模块解析之前运行
  • onLoad():在模块加载之前运行
  • onBeforeParse():在文件解析之前在解析器线程中运行零拷贝原生插件

参考

类型的粗略概述(请参考 Bun 的 bun.d.ts 获取完整类型定义):

ts
type PluginBuilder = {
  onStart(callback: () => void): void;
  onResolve: (
    args: { filter: RegExp; namespace?: string },
    callback: (args: { path: string; importer: string }) => {
      path: string;
      namespace?: string;
    } | void,
  ) => void;
  onLoad: (
    args: { filter: RegExp; namespace?: string },
    defer: () => Promise<void>,
    callback: (args: { path: string }) => {
      loader?: Loader;
      contents?: string;
      exports?: Record<string, any>;
    },
  ) => void;
  config: BuildConfig;
};

type Loader =
  | "js"
  | "jsx"
  | "ts"
  | "tsx"
  | "json"
  | "jsonc"
  | "toml"
  | "yaml"
  | "file"
  | "napi"
  | "wasm"
  | "text"
  | "css"
  | "html";

使用

插件定义为包含 name 属性和 setup 函数的简单 JavaScript 对象。

ts
import type { BunPlugin } from "bun";

const myPlugin: BunPlugin = {
  name: "自定义加载器",
  setup(build) {
    // 实现
  },
};

此插件可以在调用 Bun.build 时传递到 plugins 数组中。

ts
await Bun.build({
  entrypoints: ["./app.ts"],
  outdir: "./out",
  plugins: [myPlugin],
});

插件生命周期

命名空间

onLoadonResolve 接受可选的 namespace 字符串。什么是命名空间?

每个模块都有一个命名空间。命名空间用于在转译代码中前缀导入;例如,具有 filter: /\.yaml$/namespace: "yaml:" 的加载器将把导入从 ./myfile.yaml 转换为 yaml:./myfile.yaml

默认命名空间是 "file",不需要指定它,例如:import myModule from "./my-module.ts"import myModule from "file:./my-module.ts" 相同。

其他常见命名空间是:

  • "bun":用于 Bun 特定模块(例如 "bun:test""bun:sqlite"
  • "node":用于 Node.js 模块(例如 "node:fs""node:path"

onStart

ts
onStart(callback: () => void): Promise<void> | void;

注册回调在打包器开始新打包时运行。

ts
import { plugin } from "bun";

plugin({
  name: "onStart 示例",

  setup(build) {
    build.onStart(() => {
      console.log("打包开始!");
    });
  },
});

回调可以返回 Promise。打包过程初始化后,打包器等待所有 onStart() 回调完成后再继续。

例如:

ts
const result = await Bun.build({
  entrypoints: ["./app.ts"],
  outdir: "./dist",
  sourcemap: "external",
  plugins: [
    {
      name: "休眠 10 秒",
      setup(build) {
        build.onStart(async () => {
          await Bun.sleep(10_000);
        });
      },
    },
    {
      name: "将打包时间记录到文件",
      setup(build) {
        build.onStart(async () => {
          const now = Date.now();
          await Bun.$`echo ${now} > bundle-time.txt`;
        });
      },
    },
  ],
});

在上面的示例中,Bun 等待第一个 onStart()(休眠 10 秒)完成,以及第二个 onStart()(将打包时间写入文件)完成。

NOTE

`onStart()` 回调(如每个其他生命周期回调)没有修改 `build.config` 对象的能力。如果您想修改 `build.config`,必须直接在 `setup()` 函数中完成。

onResolve

ts
onResolve(
  args: { filter: RegExp; namespace?: string },
  callback: (args: { path: string; importer: string }) => {
    path: string;
    namespace?: string;
  } | void,
): void;

要打包您的项目,Bun 会遍历项目中所有模块的依赖树。对于每个导入的模块,Bun 实际上必须找到并读取该模块。"查找"部分被称为"解析"模块。

onResolve() 插件生命周期回调允许您配置如何解析模块。

onResolve() 的第一个参数是具有 filternamespace 属性的对象。filter 是在导入字符串上运行的正则表达式。实际上,这些允许您过滤自定义解析逻辑将应用于哪些模块。

onResolve() 的第二个参数是回调,为 Bun 找到的每个匹配第一个参数中定义的 filter 和 namespace 的模块导入运行。

回调接收匹配模块的路径作为输入。回调可以为模块返回新路径。Bun 将读取新路径的内容并将其解析为模块。

例如,将所有对 images/ 的导入重定向到 ./public/images/

ts
import { plugin } from "bun";

plugin({
  name: "onResolve 示例",
  setup(build) {
    build.onResolve({ filter: /.*/, namespace: "file" }, args => {
      if (args.path.startsWith("images/")) {
        return {
          path: args.path.replace("images/", "./public/images/"),
        };
      }
    });
  },
});

onLoad

ts
onLoad(
  args: { filter: RegExp; namespace?: string },
  defer: () => Promise<void>,
  callback: (args: { path: string, importer: string, namespace: string, kind: ImportKind  }) => {
    loader?: Loader;
    contents?: string;
    exports?: Record<string, any>;
  },
): void;

在 Bun 的打包器解析模块后,它需要读取模块的内容并解析它。

onLoad() 插件生命周期回调允许您在 Bun 读取和解析模块之前修改模块的内容。

onResolve() 一样,onLoad() 的第一个参数允许您过滤此 onLoad() 调用将应用于哪些模块。

onLoad() 的第二个参数是回调,在 Bun 将模块内容加载到内存之前为每个匹配模块运行。

此回调接收匹配模块的路径、模块的导入者(导入模块的模块)、模块的命名空间和模块的类型作为输入。

回调可以为模块返回新的 contents 字符串和新的 loader

例如:

ts
import { plugin } from "bun";

const envPlugin: BunPlugin = {
  name: "env 插件",
  setup(build) {
    build.onLoad({ filter: /env/, namespace: "file" }, args => {
      return {
        contents: `export default ${JSON.stringify(process.env)}`,
        loader: "js",
      };
    });
  },
};

Bun.build({
  entrypoints: ["./app.ts"],
  outdir: "./dist",
  plugins: [envPlugin],
});

// import env from "env"
// env.FOO === "bar"

此插件将所有形式为 import env from "env" 的导入转换为导出当前环境变量的 JavaScript 模块。

.defer()

传递给 onLoad 回调的参数之一是 defer 函数。此函数返回一个 Promise,当所有其他模块加载完成时解析。

这允许您延迟 onLoad 回调的执行,直到所有其他模块加载完成。

这对于返回依赖于其他模块的模块内容很有用。

示例:跟踪和报告未使用的导出

ts
import { plugin } from "bun";

plugin({
  name: "跟踪导入",
  setup(build) {
    const transpiler = new Bun.Transpiler();

    let trackedImports: Record<string, number> = {};

    // 通过此 onLoad 回调的每个模块
    // 将在 `trackedImports` 中记录其导入
    build.onLoad({ filter: /\.ts/ }, async ({ path }) => {
      const contents = await Bun.file(path).arrayBuffer();

      const imports = transpiler.scanImports(contents);

      for (const i of imports) {
        trackedImports[i.path] = (trackedImports[i.path] || 0) + 1;
      }

      return undefined;
    });

    build.onLoad({ filter: /stats\.json/ }, async ({ defer }) => {
      // 等待所有文件加载完成,确保
      // 每个文件都通过上面的 `onLoad()` 函数
      // 并跟踪它们的导入
      await defer();

      // 发出包含每个导入统计信息的 JSON
      return {
        contents: `export default ${JSON.stringify(trackedImports)}`,
        loader: "json",
      };
    });
  },
});

原生插件

Bun 打包器如此快速的原因之一是它用原生代码编写,并利用多线程并行加载和解析模块。

然而,用 JavaScript 编写的插件的一个限制是 JavaScript 本身是单线程的。

原生插件编写为 NAPI 模块,可以在多个线程上运行。这允许原生插件比 JavaScript 插件运行得更快。

此外,原生插件可以跳过不必要的工作,如传递字符串给 JavaScript 所需的 UTF-8 -> UTF-16 转换。

以下是原生插件可用的生命周期钩子:

  • onBeforeParse():在 Bun 打包器解析文件之前在任何线程上调用。

原生插件是 NAPI 模块,将生命周期钩子导出为 C ABI 函数。

要创建原生插件,您必须导出与要实现的原生生命周期钩子签名匹配的 C ABI 函数。

用 Rust 创建原生插件

原生插件是 NAPI 模块,将生命周期钩子导出为 C ABI 函数。

要创建原生插件,您必须导出与要实现的原生生命周期钩子签名匹配的 C ABI 函数。

bash
bun add -g @napi-rs/cli
napi new

然后安装此 crate:

bash
cargo add bun-native-plugin

现在,在 lib.rs 文件中,我们将使用 bun_native_plugin::bun proc 宏来定义实现原生插件的函数。

这是实现 onBeforeParse 钩子的示例:

rust
use bun_native_plugin::{define_bun_plugin, OnBeforeParse, bun, Result, anyhow, BunLoader};
use napi_derive::napi;

/// 定义插件及其名称
define_bun_plugin!("replace-foo-with-bar");

/// 这里我们将实现 `onBeforeParse`,用替换所有
/// `foo` 为 `bar` 的代码。
///
/// 我们使用 #[bun] 宏生成一些样板代码。
///
/// 函数的参数(`handle: &mut OnBeforeParse`)告诉
/// 宏此函数实现 `onBeforeParse` 钩子。
#[bun]
pub fn replace_foo_with_bar(handle: &mut OnBeforeParse) -> Result<()> {
  // 获取输入源代码。
  let input_source_code = handle.input_source_code()?;

  // 获取文件的 Loader
  let loader = handle.output_loader();

  let output_source_code = input_source_code.replace("foo", "bar");

  handle.set_output_source_code(output_source_code, BunLoader::BUN_LOADER_JSX);

  Ok(())
}

并在 Bun.build() 中使用它:

ts
import myNativeAddon from "./my-native-addon";

Bun.build({
  entrypoints: ["./app.tsx"],
  plugins: [
    {
      name: "my-plugin",

      setup(build) {
        build.onBeforeParse(
          {
            namespace: "file",
            filter: "**/*.tsx",
          },
          {
            napiModule: myNativeAddon,
            symbol: "replace_foo_with_bar",
            // external: myNativeAddon.getSharedState()
          },
        );
      },
    },
  ],
});

onBeforeParse

ts
onBeforeParse(
  args: { filter: RegExp; namespace?: string },
  callback: { napiModule: NapiModule; symbol: string; external?: unknown },
): void;

此生命周期回调在 Bun 打包器解析文件之前立即运行。

作为输入,它接收文件的内容,并可以选择返回新的源代码。

Bun学习网由www.bunjs.com.cn整理维护