Hide columns conditionally when applying severity filter

Author: airadier

dist/index.js

@@ -1,5 +1,5 @@
 import * as core from "@actions/core";
-import { FilterOptions, filterPackages, Package, Report, Rule } from "./report";
+import { FilterOptions, filterPackages, Package, Severity, SeverityNames, Report, Rule } from "./report";
 import { ActionInputs } from "./action";
 
 const EVALUATION: any = {
@@ -24,36 +24,72 @@ export async function generateSummary(opts: ActionInputs, data: Report, filters?
   await core.summary.write({ overwrite: true });
 }
 
-function addVulnTableToSummary(data: Report) {
-  // Fallback to empty object if undefined
-  const totalVuln = data.result.vulnTotalBySeverity ?? {};
-  const fixableVuln = data.result.fixableVulnTotalBySeverity ?? {};
+const SEVERITY_LABELS: Record<Severity, string> = {
+  critical: "🟣 Critical",
+  high: "🔴 High",
+  medium: "🟠 Medium",
+  low: "🟡 Low",
+  negligible: "⚪ Negligible"
+};
+
+function countVulnsBySeverity(
+  packages: Package[],
+  severities: Severity[]
+): {
+  total: Record<Severity, number>;
+  fixable: Record<Severity, number>;
+} {
+  const result = {
+    total: { critical: 0, high: 0, medium: 0, low: 0, negligible: 0 },
+    fixable: { critical: 0, high: 0, medium: 0, low: 0, negligible: 0 }
+  };
+
+  for (const pkg of packages) {
+    for (const vuln of pkg.vulns ?? []) {
+      const sev = (vuln.severity.value as Severity).toLowerCase() as Severity;
+      if (severities.includes(sev)) {
+        result.total[sev]++;
+        if (vuln.fixedInVersion || pkg.suggestedFix) {
+          result.fixable[sev]++;
+        }
+      }
+    }
+  }
+  return result;
+}
+
+function getVisibleSeverities(minSeverity?: Severity): Severity[] {
+  const SEVERITY_ORDER: Severity[] = ["critical", "high", "medium", "low", "negligible"];
+  if (!minSeverity || minSeverity === "negligible") {
+    return SEVERITY_ORDER;
+  }
+  const minIndex = SEVERITY_ORDER.indexOf(minSeverity);
+  return minIndex >= 0 ? SEVERITY_ORDER.slice(0, minIndex + 1) : SEVERITY_ORDER;
+}
+
+// Updated summary function
+function addVulnTableToSummary(
+  data: Report,
+  minSeverity?: Severity
+) {
+  const pkgs = data.result.packages;
+  const visibleSeverities = getVisibleSeverities(minSeverity);
+
+  const totalVulns = countVulnsBySeverity(pkgs, visibleSeverities);
 
   core.summary.addHeading(`Vulnerabilities summary`, 2);
   core.summary.addTable([
     [
       { data: '', header: true },
-      { data: '🟣 Critical', header: true },
-      { data: '🔴 High', header: true },
-      { data: '🟠 Medium', header: true },
-      { data: '🟡 Low', header: true },
-      { data: '⚪ Negligible', header: true }
+      ...visibleSeverities.map(s => ({ data: SEVERITY_LABELS[s], header: true }))
     ],
     [
       { data: '⚠️ Total Vulnerabilities', header: true },
-      `${totalVuln.critical ?? 0}`,
-      `${totalVuln.high ?? 0}`,
-      `${totalVuln.medium ?? 0}`,
-      `${totalVuln.low ?? 0}`,
-      `${totalVuln.negligible ?? 0}`
+      ...visibleSeverities.map(s => `${totalVulns.total[s] ?? 0}`)
     ],
     [
       { data: '🔧 Fixable Vulnerabilities', header: true },
-      `${fixableVuln.critical ?? 0}`,
-      `${fixableVuln.high ?? 0}`,
-      `${fixableVuln.medium ?? 0}`,
-      `${fixableVuln.low ?? 0}`,
-      `${fixableVuln.negligible ?? 0}`
+      ...visibleSeverities.map(s => `${totalVulns.fixable[s] ?? 0}`)
     ],
   ]);
 }
@@ -166,7 +202,7 @@ function addReportToSummary(data: Report) {
         core.summary.addHeading(`Rule Bundle: ${bundle.name}`, 4)
 
         bundle.rules.forEach(rule => {
-          core.summary.addHeading(`${EVALUATION[rule.evaluationResult]} Rule: ${rule.description}`, 5)
+          core.summary.addHeading(`Rule: ${rule.description}`, 5)
 
           if (rule.evaluationResult != "passed") {
             if (rule.failureType == "pkgVulnFailure") {