From 08db3fa3c488b10030093e59b2ed97b6b943d558 Mon Sep 17 00:00:00 2001
From: Richie McIlroy <33632126+richiemcilroy@users.noreply.github.com>
Date: Sat, 23 Nov 2024 20:55:26 +0000
Subject: [PATCH] feat: screenshot route fix

---
 .github/workflows/debug.yml          | 4 +++-
 .github/workflows/main.yml           | 2 ++
 .github/workflows/publish.yml        | 4 +++-
 apps/desktop/src-tauri/src/upload.rs | 6 ++++--
 apps/web/app/api/screenshot/route.ts | 5 ++---
 5 files changed, 14 insertions(+), 7 deletions(-)

diff --git a/.github/workflows/debug.yml b/.github/workflows/debug.yml
index 513746a64..c626b20ce 100644
--- a/.github/workflows/debug.yml
+++ b/.github/workflows/debug.yml
@@ -57,7 +57,9 @@ jobs:
           echo "NEXT_PUBLIC_URL=${{ secrets.NEXT_PUBLIC_URL }}" >> .env
           echo 'NEXTAUTH_URL=${NEXT_PUBLIC_URL}' >> .env
           echo 'VITE_SERVER_URL=${NEXT_PUBLIC_URL}' >> .env
-
+          echo "NEXT_PUBLIC_CAP_AWS_REGION=${{ secrets.NEXT_PUBLIC_CAP_AWS_REGION }}" >> .env
+          echo "NEXT_PUBLIC_CAP_AWS_BUCKET=${{ secrets.NEXT_PUBLIC_CAP_AWS_BUCKET }}" >> .env
+          
       - name: Copy .env to apps/desktop
         run: cp .env apps/desktop/.env
 
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index 5d70f3b2e..ae77b50a9 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -51,6 +51,8 @@ jobs:
           echo "NEXT_PUBLIC_ENVIRONMENT=production" >> .env
           echo "NEXT_PUBLIC_LOCAL_MODE=false" >> .env
           echo "CAP_DESKTOP_SENTRY_URL=https://6a3b6a09e6ae976c2ad6fff710e88748@o4506859771527168.ingest.us.sentry.io/4508330917101568" >> .env
+          echo "NEXT_PUBLIC_CAP_AWS_REGION=${{ secrets.NEXT_PUBLIC_CAP_AWS_REGION }}" >> .env
+          echo "NEXT_PUBLIC_CAP_AWS_BUCKET=${{ secrets.NEXT_PUBLIC_CAP_AWS_BUCKET }}" >> .env
 
       - name: Copy .env to apps/desktop
         run: cp .env apps/desktop/.env
diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
index 70896056d..ac3c29a89 100644
--- a/.github/workflows/publish.yml
+++ b/.github/workflows/publish.yml
@@ -132,7 +132,9 @@ jobs:
           echo "NEXT_PUBLIC_URL=${{ secrets.NEXT_PUBLIC_URL }}" >> .env
           echo 'NEXTAUTH_URL=${NEXT_PUBLIC_URL}' >> .env
           echo 'VITE_SERVER_URL=${NEXT_PUBLIC_URL}' >> .env
-
+          echo "NEXT_PUBLIC_CAP_AWS_REGION=${{ secrets.NEXT_PUBLIC_CAP_AWS_REGION }}" >> .env
+          echo "NEXT_PUBLIC_CAP_AWS_BUCKET=${{ secrets.NEXT_PUBLIC_CAP_AWS_BUCKET }}" >> .env
+          
       - name: Copy .env to apps/desktop
         run: cp .env apps/desktop/.env
 
diff --git a/apps/desktop/src-tauri/src/upload.rs b/apps/desktop/src-tauri/src/upload.rs
index 37996fd62..2a77eddd5 100644
--- a/apps/desktop/src-tauri/src/upload.rs
+++ b/apps/desktop/src-tauri/src/upload.rs
@@ -93,10 +93,12 @@ impl S3UploadMeta {
 
     pub fn ensure_defaults(&mut self) {
         if self.aws_region.is_empty() {
-            self.aws_region = std::env::var("NEXT_PUBLIC_CAP_AWS_REGION").unwrap_or_default();
+            self.aws_region = std::env::var("NEXT_PUBLIC_CAP_AWS_REGION")
+                .unwrap_or_else(|_| "us-east-1".to_string());
         }
         if self.aws_bucket.is_empty() {
-            self.aws_bucket = std::env::var("NEXT_PUBLIC_CAP_AWS_BUCKET").unwrap_or_default();
+            self.aws_bucket =
+                std::env::var("NEXT_PUBLIC_CAP_AWS_BUCKET").unwrap_or_else(|_| "capso".to_string());
         }
     }
 }
diff --git a/apps/web/app/api/screenshot/route.ts b/apps/web/app/api/screenshot/route.ts
index 78a24ea59..60a79a11c 100644
--- a/apps/web/app/api/screenshot/route.ts
+++ b/apps/web/app/api/screenshot/route.ts
@@ -49,9 +49,9 @@ export async function GET(request: NextRequest) {
   }
 
   const result = query[0];
-  if (!result?.video || !result?.bucket) {
+  if (!result?.video) {
     return new Response(
-      JSON.stringify({ error: true, message: "Video or bucket not found" }),
+      JSON.stringify({ error: true, message: "Video not found" }),
       { status: 401, headers: getHeaders(origin) }
     );
   }
@@ -113,7 +113,6 @@ export async function GET(request: NextRequest) {
       headers: getHeaders(origin),
     });
   } catch (error) {
-    console.error("Error generating screenshot URL", error);
     return new Response(
       JSON.stringify({
         error: error,