wip

Author: jakoschiko

imap-codec/examples/client.rs

@@ -1,4 +1,4 @@
-use imap_codec::{fragmentizer::Fragmentizer, GreetingCodec};
+use imap_codec::{fragmentizer::Fragmentizer, GreetingCodec, ResponseCodec};
 
 #[path = "common/common.rs"]
 mod common;
@@ -32,10 +32,7 @@ fn main() {
 
     loop {
         // Progress next message.
-        fragmentizer.progress();
-
-        // Does the Fragmentizer need more bytes?
-        if fragmentizer.is_message_complete() {
+        let Some(_fragment_info) = fragmentizer.progress() else {
             // Read more bytes ...
             read_more(&mut buffer, Role::Server);
 
@@ -45,6 +42,10 @@ fn main() {
 
             // ... and try again.
             continue;
+        };
+
+        if !fragmentizer.is_message_complete() {
+            continue;
         }
 
         // The Fragmentizer detected a complete message.
@@ -64,7 +65,7 @@ fn main() {
                 };
             }
             State::Response => {
-                match fragmentizer.decode_message(&GreetingCodec::default()) {
+                match fragmentizer.decode_message(&ResponseCodec::default()) {
                     Ok(response) => {
                         // Do something with the response.
                         println!("{:#?}", response);

imap-codec/examples/server.rs

@@ -1,13 +1,13 @@
 use imap_codec::{
-    decode::{CommandDecodeError, Decoder},
-    fragmentizer::{FragmentInfo, Fragmentizer},
+    fragmentizer::{FragmentInfo, Fragmentizer, LiteralAnnouncement},
     CommandCodec,
 };
 
 #[path = "common/common.rs"]
 mod common;
 
 use common::{read_more, COLOR_SERVER, RESET};
+use imap_types::core::LiteralMode;
 
 use crate::common::Role;
 
@@ -35,14 +35,27 @@ fn main() {
 
     loop {
         // Progress next message.
-        let fragment_info = fragmentizer.progress();
+        let Some(fragment_info) = fragmentizer.progress() else {
+            // Read more bytes ...
+            read_more(&mut buffer, Role::Client);
+
+            // ... and pass the bytes to the Fragmentizer ...
+            fragmentizer.enqueue_bytes(&buffer);
+            buffer.clear();
+
+            // ... and try again.
+            continue;
+        };
 
         // The Fragmentizer detected a line that announces a literal.
         // A command continuation request is expected.
-        if let Some(FragmentInfo::Line {
-            announcement: Some(_),
+        if let FragmentInfo::Line {
+            announcement: Some(LiteralAnnouncement {
+                mode: LiteralMode::Sync,
+                ..
+            }),
             ..
-        }) = fragment_info
+        } = fragment_info
         {
             // Simulate a command continuation request ...
             println!("S: {COLOR_SERVER}+ {RESET}");
@@ -51,16 +64,7 @@ fn main() {
             continue;
         }
 
-        // Does the Fragmentizer need more bytes?
-        if fragmentizer.is_message_complete() {
-            // Read more bytes ...
-            read_more(&mut buffer, Role::Server);
-
-            // ... and pass the bytes to the Fragmentizer ...
-            fragmentizer.enqueue_bytes(&buffer);
-            buffer.clear();
-
-            // ... and try again.
+        if !fragmentizer.is_message_complete() {
             continue;
         }