116 lines
3.8 KiB
TypeScript
116 lines
3.8 KiB
TypeScript
import { spawn } from "child_process";
|
||
import fs from "fs";
|
||
import { Plugin, ResolvedConfig } from "vite";
|
||
import pkg from "../../package.json";
|
||
|
||
/**
|
||
* 项目 logo
|
||
* @description 内容：APPTIFY
|
||
*/
|
||
const LOGO = `
|
||
________ ______ ______ _________ ________ ______ __ __
|
||
/_______/\\\\ /_____/\\\\ /_____/\\\\ /________/\\\\/_______/\\\\/_____/\\\\ /_/\\\\/_/\\\\
|
||
\\\\::: _ \\\\ \\\\\\\\:::_ \\\\ \\\\\\\\:::_ \\\\ \\\\\\\\__.::.__\\\\/\\\\__.::._\\\\/\\\\::::_\\\\/_\\\\ \\\\ \\\\ \\\\ \\\\
|
||
\\\\::(_) \\\\ \\\\\\\\:(_) \\\\ \\\\\\\\:(_) \\\\ \\\\ \\\\::\\\\ \\\\ \\\\::\\\\ \\\\ \\\\:\\\\/___/\\\\\\\\:\\\\_\\\\ \\\\ \\\\
|
||
\\\\:: __ \\\\ \\\\\\\\: ___\\\\/ \\\\: ___\\\\/ \\\\::\\\\ \\\\ _\\\\::\\\\ \\\\__\\\\:::._\\\\/ \\\\::::_\\\\/
|
||
\\\\:.\\\\ \\\\ \\\\ \\\\\\\\ \\\\ \\\\ \\\\ \\\\ \\\\ \\\\::\\\\ \\\\ /__\\\\::\\\\__/\\\\\\\\:\\\\ \\\\ \\\\::\\\\ \\\\
|
||
\\\\__\\\\/\\\\__\\\\/ \\\\_\\\\/ \\\\_\\\\/ \\\\__\\\\/ \\\\________\\\\/ \\\\_\\\\/ \\\\__\\\\/
|
||
`;
|
||
|
||
/**
|
||
* 以 shell 形式执行命令，成功返回输出的字符串
|
||
* @param cmd 命令
|
||
* @returns Promise<string | void>
|
||
*/
|
||
const exec = (cmd: string) => {
|
||
return new Promise<string | void>((resolve) => {
|
||
if (!cmd) {
|
||
return resolve();
|
||
}
|
||
const child = spawn(cmd, [], { shell: true });
|
||
child.stdout.once("data", (data) => {
|
||
resolve(data.toString().replace(/"|\n/g, ""));
|
||
});
|
||
child.stderr.once("data", () => {
|
||
resolve();
|
||
});
|
||
});
|
||
};
|
||
|
||
/**
|
||
* 获取构建信息
|
||
* @returns Promise<string>
|
||
*/
|
||
const getBuildInfo = async () => {
|
||
const hash = await exec("git log --format=%h -n 1");
|
||
const time = new Date().toLocaleString("zh-Hans-CN");
|
||
const latestTag = await exec("git describe --tags --abbrev=0");
|
||
const commits = await exec(`git rev-list --count ${latestTag}..HEAD`);
|
||
const version = commits ? `${latestTag}.${commits}` : `v${pkg.version}`;
|
||
const content = ` 欢迎访问！版本: ${version} 标识: ${hash} 构建时间: ${time}`;
|
||
const style = `"color: #09f; font-weight: 900;", "font-size: 12px; color: #09f; font-family: ''"`;
|
||
const script = `console.log(\`%c${LOGO} \n%c${content}\n\`, ${style});\n`;
|
||
return script;
|
||
};
|
||
|
||
/**
|
||
* 项目构建插件
|
||
* @returns Plugin
|
||
*/
|
||
export default function plugin(): Plugin {
|
||
let config: ResolvedConfig;
|
||
let extension: string;
|
||
|
||
return {
|
||
name: "vite:customizer",
|
||
enforce: "pre",
|
||
|
||
configResolved(resolvedConfig) {
|
||
config = resolvedConfig;
|
||
const defaultExt = config.mode === "development" ? "dev" : "prod";
|
||
extension = config.env.VITE_BUILD_EXTENTION || defaultExt;
|
||
},
|
||
|
||
async transformIndexHtml(html) {
|
||
const script = await getBuildInfo();
|
||
const replacedHtml = html.replace(/__((\w|_|-)+)__/g, (match, p1) => {
|
||
return config.env[`VITE_${p1}`] || "";
|
||
});
|
||
return {
|
||
html: replacedHtml,
|
||
tags: [
|
||
{
|
||
tag: "script",
|
||
injectTo: "body",
|
||
children: script,
|
||
},
|
||
],
|
||
};
|
||
},
|
||
|
||
async resolveId(id, importer, options) {
|
||
if (!extension || !id.startsWith("/src")) {
|
||
return;
|
||
}
|
||
const resolution = await this.resolve(id, importer, { skipSelf: true, ...options });
|
||
const targetPath = resolution?.id.replace(/\.([^.]*?)$/, `.${extension}.$1`);
|
||
if (targetPath && fs.existsSync(targetPath)) {
|
||
return targetPath;
|
||
}
|
||
},
|
||
|
||
load(id) {
|
||
if (!extension || !id.includes("src")) {
|
||
return;
|
||
}
|
||
if (id.includes("?")) {
|
||
return;
|
||
}
|
||
const targetPath = id.replace(/\.([^.]*?)$/, `.${extension}.$1`);
|
||
if (targetPath && fs.existsSync(targetPath)) {
|
||
return fs.readFileSync(targetPath, "utf-8");
|
||
}
|
||
},
|
||
};
|
||
}
|