From a11e1ee3f816bd27fc199570cd0abcacadf51c5d Mon Sep 17 00:00:00 2001
From: Daniel Supernault <danielsupernault@gmail.com>
Date: Wed, 17 May 2023 04:44:35 -0600
Subject: [PATCH] Update AutospamUpdateCachedDataPipeline

---
 .../AutospamUpdateCachedDataPipeline.php      | 40 +++++++++++--------
 1 file changed, 23 insertions(+), 17 deletions(-)

diff --git a/app/Jobs/AutospamPipeline/AutospamUpdateCachedDataPipeline.php b/app/Jobs/AutospamPipeline/AutospamUpdateCachedDataPipeline.php
index c836c79a2..4ac025bb8 100644
--- a/app/Jobs/AutospamPipeline/AutospamUpdateCachedDataPipeline.php
+++ b/app/Jobs/AutospamPipeline/AutospamUpdateCachedDataPipeline.php
@@ -52,9 +52,11 @@ class AutospamUpdateCachedDataPipeline implements ShouldQueue
 			}
 		}
 		$newSpamCount = count($spam['words']['spam']);
-		$spam['documents']['spam'] = $newSpamCount;
-		arsort($spam['words']['spam']);
-		Storage::put(AutospamService::MODEL_SPAM_PATH, json_encode($spam, JSON_UNESCAPED_SLASHES|JSON_PRETTY_PRINT));
+		if($newSpamCount) {
+			$spam['documents']['spam'] = $newSpamCount;
+			arsort($spam['words']['spam']);
+			Storage::put(AutospamService::MODEL_SPAM_PATH, json_encode($spam, JSON_UNESCAPED_SLASHES|JSON_PRETTY_PRINT));
+		}
 
 		$hamExists = Storage::exists(AutospamService::MODEL_HAM_PATH);
 		if($hamExists) {
@@ -80,23 +82,27 @@ class AutospamUpdateCachedDataPipeline implements ShouldQueue
 		}
 
 		$newHamCount = count($ham['words']['ham']);
-		$ham['documents']['ham'] = $newHamCount;
-		arsort($ham['words']['ham']);
+		if($newHamCount) {
+			$ham['documents']['ham'] = $newHamCount;
+			arsort($ham['words']['ham']);
+			Storage::put(AutospamService::MODEL_HAM_PATH, json_encode($ham, JSON_UNESCAPED_SLASHES|JSON_PRETTY_PRINT));
+		}
 
-		Storage::put(AutospamService::MODEL_HAM_PATH, json_encode($ham, JSON_UNESCAPED_SLASHES|JSON_PRETTY_PRINT));
+		if($newSpamCount && $newHamCount) {
+			$combined = [
+				'documents' => [
+					'spam' => $newSpamCount,
+					'ham' => $newHamCount,
+				],
+				'words' => [
+					'spam' => $spam['words']['spam'],
+					'ham' => $ham['words']['ham']
+				]
+			];
 
-		$combined = [
-			'documents' => [
-				'spam' => $newSpamCount,
-				'ham' => $newHamCount,
-			],
-			'words' => [
-				'spam' => $spam['words']['spam'],
-				'ham' => $ham['words']['ham']
-			]
-		];
+			Storage::put(AutospamService::MODEL_FILE_PATH, json_encode($combined, JSON_PRETTY_PRINT,JSON_UNESCAPED_SLASHES));
+		}
 
-		Storage::put(AutospamService::MODEL_FILE_PATH, json_encode($combined, JSON_PRETTY_PRINT,JSON_UNESCAPED_SLASHES));
 		Cache::forget(AutospamService::MODEL_CACHE_KEY);
 		Cache::forget(AutospamService::CHCKD_CACHE_KEY);
 	}