Stabilise embed format

butane_cli/src/lib.rs

@@ -193,16 +193,20 @@ pub fn embed(base_dir: &Path) -> Result<()> {
         copy_migration(&m, &mut new_m)?;
         mem_ms.add_migration(new_m)?;
     }
-    let json = serde_json::to_string(&mem_ms)?;
+    let json = serde_json::to_string_pretty(&mem_ms)?;
 
     let src = format!(
-        "
-use butane::migrations::MemMigrations;
+        "//! Butane migrations embedded in Rust.
 use std::result::Result;
+
+use butane::migrations::MemMigrations;
+
+/// Load the butane migrations embedded in Rust.
 pub fn get_migrations() -> Result<MemMigrations, butane::Error> {{
     let json = r#\"{json}\"#;
     MemMigrations::from_json(json)
-}}"
+}}
+"
     );
 
     let mut f = std::fs::File::create(path)?;

butane_cli/tests/embed.rs

@@ -11,8 +11,4 @@ fn embed() {
         .join("../examples/getting_started/.butane");
     assert!(example_dir.exists());
     butane_cli::embed(&example_dir).unwrap();
-    for filename in ["../src/butane_migrations.rs", "clistate.json"] {
-        let path = example_dir.join(filename);
-        std::fs::remove_file(path).unwrap();
-    }
 }

butane_core/src/migrations/adb.rs

@@ -5,7 +5,7 @@
 use crate::{Error, Result, SqlType, SqlVal};
 use serde::{de::Deserializer, de::Visitor, ser::Serializer, Deserialize, Serialize};
 use std::cmp::Ordering;
-use std::collections::{BTreeSet, HashMap};
+use std::collections::{BTreeMap, BTreeSet, HashMap};
 
 /// Suffix added to [`crate::many::Many`] tables.
 pub const MANY_SUFFIX: &str = "_Many";
@@ -145,14 +145,14 @@ impl TypeResolver {
 /// Abstract representation of a database schema.
 #[derive(Clone, Debug, Default, Deserialize, Serialize)]
 pub struct ADB {
-    tables: HashMap<String, ATable>,
-    extra_types: HashMap<TypeKey, DeferredSqlType>,
+    tables: BTreeMap<String, ATable>,
+    extra_types: BTreeMap<TypeKey, DeferredSqlType>,
 }
 impl ADB {
     pub fn new() -> Self {
         ADB {
-            tables: HashMap::new(),
-            extra_types: HashMap::new(),
+            tables: BTreeMap::new(),
+            extra_types: BTreeMap::new(),
         }
     }
     pub fn tables(&self) -> impl Iterator<Item = &ATable> {
@@ -161,7 +161,7 @@ impl ADB {
     pub fn get_table<'a>(&'a self, name: &str) -> Option<&'a ATable> {
         self.tables.get(name)
     }
-    pub fn types(&self) -> &HashMap<TypeKey, DeferredSqlType> {
+    pub fn types(&self) -> &BTreeMap<TypeKey, DeferredSqlType> {
         &self.extra_types
     }
     pub fn replace_table(&mut self, table: ATable) {
@@ -460,8 +460,8 @@ impl AColumn {
     /// Resolve a column constraints target.
     fn resolve_reference_target(
         &mut self,
-        extra_types: &HashMap<TypeKey, DeferredSqlType>,
-        tables: &HashMap<String, ATable>,
+        extra_types: &BTreeMap<TypeKey, DeferredSqlType>,
+        tables: &BTreeMap<String, ATable>,
     ) {
         match &self.reference {
             None | Some(ARef::Literal(_)) => {}

butane_core/src/migrations/memmigrations.rs

@@ -4,16 +4,16 @@ use crate::query::BoolExpr;
 use crate::{ConnectionMethods, DataObject, Result};
 use serde::{Deserialize, Serialize};
 use std::borrow::Cow;
-use std::collections::HashMap;
+use std::collections::BTreeMap;
 
 /// A migration stored in memory.
 #[derive(Clone, Debug, Deserialize, Serialize)]
 pub struct MemMigration {
     name: String,
     db: ADB,
     from: Option<String>,
-    up: HashMap<String, String>,
-    down: HashMap<String, String>,
+    up: BTreeMap<String, String>,
+    down: BTreeMap<String, String>,
 }
 
 impl MemMigration {
@@ -22,8 +22,8 @@ impl MemMigration {
             name,
             db: ADB::new(),
             from: None,
-            up: HashMap::new(),
-            down: HashMap::new(),
+            up: BTreeMap::new(),
+            down: BTreeMap::new(),
         }
     }
 }
@@ -93,15 +93,15 @@ impl MigrationMut for MemMigration {
 /// A collection of migrations stored in memory.
 #[derive(Clone, Debug, Deserialize, Serialize)]
 pub struct MemMigrations {
-    migrations: HashMap<String, MemMigration>,
+    migrations: BTreeMap<String, MemMigration>,
     current: MemMigration,
     latest: Option<String>,
 }
 
 impl MemMigrations {
     pub fn new() -> Self {
         MemMigrations {
-            migrations: HashMap::new(),
+            migrations: BTreeMap::new(),
             current: MemMigration::new("current".to_string()),
             latest: None,
         }

examples/getting_started/.butane/clistate.json

@@ -0,0 +1 @@
+{"embedded":true}