tools/testgen/src/flags.rs

// Copyright 2023 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

/// NOTE: The doc comments on `Flags` and its fields appear as the helptext of
/// `fx testgen`. Please run that command to make sure the output looks correct before
/// submitting changes.
use argh::FromArgs;
use tracing::Level;

/// testgen generates a Fuchsia test.
#[derive(FromArgs, Debug)]
pub(crate) struct Flags {
    #[argh(subcommand)]
    pub subcommand: Subcommand,

    /// if true, all logs are printed. Otherwise only errors are shown.
    #[argh(option, short = 'v', default = "false")]
    pub verbose_logging: bool,
}

#[derive(FromArgs, PartialEq, Debug)]
#[argh(subcommand)]
pub(crate) enum Subcommand {
    IntegrationTest(crate::cmd_integration_test::IntegrationTestCmd),
}

impl Flags {
    #[must_use]
    pub fn setup_logging(&self) -> impl Drop {
        let subscriber = tracing_subscriber::fmt::Subscriber::builder()
            .with_max_level(match self.verbose_logging {
                true => Level::INFO,
                false => Level::ERROR,
            })
            .finish();
        tracing::subscriber::set_default(subscriber)
    }
}