use std::path::PathBuf;
use formatx::formatx;
use clap::{Parser, Subcommand};
use mwbot::parsoid::WikiMultinode;
use dirs::config_dir;

const CONFIG_TEMPLATE_PATH: &'static str = "mwbot_template.toml";
const BOT_CONFIG_FILENAME: &'static str = "mwbot.toml";

fn get_bot_config_path() -> PathBuf {
    config_dir().unwrap().join(BOT_CONFIG_FILENAME)
}

#[derive(Parser)]
#[command(version, about, long_about = None)]
struct Cli {
    /// What action to perform
    #[command(subcommand)]
    action: Action,
}

#[derive(Parser, Clone, Eq, PartialEq, PartialOrd, Ord)]
struct SetupArgs {
    /// Specify the username of the bot
    #[arg(long, env = "MW_USERNAME")]
    username: String,
    /// Specify the botpassword of the bot
    #[arg(long, env = "MW_BOTPASSWORD")]
    botpassword: String,
    /// Specify the OAuth2 token of the bot
    #[arg(long, env = "MW_OAUTH2")]
    oauth2_token: String,
    /// Specify the API URL of the bot
    #[arg(long, env = "MW_API_URL")]
    api_url: String,
    /// Specify the REST URL of the bot
    #[arg(long, env = "MW_REST_URL")]
    rest_url: String,
    
}

#[derive(Clone, PartialEq, Eq, PartialOrd, Ord, Subcommand)]
enum Action {
    /// Move config to ~/.config
    Setup(SetupArgs),
    /// Run the bot
    Run,
}

fn setup(args: SetupArgs) -> Result<(), std::io::Error> {
    let config_template = std::fs::read_to_string(CONFIG_TEMPLATE_PATH).or(Err(std::io::Error::new(std::io::ErrorKind::NotFound, format!("Unable to find {}; did you execute the script from the same directory?", CONFIG_TEMPLATE_PATH))))?;
    let filled_in_config = formatx!(config_template, args.api_url, args.rest_url, args.username, args.botpassword, args.oauth2_token).unwrap();
    let path = get_bot_config_path();
    std::fs::write(&path, filled_in_config).unwrap();
    // Fix permissions on UNIX-like systems, since mwbot-rs doesn't like to read configs with loose permissions.
    {
        use std::os::unix::fs::PermissionsExt;
        std::fs::set_permissions(path, std::fs::Permissions::from_mode(0o600))?;
    }
    Ok(())
}

#[tokio::main]
async fn main() -> Result<(), std::io::Error> {
    dotenvy::dotenv().inspect_err(|_| eprintln!("Couldn't load .env file; please make sure to create one in the same directory as executing from!")).ok();

    let cli = Cli::parse();

    match cli.action {
        Action::Setup(args) => {
            setup(args)?;
            eprintln!("Successfully set up {}.", get_bot_config_path().to_str().unwrap())
        },
        Action::Run => {
            let bot = mwbot::Bot::from_default_config().await.unwrap();
            let page = bot.page("Bulgaria").unwrap();
            let html = page.html().await.unwrap().into_mutable();
            println!("{:?}", html);
            println!("{:?}", html.as_nodes());
        }
    }
    Ok(())
}