use anyhow::Result;
use clap::Parser;

use crate::{event, module, debug, apk_sign};

/// KernelSU userspace cli
#[derive(Parser, Debug)]
#[command(author, version, about, long_about = None)]
struct Args {
    #[command(subcommand)]
    command: Commands,
}

#[derive(clap::Subcommand, Debug)]
enum Commands {
    /// Start KernelSU userspace daemon
    Daemon,

    /// Manage KernelSU modules
    Module {
        #[command(subcommand)]
        command: Module,
    },

    /// Trigger `post-fs-data` event
    PostFsData,

    /// Trigger `service` event
    Services,

    /// Trigger `boot-complete` event
    BootCompleted,

    /// Install KernelSU userspace component to system
    Install,

    /// SELinux policy Patch tool
    Sepolicy,

    /// For developers
    Debug {
        #[command(subcommand)]
        command: Debug,
    },
}
#[derive(clap::Subcommand, Debug)]
enum Debug {
    /// Set the manager app, kernel CONFIG_KSU_DEBUG should be enabled.
    SetManager {
        /// manager package name
        #[arg(default_value_t = String::from("me.weishu.kernelsu"))]
        apk: String,
    },

    /// Get apk size and hash
    GetSign {
        /// apk path
        apk: String,
    },

    /// For testing
    Test,
}

#[derive(clap::Subcommand, Debug)]
enum Module {
    /// Install module <ZIP>
    Install {
        /// module zip file path
        zip: String,
    },

    /// Uninstall module <id>
    Uninstall {
        /// module id
        id: String,
    },

    /// enable module <id>
    Enable {
        /// module id
        id: String,
    },

    /// disable module <id>
    Disable {
        // module id
        id: String,
    },

    /// list all modules
    List,
}

pub fn run() -> Result<()> {
    let cli = Args::parse();

    let result = match cli.command {
        Commands::Daemon => event::daemon(),
        Commands::PostFsData => event::on_post_data_fs(),
        Commands::BootCompleted => event::on_boot_completed(),

        Commands::Module { command } => {
            env_logger::init();

            match command {
                Module::Install { zip } => module::install_module(zip),
                Module::Uninstall { id } => module::uninstall_module(id),
                Module::Enable { id } => module::enable_module(id),
                Module::Disable { id } => module::disable_module(id),
                Module::List => module::list_modules(),
            }
        }
        Commands::Install => event::install(),
        Commands::Sepolicy => todo!(),
        Commands::Services => event::on_services(),

        Commands::Debug { command } => {
            match command {
                Debug::SetManager { apk } => debug::set_manager(&apk),
                Debug::GetSign { apk } => { 
                    let sign = apk_sign::get_apk_signature(&apk)?;
                    println!("size: {:#x}, hash: {:#x}", sign.0, sign.1);
                    Ok(())
                },
                Debug::Test => todo!(),
            }
        }
    };

    if let Err(e) = &result {
        log::error!("Error: {}", e);
    }
    result
}