cosmoline/src/main.rs

use clap::{load_yaml, App};
use ed25519_dalek::Keypair;

use async_std::fs;
use async_std::net::UdpSocket;
use async_std::path::PathBuf;

mod keypair;

use keypair::SSBKeypair;

type Config = toml::map::Map<String, toml::Value>;

#[derive(Debug)]
struct Host {
    protocol: String,
    host: String,
    port: String,
    pubkey: String,
}

fn parse_packet(packet: String) -> Host {
    let mut packet = packet.splitn(4, ':');
    let protocol = packet.next().unwrap();
    let host = packet.next().unwrap();
    let port = packet.next().unwrap().splitn(2, '~').next().unwrap();
    let pubkey = packet.next().unwrap();
    Host {
        protocol: protocol.to_string(),
        host: host.to_string(),
        port: port.to_string(),
        pubkey: pubkey.to_string(),
    }
}

#[async_std::main]
async fn main() -> std::io::Result<()> {
    let options = load_yaml!("options.yaml");
    let options = App::from(options).get_matches();

    let config_file = match options.value_of("config") {
        Some(path) => PathBuf::from(path),
        None => PathBuf::from(dirs::config_dir()
            .unwrap()
            .join("cosmoline")
            .join("config.toml")),
    };
    let config = fs::read_to_string(config_file).await?;
    let config: Config = toml::from_str(config.as_str()).unwrap();

    let path = match options.value_of("path") {
        Some(path) => PathBuf::from(path),
        None => match config.get("path") {
            Some(path) => PathBuf::from(path.as_str().unwrap()),
            None => PathBuf::from(dirs::home_dir().unwrap().join(".cosmoline")),
        },
    };
    let keypair = Keypair::read_or_generate(path.join("secret")).await;
    println!("{}", keypair.to_json().pretty(2));

    let socket = UdpSocket::bind("0.0.0.0:8008").await?;
    let mut buf = [0u8; 1024];

    loop {
        let (amt, peer) = socket.recv_from(&mut buf).await?;
        let buf = &mut buf[..amt];
        let packet = String::from_utf8(buf.to_vec()).unwrap();
        println!("{:?}", (peer, parse_packet(packet)));
    }

}