From ce3aaa8b76d4213721afbc80df3402371c22e993 Mon Sep 17 00:00:00 2001
From: TheGeneralist <180094941+thegeneralist01@users.noreply.github.com>
Date: Fri, 29 May 2026 16:41:02 +0200
Subject: [PATCH] Finalize archive runs at command level

---
 src/main.rs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 7923c74..42c87d0 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -518,7 +518,6 @@ fn record_media_entry(
         },
     )?;
     database::complete_archive_run_item(conn, item.id, entry.id)?;
-    database::finish_archive_run(conn, run.id)?;
     Ok(entry)
 }
 
@@ -602,7 +601,6 @@ fn record_tweet_entry(
     }
 
     database::complete_archive_run_item(conn, item.id, entry.id)?;
-    database::finish_archive_run(conn, run.id)?;
     Ok(entry)
 }
 
@@ -719,6 +717,7 @@ fn main() -> Result<()> {
                                 source,
                                 &tweet_id,
                             )?;
+                            database::finish_archive_run(&conn, run.id)?;
                             println!(
                                 "Tweet archived successfully to {}",
                                 store_path.join("raw_tweets").display()
@@ -736,6 +735,7 @@ fn main() -> Result<()> {
                                 source,
                                 &tweet_id,
                             )?;
+                            database::finish_archive_run(&conn, run.id)?;
                             println!(
                                 "Tweet already archived in {}",
                                 store_path.join("raw_tweets").display()
@@ -856,6 +856,7 @@ fn main() -> Result<()> {
                 &file_extension,
                 byte_size,
             )?;
+            database::finish_archive_run(&conn, run.id)?;
 
             Ok(())
         }