From 7566f5c1677f06c988413fa7f9567fb67744324b Mon Sep 17 00:00:00 2001
From: Myles McNamara <myles@smyl.es>
Date: Tue, 8 Jul 2025 13:48:29 -0400
Subject: [PATCH 1/2] fix Estimator has not been trained fatal error

---
 src/Nunil_Manipulate_DOM.php | 37 ++++++++++++++++++++----------------
 1 file changed, 21 insertions(+), 16 deletions(-)

diff --git a/src/Nunil_Manipulate_DOM.php b/src/Nunil_Manipulate_DOM.php
index 6ededb7..5de9f55 100755
--- a/src/Nunil_Manipulate_DOM.php
+++ b/src/Nunil_Manipulate_DOM.php
@@ -986,24 +986,29 @@ private function check_cluster_whitelist( $lsh_hex_digest, $tagname, $event = nu
 		$samples[] = $lsh_hex_digest;
 		$dataset   = new Unlabeled( $samples );
 
-		if ( is_null( $event ) && 'script' === $tagname && isset( $this->inline_scripts_classifier ) ) {
-			$predicted_labels = $this->inline_scripts_classifier->predict( $dataset );
-			$predicted_label  = $predicted_labels[0];
-			$list             = $this->inline_rows;
-		} elseif ( is_null( $event ) && 'script' !== $tagname && isset( $this->internal_css_classifier ) ) {
-			$predicted_labels = $this->internal_css_classifier->predict( $dataset );
-			$predicted_label  = $predicted_labels[0];
-			$list             = $this->inline_rows;
-		} elseif ( null !== $event && isset( $this->event_handlers_classifier ) ) {
-			$predicted_labels         = $this->event_handlers_classifier->predict( $dataset );
-			$combined_predicted_label = $predicted_labels[0];
-			$label_parts              = explode( '#', $combined_predicted_label, 2 );
-			$my_evh                   = $label_parts[0];
-			$predicted_label          = $label_parts[1];
-			$list                     = $this->events_rows;
-		} else {
+		try {
+			if ( is_null( $event ) && 'script' === $tagname && isset( $this->inline_scripts_classifier ) ) {
+				$predicted_labels = $this->inline_scripts_classifier->predict( $dataset );
+				$predicted_label  = $predicted_labels[0];
+				$list             = $this->inline_rows;
+			} elseif ( is_null( $event ) && 'script' !== $tagname && isset( $this->internal_css_classifier ) ) {
+				$predicted_labels = $this->internal_css_classifier->predict( $dataset );
+				$predicted_label  = $predicted_labels[0];
+				$list             = $this->inline_rows;
+			} elseif ( null !== $event && isset( $this->event_handlers_classifier ) ) {
+				$predicted_labels         = $this->event_handlers_classifier->predict( $dataset );
+				$combined_predicted_label = $predicted_labels[0];
+				$label_parts              = explode( '#', $combined_predicted_label, 2 );
+				$my_evh                   = $label_parts[0];
+				$predicted_label          = $label_parts[1];
+				$list                     = $this->events_rows;
+			} else {
+				return false;
+			}
+		} catch ( \Rubix\ML\Exceptions\RuntimeException $e ) {
 			return false;
 		}
+
 		if ( is_array( $list ) ) {
 			if ( 'Unclustered' !== $predicted_label ) {
 				if ( null === $event ) {

From ca5cd5d8691a281583e3a15f1d410d06a0e555cb Mon Sep 17 00:00:00 2001
From: Myles McNamara <myles@smyl.es>
Date: Thu, 10 Jul 2025 14:46:26 -0400
Subject: [PATCH 2/2] add nunil_output_csp_headers_header_csp filter for
 $header_csp

---
 public/class-no-unsafe-inline-public.php | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/public/class-no-unsafe-inline-public.php b/public/class-no-unsafe-inline-public.php
index 739f38c..7bf02dd 100755
--- a/public/class-no-unsafe-inline-public.php
+++ b/public/class-no-unsafe-inline-public.php
@@ -293,6 +293,8 @@ public function output_csp_headers( $htmlsource ) {
 						}
 					}
 
+					$header_csp = apply_filters( 'nunil_output_csp_headers_header_csp', $header_csp );
+
 					$report_uri                 = '';
 					$header_report_to           = '';
 					$report_to                  = '';