修复文件校验

This commit is contained in:
whyour 2024-07-07 21:26:11 +08:00
parent ca5cb88eca
commit f4337008eb

View File

@ -144,8 +144,8 @@ async function runNormal(fileParam, scriptParams) {
async function runConcurrent(fileParam, envParam, numParam, scriptParams) { async function runConcurrent(fileParam, envParam, numParam, scriptParams) {
if (!envParam || !numParam) { if (!envParam || !numParam) {
console.log(`\n 缺少并发运行的环境变量参数 task xxx.js conc Test 1 3`); console.log(`缺少并发运行的环境变量参数 task xxx.js conc Test 1 3`);
process.exit(1); return;
} }
const array = (process.env[envParam] || '').split('&'); const array = (process.env[envParam] || '').split('&');
@ -186,8 +186,8 @@ async function runConcurrent(fileParam, envParam, numParam, scriptParams) {
async function runDesignated(fileParam, envParam, numParam, scriptParams) { async function runDesignated(fileParam, envParam, numParam, scriptParams) {
if (!envParam || !numParam) { if (!envParam || !numParam) {
console.log(`\n 缺少单独运行的参数 task xxx.js desi Test 1 3`); console.log(`缺少单独运行的参数 task xxx.js desi Test 1 3`);
process.exit(1); return;
} }
const array = (process.env[envParam] || '').split('&'); const array = (process.env[envParam] || '').split('&');
@ -200,13 +200,11 @@ async function runDesignated(fileParam, envParam, numParam, scriptParams) {
cd(relativePath); cd(relativePath);
fileParam = fileParam.replace(`${relativePath}/`, ''); fileParam = fileParam.replace(`${relativePath}/`, '');
} }
console.log('cookieStr', cookieStr.length, arrayRun.length) console.log('cookieStr', cookieStr.length, arrayRun.length);
// ${envParam}="${cookieStr.replace('"', '\\"')}" // ${envParam}="${cookieStr.replace('"', '\\"')}"
await runWithTimeout( await runWithTimeout(
`${ `${globalState.whichProgram} ${fileParam} ${scriptParams}`,
globalState.whichProgram
} ${fileParam} ${scriptParams}`,
); );
} }
@ -244,6 +242,13 @@ async function main(taskShellParams, scriptParams) {
return await usage(); return await usage();
} }
if (fileParam && /\.(js|py|pyc|sh|ts)$/.test(fileParam)) { if (fileParam && /\.(js|py|pyc|sh|ts)$/.test(fileParam)) {
const filePath = fileParam.startsWith('/')
? fileParam
: path.join(dirScripts, fileParam);
if (!(await fs.exists(filePath))) {
console.log(`文件不存在 ${fileParam}`);
return;
}
switch (action) { switch (action) {
case undefined: case undefined:
return await runNormal(fileParam, scriptParams); return await runNormal(fileParam, scriptParams);