databox: removebg fix cache on error

alchemy-fr · Nov 18, 2024 · 51edc61 · 51edc61
1 parent 5028a63
commit 51edc61
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/databox/api/src/Integration/RemoveBg/RemoveBgClient.php b/databox/api/src/Integration/RemoveBg/RemoveBgClient.php
@@ -24,7 +24,7 @@ public function getBgRemoved(File $file, string $apiKey): string
         }
 
         $path = $this->fileFetcher->getFile($file);
-        $md5 = md5_file($path);
+        $md5 = md5_file($path).'-v2';
 
         $cacheFile = sprintf('%s/%s', $this->cacheDir, $md5);
         if (file_exists($cacheFile)) {
@@ -41,11 +41,17 @@ public function getBgRemoved(File $file, string $apiKey): string
             ],
         ]);
 
-        $fileHandler = fopen($cacheFile, 'w');
-        foreach ($this->removeBgClient->stream($res) as $chunk) {
-            fwrite($fileHandler, $chunk->getContent());
+        try {
+            $fileHandler = fopen($cacheFile, 'w');
+            foreach ($this->removeBgClient->stream($res) as $chunk) {
+                fwrite($fileHandler, $chunk->getContent());
+            }
+            fclose($fileHandler);
+        } catch (\Throwable $e) {
+            fclose($fileHandler);
+            @unlink($cacheFile);
+            throw $e;
         }
-        fclose($fileHandler);
 
         return $cacheFile;
     }