add support for importing module from python:*

Author: codehz

examples/import_plugin.ts

@@ -0,0 +1,3 @@
+import { api_version, version } from "python:sys";
+console.log(version);
+console.log(api_version);

plugin.ts

@@ -5,19 +5,33 @@ import python from "./index";
 const { dir } = python.import("builtins");
 const { SourceFileLoader } = python.import("importlib.machinery");
 
+function convertExports(raw: any) {
+  return Object.fromEntries(
+    dir(raw)
+      .valueOf()
+      .map((key: string) => [key, raw[key]])
+  )
+}
+
 plugin({
   name: "python",
   target: "bun",
-  setup(build) {
-    // const { default: python } = await import("./index");
-    build.onLoad({ filter: /\.(py)$/ }, ({ path }) => {
+  setup({ onResolve, onLoad }) {
+    onResolve({ filter: /.+/, namespace: "python" }, ({ path }) => {
+      return { path, namespace: "python" };
+    });
+    onLoad({ filter: /.+/, namespace: "python" }, ({ path }) => {
+      const raw = python.import(path);
+      const exports = convertExports(raw);
+      return {
+        loader: "object",
+        exports,
+      };
+    })
+    onLoad({ filter: /\.(py)$/ }, ({ path }) => {
       const { name } = parse(path);
       const raw = SourceFileLoader(name, path).load_module();
-      const exports = Object.fromEntries(
-        dir(raw)
-          .valueOf()
-          .map((key: string) => [key, raw[key]])
-      );
+      const exports = convertExports(raw);
       return {
         loader: "object",
         exports,

python.d.ts

@@ -0,0 +1,3 @@
+declare module "python:*";
+
+declare module "*.py";

test/plugin.test.js

@@ -0,0 +1,11 @@
+import { expect, test } from "bun:test";
+import { max } from "python:builtins";
+import { hello } from "./test.py";
+
+test("load builtins module", () => {
+  expect(max(1, 2).valueOf()).toBe(2);
+});
+
+test("load py as module", () => {
+  expect(hello("world").valueOf()).toBe("hello world");
+});