From fcd8efa599fbad2f3b7703802310a72e3ac3c2c2 Mon Sep 17 00:00:00 2001
From: Pablo Pagani <79593935+pablopagani@users.noreply.github.com>
Date: Sun, 3 Nov 2024 15:51:54 +0100
Subject: [PATCH 1/2] add export-csv option

---
 hardeneks/__init__.py | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/hardeneks/__init__.py b/hardeneks/__init__.py
index 90839f3..be031a1 100644
--- a/hardeneks/__init__.py
+++ b/hardeneks/__init__.py
@@ -5,6 +5,7 @@
 import yaml
 import json
 from collections import defaultdict
+import csv
 
 from botocore.exceptions import EndpointConnectionError
 import boto3
@@ -101,6 +102,27 @@ def ndd():
     with open(json_path, "w", encoding="utf-8") as f:
         json.dump(json_blob, f, ensure_ascii=False, indent=4)
 
+def _export_csv(rules: list, csv_path=str):
+    csv_data = []
+
+    for rule in rules:
+        csv_row = {
+           "Type": rule._type,
+            "Pillar": rule.pillar,
+            "Section": rule.section,
+            "Message": rule.message,
+            "Status": rule.result.status,
+            "Resources": ', '.join(rule.result.resources) if rule.result.resources else '',
+            "Resource Type": rule.result.resource_type,
+            "Namespace": rule.result.namespace,
+            "Resolution": rule.url,
+        }
+        csv_data.append(csv_row)
+
+    with open(csv_path, "w", encoding="utf-8") as f:
+        writer = csv.DictWriter(f, fieldnames=csv_data[0].keys())
+        writer.writeheader()
+        writer.writerows(csv_data)
 
 def print_consolidated_results(rules: list):
 
@@ -159,6 +181,10 @@ def run_hardeneks(
         default=None,
         help="Export the report in txt format",
     ),
+    export_csv: str = typer.Option(
+        default=None,
+        help="Export the report in csv format",
+    ),
     export_html: str = typer.Option(
         default=None,
         help="Export the report in html format",
@@ -188,6 +214,7 @@ def run_hardeneks(
         namespace (str): Specific namespace to be checked
         config (str): Path to hardeneks config file
         export-txt (str): Export the report in txt format
+        export-csv (str): Export the report in csv format
         export-html (str): Export the report in html format
         export-json (str): Export the report in json format
         insecure-skip-tls-verify (str): Skip tls verification
@@ -254,6 +281,8 @@ def run_hardeneks(
 
     if export_txt:
         console.save_text(export_txt)
+    if export_csv:
+        _export_csv(results, export_csv)
     if export_html:
         console.save_html(export_html)
     if export_json:

From 4639c209f6f3a392a4f83fa3dd111003caec582a Mon Sep 17 00:00:00 2001
From: Pablo Pagani <79593935+pablopagani@users.noreply.github.com>
Date: Sun, 3 Nov 2024 16:25:13 +0100
Subject: [PATCH 2/2] update export-csv

---
 README.md | 1 +
 1 file changed, 1 insertion(+)

diff --git a/README.md b/README.md
index 05ade55..546de6b 100644
--- a/README.md
+++ b/README.md
@@ -33,6 +33,7 @@ hardeneks [OPTIONS]
 * `--namespace TEXT`: Namespace to be checked (default is all namespaces)
 * `--config TEXT`: Path to a hardeneks config file
 * `--export-txt TEXT`: Export the report in txt format
+* `--export-csv TEXT`: Export the report in csv format
 * `--export-html TEXT`: Export the report in html format
 * `--export-json TEXT`: Export the report in json format
 * `--insecure-skip-tls-verify`: Skip TLS verification