vite-inset-loader/dist/index.js

在Vite构建的Vue应用中，编译阶段在SFC模板指定位置插入自定义内容，适用于需要在每个页面引入组件的小程序场景。

// src/plugin.ts
import { parse } from "@vue/compiler-sfc";

// src/utils.ts
import fs from "fs";

// src/module/strip-json-comments/index.js
var singleComment = Symbol("singleComment");
var multiComment = Symbol("multiComment");
var stripWithoutWhitespace = () => "";
var stripWithWhitespace = (string, start, end) => string.slice(start, end).replace(/\S/g, " ");
var isEscaped = (jsonString, quotePosition) => {
  let index = quotePosition - 1;
  let backslashCount = 0;
  while (jsonString[index] === "\\") {
    index -= 1;
    backslashCount += 1;
  }
  return Boolean(backslashCount % 2);
};
function stripJsonComments(jsonString, { whitespace = true, trailingCommas = false } = {}) {
  if (typeof jsonString !== "string") {
    throw new TypeError(`Expected argument \`jsonString\` to be a \`string\`, got \`${typeof jsonString}\``);
  }
  const strip = whitespace ? stripWithWhitespace : stripWithoutWhitespace;
  let isInsideString = false;
  let isInsideComment = false;
  let offset = 0;
  let buffer = "";
  let result = "";
  let commaIndex = -1;
  for (let index = 0; index < jsonString.length; index++) {
    const currentCharacter = jsonString[index];
    const nextCharacter = jsonString[index + 1];
    if (!isInsideComment && currentCharacter === '"') {
      const escaped = isEscaped(jsonString, index);
      if (!escaped) {
        isInsideString = !isInsideString;
      }
    }
    if (isInsideString) {
      continue;
    }
    if (!isInsideComment && currentCharacter + nextCharacter === "//") {
      buffer += jsonString.slice(offset, index);
      offset = index;
      isInsideComment = singleComment;
      index++;
    } else if (isInsideComment === singleComment && currentCharacter + nextCharacter === "\r\n") {
      index++;
      isInsideComment = false;
      buffer += strip(jsonString, offset, index);
      offset = index;
      continue;
    } else if (isInsideComment === singleComment && currentCharacter === "\n") {
      isInsideComment = false;
      buffer += strip(jsonString, offset, index);
      offset = index;
    } else if (!isInsideComment && currentCharacter + nextCharacter === "/*") {
      buffer += jsonString.slice(offset, index);
      offset = index;
      isInsideComment = multiComment;
      index++;
      continue;
    } else if (isInsideComment === multiComment && currentCharacter + nextCharacter === "*/") {
      index++;
      isInsideComment = false;
      buffer += strip(jsonString, offset, index + 1);
      offset = index + 1;
      continue;
    } else if (trailingCommas && !isInsideComment) {
      if (commaIndex !== -1) {
        if (currentCharacter === "}" || currentCharacter === "]") {
          buffer += jsonString.slice(offset, index);
          result += strip(buffer, 0, 1) + buffer.slice(1);
          buffer = "";
          offset = index;
          commaIndex = -1;
        } else if (currentCharacter !== " " && currentCharacter !== "	" && currentCharacter !== "\r" && currentCharacter !== "\n") {
          buffer += jsonString.slice(offset, index);
          offset = index;
          commaIndex = -1;
        }
      } else if (currentCharacter === ",") {
        result += buffer + jsonString.slice(offset, index);
        buffer = "";
        offset = index;
        commaIndex = index;
      }
    }
  }
  return result + buffer + (isInsideComment ? strip(jsonString.slice(offset)) : jsonString.slice(offset));
}

// src/utils.ts
import path from "path";
var pagesJson = {};
var insetLoader = {};
var rootPath = process.env.UNI_INPUT_DIR || process.env.INIT_CWD + "\\src";
var generateLabelCode = (labelArr) => labelArr.map((e) => {
  var _a;
  return ((_a = insetLoader == null ? void 0 : insetLoader.config) == null ? void 0 : _a[e]) || "";
}).join("");
var initPages = (that) => {
  let pagesPath = ((that == null ? void 0 : that.query) || {}).pagesPath;
  if (!pagesPath) {
    pagesPath = path.resolve(rootPath, "pages.json");
  } else {
    rootPath = path.resolve(path.dirname(pagesPath));
  }
  pagesJson = JSON.parse(stripJsonComments(fs.readFileSync(pagesPath, "utf8")));
  return initInsetLoader();
};
var getPagesMap = () => {
  const pages = pagesJson.pages || [];
  const subpackages = pagesJson.subpackages || pagesJson.subPackages || [];
  return pages.reduce(
    (obj, item) => {
      const curPage = getLabelConfig(item);
      curPage.label && (obj["/" + item.path] = curPage);
      return obj;
    },
    subpackages.reduce((obj, item) => {
      const root = item.root;
      item.pages.forEach((item2) => {
        const curPage = getLabelConfig(item2);
        curPage.label && (obj["/" + root + "/" + item2.path] = curPage);
      });
      return obj;
    }, {})
  );
};
var getLabelConfig = (json) => {
  return {
    label: json.style && json.style.label || insetLoader.label,
    package: json.style && json.style.package || insetLoader.package || null
  };
};
var initInsetLoader = () => {
  insetLoader = pagesJson.insetLoader || {};
  insetLoader.label = insetLoader.label || [];
  const effective = typeof insetLoader.config === "object" && Object.keys(insetLoader.config).length > 0;
  return effective;
};
var generateHtmlCode = (template, labelCode, packageEle) => {
  const hasClosingTag = ["</pageMeta>", "</PageMeta>", "</page-meta>"].some((label2) => template.includes(label2));
  const regex = hasClosingTag ? /<(page-meta|PageMeta|pageMeta)\b[^>]*>([\s\S]*?)<\/\1>/gi : /<(page-meta|PageMeta|pageMeta)\b[^>]*\/>/gi;
  const renderHtml = (content) => {
    const regClean = /<!--(?!.*?(#ifdef|#ifndef|#endif)).*?-->|^\s+|\s+$/g;
    return `${labelCode}
${content.replace(regClean, "").trim()}
`;
  };
  const html = renderHtml(containsPageMetaTag(template) ? template.replace(regex, "") : template);
  if (!template)
    return "";
  if (!packageEle)
    return html;
  const { label = "div", options } = packageEle;
  const { class: className = "", id = "", style = {}, ...otherOptions } = options;
  const styleAttr = Object.keys(style).length > 0 ? `style="${Object.entries(style).map(([key, value]) => `${key}:${value}`).join(";")}"` : "";
  const otherAttr = Object.entries(otherOptions).map(([key, value]) => `${key}="${value}"`).join(" ");
  return `<${label} class="${className}" id="${id}" ${styleAttr} ${otherAttr}>${html}</${label}>`;
};
var generateStyleCode = (styles) => styles.reduce((str, item, _i) => {
  return str += `<style ${item.lang ? "lang='" + item.lang + "'" : ""} ${item.scoped ? "scoped='" + item.scoped + "'" : ""}>
		${item.content}
	</style>`;
}, "");
var generateScriptCode = (script) => {
  return `<script ${(script == null ? void 0 : script.lang) ? `lang='${script == null ? void 0 : script.lang}'` : ""} ${script.setup ? "setup" : null}>
  ${script.content}
</script>`;
};
var getRoute = (resourcePath) => {
  const pwd = rootPath.replace(/\\/g, "/");
  const relativePath = resourcePath.replace(pwd, "").replace(/\\/g, "/");
  if (relativePath.endsWith(".vue")) {
    return relativePath.slice(0, -4);
  }
  return relativePath;
};
var filterDirectoriesByInclude = (rootDir2, options) => {
  const { include } = options;
  if (Array.isArray(include)) {
    const arrUrl = include == null ? void 0 : include.map((url) => path.resolve(rootDir2, url).replace(/\\/g, "/"));
    return arrUrl;
  } else {
    return [path.resolve(rootDir2, include || "src").replace(/\\/g, "/")];
  }
};
var getTemplatePageMeta = (template) => {
  const hasClosingTag = ["</pageMeta>", "</PageMeta>", "</page-meta>"].some((label) => template.includes(label));
  const regex = hasClosingTag ? /<(page-meta|PageMeta|pageMeta)\b[^>]*>([\s\S]*?)<\/\1>/gi : /<(page-meta|PageMeta|pageMeta)\b[^>]*\/>/gi;
  const match = regex.exec(template);
  return match ? match[0] : "";
};
var containsPageMetaTag = (htmlString) => {
  const pageMateTagPattern = /<page-meta\b[^>]*>/i;
  return pageMateTagPattern.test(htmlString);
};

// src/plugin.ts
var pagesMap = {};
var initialized = false;
var shouldHandle = false;
var rootDir;
var initializePages = (that) => {
  shouldHandle = initPages(that);
  if (shouldHandle) {
    pagesMap = getPagesMap();
  }
};
var viteInsetLoader = (options) => ({
  name: "vite-inset-loader",
  // 插件名称
  configResolved(config) {
    rootDir = config.root;
  },
  transform: (content, id) => {
    var _a, _b;
    const allDirectories = filterDirectoriesByInclude(rootDir, options || { include: "src" });
    if (!allDirectories.some((path2) => id.includes(path2)))
      return;
    if (!initialized) {
      initialized = true;
      initializePages(void 0);
    }
    const route = getRoute(id);
    if (route == null)
      return content;
    const curPage = pagesMap[route];
    if (curPage == void 0)
      return content;
    const regex = /<script\s+(?:module="(?:utils|test)"\s+lang="(?:wxs|sjs|filter\.js)")(?:\s+src=".*?")?\s*(?:\/>|>(?:[\s\S]*?)<\/script>)/g;
    const matches = content.match(regex);
    const skipScript = (matches == null ? void 0 : matches.length) == 0 ? null : matches == null ? void 0 : matches[0];
    const { descriptor } = parse(content);
    const labelCode = generateLabelCode(curPage.label);
    const template = generateHtmlCode(((_a = descriptor.template) == null ? void 0 : _a.content) || "", labelCode, curPage.package);
    const pageMete = getTemplatePageMeta(((_b = descriptor.template) == null ? void 0 : _b.content) || "");
    const style = generateStyleCode((descriptor == null ? void 0 : descriptor.styles) || []);
    const scriptSetup = (descriptor == null ? void 0 : descriptor.scriptSetup) == null ? null : generateScriptCode(descriptor == null ? void 0 : descriptor.scriptSetup);
    const script = (descriptor == null ? void 0 : descriptor.script) == null ? null : generateScriptCode(descriptor == null ? void 0 : descriptor.script);
    console.log(`
<template>
${pageMete}
${template}
</template>
${skipScript || ""}
${scriptSetup || ""}
${script || ""}
${style || ""}
    `);
    return `
<template>
${pageMete}
${template}
</template>
${skipScript || ""}
${scriptSetup || ""}
${script || ""}
${style || ""}
    `;
  }
});

// src/index.ts
var src_default = viteInsetLoader;
export {
  src_default as default,
  filterDirectoriesByInclude,
  generateHtmlCode,
  generateLabelCode,
  generateScriptCode,
  generateStyleCode,
  getPagesMap,
  getRoute,
  getTemplatePageMeta,
  initPages
};