/**
 * @license
 * Copyright 2025 BrowserOS
 * SPDX-License-Identifier: AGPL-3.0-or-later
 */

/**
 * Bun plugin to handle esbuild-style `?binary` WASM imports.
 *
 * Transforms imports like:
 *   import('web-tree-sitter/tree-sitter.wasm?binary')
 *
 * Into inline Uint8Array exports that work in compiled Bun binaries.
 */

import { createRequire } from 'node:module'
import { isAbsolute, resolve } from 'node:path'
import type { BunPlugin } from 'bun'

export function wasmBinaryPlugin(): BunPlugin {
  const require = createRequire(import.meta.url)

  return {
    name: 'wasm-binary',
    setup(build) {
      build.onResolve({ filter: /\.wasm\?binary$/ }, (args) => {
        const specifier = args.path.replace(/\?binary$/, '')
        const resolveDir = args.resolveDir || process.cwd()

        const isBareSpecifier =
          !isAbsolute(specifier) &&
          !specifier.startsWith('./') &&
          !specifier.startsWith('../')

        let resolvedPath: string
        if (isBareSpecifier) {
          resolvedPath = require.resolve(specifier, {
            paths: [resolveDir, process.cwd()],
          })
        } else {
          resolvedPath = isAbsolute(specifier)
            ? specifier
            : resolve(resolveDir, specifier)
        }

        return {
          path: resolvedPath,
          namespace: 'wasm-binary',
        }
      })

      build.onLoad({ filter: /.*/, namespace: 'wasm-binary' }, async (args) => {
        const bytes = await Bun.file(args.path).arrayBuffer()
        const uint8 = new Uint8Array(bytes)

        return {
          contents: `export default new Uint8Array([${uint8.join(',')}]);`,
          loader: 'js',
        }
      })
    },
  }
}