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

//! A CLI tool for [`rustypaste`].
//!
//! [`rustypaste`]: https://github.com/orhun/rustypaste
#![warn(missing_docs, clippy::unwrap_used)]

/// Command-line argument parser.
pub mod args;
/// Configuration file parser.
pub mod config;
/// Custom error implementation.
pub mod error;
/// Upload handler.
pub mod upload;

use crate::args::Args;
use crate::config::Config;
use crate::error::Result;
use crate::upload::Uploader;
use colored::Colorize;
use std::fs;
use std::io::{self, Read};

/// Runs `rpaste`.
pub fn run(args: Args) -> Result<()> {
    let mut config = if let Some(ref config_path) = args.config {
        toml::from_str(&fs::read_to_string(&config_path)?)?
    } else if let Some(path) = dirs_next::home_dir()
        .map(|p| p.join(".rustypaste").join("config.toml"))
        .map(|p| p.exists().then(|| p))
        .flatten()
    {
        toml::from_str(&fs::read_to_string(&path)?)?
    } else {
        Config::default()
    };
    config.update_from_args(&args);

    let mut results = Vec::new();
    let uploader = Uploader::new(&config);
    if let Some(ref url) = args.url {
        results.push(uploader.upload_url(url));
    } else if args.files.contains(&String::from("-")) {
        let mut buffer = String::new();
        io::stdin().read_to_string(&mut buffer)?;
        results.push(uploader.upload_stream(buffer.as_bytes()));
    } else {
        for file in args.files.iter() {
            results.push(uploader.upload_file(file))
        }
    }

    let format_padding = args
        .prettify
        .then(|| results.iter().map(|v| v.0.len()).max())
        .flatten()
        .unwrap_or(1);
    for (data, result) in results.iter().map(|v| (v.0, v.1.as_ref())) {
        let data = if args.prettify {
            format!(
                "{:p$} {} ",
                data,
                if result.is_ok() {
                    "=>".green().bold()
                } else {
                    "=>".red().bold()
                },
                p = format_padding,
            )
        } else {
            String::new()
        };
        match result {
            Ok(url) => println!("{}{}", data, url.trim()),
            Err(e) => eprintln!("{}{}", data, e),
        }
    }

    Ok(())
}