api: add Client (#87)

* api: add Client
* add integration tests

Signed-off-by: Sam Batschelet <[email protected]>

Author: hexfusion

spaces-cli/Cargo.toml

@@ -8,16 +8,16 @@ description = "spacesvm cli for issuing RPC commands"
 license = "BSD-3-Clause"
 homepage = "https://avax.network"
 
+[[bin]]
+name = "spaces-cli"
+path = "src/bin/spaces-cli/main.rs"
+
 [dependencies]
-avalanche-types = { version = "0.0.138", features = ["subnet"] }
 clap = { version = "4.0", features = ["derive"] }
 hex = "0.4.3"
 jsonrpc-core = "18.0.0"
-jsonrpc-core-client = { version = "18.0.0" }
-jsonrpc-client-transports = "18.0.0"
-jsonrpc-derive = "18.0"
 log = "0.4.17"
-spacesvm = { path = "../spacesvm" }
 serde = { version = "1.0.147", features = ["derive"] }
 serde_json = "1.0.87" 
-tokio = { version = "1.21.2", features = ["full"] }
+spacesvm = { path = "../spacesvm" }
+tokio = { version = "1.22.0", features = ["full"] }

spaces-cli/src/bin/spaces-cli/main.rs

@@ -0,0 +1,110 @@
+use std::error;
+
+use clap::{Parser, Subcommand};
+use jsonrpc_core::futures;
+use spacesvm::{
+    api::{
+        client::{claim_tx, delete_tx, get_or_create_pk, set_tx, Client, Uri},
+        DecodeTxArgs, IssueTxArgs, ResolveArgs,
+    },
+    chain::tx::{decoder, unsigned::TransactionData},
+};
+
+#[derive(Subcommand, Debug)]
+enum Command {
+    Claim {
+        space: String,
+    },
+    Set {
+        space: String,
+        key: String,
+        value: String,
+    },
+    Delete {
+        space: String,
+        key: String,
+    },
+    Get {
+        space: String,
+        key: String,
+    },
+    Ping {},
+}
+
+#[derive(Parser)]
+#[command(version, about, long_about = None)]
+struct Cli {
+    /// Endpoint for RPC calls.
+    #[clap(long)]
+    endpoint: String,
+
+    /// Private key file.
+    #[clap(long, default_value = ".spacesvm-cli-pk")]
+    private_key_file: String,
+
+    /// Which subcommand to call.
+    #[command(subcommand)]
+    command: Command,
+}
+
+#[tokio::main]
+async fn main() -> Result<(), Box<dyn error::Error>> {
+    let cli = Cli::parse();
+
+    let secret_key = get_or_create_pk(&cli.private_key_file)?;
+    let uri = cli.endpoint.parse::<Uri>()?;
+    let mut client = Client::new(uri);
+
+    if let Command::Get { space, key } = &cli.command {
+        let resp = futures::executor::block_on(client.resolve(ResolveArgs {
+            space: space.as_bytes().to_vec(),
+            key: key.as_bytes().to_vec(),
+        }))
+        .map_err(|e| e.to_string())?;
+        log::debug!("resolve response: {:?}", resp);
+
+        println!("{}", serde_json::to_string(&resp)?);
+        return Ok(());
+    }
+
+    if let Command::Ping {} = &cli.command {
+        let resp = futures::executor::block_on(client.ping()).map_err(|e| e.to_string())?;
+
+        println!("{}", serde_json::to_string(&resp)?);
+        return Ok(());
+    }
+
+    // decode tx
+    let tx_data = command_to_tx(cli.command)?;
+    let resp = futures::executor::block_on(client.decode_tx(DecodeTxArgs { tx_data }))
+        .map_err(|e| e.to_string())?;
+
+    let typed_data = &resp.typed_data;
+
+    // create signature
+    let dh = decoder::hash_structured_data(typed_data)?;
+    let sig = secret_key.sign_digest(&dh.as_bytes())?;
+
+    // issue tx
+    let resp = futures::executor::block_on(client.issue_tx(IssueTxArgs {
+        typed_data: resp.typed_data,
+        signature: sig.to_bytes().to_vec(),
+    }))
+    .map_err(|e| e.to_string())?;
+    println!("{}", serde_json::to_string(&resp)?);
+
+    Ok(())
+}
+
+/// Takes a TX command and returns transaction data.
+fn command_to_tx(command: Command) -> std::io::Result<TransactionData> {
+    match command {
+        Command::Claim { space } => Ok(claim_tx(space)),
+        Command::Set { space, key, value } => Ok(set_tx(space, key, value.as_bytes().to_vec())),
+        Command::Delete { space, key } => Ok(delete_tx(space, key)),
+        _ => Err(std::io::Error::new(
+            std::io::ErrorKind::Other,
+            "not a supported tx",
+        )),
+    }
+}

spaces-cli/src/main.rs

@@ -15,7 +15,7 @@ name = "spacesvm"
 path = "src/bin/spaces/main.rs"
 
 [dependencies]
-avalanche-types = { version = "0.0.138", features = ["subnet"] }
+avalanche-types = { version = "0.0.140", features = ["subnet"] }
 byteorder = "1.4.3"
 chrono = "0.4.22"
 crossbeam-channel = "0.5.6"
@@ -27,6 +27,7 @@ eip-712 = "0.1.0"
 env_logger = "0.9.3"
 hex = "0.4.3"
 http = "0.2.8"
+hyper = "0.14.23"
 jsonrpc-core = "18.0.0"
 jsonrpc-core-client = { version = "18.0.0" }
 jsonrpc-derive = "18.0"
@@ -47,4 +48,8 @@ typetag = "0.2"
 
 [dev-dependencies] 
 jsonrpc-tcp-server = "18.0.0"
-futures-test = "0.3.24"
+futures-test = "0.3.24"
+
+[[test]]
+name = "integration"
+path = "tests/integration_tests.rs"