oas3-gen 0.26.2

use std::path::PathBuf;

use clap::{ArgAction, Args, Parser, Subcommand, ValueEnum, ValueHint, builder::Styles};

use super::colors::{ColorMode, ThemeMode};
use crate::{generator::codegen::Visibility, ui::Colors};

const DARK_STYLE: Styles = Colors::clap_styles();

#[derive(Parser, Debug)]
#[command(name = "oas3-gen")]
#[command(author, version, about = "OpenAPI to Rust code generator")]
#[command(propagate_version = true)]
#[command(styles = DARK_STYLE)]
pub struct Cli {
  #[command(subcommand)]
  pub command: Commands,

  /// Coloring
  #[arg(
    long,
    value_enum,
    value_name = "WHEN",
    default_value = "auto",
    global = true,
    display_order = 100,
    help_heading = "Terminal Output"
  )]
  pub color: ColorMode,

  /// Theme
  #[arg(
    long,
    value_enum,
    value_name = "THEME",
    default_value = "auto",
    global = true,
    display_order = 100,
    help_heading = "Terminal Output"
  )]
  pub theme: ThemeMode,
}

#[derive(Subcommand, Debug)]
pub enum Commands {
  /// List information from OpenAPI specification
  List {
    #[command(subcommand)]
    list_command: ListCommands,
  },
  /// Generates idiomatic, type-safe Rust code from an OpenAPI v3.1 (OAS31) specification.
  Generate(GenerateCommand),
}

#[derive(Args, Debug, Clone)]
#[allow(clippy::struct_excessive_bools)]
pub struct GenerateCommand {
  /// Sets the generation mode
  #[arg(value_enum, default_value = "types")]
  pub mode: GenerateMode,

  /// Path to the OpenAPI specification file
  #[arg(
    short,
    long,
    value_name = "FILE",
    value_hint = ValueHint::AnyPath,
    display_order = 0,
    help_heading = "Required"
  )]
  pub input: PathBuf,

  /// Path for generated output (file for types/client, directory for client-mod/server-mod)
  #[arg(
    short,
    long,
    value_name = "PATH",
    value_hint = ValueHint::AnyPath,
    display_order = 0,
    help_heading = "Required"
  )]
  pub output: Option<PathBuf>,

  /// Module visibility for generated items
  #[arg(
    short = 'C',
    long,
    value_name = "PUB",
    default_value = "public",
    display_order = 10,
    help_heading = "Code Generation"
  )]
  pub visibility: Visibility,

  /// Enable OData-specific field optionality rules (makes @odata.* fields optional on concrete types)
  #[arg(long, default_value_t = false, display_order = 11, help_heading = "Code Generation")]
  pub odata_support: bool,

  /// Specifies how to handle enum case sensitivity and duplicates
  #[arg(
    long,
    value_enum,
    default_value_t,
    display_order = 12,
    help_heading = "Code Generation"
  )]
  pub enum_mode: EnumCaseMode,

  /// Controls how enum variants are ordered in generated code
  #[arg(
    long,
    value_enum,
    default_value_t,
    display_order = 12,
    help_heading = "Code Generation"
  )]
  pub enum_layout: EnumLayout,

  /// Disable generation of ergonomic helper methods for enum variants
  #[arg(long, default_value_t = false, display_order = 13, help_heading = "Code Generation")]
  pub no_helpers: bool,

  /// Custom serde_as type overrides (format: type_name=custom::Path)
  #[arg(
    short = 'c',
    long = "customize",
    action = ArgAction::Append,
    value_name = "TYPE=PATH",
    display_order = 14,
    help_heading = "Code Generation"
  )]
  pub customize: Option<Vec<String>>,

  /// Generate all schemas, even those unreferenced by selected operations
  #[arg(
    group = "filter",
    long,
    default_value_t = false,
    display_order = 22,
    help_heading = "Operation Filtering"
  )]
  pub all_schemas: bool,

  /// Emit header constants for all parameters defined in components, not just those used in operations
  #[arg(long, default_value_t = false, display_order = 15, help_heading = "Code Generation")]
  pub all_headers: bool,

  /// Enable bon builder derives on schema structs and builder methods on request structs
  #[arg(long, default_value_t = false, display_order = 16, help_heading = "Code Generation")]
  pub enable_builders: bool,

  /// Emit `std::collections::HashMap` and `Vec` instead of `indexmap::IndexMap`/`IndexSet`
  /// for generated map fields and `uniqueItems` arrays. JSON key/element order is no longer
  /// preserved at runtime when this is enabled.
  #[arg(long, default_value_t = false, display_order = 18, help_heading = "Code Generation")]
  pub no_ordered_collections: bool,

  /// Format documentation comments using mdformat (requires mdformat installed)
  #[arg(long, default_value_t = false, display_order = 17, help_heading = "Code Generation")]
  pub doc_format: bool,

  /// Include only the specified comma-separated operation IDs
  #[arg(
    group = "filter",
    long, action = ArgAction::Append,
    value_name = "id_1,id_2,...",
    value_delimiter = ',',
    display_order = 20,
    help_heading = "Operation Filtering"
  )]
  pub only: Option<Vec<String>>,

  /// Exclude the specified comma-separated operation IDs
  #[arg(
    group = "filter",
    long, action = ArgAction::Append,
    value_name = "id_1,id_2,...",
    value_delimiter = ',',
    display_order = 21,
    help_heading = "Operation Filtering"
  )]
  pub exclude: Option<Vec<String>>,

  /// Enable verbose output with detailed progress information
  #[arg(
    short,
    long,
    default_value_t = false,
    global = true,
    display_order = 100,
    help_heading = "Terminal Output"
  )]
  pub verbose: bool,

  /// Suppress non-essential output (errors only)
  #[arg(
    short,
    long,
    default_value_t = false,
    global = true,
    display_order = 101,
    help_heading = "Terminal Output"
  )]
  pub quiet: bool,
}

#[derive(ValueEnum, Clone, Debug, PartialEq, Eq)]
pub enum GenerateMode {
  Types,
  Client,
  #[value(name = "client-mod")]
  ClientMod,
  #[value(name = "server-mod")]
  ServerMod,
}

#[derive(ValueEnum, Clone, Debug, Default)]
pub enum EnumCaseMode {
  #[default]
  /// Merge duplicates with strict matching (e.g., "ITEM" and "item" become "Item")
  Merge,
  /// Preserve case-variant duplicates as separate enum values
  Preserve,
  /// Merge duplicates and enable relaxed (case-insensitive) deserialization
  Relaxed,
}

#[derive(ValueEnum, Clone, Copy, Debug, Default, PartialEq, Eq)]
pub enum EnumLayout {
  #[default]
  /// Preserve enum variant order from the OpenAPI spec
  Spec,
  /// Sort enum variants alphabetically by Rust variant name
  Sorted,
}

#[derive(Subcommand, Debug)]
pub enum ListCommands {
  /// List all operations defined in the OpenAPI specification
  Operations {
    /// Path to the OpenAPI JSON specification file
    #[arg(short, long, value_name = "FILE", value_hint = ValueHint::AnyPath)]
    input: PathBuf,
  },
}