summaryrefslogtreecommitdiffstats
path: root/ipfs-cli/src/command/block.rs
blob: c477583464edda7a7e58fc0a50850025860a6d3e (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
80
81
82
83
84
85
86
87
88
89
90
91
92
// 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 crate::command::{verify_file, CliCommand, EXPECTED_FILE};
use clap::App;
use futures::{Future, Stream};
use std::fs::File;
use std::io::{self, Write};

pub struct Command;

impl CliCommand for Command {
    const NAME: &'static str = "block";

    fn signature<'a, 'b>() -> App<'a, 'b> {
        clap_app!(
            @subcommand block =>
                (@setting SubcommandRequiredElseHelp)
                (@subcommand get =>
                    (about: "Get a raw IPFS block")
                    (@arg KEY: +required "The base58 multihash of an existing block")
                )
                (@subcommand put =>
                    (about: "Store a file as an IPFS block")
                    (@arg INPUT: +required {verify_file} "The file to store as an IPFS block")
                )
                (@subcommand rm =>
                    (about: "Removes an IPFS block")
                    (@arg KEY: +required "The base58 multihash of a block to remove")
                )
                (@subcommand stat =>
                    (about: "Get information about a raw IPFS block")
                    (@arg KEY: +required "The base58 multihash of the block to retrieve")
                )
        )
    }

    handle!(
        client;
        ("get", args) => {
            let key = args.value_of("KEY").unwrap();

            client
                .block_get(key)
                .for_each(|chunk| io::stdout().write_all(&chunk).map_err(From::from))
        },
        ("put", args) => {
            let path = args.value_of("INPUT").unwrap();
            let file = File::open(path).expect(EXPECTED_FILE);

            client
                .block_put(file)
                .map(|block| {
                    println!();
                    println!("  key     : {}", block.key);
                    println!("  size    : {}", block.size);
                    println!();
                })
        },
        ("rm", args) => {
            let key = args.value_of("KEY").unwrap();

            client
                .block_rm(key)
                .map(|rm| {
                    println!();
                    println!("  hash    : {}", rm.hash);
                    if let Some(error) = rm.error {
                        println!("  error   : {}", error);
                    }
                    println!();
                })
        },
        ("stat", args) => {
            let key = args.value_of("KEY").unwrap();

            client
                .block_stat(key)
                .map(|stat| {
                    println!();
                    println!("  key     : {}", stat.key);
                    println!("  size    : {}", stat.size);
                    println!();
                })
        }
    );
}