// SPDX-License-Identifier: Apache-2.0
// SPDX-FileCopyrightText: 2020 The Gleam contributors

use ecow::EcoString;
use gleam_core::{
    build::Telemetry,
    error::{Error, StandardIoAction},
    manifest::{Changed, ChangedGit, PackageChanges},
};
use hexpm::version::Version;
use itertools::Itertools as _;
use std::{
    io::{IsTerminal, Write},
    time::{Duration, Instant},
};
use termcolor::{BufferWriter, Color, ColorChoice, ColorSpec, WriteColor};

#[derive(Debug, Default, Clone)]
pub struct Reporter;

impl Reporter {
    pub fn new() -> Self {
        Self
    }
}

impl Telemetry for Reporter {
    fn compiled_package(&self, duration: Duration) {
        print_compiled(duration);
    }

    fn compiling_package(&self, name: &str) {
        print_compiling(name);
    }

    fn checked_package(&self, duration: Duration) {
        print_checked(duration);
    }

    fn checking_package(&self, name: &str) {
        print_checking(name);
    }

    fn downloading_package(&self, name: &str) {
        print_downloading(name);
    }

    fn packages_downloaded(&self, start: Instant, count: usize) {
        print_packages_downloaded(start, count);
    }

    fn resolving_package_versions(&self) {
        print_resolving_versions();
    }

    fn running(&self, name: &str) {
        print_running(name);
    }

    fn waiting_for_build_directory_lock(&self) {
        print_waiting_for_build_directory_lock();
    }

    fn resolved_package_versions(&self, changes: &PackageChanges) {
        print_package_changes(changes);
    }
}

pub fn ask(question: &str) -> Result<String, Error> {
    print!("{question}: ");
    std::io::stdout().flush().expect("ask stdout flush");
    let mut answer = String::new();
    let _ = std::io::stdin()
        .read_line(&mut answer)
        .map_err(|e| Error::StandardIo {
            action: StandardIoAction::Read,
            err: Some(e.kind()),
        })?;
    Ok(answer.trim().to_string())
}

pub fn confirm(question: &str) -> Result<bool, Error> {
    let answer = ask(&format!("{question} [y/n]"))?;
    match answer.as_str() {
        "y" | "yes" | "Y" | "YES" => Ok(true),
        _ => Ok(false),
    }
}

pub fn confirm_with_text(response: &str) -> Result<bool, Error> {
    let answer = ask(&format!("Type '{response}' to continue"))?;
    Ok(response == answer)
}

pub fn ask_password(question: &str) -> Result<EcoString, Error> {
    let prompt = format!("{question} (will not be printed as you type): ");
    rpassword::prompt_password(prompt)
        .map_err(|e| Error::StandardIo {
            action: StandardIoAction::Read,
            err: Some(e.kind()),
        })
        .map(|s| EcoString::from(s.trim()))
}

pub fn print_publishing(name: &str, version: &Version) {
    print_colourful_prefix("Publishing", &format!("{name} v{version}"));
}

pub fn print_published(detail: &str) {
    print_colourful_prefix("Published", detail);
}

pub fn print_retired(package: &str, version: &str) {
    print_colourful_prefix("Retired", &format!("{package} {version}"));
}

pub fn print_unretired(package: &str, version: &str) {
    print_colourful_prefix("Unretired", &format!("{package} {version}"));
}

pub fn print_publishing_documentation() {
    print_colourful_prefix("Publishing", "documentation");
}

fn print_downloading(text: &str) {
    print_colourful_prefix("Downloading", text);
}

fn print_waiting_for_build_directory_lock() {
    print_colourful_prefix("Waiting", "for build directory lock");
}

fn print_resolving_versions() {
    print_colourful_prefix("Resolving", "versions");
}

fn print_compiling(text: &str) {
    print_colourful_prefix("Compiling", text);
}

pub(crate) fn print_exported(text: &str) {
    print_colourful_prefix("Exported", text);
}

pub(crate) fn print_checking(text: &str) {
    print_colourful_prefix("Checking", text);
}

pub(crate) fn print_compiled(duration: Duration) {
    print_colourful_prefix("Compiled", &format!("in {}", seconds(duration)));
}

pub(crate) fn print_checked(duration: Duration) {
    print_colourful_prefix("Checked", &format!("in {}", seconds(duration)));
}

pub(crate) fn print_running(text: &str) {
    print_colourful_prefix("Running", text);
}

pub(crate) fn print_package_changes(changes: &PackageChanges) {
    for (name, version) in changes.added.iter().sorted() {
        print_added(&format!("{name} v{version}"));
    }
    for Changed { name, old, new } in changes.changed.iter().sorted_by_key(|p| &p.name) {
        print_changed(&format!("{name} v{old} -> v{new}"));
    }
    for ChangedGit {
        name,
        old_hash,
        new_hash,
    } in changes.changed_git.iter().sorted_by_key(|p| &p.name)
    {
        print_changed(&format!("{name} {old_hash} -> {new_hash}"));
    }
    for name in changes.removed.iter().sorted() {
        print_removed(name);
    }
}

fn print_added(text: &str) {
    print_colourful_prefix("Added", text);
}

fn print_changed(text: &str) {
    print_colourful_prefix("Changed", text);
}

fn print_removed(text: &str) {
    print_colourful_prefix("Removed", text);
}

pub(crate) fn print_generating_documentation() {
    print_colourful_prefix("Generating", "documentation");
}

pub(crate) fn print_transferring_ownership() {
    print_colourful_prefix("Transferring", "ownership");
}

pub(crate) fn print_transferred_ownership() {
    print_colourful_prefix("Transferred", "ownership");
}

pub(crate) fn print_adding_owner() {
    print_colourful_prefix("Adding", "owner");
}

pub(crate) fn print_added_owner() {
    print_colourful_prefix("Added", "owner");
}

fn print_packages_downloaded(start: Instant, count: usize) {
    let elapsed = seconds(start.elapsed());
    let msg = match count {
        1 => format!("1 package in {elapsed}"),
        _ => format!("{count} packages in {elapsed}"),
    };
    print_colourful_prefix("Downloaded", &msg);
}

pub fn seconds(duration: Duration) -> String {
    format!("{:.2}s", duration.as_millis() as f32 / 1000.)
}

pub fn print_colourful_prefix(prefix: &str, text: &str) {
    let buffer_writer = stderr_buffer_writer();
    let mut buffer = buffer_writer.buffer();
    buffer
        .set_color(
            ColorSpec::new()
                .set_intense(true)
                .set_fg(Some(Color::Magenta)),
        )
        .expect("print_green_prefix");
    write!(buffer, "{prefix: >11}").expect("print_green_prefix");
    buffer
        .set_color(&ColorSpec::new())
        .expect("print_green_prefix");
    writeln!(buffer, " {text}").expect("print_green_prefix");
    buffer_writer.print(&buffer).expect("print_green_prefix");
}

pub fn stderr_buffer_writer() -> BufferWriter {
    // Don't add color codes to the output if standard error isn't connected to a terminal
    BufferWriter::stderr(color_choice())
}

fn colour_forced() -> bool {
    if let Ok(force) = std::env::var("FORCE_COLOR") {
        !force.is_empty()
    } else {
        false
    }
}

fn color_choice() -> ColorChoice {
    if colour_forced() {
        ColorChoice::Always
    } else if std::io::stderr().is_terminal() {
        ColorChoice::Auto
    } else {
        ColorChoice::Never
    }
}
