Add a script to transfer crates between users

Author: alexcrichton

Cargo.toml

@@ -34,6 +34,10 @@ test = false
 name = "populate"
 test = false
 
+[[bin]]
+name = "transfer-crates"
+test = false
+
 [[test]]
 name = "all"
 path = "src/tests/all.rs"

src/bin/transfer-crates.rs

@@ -0,0 +1,96 @@
+// Transfer all crates from one user to another.
+//
+// Usage:
+//      cargo run --bin transfer-crates from-user to-user
+
+#![deny(warnings)]
+
+extern crate cargo_registry;
+extern crate postgres;
+extern crate time;
+extern crate semver;
+
+use std::env;
+use std::io;
+use std::io::prelude::*;
+
+use cargo_registry::{Crate, env, User};
+use cargo_registry::Model;
+
+#[allow(dead_code)]
+fn main() {
+    let conn = postgres::Connection::connect(&env("DATABASE_URL")[..],
+                                             postgres::SslMode::None).unwrap();
+    {
+        let tx = conn.transaction().unwrap();
+        transfer(&tx);
+        tx.set_commit();
+        tx.finish().unwrap();
+    }
+}
+
+fn transfer(tx: &postgres::Transaction) {
+    let from = match env::args().nth(1) {
+        None => { println!("needs a from-user argument"); return }
+        Some(s) => s,
+    };
+    let to = match env::args().nth(2) {
+        None => { println!("needs a to-user argument"); return }
+        Some(s) => s,
+    };
+
+    let from = User::find_by_login(tx, &from).unwrap();
+    let to = User::find_by_login(tx, &to).unwrap();
+
+    if from.avatar != to.avatar {
+        println!("====================================================");
+        println!("WARNING");
+        println!("");
+        println!("this may not be the same github user, different avatar urls");
+        println!("");
+        println!("from: {:?}", from.avatar);
+        println!("to:   {:?}", to.avatar);
+
+        get_confirm("continue?");
+    }
+
+    println!("Are you sure you want to transfer crates from {} to {}",
+             from.gh_login, to.gh_login);
+    get_confirm("continue");
+
+
+    let stmt = tx.prepare("SELECT * FROM crates WHERE user_id = $1")
+                 .unwrap();
+    let crates = stmt.query(&[&from.id]).unwrap();
+    for krate in crates.iter() {
+        let krate = Crate::from_row(&krate);
+        println!("transferring {}", krate.name);
+        let owners = krate.owners(tx).unwrap();
+        if owners.len() == 1 {
+            let n = tx.execute("UPDATE crate_owners SET owner_id = $1
+                                 WHERE owner_id = $2 AND crate_id = $3",
+                               &[&to.id, &from.id, &krate.id]).unwrap();
+            assert_eq!(n, 1);
+        } else {
+            println!("error: not exactly one owner for {}", krate.name);
+            return
+        }
+
+        let n = tx.execute("UPDATE crates SET user_id = $1
+                             WHERE id = $2",
+                           &[&to.id, &krate.id]).unwrap();
+        assert_eq!(n, 1);
+    }
+
+    get_confirm("commit?");
+}
+
+fn get_confirm(msg: &str) {
+    print!("{} [y/N]: ", msg);
+    io::stdout().flush().unwrap();
+    let mut line = String::new();
+    io::stdin().read_line(&mut line).unwrap();
+    if !line.starts_with("y") {
+        std::process::exit(0);
+    }
+}