cargo-binstall 1.18.1

use std::{
    fs::{create_dir_all, File},
    io::{Read, Seek as _, Write as _},
    path::{Path, PathBuf},
};

use fs_lock::FileLock;
use miette::{miette, IntoDiagnostic, Result, WrapErr};
use serde::{Deserialize, Serialize};
use tempfile::NamedTempFile;
use tracing::{debug, warn};

use crate::args::{Args, StrategyWrapped};

#[derive(Clone, Debug, Deserialize, Serialize)]
#[serde(default)]
pub struct Settings {
    pub confirm: bool,
    pub install_path: Option<PathBuf>,
    pub track_installs: bool,
    pub continue_on_failure: bool,
    pub targets: Option<Vec<String>>,
    pub strategies: Vec<StrategyWrapped>,
    pub telemetry: Telemetry,
}

impl Default for Settings {
    fn default() -> Self {
        Self {
            confirm: true,
            install_path: None,
            track_installs: true,
            continue_on_failure: false,
            targets: None,
            strategies: vec![],
            telemetry: Telemetry::default(),
        }
    }
}

#[derive(Clone, Debug, Deserialize, Serialize)]
pub struct Telemetry {
    pub enabled: bool,
    pub consent_asked: bool,
}

impl Default for Telemetry {
    fn default() -> Self {
        Self {
            enabled: true,
            consent_asked: false,
        }
    }
}

impl Settings {
    pub(crate) fn telemetry_consent(&mut self, enable: bool) {
        self.telemetry.consent_asked = true;
        self.telemetry.enabled = enable;
    }

    pub(crate) fn merge_args(mut self, args: &Args) -> Self {
        if let Some(path) = &args.install_path {
            self.install_path = Some(path.clone());
        }
        if args.no_confirm {
            self.confirm = false;
        }
        if args.disable_telemetry {
            self.telemetry.enabled = false;
        }
        if args.no_track {
            self.track_installs = false;
        }
        if args.continue_on_failure {
            self.continue_on_failure = true;
        }
        if let Some(targets) = &args.targets {
            self.targets = Some(targets.clone());
        }
        if !args.strategies.is_empty() {
            self.strategies = args.strategies.clone();
        }
        if !self.telemetry.consent_asked {
            self.telemetry.enabled = false;
        }
        self
    }

    fn write(&self, file: &mut File) -> Result<()> {
        let mut write_all_and_set_len = |data| {
            file.write_all(data)?;
            file.set_len(data.len().try_into().unwrap())
        };
        write_all_and_set_len(
            toml::to_string_pretty(self)
                .into_diagnostic()
                .wrap_err("serialise default settings")?
                .as_bytes(),
        )
        .into_diagnostic()
        .wrap_err("write settings")
    }

    pub(crate) fn save(&self, path: &Path) -> Result<()> {
        let mut file = File::options()
            .read(true)
            .write(true)
            .append(false)
            .open(path)
            .and_then(FileLock::new_exclusive)
            .into_diagnostic()
            .wrap_err("open settings file")?;
        let mut settings = Self::read_from_file(&mut file)?;
        settings.telemetry = self.telemetry.clone();
        file.rewind()
            .into_diagnostic()
            .wrap_err("rewinding settings file for writing")?;
        settings.write(&mut file)
    }

    fn read_from_file(file: &mut File) -> Result<Self> {
        let mut contents = String::new();
        file.read_to_string(&mut contents)
            .into_diagnostic()
            .wrap_err("read existing settings file")?;

        toml::from_str(&contents)
            .into_diagnostic()
            .wrap_err("parse existing settings file")
    }
}

pub fn load(error_if_inaccessible: bool, path: &Path) -> Result<Settings> {
    fn inner(path: &Path) -> Result<Settings> {
        let parent = path
            .parent()
            .ok_or_else(|| miette!("settings path has no parent"))?;

        create_dir_all(parent)
            .into_diagnostic()
            .wrap_err("create settings directory")?;

        debug!(?path, "checking if settings file exists");
        if !path.exists() {
            debug!(?path, "trying to create new settings file");

            let mut tempfile = NamedTempFile::new_in(parent)
                .into_diagnostic()
                .wrap_err("creating new temporary settings file")?;

            let settings = Settings::default();
            settings
                .write(tempfile.as_file_mut())
                .wrap_err("for new temporary settings file")?;

            if tempfile.persist_noclobber(path).is_ok() {
                return Ok(settings);
            }
        }

        debug!(?path, "loading binstall settings");
        let mut file = File::open(path)
            .and_then(FileLock::new_shared)
            .into_diagnostic()
            .wrap_err("open existing settings file")?;

        let settings = Settings::read_from_file(&mut file)?;

        debug!(?settings, "loaded binstall settings");
        Ok(settings)
    }

    let settings = inner(path);
    if error_if_inaccessible {
        settings
    } else {
        Ok(settings
            .inspect_err(|err| {
                warn!("{err:?}");
            })
            .unwrap_or_default())
    }
}