76 lines
2.4 KiB
TypeScript
76 lines
2.4 KiB
TypeScript
import { spawn } from 'child_process';
|
||
import { Plugin } 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 vString = `var __APP_VERSION__ = '${version}';\n`;
|
||
const hString = `var __APP_HASH__ = '${hash}';\n`;
|
||
const dString = `var __APP_DATE__ = '${time}';\n`;
|
||
const lString = `console.log(\`%c${LOGO} \n%c${content}\n\`, ${style});\n`;
|
||
return vString + hString + dString + lString;
|
||
};
|
||
|
||
/**
|
||
* 项目构建插件
|
||
* @returns Plugin
|
||
*/
|
||
export default function plugin(): Plugin {
|
||
return {
|
||
name: 'vite:info',
|
||
enforce: 'pre',
|
||
async transformIndexHtml() {
|
||
const script = await getBuildInfo();
|
||
return [
|
||
{
|
||
tag: 'script',
|
||
injectTo: 'body',
|
||
children: script,
|
||
},
|
||
];
|
||
},
|
||
};
|
||
}
|