gflags 0.3.12

use crate::atomic::StaticAtomicPtr;
use ref_cast::RefCast;
use std::ops::Deref;
use std::sync::atomic::{AtomicUsize, Ordering};

/// The state associated with a single flag.
///
/// An invocation of [`gflags::define!`] with flag long name `--the_name` and
/// flag type `T` declares an item `static the_name: Flag<T>` through which the
/// state of the flag can be accessed.
///
/// After [`gflags::parse()`] has been called, the value of a flag is available
/// through its `.flag` field which is of type `T`.
///
/// [`gflags::define!`]: macro.define.html
/// [`gflags::parse()`]: fn.parse.html
///
/// # Examples
///
/// ```
/// use std::path::Path;
///
/// gflags::define! {
///     /// Search for patterns from the given file, with one pattern per line.
///     -f, --file: &Path
/// }
///
/// fn main() {
///     let patterns = gflags::parse();
///
///     if FILE.is_present() {
///         let path = FILE.flag;
///         println!("searching for patterns from file: {}", path.display());
///     } else {
///         println!("searching for patterns given on command line: {:?}", patterns);
///     }
/// }
/// ```
pub struct Flag<T> {
    atomic: StaticAtomicPtr<T>,
    present: AtomicUsize,
}

impl<T: 'static> Flag<T> {
    /// Whether this flag was provided on the command line.
    ///
    /// When using flags for which a default value is not provided, be sure to
    /// check `.is_present()` because accessing `.flag` when not present will
    /// cause a panic.
    ///
    /// When a flag has a default value and is not passed on the command line,
    /// `is_present()` will be false and `.flag` will refer to the default
    /// value.
    pub fn is_present(&self) -> bool {
        self.present.load(Ordering::Acquire) != 0
    }

    /// Count number of times an option is repeated on the command line.
    ///
    /// Useful to display verbosity or debug level by repeating a boolean flag
    /// several times. For example, `-vv` for "very verbose" (repeat count 2)
    /// or `-ddd` for debug level 3.
    pub fn repeat_count(&self) -> u32 {
        self.present.load(Ordering::Acquire) as u32
    }
}

#[derive(RefCast)]
#[repr(transparent)]
pub struct Accessor<T> {
    /// Value of the flag.
    pub flag: T,
}

impl<T: 'static> Flag<T> {
    // Not public API. Called from generated code.
    #[doc(hidden)]
    pub const fn new(default: &'static T) -> Self {
        Flag {
            atomic: StaticAtomicPtr::new(default),
            present: AtomicUsize::new(0),
        }
    }

    // Not public API. Called from generated code.
    #[doc(hidden)]
    pub const fn null() -> Self {
        Flag {
            atomic: StaticAtomicPtr::null(),
            present: AtomicUsize::new(0),
        }
    }

    pub(crate) fn set(&self, value: T) {
        let ptr = Box::leak(Box::new(value));
        self.atomic.store(ptr, Ordering::Release);
        self.present.fetch_add(1, Ordering::Release);
    }
}

impl Flag<bool> {
    #[allow(clippy::trivially_copy_pass_by_ref)]
    pub(crate) fn set_bool(&self, value: &'static bool) {
        self.atomic.store(value, Ordering::Release);
        self.present.fetch_add(1, Ordering::Release);
    }
}

impl<T: 'static> Deref for Flag<T> {
    type Target = Accessor<T>;

    fn deref(&self) -> &Self::Target {
        Accessor::ref_cast(self.atomic.load(Ordering::Acquire))
    }
}