fix(html): fix error html generator (#1530)

Co-authored-by: sergargar <sergio@verica.io>
Co-authored-by: Toni de la Fuente <toni@blyx.com>

lib/check/check.py

@@ -140,25 +140,26 @@ def print_compliance_frameworks(
 
 
 def print_compliance_requirements(
-    bulk_compliance_frameworks: dict, compliance_framework: str
+    bulk_compliance_frameworks: dict, compliance_frameworks: list
 ):
-    for compliance in bulk_compliance_frameworks.values():
-        # Workaround until we have more Compliance Frameworks
-        split_compliance = compliance_framework.split("_")
-        framework = split_compliance[0].upper()
-        version = split_compliance[1].upper()
-        provider = split_compliance[2].upper()
-        if compliance.Framework == framework and compliance.Version == version:
-            print(
-                f"Listing {framework} {version} {provider} Compliance Requirements:\n"
-            )
-            for requirement in compliance.Requirements:
-                checks = ""
-                for check in requirement.Checks:
-                    checks += f" {Fore.YELLOW}\t\t{check}\n{Style.RESET_ALL}"
+    for compliance_framework in compliance_frameworks:
+        for compliance in bulk_compliance_frameworks.values():
+            # Workaround until we have more Compliance Frameworks
+            split_compliance = compliance_framework.split("_")
+            framework = split_compliance[0].upper()
+            version = split_compliance[1].upper()
+            provider = split_compliance[2].upper()
+            if framework in compliance.Framework and compliance.Version == version:
                 print(
-                    f"Requirement Id: {Fore.MAGENTA}{requirement.Id}{Style.RESET_ALL}\n\t- Description: {requirement.Description}\n\t- Checks:\n{checks}"
+                    f"Listing {framework} {version} {provider} Compliance Requirements:\n"
                 )
+                for requirement in compliance.Requirements:
+                    checks = ""
+                    for check in requirement.Checks:
+                        checks += f" {Fore.YELLOW}\t\t{check}\n{Style.RESET_ALL}"
+                    print(
+                        f"Requirement Id: {Fore.MAGENTA}{requirement.Id}{Style.RESET_ALL}\n\t- Description: {requirement.Description}\n\t- Checks:\n{checks}"
+                    )
 
 
 def print_checks(

lib/outputs/outputs.py

@@ -16,7 +16,6 @@ from config.config import (
     json_asff_file_suffix,
     json_file_suffix,
     orange_color,
-    output_file_timestamp,
     prowler_version,
     timestamp,
     timestamp_iso,
@@ -304,7 +303,6 @@ def initialize_file_descriptor(
 
             if output_mode in ("json", "json-asff"):
                 file_descriptor.write("[")
-
             if "html" in output_mode:
                 add_html_header(file_descriptor, audit_info)
     except Exception as error:
@@ -463,34 +461,6 @@ def fill_json_asff(finding_output, audit_info, finding):
     return finding_output
 
 
-def fill_html(file_descriptor, audit_info, finding):
-    row_class = "p-3 mb-2 bg-success-custom"
-    if finding.status == "INFO":
-        row_class = "table-info"
-    elif finding.status == "FAIL":
-        row_class = "table-danger"
-    elif finding.status == "WARNING":
-        row_class = "table-warning"
-    file_descriptor.write(
-        f"""
-            <tr class="{row_class}">
-                <td>{finding.status}</td>
-                <td>{finding.check_metadata.Severity}</td>
-                <td>{audit_info.audited_account}</td>
-                <td>{finding.region}</td>
-                <td>{finding.check_metadata.ServiceName}</td>
-                <td>{finding.check_metadata.CheckID}</td>
-                <td>{finding.check_metadata.CheckTitle}</td>
-                <td>{finding.status_extended}</td>
-                <td><p class="show-read-more">{finding.check_metadata.Risk}</p></td>
-                <td><p class="show-read-more">{finding.check_metadata.Remediation.Recommendation.Text}</p></td>
-                <td><a class="read-more" href="{finding.check_metadata.Remediation.Recommendation.Url}"><i class="fas fa-external-link-alt"></i></a></td>
-                <td>{finding.resource_id}</td>
-            </tr>
-            """
-    )
-
-
 def close_json(output_filename, output_directory, mode):
     try:
         suffix = json_file_suffix
@@ -638,8 +608,10 @@ def display_summary_table(
                 print(f" - HTML: {output_directory}/{output_filename}.html")
             if "json-asff" in output_options.output_modes:
                 print(f" - JSON-ASFF: {output_directory}/{output_filename}.asff.json")
-            print(f" - CSV: {output_directory}/{output_filename}.csv")
-            print(f" - JSON: {output_directory}/{output_filename}.json")
+            if "csv" in output_options.output_modes:
+                print(f" - CSV: {output_directory}/{output_filename}.csv")
+            if "json" in output_options.output_modes:
+                print(f" - JSON: {output_directory}/{output_filename}.json")
 
         else:
             print(
@@ -887,8 +859,12 @@ def display_compliance_table(
 
 def add_html_header(file_descriptor, audit_info):
     try:
+        if not audit_info.profile:
+            audit_info.profile = "ENV"
         if isinstance(audit_info.audited_regions, list):
             audited_regions = " ".join(audit_info.audited_regions)
+        elif not audit_info.audited_regions:
+            audited_regions = "All Regions"
         else:
             audited_regions = audit_info.audited_regions
         file_descriptor.write(
@@ -901,7 +877,8 @@ def add_html_header(file_descriptor, audit_info):
     <meta name="viewport" content="width=device-width, initial-scale=1, shrink-to-fit=no">
     <style>
     .read-more {color:#00f;}
-    .bg-success-custom {background-color: #70dc88 !important;}
+    .bg-success-custom {background-color: #98dea7 !important;}
+    .bg-danger {background-color: #f28484 !important;}
     </style>
     <!-- Bootstrap CSS -->
     <link rel="stylesheet" href="https://stackpath.bootstrapcdn.com/bootstrap/4.5.0/css/bootstrap.min.css" integrity="sha384-9aIt2nRpC12Uk9gS9baDl411NQApFmC26EwAOH8WgZl5MYYxFfc+NcPb1dKGj7Sk" crossorigin="anonymous">
@@ -913,12 +890,9 @@ def add_html_header(file_descriptor, audit_info):
             display: none;
         }
     </style>
-    <title>Prowler - AWS Security Assessments</title>
+    <title>Prowler - The Handy Cloud Security Tool</title>
     </head>
     <body>
-    <nav class="navbar navbar-expand-xl sticky-top navbar-dark bg-dark">
-        <a class="navbar-brand" href="#">Prowler - Security Assessments in AWS</a>
-    </nav>
     <div class="container-fluid">
         <div class="row mt-3">
         <div class="col-md-4">
@@ -927,6 +901,14 @@ def add_html_header(file_descriptor, audit_info):
                 Report Information:
             </div>
             <ul class="list-group list-group-flush">
+             <li class="list-group-item text-center">
+            <a href="""
+            + html_logo_url
+            + """><img src="""
+            + html_logo_img
+            + """
+            alt="prowler-logo"></a>
+            </li>
                 <li class="list-group-item">
                 <div class="row">
                     <div class="col-md-auto">
@@ -943,17 +925,9 @@ def add_html_header(file_descriptor, audit_info):
                 </li>
                 <li class="list-group-item">
                 <b>Date:</b> """
-            + output_file_timestamp
+            + timestamp.isoformat()
             + """
                 </li>
-                <li class="list-group-item text-center">
-                <a href="""
-            + html_logo_url
-            + """><img src="""
-            + html_logo_img
-            + """
-                    alt="prowler-logo"></a>
-                </li>
             </ul>
             </div>
         </div>
@@ -998,16 +972,16 @@ def add_html_header(file_descriptor, audit_info):
                 <tr>
                 <th scope="col">Status</th>
                 <th scope="col">Severity</th>
-                <th scope="col">Account ID</th>
+                <th scope="col">Service Name</th>
                 <th scope="col">Region</th>
-                <th scope="col">Service</th>
-                <th scope="col">Check ID</th>
                 <th style="width:20%" scope="col">Check Title</th>
-                <th style="width:20%" scope="col">Check Output</th>
-                <th scope="col">Risk</th>
-                <th scope="col">Remediation</th>
-                <th scope="col">Related URL</th>
                 <th scope="col">Resource ID</th>
+                <th scope="col">Check Description</th>
+                <th scope="col">Check ID</th>
+                <th scope="col">Status Extended</th>
+                <th scope="col">Risk</th>
+                <th scope="col">Recomendation</th>
+                <th style="5% width" scope="col">Recomendation URL</th>
                 </tr>
             </thead>
             <tbody>
@@ -1019,6 +993,34 @@ def add_html_header(file_descriptor, audit_info):
         )
 
 
+def fill_html(file_descriptor, audit_info, finding):
+    row_class = "p-3 mb-2 bg-success-custom"
+    if finding.status == "INFO":
+        row_class = "table-info"
+    elif finding.status == "FAIL":
+        row_class = "table-danger"
+    elif finding.status == "WARNING":
+        row_class = "table-warning"
+    file_descriptor.write(
+        f"""
+            <tr class="{row_class}">
+                <td>{finding.status}</td>
+                <td>{finding.check_metadata.Severity}</td>
+                <td>{finding.check_metadata.ServiceName}</td>
+                <td>{finding.region}</td>
+                <td>{finding.check_metadata.CheckTitle}</td>
+                <td>{finding.resource_id}</td>
+                <td>{finding.check_metadata.Description}</td>
+                <td>{finding.check_metadata.CheckID}</td>
+                <td>{finding.status_extended}</td>
+                <td><p class="show-read-more">{finding.check_metadata.Risk}</p></td>
+                <td><p class="show-read-more">{finding.check_metadata.Remediation.Recommendation.Text}</p></td>
+                <td><a class="read-more" href="{finding.check_metadata.Remediation.Recommendation.Url}"><i class="fas fa-external-link-alt"></i></a></td>
+            </tr>
+            """
+    )
+
+
 def add_html_footer(output_filename, output_directory):
     try:
         filename = f"{output_directory}/{output_filename}{html_file_suffix}"
@@ -1045,6 +1047,8 @@ def add_html_footer(output_filename, output_directory):
     $(document).ready(function(){
       // Initialise the table with 50 rows, and some search/filtering panes
       $('#findingsTable').DataTable( {
+        lengthChange: true,
+        buttons: [ 'copy', 'excel', 'pdf' ],
         lengthMenu: [ [50, 100, -1], [50, 100, "All"] ],
         searchPanes: {
             cascadePanes: true,
@@ -1054,10 +1058,11 @@ def add_html_footer(output_filename, output_directory):
         columnDefs: [
           {
               searchPanes: {
-                  show: false
+                  show: true,
+                  pagingType: 'numbers',
+                  searching: true
               },
-              // Hide Compliance, Check ID (in favour of Check Title), CAF Epic, Risk, Remediation, Link
-              targets: [4, 6, 9, 10, 11, 12]
+              targets: [0, 1, 2, 3, 4]
           }
         ]
       });