| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361 |
- import {
- __spreadValues,
- presets
- } from "./chunk-EZINZJYF.js";
- // src/core/unplugin.ts
- import { minimatch } from "minimatch";
- import { slash as slash2 } from "@antfu/utils";
- import { createUnplugin } from "unplugin";
- // src/core/ctx.ts
- import { dirname, isAbsolute, join, relative, resolve } from "path";
- import { existsSync, promises as fs } from "fs";
- import process from "process";
- import { slash, throttle, toArray as toArray2 } from "@antfu/utils";
- import { createFilter } from "@rollup/pluginutils";
- import { isPackageExists } from "local-pkg";
- import { createUnimport, resolvePreset, scanExports } from "unimport";
- import fg from "fast-glob";
- import { vueTemplateAddon } from "unimport/addons";
- import MagicString from "magic-string";
- // src/core/eslintrc.ts
- function generateESLintConfigs(imports, eslintrc, globals = {}) {
- const eslintConfigs = { globals };
- imports.map((i) => {
- var _a;
- return (_a = i.as) != null ? _a : i.name;
- }).filter(Boolean).sort().forEach((name) => {
- eslintConfigs.globals[name] = eslintrc.globalsPropValue;
- });
- const jsonBody = JSON.stringify(eslintConfigs, null, 2);
- return jsonBody;
- }
- // src/core/resolvers.ts
- import { toArray } from "@antfu/utils";
- function normalizeImport(info, name) {
- if (typeof info === "string") {
- return {
- name: "default",
- as: name,
- from: info
- };
- }
- if ("path" in info) {
- return {
- from: info.path,
- as: info.name,
- name: info.importName,
- sideEffects: info.sideEffects
- };
- }
- return __spreadValues({
- name,
- as: name
- }, info);
- }
- async function firstMatchedResolver(resolvers, fullname) {
- let name = fullname;
- for (const resolver of resolvers) {
- if (typeof resolver === "object" && resolver.type === "directive") {
- if (name.startsWith("v"))
- name = name.slice(1);
- else
- continue;
- }
- const resolved = await (typeof resolver === "function" ? resolver(name) : resolver.resolve(name));
- if (resolved)
- return normalizeImport(resolved, fullname);
- }
- }
- function resolversAddon(resolvers) {
- return {
- async matchImports(names, matched) {
- if (!resolvers.length)
- return;
- const dynamic = [];
- const sideEffects = [];
- await Promise.all([...names].map(async (name) => {
- const matchedImport = matched.find((i) => i.as === name);
- if (matchedImport) {
- if ("sideEffects" in matchedImport)
- sideEffects.push(...toArray(matchedImport.sideEffects).map((i) => normalizeImport(i, "")));
- return;
- }
- const resolved = await firstMatchedResolver(resolvers, name);
- if (resolved)
- dynamic.push(resolved);
- if (resolved == null ? void 0 : resolved.sideEffects)
- sideEffects.push(...toArray(resolved == null ? void 0 : resolved.sideEffects).map((i) => normalizeImport(i, "")));
- }));
- if (dynamic.length) {
- this.dynamicImports.push(...dynamic);
- this.invalidate();
- }
- if (dynamic.length || sideEffects.length)
- return [...matched, ...dynamic, ...sideEffects];
- }
- };
- }
- // src/core/ctx.ts
- function resolveGlobsExclude(root, glob) {
- const excludeReg = /^!/;
- return `${excludeReg.test(glob) ? "!" : ""}${resolve(root, glob.replace(excludeReg, ""))}`;
- }
- async function scanDirExports(dirs, root) {
- const result = await fg(dirs, {
- absolute: true,
- cwd: root,
- onlyFiles: true,
- followSymbolicLinks: true
- });
- const files = Array.from(new Set(result.flat())).map(slash);
- return (await Promise.all(files.map((i) => scanExports(i, false)))).flat();
- }
- function createContext(options = {}, root = process.cwd()) {
- var _a;
- const {
- dts: preferDTS = isPackageExists("typescript")
- } = options;
- const dirs = (_a = options.dirs) == null ? void 0 : _a.concat(options.dirs.map((dir) => join(dir, "*.{tsx,jsx,ts,js,mjs,cjs,mts,cts}"))).map((dir) => slash(resolveGlobsExclude(root, dir)));
- const eslintrc = options.eslintrc || {};
- eslintrc.enabled = eslintrc.enabled === void 0 ? false : eslintrc.enabled;
- eslintrc.filepath = eslintrc.filepath || "./.eslintrc-auto-import.json";
- eslintrc.globalsPropValue = eslintrc.globalsPropValue === void 0 ? true : eslintrc.globalsPropValue;
- const resolvers = options.resolvers ? [options.resolvers].flat(2) : [];
- const injectAtEnd = options.injectAtEnd !== false;
- const unimport = createUnimport({
- imports: [],
- presets: [],
- injectAtEnd,
- addons: [
- ...options.vueTemplate ? [vueTemplateAddon()] : [],
- resolversAddon(resolvers),
- {
- declaration(dts2) {
- return `${`
- /* eslint-disable */
- /* prettier-ignore */
- // @ts-nocheck
- // noinspection JSUnusedGlobalSymbols
- // Generated by unplugin-auto-import
- ${dts2}`.trim()}
- `;
- }
- }
- ]
- });
- const importsPromise = flattenImports(options.imports).then((imports) => {
- var _a2;
- if (!imports.length && !resolvers.length && !(dirs == null ? void 0 : dirs.length))
- console.warn("[auto-import] plugin installed but no imports has defined, see https://github.com/antfu/unplugin-auto-import#configurations for configurations");
- (_a2 = options.ignore) == null ? void 0 : _a2.forEach((name) => {
- const i = imports.find((i2) => i2.as === name);
- if (i)
- i.disabled = true;
- });
- return unimport.getInternalContext().replaceImports(imports);
- });
- const filter = createFilter(
- options.include || [/\.[jt]sx?$/, /\.vue$/, /\.vue\?vue/, /\.svelte$/],
- options.exclude || [/[\\/]node_modules[\\/]/, /[\\/]\.git[\\/]/]
- );
- const dts = preferDTS === false ? false : preferDTS === true ? resolve(root, "auto-imports.d.ts") : resolve(root, preferDTS);
- const multilineCommentsRE = new RegExp("\\/\\*.*?\\*\\/", "gms");
- const singlelineCommentsRE = /\/\/.*$/gm;
- const dtsReg = new RegExp("declare\\s+global\\s*{(.*?)[\\n\\r]}", "s");
- function parseDTS(dts2) {
- var _a2;
- dts2 = dts2.replace(multilineCommentsRE, "").replace(singlelineCommentsRE, "");
- const code = (_a2 = dts2.match(dtsReg)) == null ? void 0 : _a2[0];
- if (!code)
- return;
- return Object.fromEntries(Array.from(code.matchAll(/['"]?(const\s*[^\s'"]+)['"]?\s*:\s*(.+?)[,;\r\n]/g)).map((i) => [i[1], i[2]]));
- }
- async function generateDTS(file) {
- await importsPromise;
- const dir = dirname(file);
- const originalContent = existsSync(file) ? await fs.readFile(file, "utf-8") : "";
- const originalDTS = parseDTS(originalContent);
- const currentContent = await unimport.generateTypeDeclarations({
- resolvePath: (i) => {
- if (i.from.startsWith(".") || isAbsolute(i.from)) {
- const related = slash(relative(dir, i.from).replace(/\.ts(x)?$/, ""));
- return !related.startsWith(".") ? `./${related}` : related;
- }
- return i.from;
- }
- });
- const currentDTS = parseDTS(currentContent);
- if (originalDTS) {
- Object.keys(currentDTS).forEach((key) => {
- originalDTS[key] = currentDTS[key];
- });
- const dtsList = Object.keys(originalDTS).sort().map((k) => ` ${k}: ${originalDTS[k]}`);
- return currentContent.replace(dtsReg, () => `declare global {
- ${dtsList.join("\n")}
- }`);
- }
- return currentContent;
- }
- async function parseESLint() {
- const configStr = existsSync(eslintrc.filepath) ? await fs.readFile(eslintrc.filepath, "utf-8") : "";
- const config = JSON.parse(configStr || '{ "globals": {} }');
- return config.globals;
- }
- async function generateESLint() {
- return generateESLintConfigs(await unimport.getImports(), eslintrc, await parseESLint());
- }
- const writeConfigFilesThrottled = throttle(500, writeConfigFiles, { noLeading: false });
- async function writeFile(filePath, content = "") {
- await fs.mkdir(dirname(filePath), { recursive: true });
- return await fs.writeFile(filePath, content, "utf-8");
- }
- let lastDTS;
- let lastESLint;
- async function writeConfigFiles() {
- const promises = [];
- if (dts) {
- promises.push(
- generateDTS(dts).then((content) => {
- if (content !== lastDTS) {
- lastDTS = content;
- return writeFile(dts, content);
- }
- })
- );
- }
- if (eslintrc.enabled && eslintrc.filepath) {
- promises.push(
- generateESLint().then((content) => {
- content = `${content}
- `;
- if (content.trim() !== (lastESLint == null ? void 0 : lastESLint.trim())) {
- lastESLint = content;
- return writeFile(eslintrc.filepath, content);
- }
- })
- );
- }
- return Promise.all(promises);
- }
- async function scanDirs() {
- if (dirs == null ? void 0 : dirs.length) {
- await unimport.modifyDynamicImports(async (imports) => {
- const exports_ = await scanDirExports(dirs, root);
- exports_.forEach((i) => i.__source = "dir");
- return modifyDefaultExportsAlias([
- ...imports.filter((i) => i.__source !== "dir"),
- ...exports_
- ], options);
- });
- }
- writeConfigFilesThrottled();
- }
- async function transform(code, id) {
- await importsPromise;
- const s = new MagicString(code);
- await unimport.injectImports(s, id);
- if (!s.hasChanged())
- return;
- writeConfigFilesThrottled();
- return {
- code: s.toString(),
- map: s.generateMap({ source: id, includeContent: true, hires: true })
- };
- }
- return {
- root,
- dirs,
- filter,
- scanDirs,
- writeConfigFiles,
- writeConfigFilesThrottled,
- transform,
- generateDTS,
- generateESLint
- };
- }
- async function flattenImports(map) {
- const promises = await Promise.all(toArray2(map).map(async (definition) => {
- if (typeof definition === "string") {
- if (!presets[definition])
- throw new Error(`[auto-import] preset ${definition} not found`);
- const preset = presets[definition];
- definition = typeof preset === "function" ? preset() : preset;
- }
- if ("from" in definition && "imports" in definition) {
- return await resolvePreset(definition);
- } else {
- const resolved = [];
- for (const mod of Object.keys(definition)) {
- for (const id of definition[mod]) {
- const meta = {
- from: mod
- };
- if (Array.isArray(id)) {
- meta.name = id[0];
- meta.as = id[1];
- } else {
- meta.name = id;
- meta.as = id;
- }
- resolved.push(meta);
- }
- }
- return resolved;
- }
- }));
- return promises.flat();
- }
- function modifyDefaultExportsAlias(imports, options) {
- if (options.defaultExportByFilename) {
- imports.forEach((i) => {
- var _a, _b, _c;
- if (i.name === "default")
- i.as = (_c = (_b = (_a = i.from.split("/").pop()) == null ? void 0 : _a.split(".")) == null ? void 0 : _b.shift()) != null ? _c : i.as;
- });
- }
- return imports;
- }
- // src/core/unplugin.ts
- var unplugin_default = createUnplugin((options) => {
- let ctx = createContext(options);
- return {
- name: "unplugin-auto-import",
- enforce: "post",
- transformInclude(id) {
- return ctx.filter(id);
- },
- async transform(code, id) {
- return ctx.transform(code, id);
- },
- async buildStart() {
- await ctx.scanDirs();
- },
- async buildEnd() {
- await ctx.writeConfigFiles();
- },
- vite: {
- async handleHotUpdate({ file }) {
- var _a;
- if ((_a = ctx.dirs) == null ? void 0 : _a.some((glob) => minimatch(slash2(file), slash2(glob))))
- await ctx.scanDirs();
- },
- async configResolved(config) {
- if (ctx.root !== config.root) {
- ctx = createContext(options, config.root);
- await ctx.scanDirs();
- }
- }
- }
- };
- });
- export {
- unplugin_default
- };
|