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 = [];
|
const remainingArgs = [];
|
||||||
|
|
||||||
for (const arg of args) {
|
for (const arg of args) {
|
||||||
if (arg.startsWith(SAFE_CHAIN_ARG_PREFIX)) {
|
if (arg.toLowerCase().startsWith(SAFE_CHAIN_ARG_PREFIX)) {
|
||||||
safeChainArgs.push(arg);
|
safeChainArgs.push(arg);
|
||||||
|
|
||||||
if (arg.startsWith(SAFE_CHAIN_ARG_PREFIX + "malware-action=")) {
|
|
||||||
state.malwareAction = arg.substring(
|
|
||||||
(SAFE_CHAIN_ARG_PREFIX + "malware-action=").length
|
|
||||||
);
|
|
||||||
}
|
|
||||||
} else {
|
} else {
|
||||||
remainingArgs.push(arg);
|
remainingArgs.push(arg);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
setMalwareAction(safeChainArgs);
|
||||||
|
|
||||||
return remainingArgs;
|
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() {
|
export function getMalwareAction() {
|
||||||
return state.malwareAction;
|
return state.malwareAction;
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -7,7 +7,7 @@ export function getMalwareAction() {
|
||||||
return MALWARE_ACTION_PROMPT;
|
return MALWARE_ACTION_PROMPT;
|
||||||
}
|
}
|
||||||
|
|
||||||
return "block";
|
return MALWARE_ACTION_BLOCK;
|
||||||
}
|
}
|
||||||
|
|
||||||
export const MALWARE_ACTION_BLOCK = "block";
|
export const MALWARE_ACTION_BLOCK = "block";
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue