finish transfer_key

web3_proxy/src/bin/web3_proxy_cli/transfer_key.rs

@@ -0,0 +1,64 @@
+use anyhow::Context;
+use argh::FromArgs;
+use entities::{rpc_key, user};
+use ethers::types::Address;
+use log::{debug, info};
+use migration::sea_orm::{
+    self, ActiveModelTrait, ColumnTrait, DatabaseConnection, EntityTrait, IntoActiveModel,
+    QueryFilter,
+};
+use sea_orm::prelude::Uuid;
+use web3_proxy::frontend::authorization::RpcSecretKey;
+
+/// change a key's owner.
+#[derive(FromArgs, PartialEq, Eq, Debug)]
+#[argh(subcommand, name = "transfer_key")]
+pub struct TransferKeySubCommand {
+    #[argh(positional)]
+    /// the RPC key that you want to transfer.
+    rpc_secret_key: RpcSecretKey,
+
+    /// the new owner for the key.
+    #[argh(positional)]
+    new_address: String,
+}
+
+impl TransferKeySubCommand {
+    pub async fn main(self, db_conn: &DatabaseConnection) -> anyhow::Result<()> {
+        let rpc_secret_key: Uuid = self.rpc_secret_key.into();
+
+        let new_address: Address = self.new_address.parse()?;
+
+        let new_address: Vec<u8> = new_address.to_fixed_bytes().into();
+
+        let uk = rpc_key::Entity::find()
+            .filter(rpc_key::Column::SecretKey.eq(rpc_secret_key))
+            .one(db_conn)
+            .await?
+            .context("No key found")?;
+
+        debug!("user key: {}", serde_json::to_string(&uk)?);
+
+        let new_u = user::Entity::find()
+            .filter(user::Column::Address.eq(new_address))
+            .one(db_conn)
+            .await?
+            .context("No user found with that key")?;
+
+        debug!("new user: {}", serde_json::to_string(&new_u)?);
+
+        if new_u.id == uk.user_id {
+            info!("user already owns that key");
+        } else {
+            let mut uk = uk.into_active_model();
+
+            uk.user_id = sea_orm::Set(new_u.id);
+
+            let _uk = uk.save(db_conn).await?;
+
+            info!("changed the key's owner");
+        }
+
+        Ok(())
+    }
+}