ph-dl 1.3.2

Get highest quality direct link to download PH videos, or download them automatically.
Documentation 
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

use clap::{crate_version, App, Arg};
use regex::Regex;
use std::{
    fs::File,
    io::{BufRead, BufReader},
    process,
};

fn main() {
    let matches = App::new("ph-dl")
        .version(crate_version!())
        .author("Miika Launiainen <miika.launiainen@gmail.com>")
        .about("Get highest quality direct link to download PH videos, or download them automatically.")
        .arg(
            Arg::new("url")
                .about("view key to the video or URL to the video/playlist")
                .required(true)
                .conflicts_with("file"),
        )
        .arg(
            Arg::new("download")
                .short('d')
                .long("download")
                .about("Download the video automatically"),
        )
        .arg(
            Arg::new("show_quality")
                .short('Q')
                .long("show_quality")
                .about("Print all available resolutions"),
        )
        .arg(
            Arg::new("quality")
                .short('q')
                .long("quality")
                .takes_value(true)
                .possible_values(&["1", "2", "3", "4"])
                .about("Use specific resolution"),
        )
        .arg(
            Arg::new("file")
                .short('f')
                .long("file")
                .takes_value(true)
                .about("Download videos from file")
                .conflicts_with("url"),
        )
        .get_matches();

    if matches.is_present("file") {
        download_from_file(matches.value_of_t_or_exit("file"));
        process::exit(0);
    }

    let url: String = matches.value_of_t_or_exit("url");
    let download = matches.is_present("download");
    let show_quality = matches.is_present("show_quality");
    let quality: String = if matches.is_present("quality") {
        matches.value_of_t_or_exit("quality")
    } else {
        "none".to_string()
    };

    let url = check_for_ph_url(&url);

    ph_dl::run(url, download, show_quality, quality);
}

fn check_for_ph_url(url: &str) -> String {
    let normal_url =
        Regex::new(r"(https?://(www\.)?pornhub\.com/view_video\.php\?viewkey=[0-9a-รถ]*)[/&]?")
            .unwrap();
    let playlist = Regex::new(r"https?://(www\.)pornhub.com/playlist/\d+").unwrap();

    if normal_url.is_match(url) {
        let url = match normal_url.captures(url) {
            Some(i) => i.get(1).unwrap().as_str().to_string(),
            None => "Error".to_string(),
        };
        url
    } else if playlist.is_match(url) {
        url.to_string()
    } else {
        return format!("https://www.pornhub.com/view_video.php?viewkey={}", url);
    }
}

fn download_from_file(path: String) {
    let file = File::open(&path);

    match file {
        Ok(file) => {
            let reader = BufReader::new(&file);
            for line in reader.lines() {
                let url = check_for_ph_url(line.unwrap().as_str());
                ph_dl::run(url, true, false, "none".to_string());
            }
        }
        Err(e) => {
            eprintln!("{}", e);
            process::exit(1);
        }
    };
}