fakecloud_s3/
state.rs

1use bytes::Bytes;
2use chrono::{DateTime, Utc};
3use parking_lot::RwLock;
4use std::collections::{BTreeMap, HashMap};
5use std::sync::Arc;
6
7/// An ACL grant entry.
8#[derive(Debug, Clone)]
9pub struct AclGrant {
10    pub grantee_type: String, // "CanonicalUser" or "Group"
11    pub grantee_id: Option<String>,
12    pub grantee_display_name: Option<String>,
13    pub grantee_uri: Option<String>,
14    pub permission: String, // READ, WRITE, READ_ACP, WRITE_ACP, FULL_CONTROL
15}
16
17#[derive(Debug, Clone)]
18pub struct S3Object {
19    pub key: String,
20    pub data: Bytes,
21    pub content_type: String,
22    pub etag: String,
23    pub size: u64,
24    pub last_modified: DateTime<Utc>,
25    pub metadata: HashMap<String, String>,
26    pub storage_class: String,
27    pub tags: HashMap<String, String>,
28    pub acl_grants: Vec<AclGrant>,
29    pub acl_owner_id: Option<String>,
30    /// If created from multipart upload, the number of parts.
31    pub parts_count: Option<u32>,
32    /// Per-part sizes for multipart objects (part_number, size).
33    pub part_sizes: Option<Vec<(u32, u64)>>,
34    /// Server-side encryption algorithm.
35    pub sse_algorithm: Option<String>,
36    /// KMS key ID for SSE-KMS.
37    pub sse_kms_key_id: Option<String>,
38    /// Whether bucket key is enabled.
39    pub bucket_key_enabled: Option<bool>,
40    pub version_id: Option<String>,
41    pub is_delete_marker: bool,
42    pub content_encoding: Option<String>,
43    pub website_redirect_location: Option<String>,
44    /// Glacier restore: ongoing request status.
45    pub restore_ongoing: Option<bool>,
46    /// Glacier restore: expiry date string.
47    pub restore_expiry: Option<String>,
48    /// Checksum algorithm (CRC32, SHA1, SHA256).
49    pub checksum_algorithm: Option<String>,
50    /// Base64-encoded checksum value.
51    pub checksum_value: Option<String>,
52    /// Object lock mode (GOVERNANCE or COMPLIANCE).
53    pub lock_mode: Option<String>,
54    /// Object lock retain-until date (ISO 8601).
55    pub lock_retain_until: Option<DateTime<Utc>>,
56    /// Legal hold status (ON or OFF).
57    pub lock_legal_hold: Option<String>,
58}
59
60/// A part uploaded via the multipart upload API.
61#[derive(Debug, Clone)]
62pub struct UploadPart {
63    pub part_number: u32,
64    pub data: Bytes,
65    pub etag: String,
66    pub size: u64,
67    pub last_modified: DateTime<Utc>,
68}
69
70/// An in-progress multipart upload.
71#[derive(Debug, Clone)]
72pub struct MultipartUpload {
73    pub upload_id: String,
74    pub key: String,
75    pub initiated: DateTime<Utc>,
76    /// Parts keyed by part number.
77    pub parts: BTreeMap<u32, UploadPart>,
78    /// Metadata provided at CreateMultipartUpload time.
79    pub metadata: HashMap<String, String>,
80    pub content_type: String,
81    pub storage_class: String,
82    pub sse_algorithm: Option<String>,
83    pub sse_kms_key_id: Option<String>,
84    pub tagging: Option<String>,
85    pub acl_grants: Vec<AclGrant>,
86    pub checksum_algorithm: Option<String>,
87}
88
89#[derive(Debug, Clone)]
90pub struct S3Bucket {
91    pub name: String,
92    pub creation_date: DateTime<Utc>,
93    pub region: String,
94    /// Objects keyed by their full key path.
95    pub objects: BTreeMap<String, S3Object>,
96    pub tags: HashMap<String, String>,
97    pub acl_grants: Vec<AclGrant>,
98    pub acl_owner_id: String,
99    /// In-progress multipart uploads keyed by upload ID.
100    pub multipart_uploads: HashMap<String, MultipartUpload>,
101    /// Versioning status: None = never enabled, Some("Enabled"), Some("Suspended").
102    pub versioning: Option<String>,
103    /// Object versions keyed by key, each value is a list of versions.
104    pub object_versions: HashMap<String, Vec<S3Object>>,
105    /// Bucket ACL (canned or XML).
106    pub acl: Option<String>,
107    pub encryption_config: Option<String>,
108    pub lifecycle_config: Option<String>,
109    pub policy: Option<String>,
110    pub cors_config: Option<String>,
111    pub notification_config: Option<String>,
112    pub logging_config: Option<String>,
113    pub website_config: Option<String>,
114    pub accelerate_status: Option<String>,
115    pub public_access_block: Option<String>,
116    pub object_lock_config: Option<String>,
117    pub replication_config: Option<String>,
118    pub ownership_controls: Option<String>,
119    pub inventory_configs: HashMap<String, String>,
120}
121
122impl S3Bucket {
123    pub fn new(name: &str, region: &str, owner_id: &str) -> Self {
124        Self {
125            name: name.to_string(),
126            creation_date: Utc::now(),
127            region: region.to_string(),
128            objects: BTreeMap::new(),
129            tags: HashMap::new(),
130            acl_grants: vec![AclGrant {
131                grantee_type: "CanonicalUser".to_string(),
132                grantee_id: Some(owner_id.to_string()),
133                grantee_display_name: Some(owner_id.to_string()),
134                grantee_uri: None,
135                permission: "FULL_CONTROL".to_string(),
136            }],
137            acl_owner_id: owner_id.to_string(),
138            multipart_uploads: HashMap::new(),
139            versioning: None,
140            object_versions: HashMap::new(),
141            acl: None,
142            encryption_config: None,
143            lifecycle_config: None,
144            policy: None,
145            cors_config: None,
146            notification_config: None,
147            logging_config: None,
148            website_config: None,
149            accelerate_status: None,
150            public_access_block: None,
151            object_lock_config: None,
152            replication_config: None,
153            ownership_controls: None,
154            inventory_configs: HashMap::new(),
155        }
156    }
157}
158
159pub struct S3State {
160    pub account_id: String,
161    pub region: String,
162    pub buckets: HashMap<String, S3Bucket>,
163}
164
165impl S3State {
166    pub fn new(account_id: &str, region: &str) -> Self {
167        Self {
168            account_id: account_id.to_string(),
169            region: region.to_string(),
170            buckets: HashMap::new(),
171        }
172    }
173
174    pub fn reset(&mut self) {
175        self.buckets.clear();
176    }
177}
178
179pub type SharedS3State = Arc<RwLock<S3State>>;
fakecloud_s3/state.rs

fakecloud_s3/
state.rs
