putioarr 0.1.1

put.io to sonarr/radarr proxy
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
120
121
122
123
124
125
126
127
128
129
130

use std::time::Duration;

use actix_web::{web, App, HttpServer};
use appdata::AppData;
use clap::{Args, Parser, Subcommand};
use tokio::time::sleep;

mod appdata;
mod downloader;
mod handlers;
mod oob;
mod putio;
mod routes;
mod transmission;

/// put.io to sonarr/radarr proxy
#[derive(Parser, Debug)]
#[command(author, version, about, long_about = None)]
struct Cli {
    #[command(subcommand)]
    command: Commands,
}

#[derive(Subcommand, Debug)]
enum Commands {
    /// Run the proxy
    Run(RunArgs),
    /// Generate a put.io API token
    GetToken,
}

#[derive(Args, Debug)]
struct RunArgs {
    /// put.io API token
    #[arg(short, long)]
    api_token: String,

    /// File path where state is saved
    #[arg(short, long)]
    state_file: String,

    /// Directory where downloads are saved
    #[arg(short, long)]
    download_directory: String,

    #[arg(short, long, default_value_t = 7070)]
    port: u16,

    #[arg(short, long, default_value_t = String::from("0.0.0.0"))]
    bind_address: String,

    #[arg(short, long, default_value_t = String::from("info"))]
    loglevel: String,

    /// UID of the user owning the donwloads
    #[arg(short, long, default_value_t = 1000)]
    uid: u32,
}

#[actix_web::main]
async fn main() -> std::io::Result<()> {
    let cli = Cli::parse();

    match &cli.command {
        Commands::Run(args) => {
            std::env::set_var("RUST_LOG", args.loglevel.as_str());
            env_logger::init_from_env(env_logger::Env::new().default_filter_or("info"));

            let app_data = web::Data::new(
                AppData::new(
                    args.api_token.clone(),
                    args.state_file.clone(),
                    args.download_directory.clone(),
                    args.uid,
                )
                .await
                .unwrap(),
            );

            // Actix background jobs
            let data_for_background = app_data.clone();
            actix_rt::spawn(async { downloader::start_downloader_task(data_for_background).await });

            // let http_server_data = data.clone();
            HttpServer::new(move || {
                App::new()
                    // .wrap(Logger::new(
                    //     "%a \"%r\" %s %b \"%{Referer}i\" \"%{User-Agent}i\" %T",
                    // ))
                    .app_data(app_data.clone())
                    .service(routes::rpc_post)
                    .service(routes::rpc_get)
            })
            .bind(("0.0.0.0", args.port))?
            .run()
            .await
        }
        Commands::GetToken => {
            // Create new OOB code and prompt user to link
            let oob_code = oob::get().await.expect("fetching OOB code");
            println!(
                "Go to https://put.io/link and enter the code: {:#?}",
                oob_code
            );
            println!("Waiting for link...");

            // Every three seconds, check if the OOB code was linked to the user's account
            // If linked, update the config file
            // Stops after 10 tries (30 seconds)
            let three_seconds = Duration::from_secs(3);

            loop {
                sleep(three_seconds).await;

                let get_oauth_token_result = oob::check(oob_code.clone()).await;

                match get_oauth_token_result {
                    Ok(token) => {
                        println!("Put.io API token: {token}");
                        break;
                    }
                    Err(_error) => {
                        continue;
                    }
                };
            }
            Ok(())
        }
    }
}