@lonu/stc
Version:
A tool for converting OpenApi/Swagger/Apifox into code.
219 lines (216 loc) • 7.59 kB
JavaScript
import * as dntShim from "../_dnt.shims.js";
import { parseArgs } from "../deps/jsr.io/@std/cli/1.0.19/mod.js";
import ProgressBar from "../deps/jsr.io/@deno-library/progress/1.5.1/mod.js";
import Logs from "./console.js";
import { createAppFile } from "./utils.js";
import denoJson from "../deno.js";
import { getT } from "./i18n/index.js";
const drawLogo = () => {
console.log(`
______ _______ _______
/ _____|_______|_______)
( (____ _ _
\____ \ | | | |
_____) ) | | | |_____
(______/ |_| \______)
`);
};
/**
* 检查更新并处理更新过程（如果有新版本可用）。
*
* @return {Promise<string>} 如果进行了更新，则返回最新版本，如果未找到更新，则返回当前版本。
*/
const checkUpdate = async () => {
Logs.info(`${getT("$t(cli.checkUpdate)")}...`);
const version = Number(denoJson.version?.replace(/\./g, "") ?? 0);
const res = await fetch("https://api.github.com/repos/long-woo/stc/releases/latest");
if (res.ok) {
const data = await res.json();
const latestVersion = data.tag_name;
const _lastVersion = Number(latestVersion.replace(/\./g, "") ?? 0);
if (version < _lastVersion) {
// 非 deno compile 的可执行文件，仅提示更新
if (typeof confirm === "undefined") {
// 提示有新版本
console.log("\n");
Logs.warn(getT("$t(cli.updatePrompt)", {
version: denoJson.version,
latestVersion,
}));
console.log("\n");
return;
}
// 询问是否更新
const _needUpdate = confirm(`${getT("$t(cli.updatePrompt)", {
version: denoJson.version,
latestVersion,
})}${getT("$t(cli.updateConfirm)")}`);
if (!_needUpdate)
return;
Logs.info(`${getT("$t(cli.updating)", {
version: denoJson.version,
latestVersion,
})}...`);
const dir = dntShim.Deno.cwd();
const systemInfo = dntShim.Deno.build;
const appNameMap = {
"x86_64-apple-darwin": "stc",
"aarch64-apple-darwin": "stc-m",
"x86_64-pc-windows-msvc": "stc-win",
"x86_64-unknown-linux-gnu": "stc-linux",
};
const downloadUrl = `https://github.com/long-woo/stc/releases/download/${latestVersion}/${appNameMap[systemInfo.target]}`;
const downloadApp = await fetch(downloadUrl);
const reader = downloadApp.body?.getReader();
// 文件内容长度
const contentLength = Number(downloadApp.headers.get("content-length"));
const size = Number((contentLength / 1024 / 1024).toFixed(1));
// 接收的文件字节长度
let receivedLength = 0;
// 接收到的字节数据
const chunks = [];
if (reader) {
const progressBar = new ProgressBar({
total: size,
display: ":completed/:total M :bar :percent",
});
while (true) {
const { done, value } = await reader.read();
if (done) {
break;
}
chunks.push(value);
receivedLength += value.length;
progressBar.render(Number((receivedLength / 1024 / 1024).toFixed(1)));
}
const chunkContent = new Uint8Array(receivedLength);
let offset = 0;
for (const chunk of chunks) {
chunkContent.set(chunk, offset);
offset += chunk.length;
}
await createAppFile(`${dir}/stc`, chunkContent.buffer);
Logs.success(getT("$t(cli.updateDone)", { version: latestVersion }));
dntShim.Deno.exit(0);
}
Logs.error(downloadApp.statusText);
// const command = new Deno.Command("deno", {
// args: [
// "compile",
// "-A",
// `https://deno.land/x/stc@${latestVersion}/mod.ts`,
// "--output",
// `${dir}/stc`,
// ],
// });
// const { code, stderr } = await command.output();
// if (code === 0) {
// Logs.info("更新完成");
// return;
// }
// Logs.error(new TextDecoder().decode(stderr));
}
Logs.info(getT("$t(cli.latestVersion)"));
}
};
/**
* 打印帮助信息
*/
const printHelp = () => {
console.log(`
${getT("$t(cli.usage)")}
stc [options]
${getT("$t(cli.option)")}
-h, --help ${getT("$t(cli.option_help)")}
--url ${getT("$t(cli.option_url)")}
-o, --outDir ${getT("$t(cli.option_out)", { out: "./stc_out" })}
--client ${getT("$t(cli.option_client)")}
-l, --lang ${getT("$t(cli.option_lang)")}
-f, --filter ${getT("$t(cli.option_filter)")}
--tag ${getT("$t(cli.option_tag)")}
-c, --conjunction ${getT("$t(cli.option_conjunction)")}
--actionIndex ${getT("$t(cli.option_actionIndex)")}
--shared ${getT("$t(cli.option_shared)")}
--clean ${getT("$t(cli.option_clean)")}
-v, --version ${getT("$t(cli.option_version)")}
${getT("$t(cli.example)")}
stc -o ./stc_out --url http://petstore.swagger.io/v2/swagger.json
stc -o ./stc_out -p axios -l ts --url https://petstore3.swagger.io/api/v3/openapi.json
`);
dntShim.Deno.exit(0);
};
/**
* 主入口
*/
export const main = async () => {
// 定义命令行参数和选项的配置
const argsConfig = {
boolean: ["help", "version", "shared", "clean"],
string: [
"url",
"outDir",
"client",
"lang",
"tag",
"filter",
"conjunction",
"actionIndex",
],
alias: {
h: "help",
o: "outDir",
l: "lang",
v: "version",
f: "filter",
c: "conjunction",
},
collect: ["filter"],
default: {
outDir: "./stc_out",
lang: "ts",
client: "axios",
conjunction: "By",
actionIndex: "-1",
shared: true,
clean: true,
},
unknown: (arg) => {
Logs.error(getT("$t(cli.unknownOption)", { arg }));
printHelp();
dntShim.Deno.exit(1);
},
};
// 清空控制台信息
Logs.clear();
drawLogo();
// 解析命令行参数和选项
const args = parseArgs(dntShim.Deno.args, argsConfig);
// 检查更新
await checkUpdate();
// 帮助
if (args.help) {
printHelp();
}
// 版本
if (args.version) {
console.log(`stc v${denoJson.version}`);
dntShim.Deno.exit(0);
}
// 检查 url
if (!args.url) {
Logs.error(getT("$t(cli.requiredUrl)"));
printHelp();
}
return {
url: args.url,
outDir: args.outDir,
client: args.client,
lang: args.lang,
tag: args.tag,
filter: args.filter,
conjunction: args.conjunction,
actionIndex: args.actionIndex,
shared: args.shared,
clean: args.clean,
};
};