use std::{
    io::{stdout, Write},
    path::PathBuf,
};

use anyhow::Result;
use argp::FromArgs;

use crate::{
    cmd,
    util::{
        alf::AlfFile,
        file::{buf_writer, map_file},
        reader::{Endian, FromReader},
    },
};

#[derive(FromArgs, PartialEq, Debug)]
/// Commands for processing NVIDIA Shield TV alf files.
#[argp(subcommand, name = "alf")]
pub struct Args {
    #[argp(subcommand)]
    command: SubCommand,
}

#[derive(FromArgs, PartialEq, Debug)]
#[argp(subcommand)]
enum SubCommand {
    Info(InfoArgs),
    Hashes(HashesArgs),
}

#[derive(FromArgs, PartialEq, Debug)]
/// Prints information about an alf file. (Same as `dol info`)
#[argp(subcommand, name = "info")]
pub struct InfoArgs {
    #[argp(positional)]
    /// alf file
    file: PathBuf,
}

#[derive(FromArgs, PartialEq, Debug)]
/// Extracts symbol hashes from an alf file.
#[argp(subcommand, name = "hashes")]
pub struct HashesArgs {
    #[argp(positional)]
    /// alf file
    alf_file: PathBuf,
    #[argp(positional)]
    /// output file
    output: Option<PathBuf>,
}

pub fn run(args: Args) -> Result<()> {
    match args.command {
        SubCommand::Info(c_args) => info(c_args),
        SubCommand::Hashes(c_args) => hashes(c_args),
    }
}

fn hashes(args: HashesArgs) -> Result<()> {
    let alf_file = {
        let file = map_file(&args.alf_file)?;
        let mut reader = file.as_reader();
        AlfFile::from_reader(&mut reader, Endian::Little)?
    };
    let mut w: Box<dyn Write> = if let Some(output) = args.output {
        Box::new(buf_writer(output)?)
    } else {
        Box::new(stdout())
    };
    let mut symbols = alf_file.symbols.clone();
    symbols.sort_by_key(|s| s.address);
    for symbol in symbols {
        writeln!(
            w,
            "{:#010X} | {} | {:?} | {} | {} | {:#X}",
            symbol.address,
            symbol.section,
            symbol.kind,
            symbol.name,
            symbol.demangled_name,
            symbol.size
        )?;
    }
    w.flush()?;
    Ok(())
}

fn info(args: InfoArgs) -> Result<()> {
    cmd::dol::info(cmd::dol::InfoArgs { dol_file: args.file, selfile: None })
}