atlassian-cli 0.7.2

mod commands;
mod query;

use std::io::IsTerminal;
use std::path::PathBuf;

use anyhow::{anyhow, Result};
use atlassian_cli_api::error::ApiError;
use atlassian_cli_api::ApiClient;
use atlassian_cli_auth::BITBUCKET_API_URL;
use atlassian_cli_config::{migrate_config_if_needed, Config, MigrationResult};
use atlassian_cli_output::{OutputFormat, OutputRenderer};
use clap::{Parser, Subcommand};
use commands::auth::{self, AuthCommand};
use commands::bitbucket::utils::extract_workspace_from_url;
use tracing_subscriber::{fmt, EnvFilter};

#[derive(Parser, Debug)]
#[command(name = "atlassian-cli", version, about = "Unified Atlassian Cloud CLI", long_about = None)]
struct Cli {
    /// Profile to use from config file
    //
    // global: --format has always been accepted after the subcommand, so users
    // and most of the documentation write `... issue list --profile prod` too.
    // It was rejected, which made a large share of the published examples fail.
    #[arg(short, long, global = true)]
    profile: Option<String>,

    /// Path to config file (defaults to ~/.atlassian-cli/config.yaml)
    #[arg(long, global = true)]
    config: Option<PathBuf>,

    /// Output format for command results (table, json, yaml, csv, quiet, markdown)
    #[arg(long, short = 'f', value_enum, default_value_t = OutputFormat::Table, global = true)]
    format: OutputFormat,

    /// Wrap JSON/YAML list output in {"data": [...], "count": N} envelope.
    #[arg(long, global = true)]
    envelope: bool,

    /// Enable verbose logging
    #[arg(long)]
    debug: bool,

    #[command(subcommand)]
    command: AtlassianCommand,
}

#[derive(Subcommand, Debug, Clone)]
enum AtlassianCommand {
    /// Jira commands
    Jira(commands::jira::JiraArgs),
    /// Confluence commands
    Confluence(commands::confluence::ConfluenceArgs),
    /// Bitbucket commands
    #[command(visible_alias = "bb")]
    Bitbucket(commands::bitbucket::BitbucketArgs),
    /// Jira Service Management commands
    Jsm(commands::jsm::JsmArgs),
    /// Opsgenie commands
    Opsgenie(commands::opsgenie::OpsgenieArgs),
    /// Bamboo commands
    Bamboo(commands::bamboo::BambooArgs),
    /// Authentication commands
    #[command(subcommand)]
    Auth(AuthCommand),
}

#[tokio::main]
async fn main() {
    if let Err(err) = run().await {
        // Check for ApiError in the chain and display suggestion
        for cause in err.chain() {
            if let Some(api_err) = cause.downcast_ref::<ApiError>() {
                if let Some(hint) = api_err.suggestion() {
                    eprintln!("Hint: {hint}");
                }
                break;
            }
        }
        eprintln!("Error: {err:#}");
        std::process::exit(1);
    }
}

async fn run() -> Result<()> {
    let cli = match Cli::try_parse() {
        Ok(cli) => cli,
        Err(e) => {
            let msg = e.to_string();
            if msg.contains("bitbucket-token") || msg.contains("bitbucket_token") {
                eprintln!("Hint: Use `--bitbucket --token <TOKEN>` for Bitbucket auth.");
                eprintln!("Example: atlassian-cli auth login --profile work --bitbucket --token <TOKEN> --email <EMAIL>\n");
            }
            e.exit();
        }
    };
    init_tracing(cli.debug)?;

    // Perform config directory migration if needed (only when no custom path specified)
    if cli.config.is_none() {
        handle_migration();
    }

    let config_path = cli.config.clone();
    let mut config = Config::load(config_path.as_ref())?;
    let renderer = OutputRenderer::new(cli.format).with_envelope(cli.envelope);

    // Validate profile selection for destructive commands
    let is_destructive = is_destructive_command(&cli.command);
    validate_profile_selection(&config, cli.profile.as_deref(), is_destructive)?;

    match cli.command {
        AtlassianCommand::Jira(args) => {
            let profile = resolve_profile_for_product(&config, cli.profile.as_deref())?;
            let client = build_product_client(&profile)?;
            commands::jira::execute(args, client, &renderer).await?
        }
        AtlassianCommand::Confluence(args) => {
            let profile = resolve_profile_for_product(&config, cli.profile.as_deref())?;
            let client = build_product_client(&profile)?;
            commands::confluence::execute(args, client, &renderer).await?
        }
        AtlassianCommand::Bitbucket(args) => {
            let profile = resolve_profile_for_bitbucket(&config, cli.profile.as_deref())?;
            let client = build_bitbucket_client(&profile)?;
            commands::bitbucket::execute(
                args,
                client,
                &renderer,
                profile.workspace.as_deref(),
                profile.is_bearer,
                profile.bitbucket_remote.as_deref(),
            )
            .await?
        }
        AtlassianCommand::Jsm(args) => {
            let profile = resolve_profile_for_product(&config, cli.profile.as_deref())?;
            let client = build_product_client(&profile)?;
            commands::jsm::execute(args, client, &renderer).await?
        }
        AtlassianCommand::Opsgenie(args) => {
            let profile = resolve_profile_for_opsgenie(&config, cli.profile.as_deref())?;
            let client = build_opsgenie_client(&profile)?;
            commands::opsgenie::execute(args, client, &renderer).await?
        }
        AtlassianCommand::Bamboo(args) => {
            let profile = resolve_profile_for_bamboo(&config, cli.profile.as_deref())?;
            let client = build_bamboo_client(&profile)?;
            commands::bamboo::execute(args, client, &renderer).await?
        }
        AtlassianCommand::Auth(command) => {
            auth::handle(command, &mut config, config_path.as_deref(), &renderer).await?
        }
    }

    Ok(())
}

fn init_tracing(debug: bool) -> Result<()> {
    let default = if debug {
        "info,atlassian-cli=debug"
    } else {
        "info"
    };
    let filter = EnvFilter::try_from_default_env().unwrap_or_else(|_| EnvFilter::new(default));

    // Logs go to stderr, never stdout: commands that stream raw bytes to stdout
    // (e.g. `jira attachment download <ID> --output -`) must stay byte-exact for
    // pipes, and tracing-subscriber's default writer is stdout.
    fmt()
        .with_env_filter(filter)
        .with_target(false)
        .with_writer(std::io::stderr)
        .with_ansi(std::io::stderr().is_terminal())
        .try_init()
        .map_err(|err| anyhow!("failed to initialize logger: {err}"))
}

/// Shared profile fields used by all Atlassian products.
struct BaseProfile {
    name: String,
    email: String,
}

/// Profile for Jira/Confluence/JSM commands (requires base_url and token).
struct ProductProfile {
    base: BaseProfile,
    base_url: String,
    token: String,
}

/// Profile for Bitbucket commands (only requires email and bitbucket token).
struct BitbucketProfile {
    base: BaseProfile,
    token: String,
    workspace: Option<String>,
    /// true = Bearer auth (access tokens), false = Basic auth (API tokens)
    is_bearer: bool,
    /// Preferred git remote name for Bitbucket auto-detection
    bitbucket_remote: Option<String>,
}

fn handle_migration() {
    match migrate_config_if_needed() {
        MigrationResult::Migrated { from, to } => {
            eprintln!(
                "Config migrated from {} to {}\nThe old directory can be safely removed.",
                from.display(),
                to.display()
            );
        }
        MigrationResult::Failed(e) => {
            eprintln!("Warning: Config migration failed: {}", e);
        }
        MigrationResult::NotNeeded => {}
    }
}

/// Validates that destructive commands have explicit profile selection.
fn validate_profile_selection(
    config: &Config,
    requested: Option<&str>,
    command_is_destructive: bool,
) -> Result<()> {
    if command_is_destructive && requested.is_none() && config.default_profile.is_none() {
        return Err(anyhow!(
            "Destructive command requires explicit profile selection.\n\
             Use --profile <name> or set default with: atlassian-cli auth login --default"
        ));
    }
    Ok(())
}

/// Determines if a command is destructive (modifies or deletes data).
/// TODO: Implement proper destructive command detection by checking nested command enums.
/// For now, returns false since IndexMap already provides deterministic profile selection.
fn is_destructive_command(_command: &AtlassianCommand) -> bool {
    // The IndexMap change already ensures deterministic profile selection,
    // mitigating the main security risk. Full destructive command detection
    // requires deep pattern matching on nested enum structures and will be
    // implemented in a follow-up.
    false
}

/// Resolve base profile fields (name + email) shared by all commands.
fn resolve_base_profile<'a>(
    config: &'a Config,
    requested: Option<&'a str>,
) -> Result<(BaseProfile, &'a atlassian_cli_config::Profile)> {
    let (name, profile) = config
        .resolve_profile(requested)
        .ok_or_else(|| anyhow!("No profile configured. Run `atlassian-cli auth login` first."))?;

    let email = profile
        .email
        .clone()
        .ok_or_else(|| anyhow!("Profile '{name}' is missing an email."))?;

    Ok((
        BaseProfile {
            name: name.to_string(),
            email,
        },
        profile,
    ))
}

/// Resolve profile for Jira/Confluence/JSM commands.
/// Requires base_url and Jira/Confluence token.
fn resolve_profile_for_product(config: &Config, requested: Option<&str>) -> Result<ProductProfile> {
    let (base, profile) = resolve_base_profile(config, requested)?;

    let base_url = profile.base_url.clone().ok_or_else(|| {
        anyhow!(
            "Profile '{}' is missing a base_url. Run `atlassian-cli auth login --base-url <URL>`",
            base.name
        )
    })?;

    let token = auth::get_token(&base.name).ok_or_else(|| {
        anyhow!(
            "No token found for profile '{}'. Set ATLASSIAN_CLI_TOKEN_{} env var or run `atlassian-cli auth login --profile {}`",
            base.name,
            base.name.to_uppercase(),
            base.name
        )
    })?;

    Ok(ProductProfile {
        base,
        base_url,
        token,
    })
}

/// Resolve profile for Bitbucket commands.
/// Only requires email and Bitbucket token (falls back to general token).
/// Email is optional for Bearer auth (access tokens).
fn resolve_profile_for_bitbucket(
    config: &Config,
    requested: Option<&str>,
) -> Result<BitbucketProfile> {
    let (name, profile) = config
        .resolve_profile(requested)
        .ok_or_else(|| anyhow!("No profile configured. Run `atlassian-cli auth login` first."))?;

    let is_bearer = auth::is_bitbucket_bearer(config, name);

    // Email is required for Basic auth, optional for Bearer
    let email = profile.email.clone().unwrap_or_default();
    if !is_bearer && email.is_empty() {
        return Err(anyhow!("Profile '{name}' is missing an email."));
    }

    let base = BaseProfile {
        name: name.to_string(),
        email,
    };

    // Try Bitbucket-specific token first, then fall back to general token
    let token = auth::get_bitbucket_token(&base.name)
        .or_else(|| auth::get_token(&base.name))
        .ok_or_else(|| {
            let has_jira_token = auth::get_token(&base.name).is_some();
            if has_jira_token {
                anyhow!(
                    "Profile '{}' has no Bitbucket token.\n\n\
                    Check token status: atlassian-cli auth list\n\
                    Look for 'has_bitbucket_token: true'\n\n\
                    To add: atlassian-cli auth login --bitbucket --profile {} --token <TOKEN> --email <EMAIL>\n\
                    For access tokens: atlassian-cli auth login --bitbucket --bearer --profile {} --token <TOKEN>",
                    base.name,
                    base.name,
                    base.name
                )
            } else {
                anyhow!(
                    "No token found for profile '{}'.\n\n\
                    Check configured profiles: atlassian-cli auth list --all\n\n\
                    To add: atlassian-cli auth login --bitbucket --profile {} --token <TOKEN> --email <EMAIL>\n\
                    For access tokens: atlassian-cli auth login --bitbucket --bearer --profile {} --token <TOKEN>",
                    base.name,
                    base.name,
                    base.name
                )
            }
        })?;

    // Resolve workspace: explicit profile config, or infer from base_url if present
    let workspace = profile.workspace.clone().or_else(|| {
        profile
            .base_url
            .as_ref()
            .and_then(|url| extract_workspace_from_url(url))
    });

    let bitbucket_remote = profile.bitbucket_remote.clone();

    Ok(BitbucketProfile {
        base,
        token,
        workspace,
        is_bearer,
        bitbucket_remote,
    })
}

fn build_product_client(profile: &ProductProfile) -> Result<ApiClient> {
    Ok(ApiClient::new(&profile.base_url)?
        .with_basic_auth(profile.base.email.clone(), profile.token.clone()))
}

fn build_bitbucket_client(profile: &BitbucketProfile) -> Result<ApiClient> {
    let client = ApiClient::new(BITBUCKET_API_URL)?;
    if profile.is_bearer {
        tracing::debug!("Using Bearer auth for Bitbucket");
        Ok(client.with_bearer_token(profile.token.clone()))
    } else {
        Ok(client.with_basic_auth(profile.base.email.clone(), profile.token.clone()))
    }
}

/// Profile for OpsGenie commands (requires api_key).
struct OpsgenieProfile {
    api_key: String,
    region: commands::opsgenie::Region,
}

/// Resolve profile for OpsGenie commands.
/// Requires opsgenie_api_key in the profile or OPSGENIE_API_KEY env var.
fn resolve_profile_for_opsgenie(
    config: &Config,
    requested: Option<&str>,
) -> Result<OpsgenieProfile> {
    // Try env var first
    if let Ok(api_key) = std::env::var("OPSGENIE_API_KEY") {
        let region = std::env::var("OPSGENIE_REGION")
            .ok()
            .map(|r| match r.to_lowercase().as_str() {
                "eu" => commands::opsgenie::Region::Eu,
                _ => commands::opsgenie::Region::Us,
            })
            .unwrap_or(commands::opsgenie::Region::Us);

        return Ok(OpsgenieProfile { api_key, region });
    }

    // Fall back to config profile
    let (name, profile) = config
        .resolve_profile(requested)
        .ok_or_else(|| anyhow!("No profile configured. Run `atlassian-cli auth login` first."))?;

    let api_key = profile.opsgenie_api_key.clone().ok_or_else(|| {
        anyhow!(
            "Profile '{}' is missing opsgenie_api_key. Set OPSGENIE_API_KEY env var or add opsgenie_api_key to your profile.",
            name
        )
    })?;

    let region = profile
        .opsgenie_region
        .as_ref()
        .map(|r| match r.to_lowercase().as_str() {
            "eu" => commands::opsgenie::Region::Eu,
            _ => commands::opsgenie::Region::Us,
        })
        .unwrap_or(commands::opsgenie::Region::Us);

    Ok(OpsgenieProfile { api_key, region })
}

fn build_opsgenie_client(profile: &OpsgenieProfile) -> Result<ApiClient> {
    Ok(ApiClient::new(profile.region.base_url())?.with_genie_key(profile.api_key.clone()))
}

/// Profile for Bamboo commands.
struct BambooProfile {
    email: String,
    token: String,
    base_url: String,
}

/// Resolve profile for Bamboo commands.
/// Uses bamboo_base_url if set, otherwise falls back to base_url.
fn resolve_profile_for_bamboo(config: &Config, requested: Option<&str>) -> Result<BambooProfile> {
    let (base, profile) = resolve_base_profile(config, requested)?;

    // Use bamboo_base_url if set, otherwise fall back to base_url
    let base_url = profile
        .bamboo_base_url
        .clone()
        .or_else(|| profile.base_url.clone())
        .ok_or_else(|| {
            anyhow!(
                "Profile '{}' is missing bamboo_base_url or base_url. Configure one in your profile.",
                base.name
            )
        })?;

    let token = auth::get_token(&base.name).ok_or_else(|| {
        anyhow!(
            "No token found for profile '{}'. Run `atlassian-cli auth login --profile {}`",
            base.name,
            base.name
        )
    })?;

    Ok(BambooProfile {
        email: base.email,
        token,
        base_url,
    })
}

fn build_bamboo_client(profile: &BambooProfile) -> Result<ApiClient> {
    Ok(ApiClient::new(&profile.base_url)?
        .with_basic_auth(profile.email.clone(), profile.token.clone()))
}