bodhi-cli 2.1.2

use std::str::FromStr;

use bodhi::*;
use clap::Parser;

#[derive(Clone, Debug)]
pub enum Format {
    JSON,
    Plain,
}

impl TryFrom<&str> for Format {
    type Error = String;

    fn try_from(value: &str) -> Result<Format, String> {
        match value.to_lowercase().as_str() {
            "json" => Ok(Format::JSON),
            "plain" => Ok(Format::Plain),
            _ => Err(format!("Not a recognised value for format: {}", &value)),
        }
    }
}

impl FromStr for Format {
    type Err = String;

    fn from_str(s: &str) -> Result<Format, String> {
        TryFrom::try_from(s)
    }
}

/// bodhi-cli expects a configuration file at ~/.config/fedora.toml, with at
/// least the following contents:
///
/// """
/// [FAS]
/// username = "USERNAME"
/// """
///
/// This username is used for logging in with bodhi for authenticated requests,
/// and for determining which updates, overrides, and comments the user has
/// created themselves.
#[derive(Debug, Parser)]
#[command(disable_help_subcommand(true))]
#[clap(verbatim_doc_comment)]
pub struct BaseCommand {
    /// Use the fedora staging instance of bodhi
    #[arg(long)]
    pub staging: bool,
    /// Manually specify bodhi server URL
    #[arg(long, requires("login_url"), conflicts_with("staging"))]
    pub bodhi_url: Option<String>,
    /// Manually specify OpenID endpoint URL
    #[arg(long, requires("bodhi_url"), conflicts_with("staging"))]
    pub login_url: Option<String>,
    /// Don't store password in session keyring
    #[arg(long, short = 'n')]
    pub no_store_password: bool,
    /// Ignore password stored in session keyring
    #[arg(long, short = 'k')]
    pub ignore_keyring: bool,
    /// Make output more verbose
    #[arg(long, short = 'v')]
    pub verbose: bool,
    #[command(subcommand)]
    pub subcommand: BodhiCommand,
}

#[allow(clippy::large_enum_variant)]
#[derive(Debug, Parser)]
pub enum BodhiCommand {
    /// Comment on an update
    Comment {
        /// ID of the update to comment on
        #[arg(long)]
        update: String,
        /// Publicly visible comment text
        #[arg(long)]
        text: String,
        /// Karma submitted with this comment (-1/0/+1)
        #[arg(long)]
        karma: Option<Karma>,
    },
    /// Query bodhi for information about a compose
    ComposeInfo {
        /// release string
        release: FedoraRelease,
        /// request string ("stable" or "testing")
        request: ComposeRequest,
        /// Output format (plain, JSON)
        #[arg(long)]
        format: Option<Format>,
    },
    /// Query bodhi for running composes
    ComposeList {
        /// Output format (plain, JSON)
        #[arg(long)]
        format: Option<Format>,
    },
    /// Create a new buildroot override
    CreateOverride {
        /// NVR of the override
        nvr: String,
        /// duration (in days) it should be active
        #[arg(long)]
        duration: u32,
        /// publicly visible notes
        #[arg(long)]
        notes: String,
    },
    /// Create buildroot overrides from an update
    CreateUpdateOverride {
        /// alias of the update (i.e. "FEDORA-2022-XXXXXXXXXX")
        alias: String,
        /// duration (in days) it should be active
        #[arg(long)]
        duration: u32,
        /// publicly visible notes
        #[arg(long)]
        notes: String,
    },
    /// Create a new update
    CreateUpdate {
        /// Push to stable based on karma
        #[arg(long)]
        autokarma: Option<bool>,
        /// Push to stable based on time
        #[arg(long)]
        autotime: Option<bool>,
        /// Bugs fixed by this update
        #[arg(long)]
        bugs: Option<Vec<u32>>,
        /// Builds that are part of this update
        #[arg(long, conflicts_with = "from_tag")]
        builds: Option<Vec<String>>,
        /// Close bugs when pushed to stable
        #[arg(long)]
        close_bugs: Option<bool>,
        /// Override displayed update name
        #[arg(long)]
        display_name: Option<String>,
        /// Koji tag to create this update from
        #[arg(long, conflicts_with = "builds")]
        from_tag: Option<String>,
        /// Publicly visible update notes
        #[arg(long)]
        notes: String,
        /// Require bug feedback for karma to count
        #[arg(long)]
        require_bugs: Option<bool>,
        /// Require test case feedback for karma to count
        #[arg(long)]
        require_testcases: Option<bool>,
        /// List of required gating tests
        #[arg(long)]
        requirements: Option<Vec<String>>,
        /// Update severity
        #[arg(long)]
        severity: Option<UpdateSeverity>,
        /// Days until it can be pushed to stable
        #[arg(long)]
        stable_days: Option<u32>,
        /// Karma until it can be pushed to stable
        #[arg(long)]
        stable_karma: Option<i32>,
        /// Logout / reboot suggestion
        #[arg(long)]
        suggestion: Option<UpdateSuggestion>,
        /// Karma until it will be unpushed
        #[arg(long)]
        unstable_karma: Option<i32>,
        /// Type of the update
        #[arg(long, name = "type")]
        update_type: Option<UpdateType>,
    },
    /// Edit an existing buildroot override
    EditOverride {
        /// NVR of the override
        nvr: String,
        /// duration it will still be active
        #[arg(long)]
        duration: u32,
        /// publicly visible notes
        #[arg(long)]
        notes: String,
    },
    /// Edit an existing update
    EditUpdate {
        /// Alias of the edited update
        alias: String,
        /// Add bugs to this update
        #[arg(long)]
        add_bugs: Option<Vec<u32>>,
        /// Add builds to this update
        #[arg(long)]
        add_builds: Option<Vec<String>>,
        /// Push to stable based on karma
        #[arg(long)]
        autokarma: Option<bool>,
        /// Push to stable based on time
        #[arg(long)]
        autotime: Option<bool>,
        /// Close bugs when pushed to stable
        #[arg(long)]
        close_bugs: Option<bool>,
        /// Override displayed update name
        #[arg(long)]
        display_name: Option<String>,
        /// Publicly visible update notes
        #[arg(long)]
        notes: Option<String>,
        /// Remove bugs from this update
        #[arg(long)]
        remove_bugs: Option<Vec<u32>>,
        /// Remove builds from this update
        #[arg(long)]
        remove_builds: Option<Vec<String>>,
        /// List of required gating tests
        #[arg(long)]
        requirements: Option<Vec<String>>,
        /// Update severity
        #[arg(long)]
        severity: Option<UpdateSeverity>,
        /// Days until it can be pushed to stable
        #[arg(long)]
        stable_days: Option<u32>,
        /// Karma until it can be pushed to stable
        #[arg(long)]
        stable_karma: Option<i32>,
        /// Logout / reboot suggestion
        #[arg(long)]
        suggestion: Option<UpdateSuggestion>,
        /// Karma until it will be unpushed
        #[arg(long)]
        unstable_karma: Option<i32>,
        /// Type of the update
        #[arg(long, name = "type")]
        update_type: Option<UpdateType>,
    },
    /// Expire an existing buildroot override
    ExpireOverride {
        /// NVR of the override
        nvr: String,
    },
    /// Query bodhi for buildroot overrides
    QueryOverrides {
        /// Query for this build / these builds
        #[arg(long)]
        builds: Option<Vec<String>>,
        /// Query for expired overrides
        #[arg(long)]
        expired: Option<bool>,
        /// Output format (plain, JSON)
        #[arg(long)]
        format: Option<Format>,
        /// Query for this release / these releases
        #[arg(long)]
        releases: Option<Vec<FedoraRelease>>,
        /// Query for overrides submitted by these users
        #[arg(long)]
        users: Option<Vec<String>>,
        /// Force long-running queries
        #[arg(long, short)]
        force: bool,
    },
    /// Query bodhi for updates
    QueryUpdates {
        /// update with this alias
        #[arg(long)]
        alias: Option<String>,
        /// updates associated with these bugs
        #[arg(long)]
        bugs: Option<Vec<u32>>,
        /// updates associated with these builds
        #[arg(long)]
        builds: Option<Vec<String>>,
        /// updates for critpath packages
        #[arg(long)]
        critpath: Option<bool>,
        /// RPM / module / flatpak updates
        #[arg(long)]
        content_type: Option<ContentType>,
        /// Output format (plain, JSON)
        #[arg(long)]
        format: Option<Format>,
        /// locked updates
        #[arg(long)]
        locked: Option<bool>,
        /// updates modified before this date
        #[arg(long)]
        modified_before: Option<BodhiDate>,
        /// updates modified after this date
        #[arg(long)]
        modified_since: Option<BodhiDate>,
        /// updates for these packages
        #[arg(long)]
        packages: Option<Vec<String>>,
        /// pushed updates
        #[arg(long)]
        pushed: Option<bool>,
        /// updates pushed before this date
        #[arg(long)]
        pushed_before: Option<BodhiDate>,
        /// updates pushed after this date
        #[arg(long)]
        pushed_since: Option<BodhiDate>,
        /// updates for these releases
        #[arg(long)]
        releases: Option<Vec<FedoraRelease>>,
        /// updates with this status request
        #[arg(long)]
        request: Option<UpdateRequest>,
        /// updates with this severity
        #[arg(long)]
        severity: Option<UpdateSeverity>,
        /// updates with this status
        #[arg(long)]
        status: Option<UpdateStatus>,
        /// updates submitted before this date
        #[arg(long)]
        submitted_before: Option<BodhiDate>,
        /// updates submitted after this date
        #[arg(long)]
        submitted_since: Option<BodhiDate>,
        /// updates with logout / reboot suggestion
        #[arg(long)]
        suggestion: Option<UpdateSuggestion>,
        /// updates with this type
        #[arg(name = "type", long)]
        update_type: Option<UpdateType>,
        /// updates submitted by this user
        #[arg(long)]
        users: Option<Vec<String>>,
        /// Force long-running queries
        #[arg(long, short)]
        force: bool,
    },
    /// Query bodhi for information about a release
    ReleaseInfo {
        /// ID of the release
        release: String,
        /// Output format (plain, JSON)
        #[arg(long)]
        format: Option<Format>,
    },
    /// Query bodhi for active releases
    ReleaseList {
        /// Output format (plain, JSON)
        #[arg(long)]
        format: Option<Format>,
    },
    /// Submit an update status request
    UpdateRequest {
        /// ID of the update
        alias: String,
        /// (obsolete, revoke, stable, testing, unpush)
        request: UpdateRequest,
    },
    /// Waive an update's test results
    WaiveTests {
        /// ID of the update
        alias: String,
        /// comment submitted with the waiver
        comment: String,
        /// test results to be waived (default: empty / all)
        #[arg(long)]
        tests: Option<Vec<String>>,
    },
}

impl BaseCommand {
    pub fn authenticated(&self) -> bool {
        use BodhiCommand::*;

        match self.subcommand {
            Comment { .. } => true,
            ComposeInfo { .. } => false,
            ComposeList { .. } => false,
            CreateOverride { .. } => true,
            CreateUpdateOverride { .. } => true,
            CreateUpdate { .. } => true,
            EditOverride { .. } => true,
            EditUpdate { .. } => true,
            ExpireOverride { .. } => true,
            QueryOverrides { .. } => false,
            QueryUpdates { .. } => false,
            ReleaseInfo { .. } => false,
            ReleaseList { .. } => false,
            UpdateRequest { .. } => true,
            WaiveTests { .. } => true,
        }
    }
}