Quelle  update_work_properties.rs   Sprache: Rust

//! Example for updating an availability status.
#![expect(unused_crate_dependencies, reason = "Example triggered false positive")]
use clap::{ArgAction, Args, Parser};
use libthreema::{
    cli::config::{MinimalIdentityConfig, MinimalIdentityConfigOptions},
    https::cli::https_client_builder,
    protobuf,
    utils::logging::init_stderr_logging,
    work::properties::{
        WorkAvailabilityStatus, WorkProperties, WorkPropertiesUpdateContext, WorkPropertiesUpdateLoop,
        WorkPropertiesUpdateResponse, WorkPropertiesUpdateTask,
    },
};
use tracing::Level;

#[derive(Clone, Debug, Args)]
struct WorkAvailabilityStatusOptions {
    #[arg(long)]
    category: protobuf::d2d_sync::WorkAvailabilityStatusCategory,

    #[arg(long, requires = "category")]
    description: Option<String>,
}

#[derive(Args)]
struct WorkPropertiesOptions {
    #[command(flatten)]
    availability_status: WorkAvailabilityStatusOptions,
}
impl From<WorkPropertiesOptions> for WorkProperties {
    fn from(properties: WorkPropertiesOptions) -> Self {
        Self {
            availability_status: Some(match properties.availability_status.category {
                protobuf::d2d_sync::WorkAvailabilityStatusCategory::None => WorkAvailabilityStatus::None,
                protobuf::d2d_sync::WorkAvailabilityStatusCategory::Unavailable => {
                    WorkAvailabilityStatus::Unavailable(
                        properties.availability_status.description.unwrap_or_default(),
                    )
                },
                protobuf::d2d_sync::WorkAvailabilityStatusCategory::Busy => WorkAvailabilityStatus::Busy(
                    properties.availability_status.description.unwrap_or_default(),
                ),
            }),
        }
    }
}

#[derive(Parser)]
#[command()]
struct AvailabilityStatusCommand {
    #[command(flatten)]
    config: MinimalIdentityConfigOptions,

    #[command(flatten)]
    work_properties: WorkPropertiesOptions,

    /// Turn on debugging information.
    /// Use -v -vv to control log level.
    #[arg(short, action = ArgAction::Count)]
    verbose: u8,
}

async fn run_set_availability_status(
    http_client: reqwest::Client,
    context: WorkPropertiesUpdateContext,
    work_properties: WorkProperties,
) -> anyhow::Result<()> {
    let mut task = WorkPropertiesUpdateTask::new(context, work_properties);
    loop {
        match task.poll()? {
            WorkPropertiesUpdateLoop::Instruction(instruction) => {
                let result = instruction.request.send(&http_client).await;
                task.response(WorkPropertiesUpdateResponse { result })?;
            },

            WorkPropertiesUpdateLoop::Done(()) => return Ok(()),
        }
    }
}

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    // Parse arguments for command.
    let arguments = AvailabilityStatusCommand::parse();

    // Configure logging.
    let log_level = match arguments.verbose {
        1 => Level::DEBUG,
        2 => Level::TRACE,
        _ => Level::INFO,
    };
    init_stderr_logging(log_level);

    // Create HTTP client.
    let http_client = https_client_builder().build()?;
    let config = MinimalIdentityConfig::from_options(&http_client, arguments.config).await?;

    // Update the availability status.
    run_set_availability_status(
        http_client,
        config.work_properties_update_context()?,
        arguments.work_properties.into(),
    )
    .await?;
    Ok(())
}

#[test]
fn verify_cli() {
    use clap::CommandFactory;
    AvailabilityStatusCommand::command().debug_assert();
}