summaryrefslogtreecommitdiffstats
path: root/adenosine-cli/src/main.rs
diff options
context:
space:
mode:
authorBryan Newbold <bnewbold@robocracy.org>2022-10-27 18:27:16 -0700
committerBryan Newbold <bnewbold@robocracy.org>2022-10-27 18:27:16 -0700
commitdc24074377875b7f500b20e4f6088930f2eef4ea (patch)
treeb1636f3c59920661b78505e390d215a3d37367ad /adenosine-cli/src/main.rs
parent3b1b394e1651fdc62dcd026a9453017c11e20a2c (diff)
downloadadenosine-dc24074377875b7f500b20e4f6088930f2eef4ea.tar.gz
adenosine-dc24074377875b7f500b20e4f6088930f2eef4ea.zip
CLI: progress (identifiers, ls, delete)
Diffstat (limited to 'adenosine-cli/src/main.rs')
-rw-r--r--adenosine-cli/src/main.rs444
1 files changed, 0 insertions, 444 deletions
diff --git a/adenosine-cli/src/main.rs b/adenosine-cli/src/main.rs
deleted file mode 100644
index eea16bc..0000000
--- a/adenosine-cli/src/main.rs
+++ /dev/null
@@ -1,444 +0,0 @@
-use adenosine_cli::*;
-use anyhow::anyhow;
-use serde_json::{json, Value};
-use std::collections::HashMap;
-
-use colored_json::to_colored_json_auto;
-use log::{self, debug};
-use std::io::Write;
-use structopt::StructOpt;
-use termcolor::{Color, ColorChoice, ColorSpec, StandardStream, WriteColor};
-
-#[derive(StructOpt)]
-#[structopt(rename_all = "kebab-case", about = "CLI interface for AT Protocol")]
-struct Opt {
- #[structopt(
- global = true,
- long = "--host",
- env = "ATP_HOST",
- default_value = "https://localhost:8080"
- )]
- atp_host: String,
-
- // API auth tokens can be generated from the account page in the fatcat.wiki web interface
- #[structopt(
- global = true,
- long = "--auth-token",
- env = "ATP_AUTH_TOKEN",
- hide_env_values = true
- )]
- auth_token: Option<String>,
-
- /// Log more messages. Pass multiple times for ever more verbosity
- ///
- /// By default, it'll only report errors. Passing `-v` one time also prints
- /// warnings, `-vv` enables info logging, `-vvv` debug, and `-vvvv` trace.
- #[structopt(global = true, long, short = "v", parse(from_occurrences))]
- verbose: i8,
-
- #[structopt(long = "--shell-completions", hidden = true)]
- shell_completions: Option<structopt::clap::Shell>,
-
- #[structopt(subcommand)]
- cmd: Command,
-}
-
-#[derive(StructOpt)]
-enum AccountCommand {
- /// Register a new account
- Register {
- #[structopt(long, short)]
- email: String,
-
- #[structopt(long, short)]
- username: String,
-
- #[structopt(long, short)]
- password: String,
- },
- Delete,
- Login {
- #[structopt(long, short)]
- username: String,
-
- #[structopt(long, short)]
- password: String,
- },
- Logout,
- Info,
- // TODO: CreateRevocationKey,
-}
-
-#[derive(StructOpt)]
-enum RepoCommand {
- Root {
- #[structopt(long)]
- did: String,
- },
- Export {
- #[structopt(long)]
- did: String,
- #[structopt(long)]
- from: Option<String>,
- },
- Import {
- // XXX: either path or stdin
- #[structopt(long)]
- did: String,
- },
-}
-
-#[derive(StructOpt)]
-enum BskyCommand {
- Feed { name: Option<String> },
- Notifications,
- Post { text: String },
- Repost { uri: String },
- Like { uri: String },
- // TODO: Repost { uri: String, },
- Follow { uri: String },
- // TODO: Unfollow { uri: String, },
- /*
- Follows {
- name: String,
- },
- Followers {
- name: String,
- },
- */
- Profile { name: String },
- SearchUsers { query: String },
-}
-
-#[derive(StructOpt)]
-enum Command {
- Get {
- uri: String,
- },
-
- Ls {
- uri: String,
- },
-
- Create {
- collection: String,
- body: String,
- },
- Update {
- uri: String,
- params: String,
- },
- Delete {
- uri: String,
- },
-
- Describe {
- name: String,
- },
-
- Resolve {
- name: String,
- },
-
- Xrpc {
- method: XrpcMethod,
- nsid: String,
- params: Option<String>,
- },
-
- /// Sub-commands for managing account
- Account {
- #[structopt(subcommand)]
- cmd: AccountCommand,
- },
-
- Repo {
- #[structopt(subcommand)]
- cmd: RepoCommand,
- },
-
- Bsky {
- #[structopt(subcommand)]
- cmd: BskyCommand,
- },
-
- /// Summarize connection and authentication with API
- Status,
-}
-
-fn main() -> Result<()> {
- let opt = Opt::from_args();
-
- let log_level = match opt.verbose {
- std::i8::MIN..=-1 => "none",
- 0 => "error",
- 1 => "warn",
- 2 => "info",
- 3 => "debug",
- 4..=std::i8::MAX => "trace",
- };
- // hyper logging is very verbose, so crank that down even if everything else is more verbose
- let log_filter = format!("{},hyper=error", log_level);
- env_logger::Builder::from_env(env_logger::Env::default().default_filter_or(log_filter))
- .format_timestamp(None)
- .init();
-
- debug!("Args parsed, starting up");
-
- #[cfg(windows)]
- colored_json::enable_ansi_support();
-
- if let Some(shell) = opt.shell_completions {
- Opt::clap().gen_completions_to("adenosine", shell, &mut std::io::stdout());
- std::process::exit(0);
- }
-
- if let Err(err) = run(opt) {
- // Be graceful about some errors
- if let Some(io_err) = err.root_cause().downcast_ref::<std::io::Error>() {
- if let std::io::ErrorKind::BrokenPipe = io_err.kind() {
- // presumably due to something like writing to stdout and piped to `head -n10` and
- // stdout was closed
- debug!("got BrokenPipe error, assuming stdout closed as expected and exiting with success");
- std::process::exit(0);
- }
- }
- let mut color_stderr = StandardStream::stderr(if atty::is(atty::Stream::Stderr) {
- ColorChoice::Auto
- } else {
- ColorChoice::Never
- });
- color_stderr.set_color(ColorSpec::new().set_fg(Some(Color::Red)).set_bold(true))?;
- eprintln!("Error: {:?}", err);
- color_stderr.set_color(&ColorSpec::new())?;
- std::process::exit(1);
- }
- Ok(())
-}
-
-fn print_result_json(result: Option<Value>) -> Result<()> {
- if let Some(val) = result {
- writeln!(&mut std::io::stdout(), "{}", to_colored_json_auto(&val)?)?
- };
- Ok(())
-}
-
-fn run(opt: Opt) -> Result<()> {
- let xrpc_client = XrpcClient::new(opt.atp_host.clone(), opt.auth_token.clone())?;
- let mut params: HashMap<String, String> = HashMap::new();
- let jwt_did: Option<String> = if let Some(ref token) = opt.auth_token {
- Some(parse_did_from_jwt(token)?)
- } else {
- None
- };
-
- let result = match opt.cmd {
- Command::Status => {
- // XXX
- println!("Configuration");
- println!(" ATP_HOST: {}", opt.atp_host);
- if opt.auth_token.is_some() {
- println!(" ATP_AUTH_TOKEN: <configured>");
- } else {
- println!(" ATP_AUTH_TOKEN:");
- }
- // TODO: parse JWT?
- // TODO: connection, auth check
- // TODO: account username, did, etc
- None
- }
- Command::Describe { name } => {
- params.insert("user".to_string(), name);
- xrpc_client.get("com.atproto.repoDescribe", Some(params))?
- }
- Command::Resolve { name } => {
- let mut params: HashMap<String, String> = HashMap::new();
- params.insert("name".to_string(), name);
- xrpc_client.get("com.atproto.resolveName", Some(params))?
- }
- Command::Get { uri } => {
- println!("GET: {}", uri);
- None
- }
- Command::Ls { uri } => {
- unimplemented!()
- }
- Command::Create { collection, body } => {
- unimplemented!()
- }
- Command::Update { uri, params } => {
- unimplemented!()
- }
- Command::Delete { uri } => {
- unimplemented!()
- }
- Command::Xrpc {
- method,
- nsid,
- params,
- } => {
- let body: Value = ().into();
- match method {
- // XXX: parse params
- XrpcMethod::Get => xrpc_client.get(&nsid, None)?,
- XrpcMethod::Post => xrpc_client.post(&nsid, None, body)?,
- }
- }
- Command::Account {
- cmd:
- AccountCommand::Register {
- email,
- username,
- password,
- },
- } => xrpc_client.post(
- "com.atproto.createAccount",
- None,
- json!({
- "email": email,
- "username": username,
- "password": password,
- }),
- )?,
- Command::Account {
- cmd: AccountCommand::Login { username, password },
- } => xrpc_client.post(
- "com.atproto.createSession",
- None,
- json!({
- "username": username,
- "password": password,
- }),
- )?,
- Command::Account {
- cmd: AccountCommand::Logout,
- } => xrpc_client.post("com.atproto.deleteSession", None, json!({}))?,
- Command::Account {
- cmd: AccountCommand::Delete,
- } => xrpc_client.post("com.atproto.deleteAccount", None, json!({}))?,
- Command::Account {
- cmd: AccountCommand::Info,
- } => xrpc_client.get("com.atproto.getAccount", None)?,
- Command::Repo {
- cmd: RepoCommand::Root { did },
- } => {
- params.insert("did".to_string(), did);
- xrpc_client.get("com.atproto.syncGetRoot", Some(params))?
- }
- Command::Repo {
- cmd: RepoCommand::Export { did, from },
- } => {
- params.insert("did".to_string(), did);
- if let Some(from) = from {
- params.insert("from".to_string(), from);
- };
- xrpc_client.get_to_writer(
- "com.atproto.syncGetRepo",
- Some(params),
- &mut std::io::stdout(),
- )?;
- None
- }
- Command::Repo {
- cmd: RepoCommand::Import { did },
- } => {
- params.insert("did".to_string(), did);
- xrpc_client.post_cbor_from_reader(
- "com.atproto.syncUpdateRepo",
- Some(params),
- &mut std::io::stdin(),
- )?
- }
- Command::Bsky {
- cmd: BskyCommand::Feed { name },
- } => {
- if let Some(name) = name {
- params.insert("author".to_string(), name);
- xrpc_client.get("app.bsky.getAuthorFeed", Some(params))?
- } else {
- xrpc_client.get("app.bsky.getHomeFeed", None)?
- }
- }
- Command::Bsky {
- cmd: BskyCommand::Notifications,
- } => xrpc_client.get("app.bsky.getNotifications", None)?,
- Command::Bsky {
- cmd: BskyCommand::Post { text },
- } => {
- params.insert(
- "did".to_string(),
- jwt_did.ok_or(anyhow!("need auth token"))?,
- );
- params.insert("collection".to_string(), "app.bsky.post".to_string());
- xrpc_client.post(
- "com.atproto.repoCreateRecord",
- Some(params),
- json!({
- "text": text,
- }),
- )?
- }
- Command::Bsky {
- cmd: BskyCommand::Repost { uri },
- } => {
- params.insert(
- "did".to_string(),
- jwt_did.ok_or(anyhow!("need auth token"))?,
- );
- params.insert("collection".to_string(), "app.bsky.repost".to_string());
- xrpc_client.post(
- "com.atproto.repoCreateRecord",
- Some(params),
- json!({
- "subject": uri,
- }),
- )?
- }
- Command::Bsky {
- cmd: BskyCommand::Like { uri },
- } => {
- params.insert(
- "did".to_string(),
- jwt_did.ok_or(anyhow!("need auth token"))?,
- );
- params.insert("collection".to_string(), "app.bsky.like".to_string());
- xrpc_client.post(
- "com.atproto.repoCreateRecord",
- Some(params),
- json!({
- "subject": uri,
- }),
- )?
- }
- Command::Bsky {
- cmd: BskyCommand::Follow { uri },
- } => {
- params.insert(
- "did".to_string(),
- jwt_did.ok_or(anyhow!("need auth token"))?,
- );
- params.insert("collection".to_string(), "app.bsky.follow".to_string());
- xrpc_client.post(
- "com.atproto.repoCreateRecord",
- Some(params),
- json!({
- "subject": uri,
- }),
- )?
- }
- Command::Bsky {
- cmd: BskyCommand::Profile { name },
- } => {
- params.insert("name".to_string(), name);
- xrpc_client.get("app.bsky.getProfile", Some(params))?
- }
- Command::Bsky {
- cmd: BskyCommand::SearchUsers { query },
- } => {
- params.insert("term".to_string(), query);
- xrpc_client.get("app.bsky.getUsersSearch", Some(params))?
- }
- };
- print_result_json(result)?;
- Ok(())
-}