Added the import/cleanup functionality

shanty-config/src/lib.rs

@@ -69,6 +69,9 @@ pub struct TaggingConfig {
 
     #[serde(default = "default_confidence")]
     pub confidence: f64,
+
+    #[serde(default = "default_tag_concurrency")]
+    pub concurrency: usize,
 }
 
 #[derive(Debug, Clone, Serialize, Deserialize)]
@@ -243,10 +246,15 @@ impl Default for TaggingConfig {
             auto_tag: false,
             write_tags: true,
             confidence: default_confidence(),
+            concurrency: default_tag_concurrency(),
         }
     }
 }
 
+fn default_tag_concurrency() -> usize {
+    4
+}
+
 impl Default for DownloadConfig {
     fn default() -> Self {
         Self {

shanty-data/src/mb_import.rs

@@ -715,6 +715,9 @@ pub fn run_import(
          PRAGMA foreign_keys = OFF;",
     )?;
 
+    // Wrap entire import in a transaction so interrupted imports don't leave empty tables
+    conn.execute_batch("BEGIN;")?;
+
     let mut stats = ImportStats::default();
 
     // Import artists
@@ -818,6 +821,9 @@ pub fn run_import(
         rusqlite::params![stats.recordings.to_string()],
     )?;
 
+    // Commit the transaction — if we got here, everything succeeded
+    conn.execute_batch("COMMIT;")?;
+
     progress(&format!("Import complete: {stats}"));
     Ok(stats)
 }