mirror of
https://github.com/AikidoSec/safe-chain.git
synced 2026-05-26 12:10:49 +00:00
PR Comments: use safeChainArgs and use defined constants
This commit is contained in:
parent
2f1692e253
commit
b6404775e2
2 changed files with 25 additions and 8 deletions
|
|
@ -12,22 +12,39 @@ export function initializeCliArguments(args) {
|
|||
const remainingArgs = [];
|
||||
|
||||
for (const arg of args) {
|
||||
if (arg.startsWith(SAFE_CHAIN_ARG_PREFIX)) {
|
||||
if (arg.toLowerCase().startsWith(SAFE_CHAIN_ARG_PREFIX)) {
|
||||
safeChainArgs.push(arg);
|
||||
|
||||
if (arg.startsWith(SAFE_CHAIN_ARG_PREFIX + "malware-action=")) {
|
||||
state.malwareAction = arg.substring(
|
||||
(SAFE_CHAIN_ARG_PREFIX + "malware-action=").length
|
||||
);
|
||||
}
|
||||
} else {
|
||||
remainingArgs.push(arg);
|
||||
}
|
||||
}
|
||||
|
||||
setMalwareAction(safeChainArgs);
|
||||
|
||||
return remainingArgs;
|
||||
}
|
||||
|
||||
function setMalwareAction(args) {
|
||||
const safeChainMalwareActionArg = SAFE_CHAIN_ARG_PREFIX + "malware-action=";
|
||||
|
||||
const action = getLastArgEqualsValue(args, safeChainMalwareActionArg);
|
||||
if (!action) {
|
||||
return;
|
||||
}
|
||||
state.malwareAction = action.toLowerCase();
|
||||
}
|
||||
|
||||
function getLastArgEqualsValue(args, prefix) {
|
||||
for (var i = args.length - 1; i >= 0; i--) {
|
||||
const arg = args[i];
|
||||
if (arg.toLowerCase().startsWith(prefix)) {
|
||||
return arg.substring(prefix.length);
|
||||
}
|
||||
}
|
||||
|
||||
return undefined;
|
||||
}
|
||||
|
||||
export function getMalwareAction() {
|
||||
return state.malwareAction;
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue