path: root/src/bin/zot/main.rs

/* 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;

#[tokio::main]
async 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 verify =>
            (about: "Verify")
            (@arg raw: --raw "Display raw json payload")
        )
        (@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")
        )
        (@subcommand photos =>
            (about: "Post or get photos or albums")
            (@arg UPLOAD: --upload [FILE] requires[ALBUM] "Upload a photo")
            (@arg ALBUM: --album +takes_value "The album to upload the photo to")
            (@arg DESC: --description +takes_value "A description/title for the photo")
            (@arg BODY: --body +takes_value "A longer description/body text")
            (@arg POST: --post "Post the photo to the timeline")
            (@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() {
        ("verify", Some(m)) => {
            let r = zotapi::Channel::z();
            if m.is_present("raw") {
                println!("{}", r.fetch_raw(&client).await.unwrap());
            } else {
                println!("{:?}", r.fetch(&client).await);
            }
        }
        ("channel", Some(m)) => {
            let raw = m.is_present("raw");
            zot::channel_stream::fetch(&client, raw).await;
        }
        ("network", Some(m)) => {
            let raw = m.is_present("raw");
            zot::network_stream::fetch(&client, raw).await;
        }
        ("abconfig", _) => {
            zot::abconfig::fetch(&client).await;
        }
        ("abook", Some(m)) => {
            let r = zotapi::Abook::z();

            if m.is_present("raw") {
                println!("{}", r.fetch_raw(&client).await.unwrap());
            } else {
                for abook in r.fetch(&client).await.unwrap() {
                    println!("{},{},{}", abook.xchan.name, abook.xchan.address, abook.xchan.network);
                }
            }
        }
        ("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)
                    .await
                    .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)
                    .await
                    .unwrap();

                if m.is_present("raw") {
                    println!("{}", res);
                } else {
                    zot::group::list_members(&res);
                }
            } else {
                let res = zotapi::group().fetch(&client).await.unwrap();

                if m.is_present("raw") {
                    println!("{}", res);
                } else {
                    zot::group::list(&res);
                }
            }
        }
        ("xchan", Some(m)) => {
            let mut r = zotapi::XChan::z();
            let id = m.value_of("ID").expect("No xchan provided.");

            if m.is_present("guid") {
                r.by_guid(id)
            } else if m.is_present("hash") {
                r.by_hash(id)
            } else {
                r.by_address(id)
            };

            if m.is_present("raw") {
                println!("{}", r.fetch_raw(&client).await.unwrap());
            } else {
                println!("{:?}", r.fetch(&client).await);
            }
        }
        ("post", Some(m)) => {
            zot::item::post(&client, m).await;
        }
        _ => {
            println!("{}", matches.usage());
        }
    }
}