asca 0.10.2

use std::{ffi::OsStr, fmt::Debug, fs, io::{self, Write}, path::{Path, PathBuf}};

use colored::Colorize;
use asca::{error::ASCAError, rule::RuleGroup};

use super::seq::OldConfig;

#[cfg(windows)]
pub const LINE_ENDING: &str = "\r\n";
#[cfg(not(windows))]
pub const LINE_ENDING: &str = "\n";

pub const  RULE_FILE_EXT: &str = "rsca";
pub const  WORD_FILE_EXT: &str = "wsca";
pub const  CONF_FILE_EXT: &str = "asca";
pub const  JSON_FILE_EXT: &str = "json";
pub const ALIAS_FILE_EXT: &str = "alias";


pub(super) fn ask(question: &str, auto: Option<bool>) -> io::Result<bool> {
    if let Some(ans) = auto {
        return Ok(ans)
    }
    loop {
        print!(":: {question} [y/N]: ");
        io::stdout().flush()?;
        let mut buf = String::new();
        let _ = std::io::stdin().read_line(&mut buf);
        match buf.chars().next() {
            Some('y' | 'Y') => return Ok(true),
            Some('n' | 'N') => return Ok(false),
            Some(x) if x.is_control() => return Ok(false),
            _ => println!("{}", "   yes/no only".yellow()),
        }
    }
}

pub(super) fn get_dir_files(path: &str, valid_extensions: &[&str]) -> io::Result<Vec<PathBuf>> {
    Ok(dir_read(path)?
        // Filter out entries which we couldn't read
        .filter_map(|res| res.ok())
        // Turn entries to paths
        .map(|dir_entry| dir_entry.path())
        // Filter out files with the wrong extension
        .filter_map(|path| {
            if path.extension().is_some_and(|ext| match_exts(ext, valid_extensions)) {
                Some(path)
            } else {
                None
            }
        })
    .collect::<Vec<_>>())
}

fn match_exts(ext: &OsStr, valid_extensions: &[&str]) -> bool {
    for &ve in valid_extensions {
        if ve == ext {
            return true
        }
    }
    false
}

fn create_ext_list(valid_extensions: &[&str]) -> String {
    if valid_extensions.is_empty() {
        unreachable!()
    } else if valid_extensions.len() == 1 {
        format!(".{}", valid_extensions[0])
    } else if valid_extensions.len() == 2 {
        format!(".{} or .{}", valid_extensions[0], valid_extensions[1])
    } else {
        let mut exts_str = String::new();
        for (i, e) in valid_extensions.iter().enumerate() {
            match i.cmp(&(valid_extensions.len()-2)) {
                std::cmp::Ordering::Greater => exts_str.push_str(&format!(".{e}")),
                std::cmp::Ordering::Less    => exts_str.push_str(&format!(".{e}, ")),
                std::cmp::Ordering::Equal   => exts_str.push_str(&format!(".{e}, or ")),
            }
        }
        exts_str
    }
}

/// returns true if input is a dir, false if it a file
pub(super) fn validate_file_or_dir(maybe_path: Option<PathBuf>) -> io::Result<(PathBuf, bool)> {
    match maybe_path {
        Some(path) => {
            if path.is_dir() {
                Ok((path, true))
            } else if path.is_file() {
                Ok((path, false))
            } else {
                Err(util_err(format!("Given path {} cannot be found", format!("{path:?}").yellow())))
            }
        },
        None => Ok((PathBuf::from("."), true)),
    }
}

#[allow(unused)]
pub(super) fn validate_directory(maybe_path: Option<PathBuf>) -> io::Result<PathBuf> {
    match maybe_path {
        Some(path) => {
            if path.is_dir() {
                Ok(path)
            } else {
                Err(util_err(format!("{} is not a directory", format!("{path:?}").yellow())))
            }
        },
        None => Ok(PathBuf::from(".")),
    }
}

pub(super) fn as_file(path: &Path) -> io::Result<&Path> {
    if path.is_file() {
        Ok(path)
    } else {
        Err(util_err(format!("Given path {} is not a file or doesn't exist", format!("{path:?}").yellow())))
    }
}

pub(super) fn validate_file(path: &Path, valid_extensions: &[&str]) -> io::Result<PathBuf> {
    if path.is_file() {
        match path.extension() {
            Some(ext) => if match_exts(ext, valid_extensions) {
                Ok(path.to_path_buf())
            } else {
                let exts_str = create_ext_list(valid_extensions);
                Err(util_err(format!("File {} is not of the right type. Must be {}", format!("{path:?}").yellow(), exts_str)))
            },
            None => Err(util_err(format!("Given path {} has no extension", format!("{path:?}").yellow()))),
        }
    } else {
        Err(util_err(format!("Given path {} is not a file", format!("{path:?}").yellow())))
    }
}

pub(super) fn validate_or_get_path(maybe_path: Option<&Path>, valid_extensions: &[&str], kind: &str) -> io::Result<PathBuf> {
    match maybe_path {
        Some(path) => validate_file(path, valid_extensions),
        None => {
            let files = get_dir_files(".", valid_extensions)?;
            match files.len().cmp(&1) {
                std::cmp::Ordering::Greater => Err(util_err(format!("More than one matching {kind} file found in current directory. Please specify."))),
                std::cmp::Ordering::Less    => Err(util_err(format!("No matching {kind} files found in current directory"))),
                std::cmp::Ordering::Equal   => {
                    eprintln!("{} No {} file provided. Using {}", "Note:".bright_blue(), kind, format!("{:?}", files[0]).yellow());
                    Ok(files[0].clone())
                },
            }
        }
    }
}

pub(super) fn file_open<P: AsRef<Path> + Debug + ?Sized>(path: &P) -> io::Result<fs::File> {
    match fs::File::open(path) {
        Ok(file) => Ok(file),
        Err(e) => {
            eprintln!("{} error occured when reading file {}:", "asca:".bright_red(), format!("{path:?}").yellow());
            Err(map_io_error(e))
        }
    }
}

pub(super) fn file_write<P: AsRef<Path> + Debug + ?Sized>(path: &P, content: String) -> io::Result<()> {
    if let Err (e) = fs::write(path, content) {
        eprintln!("{} error occurred writing to file {}:", "asca:".bright_red(), format!("{path:?}").yellow());
        return Err(map_io_error(e))
    }
    eprintln!(":: Wrote to file {}", format!("{path:?}").yellow());
    Ok(())
}

/// Append a file name with an arbitrary string
/// 
/// I.e. ```"example.rsca" + "_old" -> "example_old.rsca" ```
#[allow(unused)]
pub(super) fn file_name_append<P: AsRef<Path> + Debug + ?Sized>(path: &P, content: &str) -> io::Result<()> {
    let path = path.as_ref();

    if !path.is_file() { 
        return Err(io::Error::other(format!("{} Given path {} does not exist or is not a file", "asca:".bright_red(), format!("{path:?}").yellow()))) 
    }
    
    let new_path = {
        let mut p = path.to_path_buf();
        p.set_extension("");
        p.set_file_name(p.file_name().unwrap().to_str().unwrap().to_owned() + content);

        if let Some(ext) = path.extension() {
            p.set_extension(ext);
        }

        p
    };
    
    if let Err(e) = fs::rename(path, &new_path) {
        eprintln!("{} error occurred trying to rename file {}:", "asca:".bright_red(), format!("{path:?}").yellow());
        return Err(map_io_error(e))
    }

    eprintln!(":: Renamed file {} to {}", format!("{path:?}").yellow(), format!("{new_path:?}").yellow());
    Ok(())
}

pub(super) fn file_create_write<P: AsRef<Path> + Debug + ?Sized>(path: &P, content: String) -> io::Result<()> {
    if let Err (e) = fs::write(path, content) {
        eprintln!("{} error occurred writing to file {}:", "asca:".bright_red(), format!("{path:?}").yellow());
        return Err(map_io_error(e))
    }
    eprintln!(":: Created file {} in current directory", format!("{path:?}").yellow());
    Ok(())
}

pub(super) fn file_read<P: AsRef<Path> + Debug + ?Sized>(path: &P) -> io::Result<String> {
    match fs::read_to_string(path) {
        Ok(dir) => Ok(dir),
        Err(e) => {            
            eprintln!("{} error occurred when reading file {}:", "asca:".bright_red(), format!("{path:?}").yellow());
            Err(map_io_error(e))
        },
    }
}

pub(super) fn dir_create_all<P: AsRef<Path> + Debug + ?Sized>(path: &P) -> io::Result<()> {
    if let Err(e) = fs::create_dir_all(path) {            
        eprintln!("{} error occurred when creating {}:", "asca:".bright_red(), format!("{path:?}").yellow());
        return Err(map_io_error(e))
    } 
    eprintln!(":: Created dir {}", format!("{path:?}").yellow());
    Ok(())
}

pub(super) fn dir_read<P: AsRef<Path> + Debug + ?Sized>(path: &P) -> io::Result<fs::ReadDir> {
    match fs::read_dir(path) {
        Ok(dir) => Ok(dir),
        Err(e) => {            
            eprintln!("{} error occurred when reading from {}:", "asca:".bright_red(), format!("{path:?}").yellow());
            Err(map_io_error(e))
        },
    }
}

pub(super) fn dir_create_file<P: AsRef<Path> + Debug + ?Sized>(path: &P, content: String, auto: Option<bool>) -> io::Result<()> {
    let path = path.as_ref();
    if path.exists() {
        if ask(&(format!("File {} already exists, do you wish to overwrite it?", format!("{path:?}").yellow())), auto)? {
            file_write(&path, content)
        } else {
            Ok(())
        }
    } else {
        file_create_write(&path, content)
    }
}

pub(super) fn write_to_file(path: &Path, content: String, extension: &str, auto: Option<bool>) -> io::Result<()> {
    // if path is an extant file, ask for overwrite
    // if path is a dir, create file in dir, if extant ask for overwrite
    // if path does not exist and has a valid extension, create and write
    // else error
    if path.is_file() {
        if path.extension().is_some_and(|ext| ext == extension) {
            if ask(&(format!("File {} already exists, do you wish to overwrite it?", format!("{path:?}").yellow())), auto)? {
                return file_write(path, content)
            }
            Ok(())
        } else {
            Err(util_err(format!("Provided file '{}' has the wrong extension. Must be .{}", format!("{path:?}").yellow(), extension)))
        }
    } else if path.is_dir() {
        // if path is dir, write to file of <dir>/out.<extension>
        let mut p = PathBuf::from("out");
        p.set_extension(extension);
        dir_create_file(&p, content, auto)
    } else {
        match path.extension() {
            Some(ext) => if ext == extension {
                file_write(path, content)
            } else {
                Err(util_err(format!("Provided file '{}' has the wrong extension. Must be .{}", format!("{path:?}").yellow(), extension)))
            },
            None => Err(util_err(format!("Provided dir '{}' does not exist", format!("{path:?}").yellow()))),
        } 
    }
}

/// Make sure string has no illegal filename characters
pub(super) fn sanitise_str(str: &str) -> String {
    // 26 is arbitrary, but we want to make sure that the file names don't get too long
    str.chars().take(26).filter_map(|ch| match ch { 
        '"' | '\'' | '|' | '?' | 
        '!' | '\\' | '*' | '/' | 
        ':' | '\0' | '%' | '<' | '>' => None,
        ' ' => Some('-'), 
         _  => Some(ch.to_ascii_lowercase())
    }).collect()
}

fn get_seq_input(seq: &OldConfig) -> String {
    let mut input = String::new();
    
    if let Some(from) = &seq.from {
        input.push_str(from);
        input.push(' ');
    }

    if !seq.words.is_empty()  {
        let words: String =  seq.words.iter().map(|s| {
            let s = s.to_string();
            if s.chars().rev().take(5).collect::<String>() != "acsw." {
                s + ".wsca" + " "
            } else {
                s + " "
            }
        }).collect();

        input.push_str(&words);
    }
    
    if let Some(alias) = &seq.alias {
        input.push_str(alias);

        if alias.chars().rev().take(6).collect::<String>() != "saila." {
            input.push_str(".alias");
        }

        input.push(' ');
    }
    
    if !input.is_empty() {
        input.push_str("> ");
    }

    input
}

pub(super) fn to_new_config_format(conf: Vec<OldConfig>) -> String {
    let mut result = String::new();
    for seq in conf {

        let input = get_seq_input(&seq);

        let indent = input + &seq.tag + ":\n";
        result.push_str(&indent);

        for rule in seq.entries {
            result.push_str(&format!("    {};\n", rule.verbatim));
        }
        result.push('\n');
    }

    result
}

pub(super) fn to_rsca_format(rules: Vec<RuleGroup>) -> String {
    let mut result = String::new();
    for rg in rules {
        let name_str = format!("@ {}\n", rg.name);

        let mut rule_str = String::new();
        for r in rg.rule {
            rule_str.push_str(&format!("    {r}\n"));
        }

        let mut desc_str = String::new();
        for d in rg.description.split('\n') {
            desc_str.push_str(&format!("# {d}\n"));
        }

        result.push_str(&name_str);
        result.push_str(&rule_str);
        result.push_str(&desc_str);
    }
    result
}

pub(super) fn to_alias(into: Vec<String>, from: Vec<String>) -> String {
    let mut result = String::from("@into\n");

    for to in into {
        result.push_str(&format!("    {to}\n"));
    }

    result.push_str("@from\n");

    for fr in from {
        result.push_str(&format!("    {fr}\n"));
    }

    result
}

fn util_err<S: Into<String>>(message: S) -> io::Error {
    io::Error::other(format!("{} {}", "Error:".bright_red(), message.into()))
}

pub(super) fn map_io_error(error: io::Error) -> io::Error {
    match error.kind() {
        io::ErrorKind::NotFound => util_err("File or directory was not found"),
        io::ErrorKind::PermissionDenied => util_err("Do not have the right permissions to complete this operation"),
        _ => error,
    }
}

#[inline]
// Rust's built-in padding algo doesn't account for nonspacing characters for some reason
pub(super) fn fix_combining_char_pad(string: &str) -> usize {
    let mut pad = 0;
    for s in string.chars() {
        // Not an exhaustive list
        match s {
            // Combining Diacritical Marks 
            '\u{0300}'..'\u{036F}' |
            // Combining Diacritical Marks Extended
            '\u{1AB0}'..'\u{1AFF}' |
            // Combining Diacritical Marks Supplement
            '\u{1DC0}'..'\u{1DFF}' |
            // Combining Diacritical Marks for Symbols 
            '\u{20D0}'..'\u{20FF}' |
            // Cyrillic Extended-A
            '\u{2DE0}'..'\u{2DFF}' |
            // Combining Half Marks
            '\u{FE20}'..'\u{FE2F}' |
            // Dakuten & Handakuten
            '\u{3099}' | '\u{309A}' => pad += 1,
            _ => {}
        }
    }

    pad
}

pub(super) fn get_asca_errors(err: ASCAError, rules: &[RuleGroup], into: &[String], from: &[String]) -> io::Error {
    io::Error::other(match err {
        ASCAError::AliasSyn(e) => e.format(into, from),
        ASCAError::AliasRun(e) => e.format(into, from),
        ASCAError::WordSyn(e)  => e.format(),
        ASCAError::RuleSyn(e)  => e.format(rules),
        ASCAError::RuleRun(e)  => e.format(rules),
    })
}

pub(super) fn lev(a: &str, b: &str) -> usize {
    let mut dist = 0;

    if a == b { return dist }

    let a_len = a.chars().count();
    let b_len = b.chars().count();

    debug_assert!(a_len > 0);
    debug_assert!(b_len > 0);

    let mut cache: Vec<usize> = (1..).take(a_len).collect();

    for (bi, b_ch) in b.chars().enumerate() {
        dist = bi;
        let mut a_dist = bi;

        for (ai, a_ch) in a.chars().enumerate() {
            let b_dist = a_dist + (a_ch != b_ch) as usize;

            a_dist = cache[ai];

            dist = if a_dist > dist {
                if b_dist > dist {
                    dist + 1
                } else {
                    b_dist
                }
            } else if b_dist > a_dist {
                a_dist + 1
            } else {
                b_dist
            };

            cache[ai] = dist;
        }
    }

    dist
}