transform.js 5.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", { value: true });
  3. exports.createBuildJsTransformPlugin = exports.createServerTransformPlugin = void 0;
  4. const utils_1 = require("./utils");
  5. const serverPluginSourceMap_1 = require("./server/serverPluginSourceMap");
  6. function createServerTransformPlugin(transforms, customBlockTransforms, resolver) {
  7. return ({ app }) => {
  8. if (!transforms.length && !Object.keys(customBlockTransforms).length) {
  9. return;
  10. }
  11. app.use(async (ctx, next) => {
  12. await next();
  13. if (!ctx.body ||
  14. (ctx.type === 'text/html' && !utils_1.isImportRequest(ctx)) ||
  15. resolver.isPublicRequest(ctx.path)) {
  16. return;
  17. }
  18. let { url, path, query, __notModified } = ctx;
  19. const id = resolver.requestToFile(url);
  20. path = resolver.requestToFile(path);
  21. const isImport = utils_1.isImportRequest(ctx);
  22. const isBuild = false;
  23. let code = '';
  24. for (const t of transforms) {
  25. const transformContext = {
  26. id,
  27. path,
  28. query,
  29. isImport,
  30. isBuild
  31. };
  32. if (__notModified) {
  33. transformContext.notModified = true;
  34. }
  35. if (t.test(transformContext)) {
  36. code = code || (await utils_1.readBody(ctx.body));
  37. const result = await t.transform({
  38. ...transformContext,
  39. code
  40. });
  41. if (typeof result === 'string') {
  42. code = result;
  43. }
  44. else {
  45. code = result.code;
  46. if (result.map) {
  47. ctx.map = serverPluginSourceMap_1.mergeSourceMap(ctx.map, result.map);
  48. }
  49. }
  50. ctx.type = 'js';
  51. ctx.body = code;
  52. }
  53. }
  54. // custom blocks
  55. if (path.endsWith('vue') && query.type === 'custom') {
  56. const t = customBlockTransforms[query.blockType];
  57. if (t) {
  58. ctx.type = 'js';
  59. code = code || (await utils_1.readBody(ctx.body));
  60. ctx.body = await t({
  61. code,
  62. id,
  63. path,
  64. query,
  65. isImport,
  66. isBuild
  67. });
  68. }
  69. }
  70. });
  71. };
  72. }
  73. exports.createServerTransformPlugin = createServerTransformPlugin;
  74. function createBuildJsTransformPlugin(transforms, customBlockTransforms) {
  75. return {
  76. name: 'vite:transforms',
  77. async transform(code, id) {
  78. const { path, query } = utils_1.parseWithQuery(id);
  79. let transformed = code;
  80. let map = null;
  81. const runTransform = async (t, ctx) => {
  82. const result = await t(ctx);
  83. if (typeof result === 'string') {
  84. transformed = result;
  85. }
  86. else {
  87. transformed = result.code;
  88. if (result.map) {
  89. map = serverPluginSourceMap_1.mergeSourceMap(map, result.map);
  90. }
  91. }
  92. };
  93. for (const t of transforms) {
  94. const transformContext = {
  95. code: transformed,
  96. id,
  97. path,
  98. query,
  99. isImport: true,
  100. isBuild: true
  101. };
  102. if (t.test(transformContext)) {
  103. await runTransform(t.transform, transformContext);
  104. }
  105. }
  106. // custom blocks
  107. if (query.vue != null && typeof query.type === 'string') {
  108. const t = customBlockTransforms[query.type];
  109. if (t) {
  110. // normalize lang since rollup-plugin-vue appends it as .xxx
  111. const normalizedQuery = {};
  112. for (const key in query) {
  113. if (key.startsWith(`lang.`)) {
  114. normalizedQuery.lang = key.slice(5);
  115. }
  116. else {
  117. normalizedQuery[key] = query[key];
  118. }
  119. }
  120. await runTransform(t, {
  121. code: transformed,
  122. id,
  123. path,
  124. query: normalizedQuery,
  125. isImport: true,
  126. isBuild: true
  127. });
  128. }
  129. }
  130. return {
  131. code: transformed,
  132. map
  133. };
  134. }
  135. };
  136. }
  137. exports.createBuildJsTransformPlugin = createBuildJsTransformPlugin;
  138. //# sourceMappingURL=transform.js.map