crate patches get applied during toml tweak

Author: spennydl

src/crates/mod.rs

@@ -20,10 +20,11 @@ enum CrateType {
     Local(local::Local),
 }
 
+#[derive(Clone)]
 pub struct CratePatch {
-    name: String,
-    uri: String,
-    branch: String
+    pub name: String,
+    pub uri: String,
+    pub branch: String
 }
 
 /// A Rust crate that can be used with rustwide.
@@ -74,6 +75,10 @@ impl Crate {
         patches.push(patch);
     }
 
+    pub(crate) fn patches(&self) -> &Option<Vec<CratePatch>> {
+        &self.1
+    }
+
     pub(crate) fn copy_source_to(&self, workspace: &Workspace, dest: &Path) -> Result<(), Error> {
         if dest.exists() {
             info!(

src/prepare.rs

@@ -174,6 +174,7 @@ impl<'a> TomlTweaker<'a> {
         self.remove_workspaces();
         self.remove_unwanted_cargo_features();
         self.remove_dependencies();
+        self.apply_patches();
 
         info!("finished tweaking {}", self.krate);
     }
@@ -275,6 +276,23 @@ impl<'a> TomlTweaker<'a> {
         }
     }
 
+    fn apply_patches(&mut self) {
+        if let Some(krate_patches) = self.krate.patches() {
+            if !self.table.contains_key("patch.crates-io") {
+                self.table.insert("patch.crates-io".into(), Value::Table(Table::new()));
+            }
+
+            if let Some(&mut Value::Table(ref mut patches)) = self.table.get_mut("patch.crates-io") {
+                for patch in krate_patches.iter().cloned() {
+                    let mut patch_table = Table::new();
+                    patch_table.insert("git".into(), Value::String(patch.uri));
+                    patch_table.insert("branch".into(), Value::String(patch.branch));
+                    patches.insert(patch.name, Value::Table(patch_table));
+                }
+            }
+        }
+    }
+
     // This is not a method to avoid borrow checker problems
     fn remove_dependencies_from_table(table: &mut Table, krate: &str) {
         // Convert path dependencies to registry dependencies