Skip to content

Commit

Permalink
mod: cargo fmt
Browse files Browse the repository at this point in the history
  • Loading branch information
anshalshukla committed Jan 1, 2024
1 parent b66bd53 commit d6bd208
Show file tree
Hide file tree
Showing 7 changed files with 27 additions and 63 deletions.
7 changes: 2 additions & 5 deletions src/app/config.rs
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ impl AppChainConfig {
}
pub fn fund_msg(&self) {
match &self.da_layer {
DALayer::Avail { seed: _seed, public_key} => {
DALayer::Avail { seed: _seed, public_key } => {
println!();
println!("Please fund {} with atleast 1 AVL", public_key);
}
Expand Down Expand Up @@ -61,10 +61,7 @@ pub enum RollupMode {

#[derive(Debug, Serialize, Deserialize, EnumIter, Display)]
pub enum DALayer {
Avail {
seed: String,
public_key: String,
},
Avail { seed: String, public_key: String },
Celestia,
Ethereum,
NoDA,
Expand Down
8 changes: 2 additions & 6 deletions src/cli/init.rs
Original file line number Diff line number Diff line change
Expand Up @@ -62,12 +62,8 @@ fn generate_config() -> Result<AppChainConfig, InitError> {
let config_version = ConfigVersion::Version1;

let result = match da_layer_type {
DALayer::Avail{ .. } => {
setup_and_generate_keypair(&app_chain)
}
_ => {
Ok(da_layer_type)
}
DALayer::Avail { .. } => setup_and_generate_keypair(&app_chain),
_ => Ok(da_layer_type),
};
let da_layer = match result {
Ok(da_config) => da_config,
Expand Down
2 changes: 1 addition & 1 deletion src/main.rs
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
use clap::{Parser, Subcommand};
use madara_cli::{cli};
use madara_cli::cli;

#[derive(Parser)]
#[command(author, version, about, long_about = None)]
Expand Down
41 changes: 12 additions & 29 deletions src/utils/avail.rs
Original file line number Diff line number Diff line change
@@ -1,11 +1,13 @@
use std::{fs, io};
use std::process::Stdio;
use std::string::FromUtf8Error;
use std::{fs, io};

use serde_json::{json, Value};
use thiserror::Error;

use crate::app::config::DALayer;
use crate::utils::cmd::{execute_cmd, execute_cmd_stdout};
use crate::utils::paths::{get_app_home, get_madara_home};
use serde_json::{json,Value};
use crate::app::config::{DALayer};

#[derive(Debug, Error)]
pub enum AvailKeyGenError {
Expand All @@ -25,11 +27,7 @@ pub fn setup_and_generate_keypair(app_chain: &str) -> Result<DALayer, AvailKeyGe
}
fn install_subkey() -> Result<(), AvailKeyGenError> {
let madara_home = get_madara_home()?;
execute_cmd(
"cargo",
&["install", "subkey"],
&madara_home
)?;
execute_cmd("cargo", &["install", "subkey"], &madara_home)?;

Ok(())
}
Expand All @@ -39,28 +37,16 @@ fn generate_keypair(app_chain: &str) -> Result<DALayer, AvailKeyGenError> {
let app_home = get_app_home(app_chain)?;
let full_file_path = app_home.join(keypair_file);

execute_cmd(
"cargo",
&["install", "subkey"],
&madara_home
)?;
execute_cmd("cargo", &["install", "subkey"], &madara_home)?;

let output = execute_cmd_stdout(
"subkey",
&["generate", "--output-type=json"],
&madara_home,
Stdio::piped()
)?;
let output = execute_cmd_stdout("subkey", &["generate", "--output-type=json"], &madara_home, Stdio::piped())?;

let keypair_str = String::from_utf8(output.stdout)?;
let keypair_json: Value = serde_json::from_str(&keypair_str)?;

let seed = keypair_json["secretSeed"].to_string();
let public_key = keypair_json["publicKey"].to_string();
let avail = DALayer::Avail {
seed: seed.clone(),
public_key,
};
let public_key = keypair_json["publicKey"].to_string();
let avail = DALayer::Avail { seed: seed.clone(), public_key };

if let Err(err) = fs::write(full_file_path, &keypair_str) {
panic!("Error writing to file: {}", err);
Expand Down Expand Up @@ -88,10 +74,7 @@ fn generate_avail_connect_config(app_chain: &str, seed: &str) -> Result<(), Avai
let avail_connect_string = serde_json::to_string(&avail_connect)?;

// "\"{seed}\"" is invalid so converted it to "{seed}"
let avail_connect_string_modified = avail_connect_string
.replace("\"\\\"", "\"")
.replace("\\\"\"", "\"");

let avail_connect_string_modified = avail_connect_string.replace("\"\\\"", "\"").replace("\\\"\"", "\"");

if let Err(err) = fs::write(full_file_path, &avail_connect_string_modified) {
panic!("Error writing to file: {}", err);
Expand All @@ -100,4 +83,4 @@ fn generate_avail_connect_config(app_chain: &str, seed: &str) -> Result<(), Avai
}

Ok(())
}
}
7 changes: 3 additions & 4 deletions src/utils/cmd.rs
Original file line number Diff line number Diff line change
Expand Up @@ -3,13 +3,12 @@ use std::path::PathBuf;
use std::process::{Command, Output, Stdio};

pub fn execute_cmd(program: &str, args: &[&str], dir: &PathBuf) -> Result<Output, Error> {
let output =execute_cmd_stdout(program, args, dir, Stdio::inherit())?;
let output = execute_cmd_stdout(program, args, dir, Stdio::inherit())?;
Ok(output)
}

pub fn execute_cmd_stdout(program: &str, args: &[&str], dir: &PathBuf, out: Stdio) -> Result<Output, Error> {
let result =
Command::new(program).current_dir(dir).args(args).stdout(out).stderr(Stdio::inherit()).output();
let result = Command::new(program).current_dir(dir).args(args).stdout(out).stderr(Stdio::inherit()).output();

match result {
Ok(output) => {
Expand All @@ -25,4 +24,4 @@ pub fn execute_cmd_stdout(program: &str, args: &[&str], dir: &PathBuf, out: Stdi
Err(err)
}
}
}
}
23 changes: 6 additions & 17 deletions src/utils/madara.rs
Original file line number Diff line number Diff line change
Expand Up @@ -20,11 +20,7 @@ pub fn clone_madara_and_build_repo() -> Result<(), MadaraError> {
return Err(MadaraError::FailedToCloneRepo);
}
}
execute_cmd(
"cargo",
&["build", "--release"],
&madara_path
)?;
execute_cmd("cargo", &["build", "--release"], &madara_path)?;

Ok(())
}
Expand Down Expand Up @@ -52,27 +48,20 @@ pub fn setup_and_run_madara(app_chain: &str) -> Result<(), MadaraError> {
let da_conf = format!("--da-conf={}", da_config_path);
let base_path = format!("--base-path={}", config.base_path);

let mut args = vec!["--chain=dev", "--alice", "--force-authoring", "--rpc-cors=all", "--tx-ban-seconds=0", &base_path];
let mut args =
vec!["--chain=dev", "--alice", "--force-authoring", "--rpc-cors=all", "--tx-ban-seconds=0", &base_path];

match &config.da_layer {
DALayer::Avail{ .. } => {
DALayer::Avail { .. } => {
let avail_conf = vec!["--da-layer=avail", &da_conf];
args.extend(avail_conf);
}
_ => {}
}

execute_cmd(
"cargo",
&["run", "--release", "setup", "--chain=dev", "--from-remote", &base_path],
&madara_path
)?;
execute_cmd("cargo", &["run", "--release", "setup", "--chain=dev", "--from-remote", &base_path], &madara_path)?;

execute_cmd(
"./target/release/madara",
args.as_slice(),
&madara_path,
)?;
execute_cmd("./target/release/madara", args.as_slice(), &madara_path)?;

Ok(())
}
2 changes: 1 addition & 1 deletion src/utils/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -8,5 +8,5 @@ pub mod madara;

pub mod errors;

pub mod toml;
pub mod avail;
pub mod toml;

0 comments on commit d6bd208

Please sign in to comment.