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

mod config;
pub mod error;
mod opensubs;
mod utils;

use clap::ArgMatches;
use colored::Colorize;
use config::Config;
use dialoguer::{Confirm, Input, Password};
use error::{Error, Result};
use opensubs::{get_os_token, use_opensubs};
use std::fs;

pub fn run(matches: ArgMatches) -> Result<()> {
    let config = Config::new()?;

    let mut current_lang: &str = &config.lang;
    let os_token: &str = &config.os_token;

    if let Some(lang) = matches.value_of("lang") {
        current_lang = lang;
    }

    println!(
        "{} {}",
        "Current Language:".green(),
        current_lang.to_uppercase()
    );

    match matches.values_of("file") {
        Some(files) => {
            if os_token.len() > 0 {
                use_opensubs(files, current_lang, os_token)?
            } else {
                println!("{}", "[!] No OpenSubtitles token found.".red().bold());
                authenticate_os_user()?;
                use_opensubs(files, current_lang, os_token)?
            }
        }
        None => {
            println!("{}", "[!] No files provided.".red().bold());
        }
    }

    Ok(())
}

pub fn authenticate_os_user() -> Result<()> {
    let mut config = Config::new()?;

    if Confirm::new()
        .with_prompt("Do you have an OpenSubtitles token?")
        .interact()?
    {
        let token: String = Input::new()
            .with_prompt("Input your token")
            .interact_text()?;
        config.set_os_token(token)?
    } else {
        println!("You can generate a token with your username & password.");
        let username: String = Input::new().with_prompt("Username").interact_text()?;
        let password: String = Password::new().with_prompt("Password").interact()?;
        let token = get_os_token(&username, &password)?;
        config.set_os_token(token)?
    }

    Ok(())
}

pub fn save_file(content: &str, filename: &str) -> Result<()> {
    println!("{} {}", "Saving to".green(), filename.italic());
    fs::write(filename, content).map_err(|e| Error::IO(e))
}

#[cfg(test)]
mod tests {
    use super::opensubs::hash;

    #[test]
    fn test_hash_fn() {
        assert_eq!(&hash("./test/breakdance.avi").unwrap(), "8e245d9679d31e12");
    }
}