feat(yt/storage/migrate): Add version two

2 files changed, 54 insertions, 3 deletions

diff --git a/yt/src/storage/migrate/mod.rs b/yt/src/storage/migrate/mod.rs
index 52f8c2b..767fe0f 100644
--- a/yt/src/storage/migrate/mod.rs
+++ b/yt/src/storage/migrate/mod.rs
@@ -22,8 +22,11 @@ pub enum DbVersion {
 
     /// Introduced: 2025-02-17.
     One,
+
+    /// Introduced: 2025-02-18.
+    Two,
 }
-const CURRENT_VERSION: DbVersion = DbVersion::One;
+const CURRENT_VERSION: DbVersion = DbVersion::Two;
 
 async fn add_error_context(
     function: impl Future<Output = Result<()>>,
@@ -73,15 +76,18 @@ impl DbVersion {
             DbVersion::Empty => unreachable!("A empty version does not have an associated integer"),
             DbVersion::Zero => 0,
             DbVersion::One => 1,
+            DbVersion::Two => 2,
         }
     }
     fn from_db(number: i64, namespace: &str) -> Result<Self> {
         match (number, namespace) {
             (0, "yt") => Ok(DbVersion::Zero),
             (1, "yt") => Ok(DbVersion::One),
+            (2, "yt") => Ok(DbVersion::Two),
 
             (0, other) => bail!("Db version is Zero, but got unknown namespace: '{other}'"),
             (1, other) => bail!("Db version is One, but got unknown namespace: '{other}'"),
+            (2, other) => bail!("Db version is Two, but got unknown namespace: '{other}'"),
 
             (other, "yt") => bail!("Got unkown version for 'yt' namespace: {other}"),
             (num, nasp) => bail!("Got unkown version number ({num}) and namespace ('{nasp}')"),
@@ -114,7 +120,9 @@ impl DbVersion {
                             .await
                             .context("Failed to set new version")?;
 
-                        tx.commit().await.context("Failed to commit changes")?;
+                        tx.commit()
+                            .await
+                            .context("Failed to commit changes")?;
 
                         // NOTE: This is needed, so that sqlite "sees" our changes to the table
                         // without having to reconnect. <2025-02-18>
@@ -169,8 +177,50 @@ impl DbVersion {
 
                 Box::pin(Self::One.update(app)).await
             }
-            // This is the current_version
+
             DbVersion::One => {
+                add_error_context(
+                    async {
+                        let mut tx = app
+                            .database
+                            .begin()
+                            .await
+                            .context("Failed to start the update transaction")?;
+                        debug!("Migrate: One -> Two");
+
+                        sqlx::raw_sql(include_str!("./sql/02_one_to_two.sql"))
+                            .execute(&mut *tx)
+                            .await
+                            .context("Failed to run the update sql script")?;
+
+                        set_db_version(&mut tx, Some(DbVersion::One), DbVersion::Two)
+                            .await
+                            .context("Failed to set the new version")?;
+
+                        tx.commit()
+                            .await
+                            .context("Failed to commit the update transaction")?;
+
+                        // NOTE: This is needed, so that sqlite "sees" our changes to the table
+                        // without having to reconnect. <2025-02-18>
+                        query!("VACUUM")
+                            .execute(&app.database)
+                            .await
+                            .context("Failed to vacuum database")?;
+
+                        Ok(())
+                    },
+                    DbVersion::One,
+                )
+                .await?;
+
+                Box::pin(Self::Two.update(app))
+                    .await
+                    .context("Failed to update to version: Three")
+            }
+
+            // This is the current_version
+            DbVersion::Two => {
                 assert_eq!(self, CURRENT_VERSION);
                 assert_eq!(self, get_version(app).await?);
                 Ok(())
diff --git a/yt/src/storage/migrate/sql/02_one_to_two.sql b/yt/src/storage/migrate/sql/02_one_to_two.sql
new file mode 100644
index 0000000..dd85737
--- /dev/null
+++ b/yt/src/storage/migrate/sql/02_one_to_two.sql
@@ -0,0 +1 @@
+ALTER TABLE videos DROP in_playlist;