fix(code): correct severity counting

- fix incorrect severity counts - removed spacing around arrows
instriq · Sep 10, 2024 · 59ac038 · 59ac038
1 parent ea8dd96
commit 59ac038
Showing 1 changed file with 43 additions and 43 deletions.
diff --git a/lib/SecurityGate/Engine/Code.pm b/lib/SecurityGate/Engine/Code.pm
@@ -5,55 +5,55 @@ package SecurityGate::Engine::Code {
     use Mojo::JSON;
 
     sub new {
-      my ($class, $token, $repository, $severity_limits) = @_;
-      my $alerts_endpoint   = "https://api.github.com/repos/$repository/code-scanning/alerts";      
-
-      my $userAgent = Mojo::UserAgent -> new();
-      my $alerts_request = $userAgent -> get($alerts_endpoint, {Authorization => "Bearer $token"}) -> result();
-
-      if ($alerts_request -> code() == 200) {
-        my $alerts_data = $alerts_request -> json();
-        my $open_alerts = 0;
-        my %severity_counts = map {$_ => 0} keys %$severity_limits;
-
-        foreach my $alert (@$alerts_data) {
-          if ($alert -> {state} eq "open") {
-            $open_alerts++;
-
-            my $severity = $alert -> {rule} -> {severity};
-            $severity_counts{$severity}++ if exists $severity_counts{$severity};
-          }
-        }
+        my ($class, $token, $repository, $severity_limits) = @_;
+        my $alerts_endpoint = "https://api.github.com/repos/$repository/code-scanning/alerts";
 
-        print "[!] Total of open code scanning alerts: $open_alerts\n";
-
-        foreach my $severity (keys %severity_counts) {
-          print "[-] $severity: $severity_counts{$severity}\n";
-        }
+        my $userAgent = Mojo::UserAgent->new();
+        my $alerts_request = $userAgent->get($alerts_endpoint, {Authorization => "Bearer $token"})->result();
 
-        my $threshold_exceeded = 0;
-
-        foreach my $severity (keys %severity_counts) {
-          if ($severity_counts{$severity} > $severity_limits -> {$severity}) {
-            print "[+] More than $severity_limits->{$severity} $severity code scanning alerts found.\n";
-
-            $threshold_exceeded = 1;
-          }
-        }
+        if ($alerts_request->code() == 200) {
+            my $alerts_data = $alerts_request->json();
+            my $open_alerts = 0;
+            my %severity_counts = map {$_ => 0} keys %$severity_limits;
+
+            foreach my $alert (@$alerts_data) {
+                if ($alert->{state} eq "open") {
+                    $open_alerts++;
+
+                    my $severity = $alert->{rule}->{security_severity_level} // 'unknown';
+                    $severity_counts{$severity}++ if exists $severity_counts{$severity};
+                }
+            }
+
+            print "\n[!] Total of open code scanning alerts: $open_alerts\n\n";
 
-        if ($threshold_exceeded) {
-          return 1;
+            foreach my $severity (keys %severity_counts) {
+                print "[-] $severity: $severity_counts{$severity}\n";
+            }
+
+            print "\n";
+
+            my $threshold_exceeded = 0;
+
+            foreach my $severity (keys %severity_counts) {
+                if ($severity_counts{$severity} > $severity_limits->{$severity}) {
+                    print "[+] More than $severity_limits->{$severity} $severity code scanning alerts found.\n";
+                    $threshold_exceeded = 1;
+                }
+            }
+
+            if ($threshold_exceeded) {
+                return 1;
+            }
         }
-      }
 
-      else {
-        print "Error: Unable to fetch code scanning alerts. HTTP status code: " . $alerts_request -> code() . "\n";
-
-        return 1;
-      }
+        else {
+            print "Error: Unable to fetch code scanning alerts. HTTP status code: " . $alerts_request->code() . "\n";
+            return 1;
+        }
 
-      return 0;
+        return 0;
     }
 }
 
-1;
+1;