jacquard-api 0.11.0

// @generated by jacquard-lexicon. DO NOT EDIT.
//
// Lexicon: science.alt.dataset.storageS3
//
// This file was automatically generated from Lexicon schemas.
// Any manual changes will be overwritten on the next regeneration.

#[allow(unused_imports)]
use alloc::collections::BTreeMap;

#[allow(unused_imports)]
use core::marker::PhantomData;
use jacquard_common::CowStr;

#[allow(unused_imports)]
use jacquard_common::deps::codegen::unicode_segmentation::UnicodeSegmentation;
use jacquard_common::types::string::UriValue;
use jacquard_derive::{IntoStatic, lexicon};
use jacquard_lexicon::lexicon::LexiconDoc;
use jacquard_lexicon::schema::LexiconSchema;

#[allow(unused_imports)]
use jacquard_lexicon::validation::{ConstraintError, ValidationPath};
use serde::{Serialize, Deserialize};
use crate::science_alt::dataset::entry::ShardChecksum;
use crate::science_alt::dataset::storage_s3;
/// S3 or S3-compatible storage for WebDataset tar archives. Supports custom endpoints for MinIO, Cloudflare R2, and other S3-compatible services.

#[lexicon]
#[derive(Serialize, Deserialize, Debug, Clone, PartialEq, Eq, IntoStatic)]
#[serde(rename_all = "camelCase")]
pub struct StorageS3<'a> {
    ///S3 bucket name
    #[serde(borrow)]
    pub bucket: CowStr<'a>,
    ///Custom S3-compatible endpoint URL (e.g., for MinIO, Cloudflare R2). Omit for standard AWS S3.
    #[serde(skip_serializing_if = "Option::is_none")]
    #[serde(borrow)]
    pub endpoint: Option<UriValue<'a>>,
    ///AWS region (e.g., 'us-east-1'). Optional for S3-compatible services.
    #[serde(skip_serializing_if = "Option::is_none")]
    #[serde(borrow)]
    pub region: Option<CowStr<'a>>,
    ///Array of shard entries with object key and integrity checksum
    #[serde(borrow)]
    pub shards: Vec<storage_s3::ShardEntry<'a>>,
}

/// A single S3 object shard with integrity checksum

#[lexicon]
#[derive(Serialize, Deserialize, Debug, Clone, PartialEq, Eq, IntoStatic)]
#[serde(rename_all = "camelCase")]
pub struct ShardEntry<'a> {
    ///Content hash for integrity verification
    #[serde(borrow)]
    pub checksum: ShardChecksum<'a>,
    ///S3 object key for this WebDataset tar shard
    #[serde(borrow)]
    pub key: CowStr<'a>,
}

impl<'a> LexiconSchema for StorageS3<'a> {
    fn nsid() -> &'static str {
        "science.alt.dataset.storageS3"
    }
    fn def_name() -> &'static str {
        "main"
    }
    fn lexicon_doc() -> LexiconDoc<'static> {
        lexicon_doc_science_alt_dataset_storageS3()
    }
    fn validate(&self) -> Result<(), ConstraintError> {
        {
            let value = &self.bucket;
            #[allow(unused_comparisons)]
            if <str>::len(value.as_ref()) > 255usize {
                return Err(ConstraintError::MaxLength {
                    path: ValidationPath::from_field("bucket"),
                    max: 255usize,
                    actual: <str>::len(value.as_ref()),
                });
            }
        }
        if let Some(ref value) = self.endpoint {
            #[allow(unused_comparisons)]
            if <str>::len(value.as_ref()) > 500usize {
                return Err(ConstraintError::MaxLength {
                    path: ValidationPath::from_field("endpoint"),
                    max: 500usize,
                    actual: <str>::len(value.as_ref()),
                });
            }
        }
        if let Some(ref value) = self.region {
            #[allow(unused_comparisons)]
            if <str>::len(value.as_ref()) > 50usize {
                return Err(ConstraintError::MaxLength {
                    path: ValidationPath::from_field("region"),
                    max: 50usize,
                    actual: <str>::len(value.as_ref()),
                });
            }
        }
        {
            let value = &self.shards;
            #[allow(unused_comparisons)]
            if value.len() < 1usize {
                return Err(ConstraintError::MinLength {
                    path: ValidationPath::from_field("shards"),
                    min: 1usize,
                    actual: value.len(),
                });
            }
        }
        Ok(())
    }
}

impl<'a> LexiconSchema for ShardEntry<'a> {
    fn nsid() -> &'static str {
        "science.alt.dataset.storageS3"
    }
    fn def_name() -> &'static str {
        "shardEntry"
    }
    fn lexicon_doc() -> LexiconDoc<'static> {
        lexicon_doc_science_alt_dataset_storageS3()
    }
    fn validate(&self) -> Result<(), ConstraintError> {
        {
            let value = &self.key;
            #[allow(unused_comparisons)]
            if <str>::len(value.as_ref()) > 1024usize {
                return Err(ConstraintError::MaxLength {
                    path: ValidationPath::from_field("key"),
                    max: 1024usize,
                    actual: <str>::len(value.as_ref()),
                });
            }
        }
        Ok(())
    }
}

pub mod storage_s3_state {

    pub use crate::builder_types::{Set, Unset, IsSet, IsUnset};
    #[allow(unused)]
    use ::core::marker::PhantomData;
    mod sealed {
        pub trait Sealed {}
    }
    /// State trait tracking which required fields have been set
    pub trait State: sealed::Sealed {
        type Shards;
        type Bucket;
    }
    /// Empty state - all required fields are unset
    pub struct Empty(());
    impl sealed::Sealed for Empty {}
    impl State for Empty {
        type Shards = Unset;
        type Bucket = Unset;
    }
    ///State transition - sets the `shards` field to Set
    pub struct SetShards<S: State = Empty>(PhantomData<fn() -> S>);
    impl<S: State> sealed::Sealed for SetShards<S> {}
    impl<S: State> State for SetShards<S> {
        type Shards = Set<members::shards>;
        type Bucket = S::Bucket;
    }
    ///State transition - sets the `bucket` field to Set
    pub struct SetBucket<S: State = Empty>(PhantomData<fn() -> S>);
    impl<S: State> sealed::Sealed for SetBucket<S> {}
    impl<S: State> State for SetBucket<S> {
        type Shards = S::Shards;
        type Bucket = Set<members::bucket>;
    }
    /// Marker types for field names
    #[allow(non_camel_case_types)]
    pub mod members {
        ///Marker type for the `shards` field
        pub struct shards(());
        ///Marker type for the `bucket` field
        pub struct bucket(());
    }
}

/// Builder for constructing an instance of this type
pub struct StorageS3Builder<'a, S: storage_s3_state::State> {
    _state: PhantomData<fn() -> S>,
    _fields: (
        Option<CowStr<'a>>,
        Option<UriValue<'a>>,
        Option<CowStr<'a>>,
        Option<Vec<storage_s3::ShardEntry<'a>>>,
    ),
    _lifetime: PhantomData<&'a ()>,
}

impl<'a> StorageS3<'a> {
    /// Create a new builder for this type
    pub fn new() -> StorageS3Builder<'a, storage_s3_state::Empty> {
        StorageS3Builder::new()
    }
}

impl<'a> StorageS3Builder<'a, storage_s3_state::Empty> {
    /// Create a new builder with all fields unset
    pub fn new() -> Self {
        StorageS3Builder {
            _state: PhantomData,
            _fields: (None, None, None, None),
            _lifetime: PhantomData,
        }
    }
}

impl<'a, S> StorageS3Builder<'a, S>
where
    S: storage_s3_state::State,
    S::Bucket: storage_s3_state::IsUnset,
{
    /// Set the `bucket` field (required)
    pub fn bucket(
        mut self,
        value: impl Into<CowStr<'a>>,
    ) -> StorageS3Builder<'a, storage_s3_state::SetBucket<S>> {
        self._fields.0 = Option::Some(value.into());
        StorageS3Builder {
            _state: PhantomData,
            _fields: self._fields,
            _lifetime: PhantomData,
        }
    }
}

impl<'a, S: storage_s3_state::State> StorageS3Builder<'a, S> {
    /// Set the `endpoint` field (optional)
    pub fn endpoint(mut self, value: impl Into<Option<UriValue<'a>>>) -> Self {
        self._fields.1 = value.into();
        self
    }
    /// Set the `endpoint` field to an Option value (optional)
    pub fn maybe_endpoint(mut self, value: Option<UriValue<'a>>) -> Self {
        self._fields.1 = value;
        self
    }
}

impl<'a, S: storage_s3_state::State> StorageS3Builder<'a, S> {
    /// Set the `region` field (optional)
    pub fn region(mut self, value: impl Into<Option<CowStr<'a>>>) -> Self {
        self._fields.2 = value.into();
        self
    }
    /// Set the `region` field to an Option value (optional)
    pub fn maybe_region(mut self, value: Option<CowStr<'a>>) -> Self {
        self._fields.2 = value;
        self
    }
}

impl<'a, S> StorageS3Builder<'a, S>
where
    S: storage_s3_state::State,
    S::Shards: storage_s3_state::IsUnset,
{
    /// Set the `shards` field (required)
    pub fn shards(
        mut self,
        value: impl Into<Vec<storage_s3::ShardEntry<'a>>>,
    ) -> StorageS3Builder<'a, storage_s3_state::SetShards<S>> {
        self._fields.3 = Option::Some(value.into());
        StorageS3Builder {
            _state: PhantomData,
            _fields: self._fields,
            _lifetime: PhantomData,
        }
    }
}

impl<'a, S> StorageS3Builder<'a, S>
where
    S: storage_s3_state::State,
    S::Shards: storage_s3_state::IsSet,
    S::Bucket: storage_s3_state::IsSet,
{
    /// Build the final struct
    pub fn build(self) -> StorageS3<'a> {
        StorageS3 {
            bucket: self._fields.0.unwrap(),
            endpoint: self._fields.1,
            region: self._fields.2,
            shards: self._fields.3.unwrap(),
            extra_data: Default::default(),
        }
    }
    /// Build the final struct with custom extra_data
    pub fn build_with_data(
        self,
        extra_data: BTreeMap<
            jacquard_common::deps::smol_str::SmolStr,
            jacquard_common::types::value::Data<'a>,
        >,
    ) -> StorageS3<'a> {
        StorageS3 {
            bucket: self._fields.0.unwrap(),
            endpoint: self._fields.1,
            region: self._fields.2,
            shards: self._fields.3.unwrap(),
            extra_data: Some(extra_data),
        }
    }
}

fn lexicon_doc_science_alt_dataset_storageS3() -> LexiconDoc<'static> {
    #[allow(unused_imports)]
    use jacquard_common::{CowStr, deps::smol_str::SmolStr, types::blob::MimeType};
    use jacquard_lexicon::lexicon::*;
    use alloc::collections::BTreeMap;
    LexiconDoc {
        lexicon: Lexicon::Lexicon1,
        id: CowStr::new_static("science.alt.dataset.storageS3"),
        defs: {
            let mut map = BTreeMap::new();
            map.insert(
                SmolStr::new_static("main"),
                LexUserType::Object(LexObject {
                    description: Some(
                        CowStr::new_static(
                            "S3 or S3-compatible storage for WebDataset tar archives. Supports custom endpoints for MinIO, Cloudflare R2, and other S3-compatible services.",
                        ),
                    ),
                    required: Some(
                        vec![
                            SmolStr::new_static("bucket"), SmolStr::new_static("shards")
                        ],
                    ),
                    properties: {
                        #[allow(unused_mut)]
                        let mut map = BTreeMap::new();
                        map.insert(
                            SmolStr::new_static("bucket"),
                            LexObjectProperty::String(LexString {
                                description: Some(CowStr::new_static("S3 bucket name")),
                                max_length: Some(255usize),
                                ..Default::default()
                            }),
                        );
                        map.insert(
                            SmolStr::new_static("endpoint"),
                            LexObjectProperty::String(LexString {
                                description: Some(
                                    CowStr::new_static(
                                        "Custom S3-compatible endpoint URL (e.g., for MinIO, Cloudflare R2). Omit for standard AWS S3.",
                                    ),
                                ),
                                format: Some(LexStringFormat::Uri),
                                max_length: Some(500usize),
                                ..Default::default()
                            }),
                        );
                        map.insert(
                            SmolStr::new_static("region"),
                            LexObjectProperty::String(LexString {
                                description: Some(
                                    CowStr::new_static(
                                        "AWS region (e.g., 'us-east-1'). Optional for S3-compatible services.",
                                    ),
                                ),
                                max_length: Some(50usize),
                                ..Default::default()
                            }),
                        );
                        map.insert(
                            SmolStr::new_static("shards"),
                            LexObjectProperty::Array(LexArray {
                                description: Some(
                                    CowStr::new_static(
                                        "Array of shard entries with object key and integrity checksum",
                                    ),
                                ),
                                items: LexArrayItem::Ref(LexRef {
                                    r#ref: CowStr::new_static("#shardEntry"),
                                    ..Default::default()
                                }),
                                min_length: Some(1usize),
                                ..Default::default()
                            }),
                        );
                        map
                    },
                    ..Default::default()
                }),
            );
            map.insert(
                SmolStr::new_static("shardEntry"),
                LexUserType::Object(LexObject {
                    description: Some(
                        CowStr::new_static(
                            "A single S3 object shard with integrity checksum",
                        ),
                    ),
                    required: Some(
                        vec![SmolStr::new_static("key"), SmolStr::new_static("checksum")],
                    ),
                    properties: {
                        #[allow(unused_mut)]
                        let mut map = BTreeMap::new();
                        map.insert(
                            SmolStr::new_static("checksum"),
                            LexObjectProperty::Ref(LexRef {
                                r#ref: CowStr::new_static(
                                    "science.alt.dataset.entry#shardChecksum",
                                ),
                                ..Default::default()
                            }),
                        );
                        map.insert(
                            SmolStr::new_static("key"),
                            LexObjectProperty::String(LexString {
                                description: Some(
                                    CowStr::new_static(
                                        "S3 object key for this WebDataset tar shard",
                                    ),
                                ),
                                max_length: Some(1024usize),
                                ..Default::default()
                            }),
                        );
                        map
                    },
                    ..Default::default()
                }),
            );
            map
        },
        ..Default::default()
    }
}

pub mod shard_entry_state {

    pub use crate::builder_types::{Set, Unset, IsSet, IsUnset};
    #[allow(unused)]
    use ::core::marker::PhantomData;
    mod sealed {
        pub trait Sealed {}
    }
    /// State trait tracking which required fields have been set
    pub trait State: sealed::Sealed {
        type Key;
        type Checksum;
    }
    /// Empty state - all required fields are unset
    pub struct Empty(());
    impl sealed::Sealed for Empty {}
    impl State for Empty {
        type Key = Unset;
        type Checksum = Unset;
    }
    ///State transition - sets the `key` field to Set
    pub struct SetKey<S: State = Empty>(PhantomData<fn() -> S>);
    impl<S: State> sealed::Sealed for SetKey<S> {}
    impl<S: State> State for SetKey<S> {
        type Key = Set<members::key>;
        type Checksum = S::Checksum;
    }
    ///State transition - sets the `checksum` field to Set
    pub struct SetChecksum<S: State = Empty>(PhantomData<fn() -> S>);
    impl<S: State> sealed::Sealed for SetChecksum<S> {}
    impl<S: State> State for SetChecksum<S> {
        type Key = S::Key;
        type Checksum = Set<members::checksum>;
    }
    /// Marker types for field names
    #[allow(non_camel_case_types)]
    pub mod members {
        ///Marker type for the `key` field
        pub struct key(());
        ///Marker type for the `checksum` field
        pub struct checksum(());
    }
}

/// Builder for constructing an instance of this type
pub struct ShardEntryBuilder<'a, S: shard_entry_state::State> {
    _state: PhantomData<fn() -> S>,
    _fields: (Option<ShardChecksum<'a>>, Option<CowStr<'a>>),
    _lifetime: PhantomData<&'a ()>,
}

impl<'a> ShardEntry<'a> {
    /// Create a new builder for this type
    pub fn new() -> ShardEntryBuilder<'a, shard_entry_state::Empty> {
        ShardEntryBuilder::new()
    }
}

impl<'a> ShardEntryBuilder<'a, shard_entry_state::Empty> {
    /// Create a new builder with all fields unset
    pub fn new() -> Self {
        ShardEntryBuilder {
            _state: PhantomData,
            _fields: (None, None),
            _lifetime: PhantomData,
        }
    }
}

impl<'a, S> ShardEntryBuilder<'a, S>
where
    S: shard_entry_state::State,
    S::Checksum: shard_entry_state::IsUnset,
{
    /// Set the `checksum` field (required)
    pub fn checksum(
        mut self,
        value: impl Into<ShardChecksum<'a>>,
    ) -> ShardEntryBuilder<'a, shard_entry_state::SetChecksum<S>> {
        self._fields.0 = Option::Some(value.into());
        ShardEntryBuilder {
            _state: PhantomData,
            _fields: self._fields,
            _lifetime: PhantomData,
        }
    }
}

impl<'a, S> ShardEntryBuilder<'a, S>
where
    S: shard_entry_state::State,
    S::Key: shard_entry_state::IsUnset,
{
    /// Set the `key` field (required)
    pub fn key(
        mut self,
        value: impl Into<CowStr<'a>>,
    ) -> ShardEntryBuilder<'a, shard_entry_state::SetKey<S>> {
        self._fields.1 = Option::Some(value.into());
        ShardEntryBuilder {
            _state: PhantomData,
            _fields: self._fields,
            _lifetime: PhantomData,
        }
    }
}

impl<'a, S> ShardEntryBuilder<'a, S>
where
    S: shard_entry_state::State,
    S::Key: shard_entry_state::IsSet,
    S::Checksum: shard_entry_state::IsSet,
{
    /// Build the final struct
    pub fn build(self) -> ShardEntry<'a> {
        ShardEntry {
            checksum: self._fields.0.unwrap(),
            key: self._fields.1.unwrap(),
            extra_data: Default::default(),
        }
    }
    /// Build the final struct with custom extra_data
    pub fn build_with_data(
        self,
        extra_data: BTreeMap<
            jacquard_common::deps::smol_str::SmolStr,
            jacquard_common::types::value::Data<'a>,
        >,
    ) -> ShardEntry<'a> {
        ShardEntry {
            checksum: self._fields.0.unwrap(),
            key: self._fields.1.unwrap(),
            extra_data: Some(extra_data),
        }
    }
}