mirror of
https://github.com/CyberSecurityUP/NeuroSploit.git
synced 2026-08-14 13:40:23 +02:00
feat: PR security gate, @neurosploit bot, richer NL REPL (#39)
GitHub automation - integrations: github_set_status (commit status), github_pr_review (REQUEST_CHANGES/APPROVE), github_pr_head_sha, and a shared severity gate (severity_rank / worst_confirmed_rank / gate_trips — confirmed findings only). - `neurosploit pr --fail-on <critical|high|medium|low>`: on a confirmed finding at/above the threshold, sets a failing `neurosploit/security` commit status, posts a REQUEST_CHANGES review, and exits 2 so a CI check fails — branch protection then blocks the merge. - Two ready GitHub Actions: neurosploit-pr-gate.yml (review + block every PR) and neurosploit-mention.yml (writers comment @neurosploit <text> to trigger a scan; any language; URL → black-box, else PR review). Natural-language REPL - Intent now also parses spoken toggles/knobs across PT/EN/ES: Burp/proxy, browser/MCP, subscription, "N votos/votes", recon depth (number or quick/deep/exhaustive), plus stop verbs. handle_nl returns the follow-up command (/run or /stop). Docs: README trimmed to features (version changelog stays in RELEASE.md), new automations documented in README + TUTORIAL-INTEGRATION. Tests: gate (3), NL toggles/stop (added). All green. Claude-Session: https://claude.ai/code/session_018BGLy4j5qsqqid6CoovowC Co-authored-by: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
co-authored by
Claude Opus 4.8
parent
21a62c95e5
commit
3786d7c559
@@ -258,6 +258,11 @@ enum Cmd {
|
||||
/// Post a summary comment back on the PR (needs github integration on).
|
||||
#[arg(long)]
|
||||
comment: bool,
|
||||
/// Block the PR when a confirmed finding is this severity or worse:
|
||||
/// critical|high|medium|low. Sets a failing commit status + a
|
||||
/// REQUEST_CHANGES review, and exits non-zero so CI fails the check.
|
||||
#[arg(long)]
|
||||
fail_on: Option<String>,
|
||||
/// Open a Jira card per finding (needs jira integration on).
|
||||
#[arg(long)]
|
||||
jira: bool,
|
||||
@@ -494,7 +499,7 @@ async fn main() -> anyhow::Result<()> {
|
||||
let out = run_mode(&base, cfg, false, Mode::Skills).await?;
|
||||
print_findings(&out);
|
||||
}
|
||||
Cmd::Pr { repo, number, models, vote_n, chain_depth, recon, subscription, comment, jira, verbose } => {
|
||||
Cmd::Pr { repo, number, models, vote_n, chain_depth, recon, subscription, comment, fail_on, jira, verbose } => {
|
||||
let ig = harness::integrations::Integrations::load(&repl::proj_dir());
|
||||
let owner_repo = normalize_repo(&repo);
|
||||
let path = clone_pr(&base, &ig, &owner_repo, number)?;
|
||||
@@ -510,6 +515,16 @@ async fn main() -> anyhow::Result<()> {
|
||||
let out = run_engagement(&base, cfg, false, true).await?;
|
||||
print_findings(&out);
|
||||
post_integrations(&ig, &format!("{owner_repo}#{number}"), &out, jira, comment, Some((&owner_repo, number))).await;
|
||||
// Security gate: block the PR when a confirmed finding is >= threshold.
|
||||
if let Some(thresh) = fail_on.as_deref() {
|
||||
let blocked = gate_pr(&ig, &owner_repo, number, &out, thresh).await;
|
||||
if blocked {
|
||||
eprintln!(" \x1b[1;31m⛔ PR gate: confirmed finding ≥ {thresh} — blocking (exit 2)\x1b[0m");
|
||||
std::process::exit(2);
|
||||
} else {
|
||||
println!(" \x1b[1;32m✓ PR gate: nothing ≥ {thresh} — clear\x1b[0m");
|
||||
}
|
||||
}
|
||||
}
|
||||
Cmd::Watch { repo, branch, interval, models, subscription, jira, verbose } => {
|
||||
let ig = harness::integrations::Integrations::load(&repl::proj_dir());
|
||||
@@ -972,6 +987,44 @@ fn clone_pr(base: &Path, ig: &harness::integrations::Integrations, owner_repo: &
|
||||
}
|
||||
|
||||
/// After a run, optionally open Jira cards and/or comment on a GitHub PR.
|
||||
/// Enforce the PR security gate. Sets a GitHub commit status (success/failure)
|
||||
/// on the PR head and, when it trips, submits a REQUEST_CHANGES review so branch
|
||||
/// protection blocks the merge. Best-effort on the API calls (a token may be
|
||||
/// absent locally); returns whether the gate tripped so the caller can exit 2.
|
||||
async fn gate_pr(
|
||||
ig: &harness::integrations::Integrations,
|
||||
owner_repo: &str,
|
||||
number: u64,
|
||||
out: &RunOutput,
|
||||
threshold: &str,
|
||||
) -> bool {
|
||||
use harness::integrations as gi;
|
||||
let tripped = gi::gate_trips(&out.findings, threshold);
|
||||
if ig.github.enabled {
|
||||
let (state, desc) = if tripped {
|
||||
("failure", format!("Confirmed finding ≥ {threshold} — merge blocked by NeuroSploit"))
|
||||
} else {
|
||||
("success", "No confirmed finding at/above the gate threshold".to_string())
|
||||
};
|
||||
// Attach the status to the PR head SHA (looked up from the API).
|
||||
match ig.github_pr_head_sha(owner_repo, number).await {
|
||||
Ok(sha) => {
|
||||
if let Err(e) = ig.github_set_status(owner_repo, &sha, state, "neurosploit/security", &desc, None).await {
|
||||
eprintln!(" github status: {e}");
|
||||
}
|
||||
}
|
||||
Err(e) => eprintln!(" github PR head lookup: {e}"),
|
||||
}
|
||||
if tripped {
|
||||
let body = format!("## ⛔ NeuroSploit security gate\n\nBlocking this PR: a **confirmed** finding is **{threshold}** or worse.\n\n{}", pr_comment_body(out));
|
||||
if let Err(e) = ig.github_pr_review(owner_repo, number, "REQUEST_CHANGES", &body).await {
|
||||
eprintln!(" github review: {e}");
|
||||
}
|
||||
}
|
||||
}
|
||||
tripped
|
||||
}
|
||||
|
||||
async fn post_integrations(
|
||||
ig: &harness::integrations::Integrations,
|
||||
target: &str,
|
||||
|
||||
+103
-14
@@ -466,8 +466,10 @@ pub async fn repl(base: &Path) -> anyhow::Result<()> {
|
||||
} else {
|
||||
let attached = expand_ats(line, &mut s);
|
||||
if attached > 0 { println!(" ({attached} @attachment(s) added to context)"); }
|
||||
let run = handle_nl(line, &mut s).await;
|
||||
if run { ("/run".to_string(), String::new()) } else { continue }
|
||||
match handle_nl(line, &mut s).await {
|
||||
Some(c) => (c.to_string(), String::new()), // "/run" or "/stop"
|
||||
None => continue,
|
||||
}
|
||||
};
|
||||
let (cmd, arg) = (cmd.as_str(), arg.as_str());
|
||||
match cmd {
|
||||
@@ -1700,8 +1702,9 @@ fn help() {
|
||||
println!("\n \x1b[1mNeuroSploit REPL — commands\x1b[0m");
|
||||
|
||||
println!("\n \x1b[2mNATURAL LANGUAGE (any language — just type, no slash)\x1b[0m");
|
||||
println!(" e.g. \x1b[36mtesta https://loja.com com opus, foco em SQLi, fora de escopo /admin, roda\x1b[0m");
|
||||
println!(" \x1b[2mconfigures target/models/focus/objective/out-of-scope and can launch — hands-free\x1b[0m");
|
||||
println!(" e.g. \x1b[36mtesta https://loja.com com opus, foco em SQLi, fora de escopo /admin, usa burp, roda\x1b[0m");
|
||||
println!(" \x1b[2msets target/models/focus/objective/out-of-scope + toggles (burp·browser·votes·recon),\x1b[0m");
|
||||
println!(" \x1b[2mand can launch or stop — hands-free. Ambiguous phrasing falls back to the model.\x1b[0m");
|
||||
|
||||
println!("\n \x1b[2mTARGET & SCOPE\x1b[0m");
|
||||
h("/onboard", "guided setup: pick scope (web · infra · cloud · ai/llm · skills/n8n)");
|
||||
@@ -1774,14 +1777,23 @@ struct Intent {
|
||||
out_of_scope: Option<String>,
|
||||
auth: Option<String>,
|
||||
scope: Option<&'static str>,
|
||||
// Toggles/knobs the user can ask for in words.
|
||||
mcp: Option<bool>, // "usa navegador/browser", "ativa mcp"
|
||||
proxy: Option<String>, // "manda pro burp", "usa proxy 127.0.0.1:8080"
|
||||
subscription: Option<bool>, // "usa minha assinatura/login"
|
||||
vote_n: Option<usize>, // "3 votos", "5 votes"
|
||||
recon: Option<usize>, // "recon profundo/exaustivo", "recon 4"
|
||||
run: bool,
|
||||
stop: bool, // "para", "stop", "cancela"
|
||||
}
|
||||
|
||||
impl Intent {
|
||||
fn is_empty(&self) -> bool {
|
||||
self.target.is_none() && self.repo.is_none() && self.models.is_empty()
|
||||
&& self.focus.is_none() && self.objective.is_none() && self.out_of_scope.is_none()
|
||||
&& self.auth.is_none() && self.scope.is_none() && !self.run
|
||||
&& self.auth.is_none() && self.scope.is_none()
|
||||
&& self.mcp.is_none() && self.proxy.is_none() && self.subscription.is_none()
|
||||
&& self.vote_n.is_none() && self.recon.is_none() && !self.run && !self.stop
|
||||
}
|
||||
/// Fill any field this intent is missing from `other` (deterministic wins).
|
||||
fn merge_from(&mut self, other: Intent) {
|
||||
@@ -1793,14 +1805,21 @@ impl Intent {
|
||||
if self.out_of_scope.is_none() { self.out_of_scope = other.out_of_scope; }
|
||||
if self.auth.is_none() { self.auth = other.auth; }
|
||||
if self.scope.is_none() { self.scope = other.scope; }
|
||||
if self.mcp.is_none() { self.mcp = other.mcp; }
|
||||
if self.proxy.is_none() { self.proxy = other.proxy; }
|
||||
if self.subscription.is_none() { self.subscription = other.subscription; }
|
||||
if self.vote_n.is_none() { self.vote_n = other.vote_n; }
|
||||
if self.recon.is_none() { self.recon = other.recon; }
|
||||
self.run = self.run || other.run;
|
||||
self.stop = self.stop || other.stop;
|
||||
}
|
||||
}
|
||||
|
||||
/// Resolve a natural-language line into session config. Deterministic fast-path
|
||||
/// first (0 tokens); if the phrase is ambiguous and a model is available, ask it
|
||||
/// to structure the request (works in any language). Returns whether to run now.
|
||||
async fn handle_nl(line: &str, s: &mut Session) -> bool {
|
||||
/// to structure the request (works in any language). Returns a follow-up command
|
||||
/// to execute ("/run" or "/stop"), or None when it only reconfigured.
|
||||
async fn handle_nl(line: &str, s: &mut Session) -> Option<&'static str> {
|
||||
let (mut intent, confident) = parse_intent_fast(line);
|
||||
if !confident && !s.offline {
|
||||
if let Some(mi) = parse_intent_model(line, s).await {
|
||||
@@ -1811,13 +1830,14 @@ async fn handle_nl(line: &str, s: &mut Session) -> bool {
|
||||
// Nothing structured found → treat the whole line as focus (old behavior).
|
||||
s.instructions = Some(line.to_string());
|
||||
println!(" focus set: {line}");
|
||||
return false;
|
||||
return None;
|
||||
}
|
||||
apply_intent(s, intent)
|
||||
}
|
||||
|
||||
/// Apply an intent to the session, print a summary, and return `run`.
|
||||
fn apply_intent(s: &mut Session, intent: Intent) -> bool {
|
||||
/// Apply an intent to the session, print a summary, and return the follow-up
|
||||
/// command ("/run" | "/stop") or None.
|
||||
fn apply_intent(s: &mut Session, intent: Intent) -> Option<&'static str> {
|
||||
let mut set = Vec::new();
|
||||
if let Some(t) = intent.target {
|
||||
let t = if t.starts_with("http") || t.contains("://") { t } else { format!("https://{t}") };
|
||||
@@ -1833,19 +1853,32 @@ fn apply_intent(s: &mut Session, intent: Intent) -> bool {
|
||||
if let Some(o) = intent.objective { s.objective = Some(o.clone()); set.push(format!("objective=\"{o}\"")); }
|
||||
if let Some(x) = intent.out_of_scope { s.out_of_scope = Some(x.clone()); set.push(format!("out-of-scope=\"{x}\"")); }
|
||||
if let Some(a) = intent.auth { let a = normalize_auth(&a); s.auth = Some(a.clone()); set.push("auth set".into()); let _ = a; }
|
||||
if set.is_empty() {
|
||||
if let Some(b) = intent.mcp { s.mcp = b; set.push(format!("mcp={}", onoff(b))); }
|
||||
if let Some(p) = intent.proxy {
|
||||
let p = if p.starts_with("http") { p } else { format!("http://{p}") };
|
||||
s.proxy = Some(p.clone()); set.push(format!("proxy={p}"));
|
||||
}
|
||||
if let Some(b) = intent.subscription { s.subscription = b; set.push(format!("subscription={}", onoff(b))); }
|
||||
if let Some(v) = intent.vote_n { s.vote_n = v; set.push(format!("votes={v}")); }
|
||||
if let Some(r) = intent.recon { s.recon_intensity = r.clamp(1, 4); set.push(format!("recon={}", s.recon_intensity)); }
|
||||
if set.is_empty() && !intent.stop {
|
||||
println!(" \x1b[2m(understood — nothing to change)\x1b[0m");
|
||||
} else {
|
||||
} else if !set.is_empty() {
|
||||
println!(" \x1b[36m⇢ configured\x1b[0m {}", set.join(" · "));
|
||||
}
|
||||
if intent.stop {
|
||||
println!(" \x1b[1;33m⏸ stopping\x1b[0m …");
|
||||
return Some("/stop");
|
||||
}
|
||||
if intent.run {
|
||||
if s.target.is_none() && s.repo.is_none() {
|
||||
println!(" \x1b[33m! set a target/repo first — nothing to run yet.\x1b[0m");
|
||||
return false;
|
||||
return None;
|
||||
}
|
||||
println!(" \x1b[1;35m▶ launching\x1b[0m …");
|
||||
return Some("/run");
|
||||
}
|
||||
intent.run
|
||||
None
|
||||
}
|
||||
|
||||
/// Deterministic, zero-token parse for the common phrasings (PT/EN/ES). Returns
|
||||
@@ -1872,6 +1905,27 @@ fn parse_intent_fast(line: &str) -> (Intent, bool) {
|
||||
"prueba", "probar", "escanea", "ejecuta", "corre", "lanza", "lanzar",
|
||||
];
|
||||
if RUN_VERBS.iter().any(|v| word_present(&low, v)) { it.run = true; }
|
||||
// Stop verbs.
|
||||
const STOP_VERBS: &[&str] = &["stop", "para", "pare", "parar", "cancel", "cancela", "cancelar", "aborta", "abortar", "halt", "detén", "detener", "para tudo"];
|
||||
if STOP_VERBS.iter().any(|v| word_present(&low, v)) { it.stop = true; }
|
||||
|
||||
// Spoken toggles/knobs (PT/EN/ES). Only set when clearly mentioned.
|
||||
if word_present(&low, "burp") || low.contains("intercept") { it.proxy = Some("http://127.0.0.1:8080".into()); }
|
||||
if low.contains("browser") || low.contains("navegador") || low.contains("navegou") || low.contains("playwright") || word_present(&low, "mcp") {
|
||||
it.mcp = Some(!(low.contains("sem navegador") || low.contains("no browser") || low.contains("sin navegador")));
|
||||
}
|
||||
if low.contains("assinatura") || low.contains("subscription") || low.contains("meu login") || low.contains("mi cuenta") || low.contains("suscripción") {
|
||||
it.subscription = Some(true);
|
||||
}
|
||||
// "3 votos" / "5 votes" / "2 votos de validação".
|
||||
if let Some(n) = number_before_any(&low, &["voto", "votos", "vote", "votes"]) { it.vote_n = Some(n as usize); }
|
||||
// Recon depth: explicit number 1-4 or a qualitative word.
|
||||
if let Some(n) = number_before_any(&low, &["recon"]) { if (1..=4).contains(&n) { it.recon = Some(n as usize); } }
|
||||
if it.recon.is_none() && low.contains("recon") {
|
||||
if low.contains("exausti") || low.contains("exhaust") { it.recon = Some(4); }
|
||||
else if low.contains("profund") || low.contains("deep") || low.contains("profundo") { it.recon = Some(3); }
|
||||
else if low.contains("rápid") || low.contains("rapid") || low.contains("quick") || low.contains("quick") { it.recon = Some(1); }
|
||||
}
|
||||
|
||||
// Keyworded clauses: split on commas/semicolons and classify each chunk.
|
||||
let mut residue = 0usize;
|
||||
@@ -2014,6 +2068,24 @@ fn join_opt(prev: Option<String>, add: &str) -> String {
|
||||
}
|
||||
}
|
||||
|
||||
/// First integer that appears immediately before any of `keys` (e.g. "3 votos"
|
||||
/// with keys ["votos"] → 3), scanning token pairs. Returns None if not found.
|
||||
fn number_before_any(low: &str, keys: &[&str]) -> Option<u64> {
|
||||
let toks: Vec<&str> = low.split(|c: char| !c.is_ascii_alphanumeric()).filter(|t| !t.is_empty()).collect();
|
||||
for w in toks.windows(2) {
|
||||
if keys.iter().any(|k| w[1].starts_with(k)) {
|
||||
if let Ok(n) = w[0].parse::<u64>() { return Some(n); }
|
||||
}
|
||||
}
|
||||
// Also "recon 4" (number AFTER the key).
|
||||
for w in toks.windows(2) {
|
||||
if keys.iter().any(|k| w[0].starts_with(k)) {
|
||||
if let Ok(n) = w[1].parse::<u64>() { return Some(n); }
|
||||
}
|
||||
}
|
||||
None
|
||||
}
|
||||
|
||||
/// Scan a line for @path tokens, attach each referenced file/dir to context.
|
||||
fn expand_ats(line: &str, s: &mut Session) -> usize {
|
||||
let mut n = 0;
|
||||
@@ -2177,4 +2249,21 @@ mod nl_tests {
|
||||
assert!(ids.iter().all(|i| i.contains(':')));
|
||||
assert!(ids.iter().any(|i| i.to_lowercase().contains("opus")));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn fast_parse_spoken_toggles() {
|
||||
let (it, _) = parse_intent_fast("testa loja.com com opus, usa burp, 5 votos, recon profundo, roda");
|
||||
assert_eq!(it.proxy.as_deref(), Some("http://127.0.0.1:8080"));
|
||||
assert_eq!(it.vote_n, Some(5));
|
||||
assert_eq!(it.recon, Some(3));
|
||||
assert!(it.run);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn fast_parse_stop_and_browser() {
|
||||
assert!(parse_intent_fast("para tudo agora").0.stop);
|
||||
assert_eq!(parse_intent_fast("usa o navegador").0.mcp, Some(true));
|
||||
assert_eq!(parse_intent_fast("recon 4 em example.com").0.recon, Some(4));
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@@ -56,6 +56,33 @@ fn env(name: &str) -> Option<String> {
|
||||
std::env::var(name).ok().filter(|v| !v.trim().is_empty())
|
||||
}
|
||||
|
||||
/// Severity as a numeric rank (Critical=4 … Info=0) for gate comparisons.
|
||||
pub fn severity_rank(s: &str) -> u8 {
|
||||
match s.trim().to_ascii_lowercase().as_str() {
|
||||
"critical" => 4, "high" => 3, "medium" => 2, "low" => 1, _ => 0,
|
||||
}
|
||||
}
|
||||
|
||||
/// The single worst confirmed severity across findings (skips needs-review), as
|
||||
/// a rank. 0 when nothing confirmed.
|
||||
pub fn worst_confirmed_rank(findings: &[Finding]) -> u8 {
|
||||
findings.iter()
|
||||
.filter(|f| f.review_status != "needs-review")
|
||||
.map(|f| severity_rank(&f.severity))
|
||||
.max().unwrap_or(0)
|
||||
}
|
||||
|
||||
/// Does any CONFIRMED finding meet/exceed `threshold` (a severity word)? This is
|
||||
/// the CI gate: true → the PR should be blocked. An unknown threshold disables
|
||||
/// the gate (returns false).
|
||||
pub fn gate_trips(findings: &[Finding], threshold: &str) -> bool {
|
||||
let t = severity_rank(threshold);
|
||||
if t == 0 && !matches!(threshold.trim().to_ascii_lowercase().as_str(), "low" | "info") {
|
||||
return false; // unknown threshold word → no gate
|
||||
}
|
||||
worst_confirmed_rank(findings) >= t
|
||||
}
|
||||
|
||||
fn client() -> reqwest::Client {
|
||||
reqwest::Client::builder()
|
||||
.timeout(std::time::Duration::from_secs(30))
|
||||
@@ -121,6 +148,63 @@ impl Integrations {
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Set a GitHub commit status (Checks-style) so branch protection can BLOCK a
|
||||
/// merge on a failing state. `state` ∈ success|failure|error|pending. `context`
|
||||
/// names the check (e.g. "neurosploit/security"). Requires a token with
|
||||
/// `repo:status` (or `statuses:write` on fine-grained PATs).
|
||||
pub async fn github_set_status(&self, repo: &str, sha: &str, state: &str,
|
||||
context: &str, description: &str, target_url: Option<&str>) -> Result<()> {
|
||||
let tok = self.github_token().ok_or_else(|| anyhow!("{} not set", self.github.token_env))?;
|
||||
let url = format!("{}/repos/{}/statuses/{}", self.github.api.trim_end_matches('/'), repo, sha);
|
||||
// GitHub caps status description at 140 chars.
|
||||
let desc: String = description.chars().take(140).collect();
|
||||
let mut body = serde_json::json!({ "state": state, "context": context, "description": desc });
|
||||
if let Some(u) = target_url { body["target_url"] = serde_json::json!(u); }
|
||||
let resp = client().post(&url)
|
||||
.header("User-Agent", "NeuroSploit")
|
||||
.header("Accept", "application/vnd.github+json")
|
||||
.bearer_auth(tok)
|
||||
.json(&body)
|
||||
.send().await?;
|
||||
if !resp.status().is_success() {
|
||||
return Err(anyhow!("github status failed: {} {}", resp.status(), resp.text().await.unwrap_or_default()));
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Submit a PR review. `event` ∈ APPROVE | REQUEST_CHANGES | COMMENT. Used to
|
||||
/// REQUEST_CHANGES when critical/high findings land — combined with a "require
|
||||
/// review" branch rule, this blocks the merge until a human overrides.
|
||||
pub async fn github_pr_review(&self, repo: &str, number: u64, event: &str, body: &str) -> Result<()> {
|
||||
let tok = self.github_token().ok_or_else(|| anyhow!("{} not set", self.github.token_env))?;
|
||||
let url = format!("{}/repos/{}/pulls/{}/reviews", self.github.api.trim_end_matches('/'), repo, number);
|
||||
let resp = client().post(&url)
|
||||
.header("User-Agent", "NeuroSploit")
|
||||
.header("Accept", "application/vnd.github+json")
|
||||
.bearer_auth(tok)
|
||||
.json(&serde_json::json!({ "event": event, "body": body }))
|
||||
.send().await?;
|
||||
if !resp.status().is_success() {
|
||||
return Err(anyhow!("github review failed: {} {}", resp.status(), resp.text().await.unwrap_or_default()));
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Head commit SHA of a PR (needed to attach a commit status to the PR tip).
|
||||
pub async fn github_pr_head_sha(&self, repo: &str, number: u64) -> Result<String> {
|
||||
let url = format!("{}/repos/{}/pulls/{}", self.github.api.trim_end_matches('/'), repo, number);
|
||||
let mut req = client().get(&url)
|
||||
.header("User-Agent", "NeuroSploit")
|
||||
.header("Accept", "application/vnd.github+json");
|
||||
if let Some(t) = self.github_token() { req = req.bearer_auth(t); }
|
||||
let resp = req.send().await?;
|
||||
if !resp.status().is_success() {
|
||||
return Err(anyhow!("github PR API {}: {}", resp.status(), resp.text().await.unwrap_or_default()));
|
||||
}
|
||||
let v: serde_json::Value = resp.json().await?;
|
||||
v["head"]["sha"].as_str().map(|s| s.to_string()).ok_or_else(|| anyhow!("no head.sha in response"))
|
||||
}
|
||||
|
||||
/// Latest commit SHA of a branch via the GitHub API (for `watch`).
|
||||
pub async fn github_latest_sha(&self, repo: &str, branch: &str) -> Result<String> {
|
||||
let url = format!("{}/repos/{}/commits/{}", self.github.api.trim_end_matches('/'), repo, branch);
|
||||
@@ -197,3 +281,34 @@ impl Integrations {
|
||||
]
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod gate_tests {
|
||||
use super::*;
|
||||
use crate::types::Finding;
|
||||
|
||||
fn f(sev: &str, status: &str) -> Finding {
|
||||
Finding { severity: sev.into(), review_status: status.into(), ..Default::default() }
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn gate_blocks_on_threshold_and_above() {
|
||||
let fs = vec![f("High", "confirmed"), f("Low", "confirmed")];
|
||||
assert!(gate_trips(&fs, "high"));
|
||||
assert!(gate_trips(&fs, "medium"));
|
||||
assert!(!gate_trips(&fs, "critical"));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn gate_ignores_needs_review() {
|
||||
let fs = vec![f("Critical", "needs-review")];
|
||||
assert!(!gate_trips(&fs, "critical"));
|
||||
assert_eq!(worst_confirmed_rank(&fs), 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn unknown_threshold_disables_gate() {
|
||||
let fs = vec![f("Critical", "confirmed")];
|
||||
assert!(!gate_trips(&fs, "banana"));
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user