getsb 0.1.2

Getsb is a command line tool for sending HTTP request.
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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118

#![cfg_attr(feature="clippy", feature(plugin))]
#![cfg_attr(feature="clippy", plugin(clippy))]

extern crate clap;

use clap::{Arg, App};
use std::collections::HashMap;

mod sender;
mod parser;
mod errors;

// HTTP Request structure
#[derive(Debug)]
pub struct Request {
    body: String,
    headers: HashMap<String, String>,
    method: String,
    url: String,
}

fn main() {
    // Validate and set args using clap crate for building CLI
    let matches = App::new("Getsb")
        .version("0.1.1")
        .author("Nazarii Sheremet. <nazarii.sheremet@gmail.com>")
        .about("Getsb is a command line tool for sending HTTP request.")
        .arg(Arg::with_name("request-file")
                 .short("r")
                 .long("request-file")
                 .value_name("FILE")
                 .help("Sets a custom request file")
                 .takes_value(true))
        .arg(Arg::with_name("METHOD")
                 .required_unless("request-file")
                 .possible_values(&["GET", "POST", "PUT", "PATCH", "DELETE", "OPTIONS"])
                 .help("Sets the METHOD of request")
                 .index(1))
        .arg(Arg::with_name("URL")
                 .required_unless("request-file")
                 .help("Sets the URL of request")
                 .index(2))
        .arg(Arg::with_name("body")
                 .short("b")
                 .long("body")
                 .value_name("BODY")
                 .help("Sets the body of request")
                 .takes_value(true))
        .arg(Arg::with_name("headers")
                 .short("h")
                 .long("headers")
                 .multiple(true)
                 .value_name("HEADERS")
                 .help("Sets the headers of request")
                 .takes_value(true))
        .arg(Arg::with_name("to-file")
                 .short("f")
                 .long("to-file")
                 .value_name("FILE")
                 .help("Save all response to file")
                 .takes_value(true))
        .get_matches();

    let mut to_file = String::new();

    // Checking first request-file
    if let Some(file) = matches.value_of("request-file") {
        if let Some(val) = matches.value_of("to-file") {
            to_file = val.to_string();
        }

        // Parsing file for creating request
        proccess_file(file, &to_file);
    } else {
        let mut method = String::new();
        let mut url = String::new();
        let mut body = String::new();
        let mut header = HashMap::new();

        if let Some(headers) = matches.values_of("headers") {
            for h in headers {
                let line: Vec<&str> = h.split(':').collect();
                header.insert(line[0].trim().to_string(), line[1].trim().to_string());
            }
        }

        if let Some(val) = matches.value_of("body") {
            body = val.to_string();
        }

        if let Some(val) = matches.value_of("URL") {
            url = val.to_string();
        }

        if let Some(val) = matches.value_of("METHOD") {
            method = val.to_string();
        }

        if let Some(val) = matches.value_of("to-file") {
            to_file = val.to_string();
        }

        let request = Request {
            method: method,
            url: url,
            body: body,
            headers: header,
        };

        sender::send(request, &to_file);
    }
}

// Working with request info from file
fn proccess_file(filename: &str, to_filename: &str) {
    let request = parser::parse_file(filename);
    sender::send(request, to_filename);
}