1 files changed, 112 insertions, 0 deletions
diff --git a/src/mailbox.rs b/src/mailbox.rs
new file mode 100644
index 0000000..7dcb92a
--- /dev/null
+++ b/src/mailbox.rs
@@ -0,0 +1,112 @@
+use anyhow::{anyhow, bail, Result};
+use k2v_client::{BatchDeleteOp, BatchReadOp, Filter, K2vClient, K2vValue};
+use rusoto_core::HttpClient;
+use rusoto_credential::{AwsCredentials, StaticProvider, ProvideAwsCredentials};
+use rusoto_s3::{
+    DeleteObjectRequest, GetObjectRequest, ListObjectsV2Request, PutObjectRequest, S3Client, S3,
+};
+use rusoto_signature::Region;
+use rand::prelude::*;
+
+use crate::cryptoblob::Key;
+use crate::bayou::Bayou;
+use crate::login::Credentials;
+use crate::uidindex::*;
+
+pub struct Mailbox {
+    bucket: String,
+    name: String,
+    key: Key,
+
+    k2v: K2vClient,
+    s3: S3Client,
+
+    uid_index: Bayou<UidIndex>,
+}
+
+impl Mailbox {
+    pub async fn new(k2v_region: Region, s3_region: Region, creds: Credentials,name: String) -> Result<Self> {
+        let aws_creds_provider = StaticProvider::new_minimal(
+            creds.aws_access_key_id,
+            creds.aws_secret_access_key,
+        );
+        let aws_creds = aws_creds_provider.credentials().await?;
+
+        let uid_index = Bayou::<UidIndex>::new(
+            aws_creds.clone(),
+            k2v_region.clone(),
+            s3_region.clone(),
+            creds.bucket.clone(),
+            name.clone(),
+            creds.master_key.clone(),
+        )?;
+        
+        let k2v_client = K2vClient::new(k2v_region, creds.bucket.clone(), aws_creds, None)?;
+        let s3_client = S3Client::new_with(HttpClient::new()?, aws_creds_provider, s3_region);
+
+        Ok(Self {
+            bucket: creds.bucket,
+            name,
+            key: creds.master_key,
+            k2v: k2v_client,
+            s3: s3_client,
+            uid_index,
+        })
+    }
+
+
+    pub async fn test(&mut self) -> Result<()> {
+
+        self.uid_index.sync().await?;
+
+        dump(&self.uid_index);
+
+        let mut rand_id = [0u8; 24];
+        rand_id[..16].copy_from_slice(&u128::to_be_bytes(thread_rng().gen()));
+        let add_mail_op = self.uid_index
+            .state()
+            .op_mail_add(MailUuid(rand_id), vec!["\\Unseen".into()]);
+        self.uid_index.push(add_mail_op).await?;
+
+        dump(&self.uid_index);
+
+        if self.uid_index.state().mails_by_uid.len() > 6 {
+            for i in 0..2 {
+                let (_, uuid) = self.uid_index
+                    .state()
+                    .mails_by_uid
+                    .iter()
+                    .skip(3 + i)
+                    .next()
+                    .unwrap();
+                let del_mail_op = self.uid_index.state().op_mail_del(*uuid);
+                self.uid_index.push(del_mail_op).await?;
+
+                dump(&self.uid_index);
+            }
+        }
+
+        Ok(())
+    }
+}
+
+fn dump(uid_index: &Bayou<UidIndex>) {
+    let s = uid_index.state();
+    println!("---- MAILBOX STATE ----");
+    println!("UIDVALIDITY {}", s.uidvalidity);
+    println!("UIDNEXT {}", s.uidnext);
+    println!("INTERNALSEQ {}", s.internalseq);
+    for (uid, uuid) in s.mails_by_uid.iter() {
+        println!(
+            "{} {} {}",
+            uid,
+            hex::encode(uuid.0),
+            s.mail_flags
+                .get(uuid)
+                .cloned()
+                .unwrap_or_default()
+                .join(", ")
+        );
+    }
+    println!("");
+}