// Jackson Coxson
// Just lists apps for now

use idevice::{
    lockdownd::{self, LockdowndClient},
    mounter::ImageMounter,
    pairing_file::PairingFile,
    Idevice,
};

use sha2::{Digest, Sha384};

use std::{
    net::{Ipv4Addr, SocketAddrV4},
    str::FromStr,
};

#[tokio::main]
async fn main() {
    env_logger::init();
    let mut host = None;
    let mut pairing_file = None;

    // Loop through args
    let mut i = 0;
    while i < std::env::args().len() {
        match std::env::args().nth(i).unwrap().as_str() {
            "--host" => {
                host = Some(std::env::args().nth(i + 1).unwrap().to_string());
                i += 2;
            }
            "--pairing-file" => {
                pairing_file = Some(std::env::args().nth(i + 1).unwrap().to_string());
                i += 2;
            }
            "-h" | "--help" => {
                println!("ideviceinfo - get information from the idevice");
                println!("Usage:");
                println!("  ideviceinfo [options]");
                println!("Options:");
                println!("  --host <host>");
                println!("  --pairing_file <path>");
                println!("  -h, --help");
                println!("  --about");
                println!("\n\nSet RUST_LOG to info, debug, warn, error, or trace to see more logs. Default is error.");
                return;
            }
            "--about" => {
                println!("mounter - mount developer disk images on the device. Reimplementation of pymobiledevice3's program.");
                println!("Copyright (c) 2025 Jackson Coxson");
            }
            _ => {
                i += 1;
            }
        }
    }
    if host.is_none() {
        println!("Invalid arguments! Pass the IP of the device with --host");
        return;
    }
    if pairing_file.is_none() {
        println!("Invalid arguments! Pass the path the the pairing file with --pairing-file");
        return;
    }
    let ip = Ipv4Addr::from_str(host.unwrap().as_str()).unwrap();
    let socket = SocketAddrV4::new(ip, lockdownd::LOCKDOWND_PORT);

    let socket = tokio::net::TcpStream::connect(socket).await.unwrap();
    let socket = Box::new(socket);
    let idevice = Idevice::new(socket, "mounter_client");

    let p = PairingFile::read_from_file(pairing_file.as_ref().unwrap()).unwrap();

    let mut lockdown_client = LockdowndClient { idevice };
    lockdown_client.start_session(&p).await.unwrap();

    let (port, _) = lockdown_client
        .start_service("com.apple.mobile.mobile_image_mounter")
        .await
        .unwrap();

    let socket = SocketAddrV4::new(ip, port);
    let socket = tokio::net::TcpStream::connect(socket).await.unwrap();
    let socket = Box::new(socket);
    let mut idevice = Idevice::new(socket, "mounter_client");

    let p = PairingFile::read_from_file(pairing_file.unwrap()).unwrap();

    idevice.start_session(&p).await.unwrap();

    let mut mounter_client = ImageMounter::new(idevice);
    let images = mounter_client.copy_devices().await.unwrap();
    println!("Images: {images:#?}");

    let image = std::fs::read("Image.dmg").unwrap();
    let mut hasher = Sha384::new();
    hasher.update(image);
    let hash = hasher.finalize();

    let manifest = mounter_client
        .query_personalization_manifest("DeveloperDiskImage", hash.to_vec())
        .await
        .unwrap();
    println!("len: {}", manifest.len());
}