59 lines
1.6 KiB
TypeScript
59 lines
1.6 KiB
TypeScript
import gensync from "gensync";
|
|
|
|
import loadConfig from "./config";
|
|
import type { InputOptions, ResolvedConfig } from "./config";
|
|
import { run } from "./transformation";
|
|
import type * as t from "@babel/types";
|
|
|
|
import type { FileResult, FileResultCallback } from "./transformation";
|
|
type AstRoot = t.File | t.Program;
|
|
|
|
type TransformFromAst = {
|
|
(ast: AstRoot, code: string, callback: FileResultCallback): void;
|
|
(
|
|
ast: AstRoot,
|
|
code: string,
|
|
opts: InputOptions | undefined | null,
|
|
callback: FileResultCallback,
|
|
): void;
|
|
(ast: AstRoot, code: string, opts?: InputOptions | null): FileResult | null;
|
|
};
|
|
|
|
const transformFromAstRunner = gensync<
|
|
(
|
|
ast: AstRoot,
|
|
code: string,
|
|
opts: InputOptions | undefined | null,
|
|
) => FileResult | null
|
|
>(function* (ast, code, opts) {
|
|
const config: ResolvedConfig | null = yield* loadConfig(opts);
|
|
if (config === null) return null;
|
|
|
|
if (!ast) throw new Error("No AST given");
|
|
|
|
return yield* run(config, code, ast);
|
|
});
|
|
|
|
export const transformFromAst: TransformFromAst = function transformFromAst(
|
|
ast,
|
|
code,
|
|
opts,
|
|
callback?,
|
|
) {
|
|
if (typeof opts === "function") {
|
|
callback = opts;
|
|
opts = undefined;
|
|
}
|
|
|
|
// For backward-compat with Babel 6, we allow sync transformation when
|
|
// no callback is given. Will be dropped in some future Babel major version.
|
|
if (callback === undefined) {
|
|
return transformFromAstRunner.sync(ast, code, opts);
|
|
}
|
|
|
|
transformFromAstRunner.errback(ast, code, opts, callback);
|
|
};
|
|
|
|
export const transformFromAstSync = transformFromAstRunner.sync;
|
|
export const transformFromAstAsync = transformFromAstRunner.async;
|