cao 0.1.0

Update DNS record.
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
119

mod args;
mod provider;
mod error;
mod interface;

use crate::args::{Args, RecordCmds};
use crate::provider::build_dns_provider;
use crate::error::Error;
use crate::interface::{interface_list, interface_or_value};

fn fetch_key(file_name: String) -> Result<String, Error>  {
    use std::fs::File;
    use std::io::prelude::*;

    let mut f = File::open(file_name)?;
    let mut s = String::with_capacity(64);
    f.read_to_string(&mut s)?;
    Ok(String::from(s.trim()))
}

fn main() -> Result<(), Error> {
    let args = Args::args();

    match args {
        Err(err) => {
            eprintln!("{}", err);
        },
        Ok(param) => { match param {
            Args::Record{provider, key, domain, cmd} => {
                let key = fetch_key(key)?;
                let provider = build_dns_provider(provider, key, domain)?;
                match cmd {
                    RecordCmds::Add{
                        sub_domain, record_type, record_line, value, interface
                    } => {
                        let value = interface_or_value(interface, value)?;
                        eprintln!("value: {}", value);
                        let id = provider.add_record(
                            &sub_domain,
                            &record_type,
                            &record_line,
                            &value,
                        )?;
                        print!("{}", id);
                    },
                    RecordCmds::List{offset, length, sub_domain} => {
                        let records = provider.list_record(
                            offset,
                            length,
                            sub_domain.as_ref()
                        )?;
                        for record in records {
                            println!("{}", record);
                        }
                    },
                    RecordCmds::Modify{
                        record_id, sub_domain, record_type, record_line, value, interface
                    } => {
                        let value = interface_or_value(interface, value)?;
                        provider.modify_record(
                            record_id,
                            sub_domain.as_ref(),
                            &record_type,
                            &record_line,
                            &value
                        )?;
                    },
                    RecordCmds::Delete{record_id} => {
                        provider.delete_record(record_id)?;
                    },
                    // _ => unimplemented!("Unimplemented option: {:?}", param.cmd),
                }
            },
            Args::Interface{interface} => {
                for (n, i) in interface_list(interface)? {
                    eprintln!("{}: {}", n, i);
                }

            }
            // _ => unimplemented!("Unimplemented option: {:?}", param.cmd),
        };}
    };

    Ok(())
}

// use curl::easy::Easy;

// // Capture output into a local `Vec`.
// fn main() {
//     let mut dst = Vec::new();
//     let mut easy = Easy::new();
//     easy.url("https://www.rust-lang.org/").unwrap();

//     {
//         let mut transfer = easy.transfer();
//         transfer.write_function(|data| {
//             dst.extend_from_slice(data);
//             Ok(data.len())
//         }).unwrap();
//         transfer.perform().unwrap();
//     }
//     let dst_s = String::from_utf8(dst).unwrap();
//     println!("{}", dst_s);

//     let mut dst = Vec::new();
//     easy.url("http://baidu.com/").unwrap();

//     {
//         let mut transfer = easy.transfer();
//         transfer.write_function(|data| {
//             dst.extend_from_slice(data);
//             Ok(data.len())
//         }).unwrap();
//         transfer.perform().unwrap();
//     }
//     let dst_s = String::from_utf8(dst).unwrap();
//     println!("{}", dst_s);
// }