summaryrefslogtreecommitdiffstats
path: root/ipfs-cli/src/command/bootstrap.rs
blob: 63c5925d580a4a0d0ea50447cc051608063a2c08 (plain)
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
// Copyright 2017 rust-ipfs-api Developers
//
// Licensed under the Apache License, Version 2.0, <LICENSE-APACHE or
// http://apache.org/licenses/LICENSE-2.0> or the MIT license <LICENSE-MIT or
// http://opensource.org/licenses/MIT>, at your option. This file may not be
// copied, modified, or distributed except according to those terms.
//

use clap::{App, ArgMatches};
use command::EXPECTED_API;
use ipfs_api::IpfsClient;
use tokio_core::reactor::Core;


pub fn signature<'a, 'b>() -> App<'a, 'b> {
    clap_app!(
        @subcommand bootstrap =>
            (@setting SubcommandRequiredElseHelp)
            (@subcommand add =>
                (@setting SubcommandRequiredElseHelp)
                (@subcommand default =>
                    (about: "Add default peers to the bootstrap list")
                )
            )
            (@subcommand list =>
                (about: "Show peers in the bootstrap list")
            )
            (@subcommand rm =>
                (@setting SubcommandRequiredElseHelp)
                (@subcommand all =>
                    (about: "Remove all peers from the bootstrap list")
                )
            )
    )
}


fn print_peers(peers: &Vec<String>) {
    println!("");
    println!("  peers                  :");
    for peer in peers {
        println!("    {}", peer);
    }
    println!("");
}


pub fn handle(core: &mut Core, client: &IpfsClient, args: &ArgMatches) {
    match args.subcommand() {
        ("add", Some(add)) => {
            match add.subcommand() {
                ("default", _) => {
                    let peers = core.run(client.bootstrap_add_default()).expect(
                        EXPECTED_API,
                    );

                    print_peers(&peers.peers);
                }
                _ => unreachable!(),
            }
        }
        ("list", _) => {
            let peers = core.run(client.bootstrap_list()).expect(EXPECTED_API);

            print_peers(&peers.peers);
        }
        ("rm", Some(rm)) => {
            match rm.subcommand() {
                ("all", _) => {
                    let peers = core.run(client.bootstrap_rm_all()).expect(EXPECTED_API);

                    print_peers(&peers.peers);
                }
                _ => unreachable!(),
            }
        }
        _ => unreachable!(),
    }
}