| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281 |
- import { createFilter } from '@rollup/pluginutils';
- import * as changeCase from 'change-case';
- import { init, parse } from 'es-module-lexer';
- import MagicString from 'magic-string';
- import path from 'pathe';
- import consola from 'consola';
- import { normalizePath } from 'vite';
- import fs from 'fs-extra';
- import { createRequire } from 'module';
- function resolveNodeModules(libName, ...dir) {
- const esRequire = createRequire(import.meta.url);
- let modulePath = "";
- try {
- modulePath = normalizePath(esRequire.resolve(libName));
- } catch (error) {
- modulePath = normalizePath(require.resolve(libName));
- }
- const lastIndex = modulePath.lastIndexOf(libName);
- return normalizePath(path.resolve(modulePath.substring(0, lastIndex), ...dir));
- }
- function resolvePnp(module) {
- try {
- return normalizePath(require.resolve(module));
- } catch (error) {
- return "";
- }
- }
- const isPnp = !!process.versions.pnp;
- function isRegExp(value) {
- return Object.prototype.toString.call(value) === "[object RegExp]";
- }
- function fileExists(f) {
- try {
- fs.accessSync(f, fs.constants.W_OK);
- return true;
- } catch (error) {
- return false;
- }
- }
- function isFunction(value) {
- return value != null && Object.prototype.toString.call(value) === "[object Function]";
- }
- function AntdResolve() {
- return {
- libraryName: "antd",
- esModule: true,
- resolveStyle: (name) => {
- return `antd/es/${name}/style/index`;
- }
- };
- }
- function AndDesignVueResolve() {
- return {
- ensureStyleFile: true,
- libraryName: "ant-design-vue",
- esModule: true,
- resolveStyle: (name) => {
- return `ant-design-vue/es/${name}/style/index`;
- }
- };
- }
- function ElementPlusResolve() {
- return {
- libraryName: "element-plus",
- ensureStyleFile: true,
- esModule: true,
- resolveStyle: (name) => {
- return `element-plus/theme-chalk/${name}.css`;
- },
- base: "element-plus/theme-chalk/base.css"
- };
- }
- function VantResolve() {
- return {
- libraryName: "vant",
- esModule: true,
- resolveStyle: (name) => {
- return `vant/es/${name}/style`;
- }
- };
- }
- function NutuiResolve() {
- return {
- libraryName: "@nutui/nutui",
- libraryNameChangeCase: "pascalCase",
- resolveStyle: (name) => {
- return `@nutui/nutui/dist/packages/${name}/index.scss`;
- }
- };
- }
- function VxeTableResolve() {
- return {
- ensureStyleFile: true,
- libraryName: "vxe-table",
- esModule: true,
- resolveStyle: (name) => {
- return `vxe-table/es/${name}/style.css`;
- }
- };
- }
- const ensureFileExts = [".css", ".js", ".scss", ".less", ".styl"];
- const asRE = /\s+as\s+\w+,?/g;
- consola.wrapConsole();
- function createStyleImportPlugin(options) {
- const {
- include = ["**/*.vue", "**/*.ts", "**/*.js", "**/*.tsx", "**/*.jsx"],
- exclude = "node_modules/**",
- resolves = []
- } = options;
- let { libs = [] } = options;
- libs = [...libs, ...resolves];
- const filter = createFilter(include, exclude);
- let needSourcemap = false;
- let external;
- return {
- name: "vite:style-import",
- enforce: "post",
- configResolved(resolvedConfig) {
- needSourcemap = !!resolvedConfig.build.sourcemap;
- external = resolvedConfig?.build?.rollupOptions?.external ?? void 0;
- },
- async transform(code, id) {
- if (!code || !filter(id) || !needTransform(code, libs)) {
- return null;
- }
- await init;
- let imports = [];
- try {
- imports = parse(code)[0];
- } catch (e) {
- consola.error(e);
- }
- if (!imports.length) {
- return null;
- }
- let s;
- const str = () => s || (s = new MagicString(code));
- for (let index = 0; index < imports.length; index++) {
- const { n, se, ss } = imports[index];
- if (!n)
- continue;
- const lib = getLib(n, libs, external);
- if (!lib)
- continue;
- const importStr = code.slice(ss, se);
- let importVariables = transformImportVar(importStr);
- importVariables = filterImportVariables(importVariables, lib.importTest);
- const importCssStrList = await transformComponentCss(lib, importVariables);
- const compStrList = [];
- const { base = "" } = lib;
- let baseImporter = base ? `
- import '${base}'` : "";
- if (str().toString().includes(base)) {
- baseImporter = "";
- }
- const endIndex = se + 1;
- str().prependRight(endIndex, `${baseImporter}
- ${compStrList.join("")}${importCssStrList.join("")}`);
- }
- return {
- map: needSourcemap ? str().generateMap({ hires: true }) : null,
- code: str().toString()
- };
- }
- };
- }
- function filterImportVariables(importVars, reg) {
- if (!reg) {
- return importVars;
- }
- return importVars.filter((item) => reg.test(item));
- }
- async function transformComponentCss(lib, importVariables) {
- const {
- libraryName,
- resolveStyle,
- esModule,
- libraryNameChangeCase = "paramCase",
- ensureStyleFile = false
- } = lib;
- if (!isFunction(resolveStyle) || !libraryName) {
- return [];
- }
- const set = /* @__PURE__ */ new Set();
- for (let index = 0; index < importVariables.length; index++) {
- const name = getChangeCaseFileName(importVariables[index], libraryNameChangeCase);
- let importStr = resolveStyle(name);
- if (!importStr) {
- continue;
- }
- let isAdd = true;
- if (isPnp) {
- importStr = resolvePnp(importStr);
- isAdd = !!importStr;
- } else {
- if (esModule) {
- importStr = resolveNodeModules(libraryName, importStr);
- }
- if (ensureStyleFile) {
- isAdd = ensureFileExists(libraryName, importStr, esModule);
- }
- }
- isAdd && set.add(`import '${importStr}';
- `);
- }
- return Array.from(set);
- }
- function transformImportVar(importStr) {
- if (!importStr) {
- return [];
- }
- const exportStr = importStr.replace("import", "export").replace(asRE, ",");
- let importVariables = [];
- try {
- importVariables = parse(exportStr)[1];
- } catch (error) {
- consola.error(error);
- }
- return importVariables;
- }
- function ensureFileExists(libraryName, importStr, esModule = false) {
- const extName = path.extname(importStr);
- if (!extName) {
- return tryEnsureFile(libraryName, importStr, esModule);
- }
- if (esModule) {
- return fileExists(importStr);
- }
- return true;
- }
- function tryEnsureFile(libraryName, filePath, esModule = false) {
- const filePathList = ensureFileExts.map((item) => {
- const p = `${filePath}${item}`;
- return esModule ? p : resolveNodeModules(libraryName, p);
- });
- return filePathList.some((item) => fileExists(item));
- }
- function getLib(libraryName, libs, external) {
- let libList = libs;
- if (external) {
- const isString = typeof external === "string";
- const isRE = isRegExp(external);
- if (isString) {
- libList = libList.filter((item) => item.libraryName !== external);
- } else if (isRE) {
- libList = libList.filter((item) => !external.test(item.libraryName));
- } else if (Array.isArray(external)) {
- libList = libList.filter((item) => {
- return !external.some((val) => {
- if (typeof val === "string") {
- return val === item.libraryName;
- }
- return val.test(item.libraryName);
- });
- });
- }
- }
- return libList.find((item) => item.libraryName === libraryName);
- }
- function getChangeCaseFileName(importedName, libraryNameChangeCase) {
- try {
- return changeCase[libraryNameChangeCase](importedName);
- } catch (error) {
- return importedName;
- }
- }
- function needTransform(code, libs) {
- return !libs.every(({ libraryName }) => {
- return !new RegExp(`('${libraryName}')|("${libraryName}")`).test(code);
- });
- }
- export { AndDesignVueResolve, AntdResolve, ElementPlusResolve, NutuiResolve, VantResolve, VxeTableResolve, createStyleImportPlugin, getChangeCaseFileName, transformImportVar };
|