neo-cli 1.0.0

Command-line interface for the NeoRust SDK
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92

#![allow(dead_code)]
use anyhow::{Context, Result};
use serde::{Deserialize, Serialize};
use std::{fs, path::PathBuf};

#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct CliConfig {
	pub network: NetworkConfig,
	pub wallet: WalletConfig,
	pub storage: StorageConfig,
	pub logging: LoggingConfig,
}

#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct NetworkConfig {
	pub rpc_url: String,
	pub network_magic: u32,
	pub address_version: u8,
}

#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct WalletConfig {
	pub default_path: Option<String>,
	pub auto_unlock: bool,
}

#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct StorageConfig {
	pub path: PathBuf,
}

#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct LoggingConfig {
	pub level: String,
	pub file: Option<PathBuf>,
}

impl Default for CliConfig {
	fn default() -> Self {
		Self {
			network: NetworkConfig {
				rpc_url: "http://localhost:10332".to_string(),
				network_magic: 860833102, // Mainnet
				address_version: 53,
			},
			wallet: WalletConfig { default_path: None, auto_unlock: false },
			storage: StorageConfig {
				path: dirs::data_dir().unwrap_or_else(|| PathBuf::from(".")).join("neo-cli"),
			},
			logging: LoggingConfig { level: "info".to_string(), file: None },
		}
	}
}

impl CliConfig {
	pub fn load() -> Result<Self> {
		let config_path = Self::config_path()?;

		if !config_path.exists() {
			let default_config = Self::default();
			let parent_dir = config_path.parent().ok_or_else(|| anyhow::anyhow!("Config path has no parent directory"))?;
			fs::create_dir_all(parent_dir)?;
			let config_str = serde_json::to_string_pretty(&default_config)?;
			fs::write(&config_path, config_str)?;
			return Ok(default_config);
		}

		let config_str = fs::read_to_string(&config_path)
			.context(format!("Failed to read config file: {:?}", config_path))?;
		let config: CliConfig =
			serde_json::from_str(&config_str).context("Failed to parse config file")?;

		Ok(config)
	}

	pub fn save(&self) -> Result<()> {
		let config_path = Self::config_path()?;
		let parent_dir = config_path.parent().ok_or_else(|| anyhow::anyhow!("Config path has no parent directory"))?;
		fs::create_dir_all(parent_dir)?;
		let config_str = serde_json::to_string_pretty(self)?;
		fs::write(&config_path, config_str)?;
		Ok(())
	}

	fn config_path() -> Result<PathBuf> {
		let config_dir = dirs::config_dir().unwrap_or_else(|| PathBuf::from(".")).join("neo-cli");

		fs::create_dir_all(&config_dir)?;

		Ok(config_dir.join("config.json"))
	}
}