pryr 0.4.1

use anyhow::{Context, Ok};
use clap::{Args, Parser, Subcommand};
use pryr::cli::PrettyPrint;
use pryr::config::{Config, Location};
use pryr::ipc::IpcRequest;
use pryr::ipc::{IpcResponse, connect_ipc};
use pryr::prayers::{MadhabLocal, MethodLocal};
use pryr::system::get_config_path;
use pryr::updater::run_update;
use std::io::Write;
use std::io::{BufRead, BufReader};
use std::process;

const LOCATION_API_BASE_URL: &str = "https://nominatim.openstreetmap.org/search?format=json";

#[derive(Parser)]
#[command(version, about)]
struct Cli {
    #[command(subcommand)]
    command: Commands,
}

#[derive(Subcommand)]
enum Commands {
    Status(OutputArgs),
    Schedule(OutputArgs),
    ReloadConfig,
    Configure(ConfigureArgs),
    Update,
}

#[derive(Args)]
struct OutputArgs {
    #[arg(long)]
    json: bool,
}

#[derive(Args)]
#[group(required = true, multiple = true, id = "config_group")]
struct ConfigureArgs {
    #[arg(long, group = "config_group")]
    city: Option<String>,
    #[arg(long, group = "config_group")]
    method: Option<MethodLocal>,
    #[arg(long, group = "config_group")]
    madhab: Option<MadhabLocal>,
}

fn main() -> anyhow::Result<()> {
    let cli = Cli::parse();

    let (request, json_output) = match cli.command {
        Commands::Status(args) => (IpcRequest::GetStatus, args.json),
        Commands::Schedule(args) => (IpcRequest::GetTodaySchedule, args.json),
        Commands::ReloadConfig => (IpcRequest::ReloadConfig, false),
        Commands::Configure(args) => {
            handle_configure(&args)?;
            (IpcRequest::ReloadConfig, false)
        }
        Commands::Update => {
            run_update()?;
            return Ok(());
        }
    };

    let json_response = send_ipc_request(&request)?;
    let response: IpcResponse = serde_json::from_str(&json_response)?;

    if json_output {
        println!("{json_response}");
    } else {
        response.pretty_print();
    }

    Ok(())
}

// TODO: Make a menu with selections so the user won't pick wrong location
fn get_location(city: &str) -> Option<Location> {
    let mut res = ureq::get(format!(
        "{LOCATION_API_BASE_URL}&q=${}",
        city.replace(" ", "%20")
    ))
    .call()
    .expect("Couldn't send request to Location API");
    let res = res
        .body_mut()
        .read_to_string()
        .expect("Couldn't read API response");

    let value: serde_json::Value =
        serde_json::from_str(&res).expect("Couldn't convert response to JSON");
    let locations = value.as_array()?;

    let first_match = locations.first()?;

    Some(Location {
        long: first_match["lon"]
            .as_str()?
            .parse::<f64>()
            .expect("Couldn't parse longitude"),
        lat: first_match["lat"]
            .as_str()?
            .parse::<f64>()
            .expect("Couldn't parse latitude"),
    })
}

fn handle_configure(args: &ConfigureArgs) -> anyhow::Result<()> {
    let config_path = get_config_path().context("Could not get config path")?;
    let mut config = Config::from_file(&config_path)?;

    if let Some(ref city) = args.city {
        match get_location(city) {
            Some(location) => config.location = location,
            None => {
                eprintln!("Could not find location for this city");
                process::exit(1)
            }
        }
    }

    if let Some(method) = args.method {
        config.prayer_time.method = method;
    }

    if let Some(madhab) = args.madhab {
        config.prayer_time.madhab = madhab;
    }

    config.save(&config_path)?;

    Ok(())
}

fn send_ipc_request(request: &IpcRequest) -> anyhow::Result<String> {
    let mut stream = connect_ipc().context("Could not connect to daemon. Is pryrd running?")?;

    let request_string = serde_json::to_string(&request)? + "\n";
    stream.write_all(request_string.as_bytes())?;
    stream.flush()?;

    let mut buf_reader = BufReader::new(&stream);
    let mut s = String::new();

    buf_reader.read_line(&mut s)?;

    Ok(s)
}