//! Rustemo compiler. Run `rcomp --help` for more information.

use std::path::PathBuf;

use clap::{ArgAction, Parser};
use colored::Colorize;
use rustemo_compiler::{
    BuilderType, GeneratorTableType, LexerType, ParserAlgo, Settings, TableType,
};

#[derive(Parser)]
#[cfg_attr(feature="bootstrap",
           clap(version = concat!(concat!(env!("CARGO_PKG_VERSION"),
                                          env!("GIT_HASH")), "-bootstrap")))]
#[cfg_attr(not(feature="bootstrap"),
           clap(version = concat!(env!("CARGO_PKG_VERSION"), env!("GIT_HASH"))))]
#[clap(author, about, long_about = None)]
struct Cli {
    /// Regenerate output actions file even if exists
    #[clap(short, long, action)]
    force: bool,

    /// Create DOT automata visualization
    #[clap(long, action)]
    dot: bool,

    /// Do not generate actions
    #[clap(short, long, action)]
    noactions: bool,

    /// Do not print trace logs
    #[clap(long, action)]
    notrace: bool,

    /// Grammar file or directory to process
    #[clap(value_parser, value_name="GRAMMAR FILE/DIR", value_hint = clap::ValueHint::AnyPath)]
    grammar_file_or_dir: PathBuf,

    /// Output root directory for the parser. Default is the same as input grammar file.
    #[clap(short, long, value_name="OUT DIR ROOT", value_hint = clap::ValueHint::DirPath)]
    outdir_root: Option<PathBuf>,

    /// Output directory for actions. Default is the same as input grammar file.
    #[clap(short='a', long, value_name="OUT DIR ACTIONS ROOT", value_hint = clap::ValueHint::DirPath)]
    outdir_actions_root: Option<PathBuf>,

    /// Prefer shifts in case of possible shift/reduce conflicts.
    #[clap(long)]
    prefer_shifts: bool,

    /// Do not prefer shifts over empty reductions.
    #[clap(long)]
    no_shifts_over_empty: bool,

    /// The type of LR table
    #[clap(short, long, value_enum, default_value_t)]
    table_type: TableType,

    /// Parser algorithm
    #[clap(short, long, value_enum, default_value_t)]
    parser_algo: ParserAlgo,

    /// Parser generator table type
    #[clap(short, long, value_enum, default_value_t)]
    generator_table_type: GeneratorTableType,

    /// What kind of lexer should be used.
    #[clap(short, long, value_enum, default_value_t)]
    lexer_type: LexerType,

    /// The type of the input if non-default lexer is used
    #[clap(short, long, default_value = "str")]
    input_type: String,

    /// Generated builder type.
    #[clap(short, long, value_enum, default_value_t)]
    builder_type: BuilderType,

    /// Should generated default AST builder types contain location/layout information
    /// This is only used for the default builder type.
    #[clap(long)]
    builder_loc_info: bool,

    /// Lexical disambiguation using most specific match strategy.
    #[clap(long, default_missing_value = "true", require_equals = true)]
    lexical_disamb_most_specific: Option<bool>,

    /// Lexical disambiguation using longest match strategy.
    #[clap(long, default_missing_value = "true", require_equals = true)]
    lexical_disamb_longest_match: Option<bool>,

    /// Lexical disambiguation using grammar order.
    #[clap(long, default_missing_value = "true", require_equals = true)]
    lexical_disamb_grammar_order: Option<bool>,

    /// Should fancy_regex crate be used instead of regex.
    #[clap(long)]
    fancy_regex: bool,

    /// Parser can succeed without consuming the whole input.
    #[clap(long)]
    partial_parse: bool,

    /// Should whitespace be skipped. Not used if Layout rule exists in the Grammar.
    #[clap(long)]
    no_skip_ws: bool,

    /// Print LR table
    #[clap(long)]
    print_table: bool,

    /// Exclude dirs containing these parts. Used with dir processing.
    #[clap(short, long, value_parser)]
    exclude: Vec<String>,

    /// Verbosity level 0-2
    #[clap(short, long, action = ArgAction::Count)]
    verbosity: u8,
}

fn main() {
    let cli = Cli::parse();

    let mut settings = Settings::new()
        .force(cli.force)
        .dot(cli.dot)
        .actions(!cli.noactions)
        .notrace(cli.notrace)
        .exclude(cli.exclude)
        .prefer_shifts(cli.prefer_shifts)
        .prefer_shifts_over_empty(!cli.no_shifts_over_empty)
        .fancy_regex(cli.fancy_regex)
        .partial_parse(cli.partial_parse)
        .skip_ws(!cli.no_skip_ws)
        .table_type(cli.table_type)
        .print_table(cli.print_table)
        .parser_algo(cli.parser_algo)
        .generator_table_type(cli.generator_table_type)
        .lexer_type(cli.lexer_type)
        .builder_type(cli.builder_type)
        .builder_loc_info(cli.builder_loc_info)
        .input_type(cli.input_type);

    if let Some(most_specific) = cli.lexical_disamb_most_specific {
        settings = settings.lexical_disamb_most_specific(most_specific)
    }
    if let Some(longest_match) = cli.lexical_disamb_longest_match {
        settings = settings.lexical_disamb_longest_match(longest_match)
    }
    if let Some(grammar_order) = cli.lexical_disamb_grammar_order {
        settings = settings.lexical_disamb_grammar_order(grammar_order)
    }

    if let Some(outdir_root) = cli.outdir_root {
        settings = settings.out_dir_root(outdir_root);
    }
    if let Some(outdir_actions_root) = cli.outdir_actions_root {
        settings = settings.out_dir_actions_root(outdir_actions_root);
    }

    let result = if cli.grammar_file_or_dir.is_file() {
        settings.process_grammar(&cli.grammar_file_or_dir)
    } else {
        settings.root_dir(cli.grammar_file_or_dir).process_dir()
    };

    if let Err(e) = result {
        println!("{}", e.to_string().red());
        println!("{}", "Parser(s) not generated.".red());
    }
}
