rdest 0.1.0

// Copyright 2020 Mateusz Janda.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// https://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or https://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.

use rdest::peer_id;
use rdest::{Metainfo, Session};
use std::path::PathBuf;
use structopt::StructOpt;
use tokio;

#[derive(StructOpt)]
#[structopt(
    name = "rdest",
    author = "Mateusz Janda <mateusz.janda@gmail.com>",
    about = r#"
   _i_i_     rdest is a simple BitTorrent client, currently supporting BEP3
⸝⸍/     \⸌⸜  specification. If you are interested, you can find name origin
||\  ¬  /||  at youtu.be/4el6qQ40BQw?t=4200
\_,"" ""._/
"#
)]
enum Opt {
    /// Get torrent files from p2p network
    Get(Get),
    /// Create .torrent file
    Create(Create),
}

#[derive(StructOpt)]
struct Get {
    /// Path to .torrent file
    #[structopt(parse(from_os_str), name = "PATH")]
    path: PathBuf,
}

#[derive(StructOpt)]
struct Create {
    /// Create .torrent for file
    #[structopt(parse(from_os_str), name = "FILE")]
    path: PathBuf,
    /// Tracker address
    #[structopt(short, long, name = "ADDRESS")]
    tracker_addr: String,
}

#[tokio::main]
async fn main() {
    match Opt::from_args() {
        Opt::Get(get) => get_torrent(&get.path).await,
        Opt::Create(create) => create_torrent(&create.path, &create.tracker_addr).await,
    };
}

async fn get_torrent(path: &PathBuf) {
    let metainfo = match Metainfo::from_file(path.as_path()) {
        Ok(metainfo) => metainfo,
        Err(e) => panic!("[-] Can't read metafile. Error: {}", e),
    };
    let mut session = Session::new(metainfo, peer_id::generate());
    session.run().await;
}

async fn create_torrent(path: &PathBuf, tracker_addr: &String) {
    match Metainfo::create_file(path.as_path(), tracker_addr) {
        Ok(()) => (),
        Err(e) => panic!("[-] Can't create metafile. Error: {}", e),
    }
}