diff options
Diffstat (limited to 'src/bin/zot/main.rs')
| -rw-r--r-- | src/bin/zot/main.rs | 147 |
1 files changed, 147 insertions, 0 deletions
diff --git a/src/bin/zot/main.rs b/src/bin/zot/main.rs new file mode 100644 index 0000000..5828f44 --- /dev/null +++ b/src/bin/zot/main.rs @@ -0,0 +1,147 @@ +/* Example Zot API command line utility, part of zotapi. + * Copyright (C) 2018 Harald Eilertsen <haraldei@anduin.net> + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see <http://www.gnu.org/licenses/>. + */ + +use clap::{clap_app, crate_authors, crate_version}; +use dotenv::dotenv; +use std::env; +use std::str::FromStr; + +mod zot; + +fn main() { + dotenv().ok(); + let site = env::var("HZ_SITE").expect("SITE variable expected"); + let user = env::var("HZ_USER").expect("USER variable expected"); + let password = env::var("HZ_PASSWORD").expect("PASSWORD variable expected"); + + let matches = clap_app!(app => + (name: "zot") + (version: crate_version!()) + (author: crate_authors!()) + (about: "zotapi command line client") + (@subcommand channel => + (about: "Fetch the channel stream") + (@arg raw: --raw "Display raw json payload") + ) + (@subcommand network => + (about: "Fetch the network stream") + (@arg raw: --raw "Display raw json payload") + ) + (@subcommand abook => + (about: "Fetch address book/contact info") + (@arg raw: --raw "Display raw json payload") + ) + (@subcommand abconfig => + (about: "Fetch abconfig") + ) + (@subcommand group => + (about: "Fetch privacy groups") + (@arg raw: --raw "Display raw json payload") + (@group selector => + (@arg ID: --id +takes_value "Fetch members of group <ID>") + (@arg GNAME: --name +takes_value "Fetch members of group <GNAME>") + ) + ) + (@subcommand xchan => + (about: "Fetch xchan info") + (@arg raw: --raw "Display raw json payload") + (@arg addr: --addr "ID is given as webbie (default)") + (@arg hash: --hash "ID is given as xchan hash") + (@arg guid: --guid "ID is given as a GUID") + (@arg ID: +required "id (webbie, hash or GUID) of xchan to fetch") + ) + (@subcommand post => + (about: "Post a new message") + (@arg FILE: "A text file containing the body of the message") + (@arg TITLE: --title +takes_value "Set a title for the message") + (@arg ATTACH: --attach [FILE] "Attach a file or image to the post") + (@arg GROUP: --group [groups] "Limit distribution to the specified group(s) separated by comma") + ) + ) + .get_matches(); + + let client = zotapi::client(&site, &user, &password); + + match matches.subcommand() { + ("channel", Some(m)) => { + let raw = m.is_present("raw"); + zot::channel_stream::fetch(&client, raw); + } + ("network", Some(m)) => { + let raw = m.is_present("raw"); + zot::network_stream::fetch(&client, raw); + } + ("abconfig", _) => { + zot::abconfig::fetch(&client); + } + ("abook", Some(m)) => { + let raw = m.is_present("raw"); + zot::abook::fetch(&client, raw); + } + ("group", Some(m)) => { + if let Some(id) = m.value_of("ID") { + let res = zotapi::group_members() + .by_group_id(u64::from_str(id).unwrap()) + .fetch(&client) + .unwrap(); + + if m.is_present("raw") { + println!("{}", res); + } else { + zot::group::list_members(&res); + } + } else if let Some(gname) = m.value_of("GNAME") { + let res = zotapi::group_members() + .by_group_name(gname) + .fetch(&client) + .unwrap(); + + if m.is_present("raw") { + println!("{}", res); + } else { + zot::group::list_members(&res); + } + } else { + let res = zotapi::group().fetch(&client).unwrap(); + + if m.is_present("raw") { + println!("{}", res); + } else { + zot::group::list(&res); + } + } + } + ("xchan", Some(m)) => { + let raw = m.is_present("raw"); + let t = if m.is_present("guid") { + zot::xchan::Type::GUID + } else if m.is_present("hash") { + zot::xchan::Type::Hash + } else { + zot::xchan::Type::Addr + }; + + zot::xchan::fetch(&client, raw, t, m.value_of("ID").unwrap()); + } + ("post", Some(m)) => { + zot::item::post(&client, m); + } + _ => { + println!("{}", matches.usage()); + } + } +} |