Update

app/Jobs/GenerateMemeJob.php

@@ -2,8 +2,10 @@
 
 namespace App\Jobs;
 
+use App\Helpers\FirstParty\Credits\CreditsService;
 use App\Helpers\FirstParty\Meme\MemeGenerator;
 use App\Models\User;
+use App\Models\UserMemeGeneration;
 use Illuminate\Bus\Queueable;
 use Illuminate\Contracts\Queue\ShouldQueue;
 use Illuminate\Foundation\Bus\Dispatchable;
@@ -23,7 +25,7 @@ class GenerateMemeJob implements ShouldQueue
     protected $prompt;
     protected $jobId;
 
-    public function __construct(int $userId, string $prompt, string $jobId = null)
+    public function __construct(int $userId, string $prompt, ?string $jobId)
     {
         $this->userId = $userId;
         $this->prompt = $prompt;
@@ -33,12 +35,27 @@ public function __construct(int $userId, string $prompt, string $jobId = null)
 
     public function handle(): void
     {
+        $userGeneration = UserMemeGeneration::where('job_id', $this->jobId)->first();
+
+        if (!$userGeneration) {
+            throw new \Exception("User generation record not found for job {$this->jobId}");
+        }
+
         try {
+            // Update status to processing
+            $userGeneration->update(['status' => 'processing']);
             Cache::put("meme_job_status_{$this->jobId}", 'processing', 300);
 
             $meme = MemeGenerator::generateMemeByKeyword($this->prompt);
             $meme_media = MemeGenerator::getSuitableMemeMedia($meme, 2);
 
+            // Update the generation record with the meme
+            $userGeneration->update([
+                'meme_id' => $meme->id,
+                'status' => 'completed',
+                'credits_are_processed' => true,
+            ]);
+
             $result = [
                 'generate' => [
                     'info' => $meme,
@@ -50,15 +67,61 @@ public function handle(): void
 
             Cache::put("meme_job_result_{$this->jobId}", $result, 300);
             Cache::put("meme_job_status_{$this->jobId}", 'completed', 300);
+
+            // Clear active job from cache
+            Cache::forget("user_active_job_{$this->userId}");
         } catch (\Exception $e) {
+            // Handle failure with credit refund
+            if (!$userGeneration->credits_are_processed) {
+                if ($userGeneration->credits_to_be_charged > 0) {
+                    CreditsService::depositAlacarte(
+                        $userGeneration->user_id,
+                        $userGeneration->credits_to_be_charged,
+                        'Refunded credits for failed generation due to provider issue.'
+                    );
+                }
+
+                $userGeneration->update([
+                    'credits_are_processed' => true,
+                    'status' => 'failed',
+                ]);
+            }
+
             Cache::put("meme_job_status_{$this->jobId}", 'failed', 300);
             Cache::put("meme_job_error_{$this->jobId}", $e->getMessage(), 300);
+
+            // Clear active job from cache
+            Cache::forget("user_active_job_{$this->userId}");
+
             throw $e;
         }
     }
 
+    public function failed(\Throwable $exception): void
+    {
+        $userGeneration = UserMemeGeneration::where('job_id', $this->jobId)->first();
+
+        if ($userGeneration && !$userGeneration->credits_are_processed) {
+            if ($userGeneration->credits_to_be_charged > 0) {
+                CreditsService::depositAlacarte(
+                    $userGeneration->user_id,
+                    $userGeneration->credits_to_be_charged,
+                    'Refunded credits for failed generation due to system error.'
+                );
+            }
+
+            $userGeneration->update([
+                'credits_are_processed' => true,
+                'status' => 'failed',
+            ]);
+        }
+
+        Cache::put("meme_job_status_{$this->jobId}", 'failed', 300);
+        Cache::forget("user_active_job_{$this->userId}");
+    }
+
     public function getJobId(): string
     {
         return $this->jobId;
     }
-}
+}