register.js 2.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788
  1. import * as pirates from "pirates";
  2. import { transform} from "./index";
  3. export function addHook(
  4. extension,
  5. sucraseOptions,
  6. hookOptions,
  7. ) {
  8. let mergedSucraseOptions = sucraseOptions;
  9. const sucraseOptionsEnvJSON = process.env.SUCRASE_OPTIONS;
  10. if (sucraseOptionsEnvJSON) {
  11. mergedSucraseOptions = {...mergedSucraseOptions, ...JSON.parse(sucraseOptionsEnvJSON)};
  12. }
  13. return pirates.addHook(
  14. (code, filePath) => {
  15. const {code: transformedCode, sourceMap} = transform(code, {
  16. ...mergedSucraseOptions,
  17. sourceMapOptions: {compiledFilename: filePath},
  18. filePath,
  19. });
  20. const mapBase64 = Buffer.from(JSON.stringify(sourceMap)).toString("base64");
  21. const suffix = `//# sourceMappingURL=data:application/json;charset=utf-8;base64,${mapBase64}`;
  22. return `${transformedCode}\n${suffix}`;
  23. },
  24. {...hookOptions, exts: [extension]},
  25. );
  26. }
  27. export function registerJS(hookOptions) {
  28. return addHook(".js", {transforms: ["imports", "flow", "jsx"]}, hookOptions);
  29. }
  30. export function registerJSX(hookOptions) {
  31. return addHook(".jsx", {transforms: ["imports", "flow", "jsx"]}, hookOptions);
  32. }
  33. export function registerTS(hookOptions) {
  34. return addHook(".ts", {transforms: ["imports", "typescript"]}, hookOptions);
  35. }
  36. export function registerTSX(hookOptions) {
  37. return addHook(".tsx", {transforms: ["imports", "typescript", "jsx"]}, hookOptions);
  38. }
  39. export function registerTSLegacyModuleInterop(hookOptions) {
  40. return addHook(
  41. ".ts",
  42. {
  43. transforms: ["imports", "typescript"],
  44. enableLegacyTypeScriptModuleInterop: true,
  45. },
  46. hookOptions,
  47. );
  48. }
  49. export function registerTSXLegacyModuleInterop(hookOptions) {
  50. return addHook(
  51. ".tsx",
  52. {
  53. transforms: ["imports", "typescript", "jsx"],
  54. enableLegacyTypeScriptModuleInterop: true,
  55. },
  56. hookOptions,
  57. );
  58. }
  59. export function registerAll(hookOptions) {
  60. const reverts = [
  61. registerJS(hookOptions),
  62. registerJSX(hookOptions),
  63. registerTS(hookOptions),
  64. registerTSX(hookOptions),
  65. ];
  66. return () => {
  67. for (const fn of reverts) {
  68. fn();
  69. }
  70. };
  71. }