Codex Handbook
execpolicy-legacy/src/opt.rs 77 lines
#![allow(clippy::needless_lifetimes)]use crate::ArgType;use crate::starlark::values::ValueLike;use allocative::Allocative;use derive_more::derive::Display;use starlark::any::ProvidesStaticType;use starlark::values::AllocValue;use starlark::values::Heap;use starlark::values::NoSerialize;use starlark::values::StarlarkValue;use starlark::values::UnpackValue;use starlark::values::Value;use starlark::values::starlark_value;/// Command line option that takes a value.#[derive(Clone, Debug, Display, PartialEq, Eq, ProvidesStaticType, NoSerialize, Allocative)]#[display("opt({})", opt)]pub struct Opt {    /// The option as typed on the command line, e.g., `-h` or `--help`. If    /// it can be used in the `--name=value` format, then this should be    /// `--name` (though this is subject to change).    pub opt: String,    pub meta: OptMeta,    pub required: bool,}/// When defining an Opt, use as specific an OptMeta as possible.#[derive(Clone, Debug, Display, PartialEq, Eq, ProvidesStaticType, NoSerialize, Allocative)]#[display("{}", self)]pub enum OptMeta {    /// Option does not take a value.    Flag,    /// Option takes a single value matching the specified type.    Value(ArgType),}impl Opt {    pub fn new(opt: String, meta: OptMeta, required: bool) -> Self {        Self {            opt,            meta,            required,        }    }    pub fn name(&self) -> &str {        &self.opt    }}#[starlark_value(type = "Opt")]impl<'v> StarlarkValue<'v> for Opt {    type Canonical = Opt;}impl<'v> UnpackValue<'v> for Opt {    type Error = starlark::Error;    fn unpack_value_impl(value: Value<'v>) -> starlark::Result<Option<Self>> {        // TODO(mbolin): It fels like this should be doable without cloning?        // Cannot simply consume the value?        Ok(value.downcast_ref::<Opt>().cloned())    }}impl<'v> AllocValue<'v> for Opt {    fn alloc_value(self, heap: Heap<'v>) -> Value<'v> {        heap.alloc_simple(self)    }}#[starlark_value(type = "OptMeta")]impl<'v> StarlarkValue<'v> for OptMeta {    type Canonical = OptMeta;}