AI sahit

Author: drahnr

crates/ntx-builder/src/db/migrations/2026020900000_setup/mod.rs

@@ -0,0 +1,9 @@
+use diesel::SqliteConnection;
+
+use miden_node_db::DatabaseError;
+
+pub(super) const SQL: &str = include_str!("up.sql");
+
+pub(super) fn data_migration(_conn: &mut SqliteConnection) -> Result<(), DatabaseError> {
+    Ok(())
+}

crates/store/src/db/migrations/2025062000000_setup/mod.rs

@@ -0,0 +1,9 @@
+use diesel::SqliteConnection;
+
+use crate::DatabaseError;
+
+pub(super) const SQL: &str = include_str!("up.sql");
+
+pub(super) fn data_migration(_conn: &mut SqliteConnection) -> Result<(), DatabaseError> {
+    Ok(())
+}

crates/store/src/db/migrations/2026020600000_cleanup_indices/mod.rs

@@ -0,0 +1,9 @@
+use diesel::SqliteConnection;
+
+use crate::DatabaseError;
+
+pub(super) const SQL: &str = include_str!("up.sql");
+
+pub(super) fn data_migration(_conn: &mut SqliteConnection) -> Result<(), DatabaseError> {
+    Ok(())
+}

crates/store/src/db/migrations/20260206163855_add_account_indices/mod.rs

@@ -0,0 +1,9 @@
+use diesel::SqliteConnection;
+
+use crate::DatabaseError;
+
+pub(super) const SQL: &str = include_str!("up.sql");
+
+pub(super) fn data_migration(_conn: &mut SqliteConnection) -> Result<(), DatabaseError> {
+    Ok(())
+}

crates/store/src/db/migrations/20260223160000_add_block_header_commitment/mod.rs

@@ -1,16 +1,12 @@
-//! Data migration
-//!
-//! Populate the new column `commitment` in table `block_headers` from the existing serialized
-//! representation.
-
-use diesel::migration::MigrationVersion;
 use diesel::sqlite::SqliteConnection;
 use diesel::{QueryableByName, RunQueryDsl};
 use miden_protocol::block::BlockHeader;
 use miden_protocol::utils::{Deserializable, Serializable};
 
 use crate::DatabaseError;
 
+pub(super) const SQL: &str = include_str!("up.sql");
+
 #[derive(QueryableByName)]
 struct BlockHeaderRow {
     #[diesel(sql_type = diesel::sql_types::BigInt)]
@@ -19,18 +15,12 @@ struct BlockHeaderRow {
     block_header: Vec<u8>,
 }
 
-pub(super) fn run(
-    conn: &mut SqliteConnection,
-    version: &MigrationVersion<'_>,
-) -> std::result::Result<(), DatabaseError> {
-    backfill_block_header_commitments(conn, version)
+pub(super) fn data_migration(conn: &mut SqliteConnection) -> Result<(), DatabaseError> {
+    backfill_block_header_commitments(conn)
 }
 
 /// Backfill block header commitments for existing rows in table `block_headers`
-pub(super) fn backfill_block_header_commitments(
-    conn: &mut SqliteConnection,
-    _version: &MigrationVersion<'_>,
-) -> std::result::Result<(), DatabaseError> {
+fn backfill_block_header_commitments(conn: &mut SqliteConnection) -> Result<(), DatabaseError> {
     let block_headers = diesel::sql_query(
         "SELECT block_num, block_header FROM block_headers WHERE commitment IS NULL",
     )

crates/store/src/db/migrations/data.rs

@@ -1,91 +1,15 @@
-//! Data migration declaration and representation utilities
-
 use diesel::SqliteConnection;
-use diesel::migration::MigrationVersion;
 
 use crate::DatabaseError;
 
-/// A data migration to apply post schema migration.
-pub(crate) struct DataMigration {
-    /// Identifier of the migration which matches the directory name containing the up.sql/down.sql
-    /// files
+pub(crate) struct Migration {
     pub(crate) id: &'static str,
-    /// Closure to execute given a connection
-    pub(crate) run: fn(
-        &mut SqliteConnection,
-        &MigrationVersion<'static>,
-    ) -> std::result::Result<(), DatabaseError>,
+    pub(crate) sql: &'static str,
+    pub(crate) data_migration: fn(&mut SqliteConnection) -> Result<(), DatabaseError>,
 }
 
-impl std::fmt::Debug for DataMigration {
+impl std::fmt::Debug for Migration {
     fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
         std::fmt::Debug::fmt(self.id, f)
     }
 }
-
-// Helper macro to resolve an optional explicit run identifier or default to `<name>::run`.
-#[macro_export]
-macro_rules! __data_migrations_resolve_run {
-    ($name:ident, $run:ident) => {
-        $name::$run
-    };
-    ($name:ident) => {
-        $name::run
-    };
-}
-
-#[macro_export]
-macro_rules! __data_migrations_mod {
-    ($module:ident, $path:literal) => {
-        // It'd be great if we could use `$(concat!($path, "/mod.rs"))`
-        // but it's already required at module traversal, so it _must be_
-        // a true literal string `"somewhere/mod.rs"`.
-        #[path = $path]
-        mod $module;
-    };
-}
-
-/// Helper macro to both declare the `fn run` (default), declare the module, and keeps it in a slice
-/// for later usage
-///
-/// Example:
-///
-/// ```ignore
-/// post_migrations! {
-///     const DATA_MIGRATION_CODE: &[DataMigration] = {
-///         "20260223160000_add_block_header_commitment/mod.rs" => alt_name_add_commitment,
-///         "20260223160001_bar/mod.rs" => bar::fun,
-///     };
-/// }
-/// ```
-///
-/// expands to
-///
-///```ignore
-/// mod alt_name_add_commitment { include!{ "20260223160000_add_block_header_commitment/mod.rs" } }
-/// mod bar { include!{ "20260223160001_bar/mod.rs" } }
-///
-/// const DATA_MIGRATION_CODE: &[DataMigration] = {
-///     PostMigration { id: "20260223160000_add_block_header_commitment", add_block_header::run },
-///     PostMigration { id: "20260223160001_bar", bar::fun },
-/// };
-/// ```
-#[macro_export]
-macro_rules! data_migrations {
-    (const $binding:ident : &[$what:ty] = {$($path:literal => $module:ident $(:: $run:ident)?),* $(,)?};) => {
-        $(
-            $crate::__data_migrations_mod!{$module, $path}
-        )*
-        static $binding: ::std::sync::LazyLock<Box<[$what]>> =
-            ::std::sync::LazyLock::new(|| {
-                Box::new([
-                    $(
-                        $crate::db::migrations::data::DataMigration {
-                            id: $path.split_once('/').expect("Has one path segment at least").0,
-                            run: $crate::__data_migrations_resolve_run!($module $(, $run)?),
-                        },
-                    )*
-                ])
-            });
-    };
-}