command_override.rs

use std::{
    env::current_dir,
    path::{Path, PathBuf},
};

use anyhow::{bail, Context, Result};
use cli_table::{
    format::{Border, HorizontalLine, Separator},
    print_stdout, ColorChoice, Table, WithTitle,
};
use itertools::Itertools;

use crate::{
    config_file::{load_config_db, load_mut_config_db, save_config_db, JuliaupOverride},
    global_paths::GlobalPaths,
};

#[derive(Table)]
struct OverrideRow {
    #[table(title = "Path")]
    path: String,
    #[table(title = "Channel")]
    channel: String,
}

pub fn run_command_override_status(paths: &GlobalPaths) -> Result<()> {
    let config_file = load_config_db(paths, None)
        .with_context(|| "`override status` command failed to load configuration file.")?;

    let rows_in_table: Vec<_> = config_file
        .data
        .overrides
        .iter()
        .sorted_by_key(|i| i.path.to_string())
        .map(|i| -> OverrideRow {
            OverrideRow {
                path: dunce::simplified(&PathBuf::from(&i.path))
                    .to_string_lossy()
                    .to_string(),
                channel: i.channel.to_string(),
            }
        })
        .collect();

    print_stdout(
        rows_in_table
            .with_title()
            .color_choice(ColorChoice::Never)
            .border(Border::builder().build())
            .separator(
                Separator::builder()
                    .title(Some(HorizontalLine::new('1', '2', '3', '-')))
                    .build(),
            ),
    )?;

    Ok(())
}

pub fn run_command_override_set(
    paths: &GlobalPaths,
    channel: String,
    path: Option<String>,
) -> Result<()> {
    let mut config_file = load_mut_config_db(paths)
        .with_context(|| "`override set` command failed to load configuration data.")?;

    if !config_file.data.installed_channels.contains_key(&channel) {
        bail!(
            "'{}' is not installed. Please run `juliaup add {}` to install channel or version.",
            &channel,
            &channel
        );
    }

    let path = match path {
        Some(path) => PathBuf::from(path),
        None => current_dir()?,
    }
    .canonicalize()?;

    if let Some(i) = config_file
        .data
        .overrides
        .iter_mut()
        .find(|i| i.path == path.to_string_lossy())
    {
        if i.channel == channel {
            eprintln!(
                "Override already set to '{}' for '{}'.",
                channel,
                path.to_string_lossy()
            );
            return Ok(());
        } else {
            eprintln!(
                "Override changed from '{}' to '{}' for '{}'.",
                i.channel,
                channel,
                path.to_string_lossy()
            );
            i.channel = channel.clone();

            save_config_db(&mut config_file).with_context(|| {
                "Failed to save configuration file from `override set` command."
            })?;
            return Ok(());
        }
    }

    config_file.data.overrides.push(JuliaupOverride {
        path: path.to_string_lossy().to_string(),
        channel: channel.clone(),
    });
    eprintln!(
        "Override set to '{}' for '{}'.",
        channel,
        path.to_string_lossy()
    );

    save_config_db(&mut config_file)
        .with_context(|| "Failed to save configuration file from `override set` command.")?;

    Ok(())
}

pub fn run_command_override_unset(
    paths: &GlobalPaths,
    nonexistent: bool,
    path: Option<String>,
) -> Result<()> {
    let mut config_file = load_mut_config_db(paths)
        .with_context(|| "`override unset` command failed to load configuration data.")?;

    let path = match path {
        Some(path) => PathBuf::from(path),
        None => current_dir()?,
    }
    .canonicalize()?;

    if nonexistent {
        config_file
            .data
            .overrides
            .retain(|x| Path::new(&x.path).is_dir());
    } else {
        // First remove any duplicates
        config_file
            .data
            .overrides
            .retain(|x| Path::new(&x.path) != path);
    }

    save_config_db(&mut config_file)
        .with_context(|| "Failed to save configuration file from `override add` command.")?;

    Ok(())
}