dvach 0.1.0

dvach a simple client cli tool for the 2ch.hk imageboard
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

//! dvach is a simple client cli tool for the 2ch.hk imageboard.
//!
//! Usage: dvach [board] [thread]
//! ```
//! $ dvach # list boards
//! $ dvach pr # list threads for the "pr" board
//! $ dvach pr 1299618 # show selected thread
//! ```

use env_logger;
use log::debug;
use reqwest;
use serde_json;
use structopt::StructOpt;

/// Custom println! version which exits cleanly when output was
/// interrupted, e.g. by piping to `head`.
macro_rules! println {
    ($fmt:expr, $($args:tt)*) => {
        {
            use std::io::{stdout, Write};
            use std::process::exit;

            if writeln!(&mut stdout(), $fmt, $($args)*).is_err() {
                exit(1);
            }
        }
    }
}

mod boards;
mod posts;
mod threads;
mod download;

use self::boards::list_boards;
use self::posts::list_posts;
use self::threads::list_threads;
use self::download::download;

/// Represent available cli args
#[derive(StructOpt, Debug)]
struct Cli {
    /// Board to list
    board: Option<String>,

    /// Thread to show
    thread: Option<usize>,

    /// Width of the comment in posts before wrapping
    #[structopt(short = "w", long = "comment-width", default_value = "80")]
    comment_width: usize,

    /// Default imageboard url
    #[structopt(long = "base-url", default_value = "https://2ch.hk")]
    base_url: String,

    /// Width of the comment in posts before wrapping
    #[structopt(long = "download")]
    download: Option<String>,
}

fn main() {
    env_logger::init();

    let args = Cli::from_args();
    debug!("Got args: {:#?}", &args);

    // get thumb or full image
    if let Some(path) = args.download {
        download(&args.base_url, &path);
        return
    }

    // run appropriate list action based on cli arguments
    match (args.board, args.thread) {
        (None, None) => list_boards(),
        (Some(board), None) => list_threads(&board, args.comment_width),
        (Some(board), Some(thread)) => list_posts(&board, thread, args.comment_width),
        _ => Cli::clap().print_help().expect("Cannot print help"),
    }
}