index.ts

import { PluginBuilder } from "bun"
import { UnimportOptions } from "unimport"

const getLoader = (path: string) =>
	path.endsWith("ts")
		? "ts"
		: path.endsWith("js")
		? "js"
		: path.endsWith("tsx")
		? "tsx"
		: "jsx"

export const autoImport = (
	options: Partial<
		UnimportOptions & {
			dts: string
		}
	>
) => {
	return {
		name: "auto-import",
		async setup(builder: PluginBuilder) {
			const { createUnimport } = await import("unimport")
			const { injectImports, generateTypeDeclarations } = createUnimport({
				...options,
				dts: undefined,
			} as UnimportOptions)

			const dtsContent = await generateTypeDeclarations()
			Bun.write(options.dts ?? "./auto-import.d.ts", dtsContent)

			builder.onLoad({ filter: /.*/i }, async (args) => {
				const fileContent = await Bun.file(args.path).text()
				const transformedFileContent = await injectImports(fileContent)

				return {
					contents: transformedFileContent.code,
					loader: getLoader(args.path),
				}
			})
		},
	}
}

export default autoImport