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

mod commands;
mod config;
mod data;
mod parser;

use commands::{insert, list, remove};
use config::Config;
use data::read_data_file;

use std::io::stdin;
use structopt::StructOpt;

/// Call command functions based on given options
pub fn run() {
    let opts = Config::from_args();
    let mut data = read_data_file();

    // Run based on options
    if opts.insert {
        // Read from stdin
        let mut input_url = String::new();
        stdin().read_line(&mut input_url).unwrap();
        input_url = input_url.trim().to_owned();

        insert(input_url, data, opts);
    } else if opts.list {
        list(data, opts);
    } else if opts.remove.is_some() {
        remove(&mut data, opts.remove.unwrap());
    }
}