add support in check-compliance command

Signed-off-by: NucleonGodX <[email protected]>

Author: NucleonGodX

scanpipe/management/commands/check-compliance.py

@@ -56,20 +56,40 @@ def handle(self, *args, **options):
             for model_alerts in compliance_alerts.values()
             for issues_by_severity in model_alerts.values()
         )
-        if not compliance_alerts_count:
+
+        extra_data = self.project.extra_data or {}
+        clarity_alert = extra_data.get("clarity_compliance_alert")
+
+        severity_map = {"ok": 0, "warning": 1, "error": 2}
+        fail_level_map = {"MISSING": 0, "WARNING": 1, "ERROR": 2}
+
+        clarity_alert_severity = severity_map.get(clarity_alert, 0)
+        fail_level_severity = fail_level_map.get(fail_level.upper(), 2)
+
+        clarity_issue_count = (
+            1
+            if clarity_alert_severity >= fail_level_severity and clarity_alert != "ok"
+            else 0
+        )
+
+        total_issues = compliance_alerts_count + clarity_issue_count
+
+        if total_issues == 0:
             sys.exit(0)
 
         if self.verbosity > 0:
-            msg = [
-                f"{compliance_alerts_count} compliance issues detected on this project."
-            ]
+            msg = [f"{total_issues} compliance issues detected on this project."]
             for label, issues in compliance_alerts.items():
                 msg.append(f"[{label}]")
                 for severity, entries in issues.items():
                     msg.append(f" > {severity.upper()}: {len(entries)}")
                     if self.verbosity > 1:
                         msg.append("   " + "\n   ".join(entries))
 
+            if clarity_issue_count:
+                msg.append("[License Clarity Compliance]")
+                msg.append(f" > Alert Level: {clarity_alert}")
+
             self.stderr.write("\n".join(msg))
 
         sys.exit(1)

scanpipe/tests/test_commands.py

@@ -1187,6 +1187,48 @@ def test_scanpipe_management_command_check_compliance(self):
         )
         self.assertEqual(expected, out_value)
 
+    def test_scanpipe_management_command_check_clarity_compliance_only(self):
+        project = Project.objects.create(name="my_project_clarity")
+
+        project.extra_data = {"clarity_compliance_alert": "error"}
+        project.save(update_fields=["extra_data"])
+
+        out = StringIO()
+        options = ["--project", project.name]
+        with self.assertRaises(SystemExit) as cm:
+            call_command("check-compliance", *options, stderr=out)
+        self.assertEqual(cm.exception.code, 1)
+        out_value = out.getvalue().strip()
+        expected = (
+            "1 compliance issues detected on this project."
+            "\n[License Clarity Compliance]\n > Alert Level: error"
+        )
+        self.assertEqual(expected, out_value)
+
+    def test_scanpipe_management_command_check_both_compliance_and_clarity(self):
+        project = Project.objects.create(name="my_project_both")
+
+        make_package(
+            project,
+            package_url="pkg:generic/[email protected]",
+            compliance_alert=CodebaseResource.Compliance.ERROR,
+        )
+        project.extra_data = {"clarity_compliance_alert": "warning"}
+        project.save(update_fields=["extra_data"])
+
+        out = StringIO()
+        options = ["--project", project.name, "--fail-level", "WARNING"]
+        with self.assertRaises(SystemExit) as cm:
+            call_command("check-compliance", *options, stderr=out)
+        self.assertEqual(cm.exception.code, 1)
+        out_value = out.getvalue().strip()
+        expected = (
+            "2 compliance issues detected on this project."
+            "\n[packages]\n > ERROR: 1"
+            "\n[License Clarity Compliance]\n > Alert Level: warning"
+        )
+        self.assertEqual(expected, out_value)
+
     def test_scanpipe_management_command_report(self):
         project1 = make_project("project1")
         label1 = "label1"