// Jackson Coxson

use std::path::PathBuf;

use clap::{Arg, Command, arg, value_parser};
use idevice::{IdeviceService, misagent::MisagentClient};

mod common;

#[tokio::main]
async fn main() {
    env_logger::init();

    let matches = Command::new("core_device_proxy_tun")
        .about("Start a tunnel")
        .arg(
            Arg::new("host")
                .long("host")
                .value_name("HOST")
                .help("IP address of the device"),
        )
        .arg(
            Arg::new("pairing_file")
                .long("pairing-file")
                .value_name("PATH")
                .help("Path to the pairing file"),
        )
        .arg(
            Arg::new("udid")
                .value_name("UDID")
                .help("UDID of the device (overrides host/pairing file)"),
        )
        .arg(
            Arg::new("about")
                .long("about")
                .help("Show about information")
                .action(clap::ArgAction::SetTrue),
        )
        .subcommand(
            Command::new("list")
                .about("Lists the images mounted on the device")
                .arg(
                    arg!(-s --save <FOLDER> "the folder to save the profiles to")
                        .value_parser(value_parser!(PathBuf)),
                ),
        )
        .subcommand(
            Command::new("remove")
                .about("Remove a provisioning profile")
                .arg(Arg::new("id").required(true).index(1)),
        )
        .get_matches();

    if matches.get_flag("about") {
        println!(
            "mounter - query and manage images mounted on a device. Reimplementation of libimobiledevice's binary."
        );
        println!("Copyright (c) 2025 Jackson Coxson");
        return;
    }

    let udid = matches.get_one::<String>("udid");
    let host = matches.get_one::<String>("host");
    let pairing_file = matches.get_one::<String>("pairing_file");

    let provider = match common::get_provider(udid, host, pairing_file, "misagent-jkcoxson").await {
        Ok(p) => p,
        Err(e) => {
            eprintln!("{e}");
            return;
        }
    };
    let mut misagent_client = MisagentClient::connect(&*provider)
        .await
        .expect("Unable to connect to misagent");

    if let Some(matches) = matches.subcommand_matches("list") {
        let images = misagent_client
            .copy_all()
            .await
            .expect("Unable to get images");
        if let Some(path) = matches.get_one::<PathBuf>("save") {
            tokio::fs::create_dir_all(path)
                .await
                .expect("Unable to create save DIR");

            for (index, image) in images.iter().enumerate() {
                let f = path.join(format!("{index}.pem"));
                tokio::fs::write(f, image)
                    .await
                    .expect("Failed to write image");
            }
        }
    } else if let Some(matches) = matches.subcommand_matches("remove") {
        let id = matches.get_one::<String>("id").expect("No ID passed");
        misagent_client.remove(id).await.expect("Failed to remove");
    } else {
        eprintln!("Invalid usage, pass -h for help");
    }
}