refactor: Refactor examples

Author: jakoschiko

imap-codec/examples/client.rs

@@ -0,0 +1,81 @@
+use imap_codec::{fragmentizer::Fragmentizer, GreetingCodec, ResponseCodec};
+
+#[path = "common/common.rs"]
+mod common;
+
+use common::read_more;
+
+use crate::common::Role;
+
+enum State {
+    Greeting,
+    Response,
+}
+
+const WELCOME: &str = r#"# Parsing of IMAP greeting and responses
+
+"S:" denotes the server.
+".." denotes the continuation of an (incomplete) response, e.g., due to the use of an IMAP literal.
+
+Note: "\n" will be automatically replaced by "\r\n".
+
+--------------------------------------------------------------------------------------------------
+
+Enter intial IMAP greeting followed by IMAP responses (or "exit").
+"#;
+
+fn main() {
+    println!("{}", WELCOME);
+
+    let mut fragmentizer = Fragmentizer::new(10 * 1024);
+    let mut state = State::Greeting;
+
+    loop {
+        // Progress next fragment.
+        let Some(_fragment_info) = fragmentizer.progress() else {
+            // Read more bytes ...
+            let bytes = read_more(Role::Server, fragmentizer.message_bytes().is_empty());
+
+            // ... and pass the bytes to the Fragmentizer ...
+            fragmentizer.enqueue_bytes(&bytes);
+
+            // ... and try again.
+            continue;
+        };
+
+        // Check whether the Fragmentizer detected a complete message.
+        if !fragmentizer.is_message_complete() {
+            // Read next fragment.
+            continue;
+        }
+
+        // The Fragmentizer detected a complete message.
+        match state {
+            State::Greeting => {
+                match fragmentizer.decode_message(&GreetingCodec::default()) {
+                    Ok(greeting) => {
+                        // Do something with the greeting ...
+                        println!("{:#?}", greeting);
+
+                        // ... and proceed with reponses.
+                        state = State::Response;
+                    }
+                    Err(err) => {
+                        println!("Error parsing greeting: {err:?}");
+                    }
+                };
+            }
+            State::Response => {
+                match fragmentizer.decode_message(&ResponseCodec::default()) {
+                    Ok(response) => {
+                        // Do something with the response.
+                        println!("{:#?}", response);
+                    }
+                    Err(err) => {
+                        println!("Error parsing response: {err:?}");
+                    }
+                };
+            }
+        };
+    }
+}

imap-codec/examples/common/common.rs

@@ -12,8 +12,8 @@ pub enum Role {
     Server,
 }
 
-pub fn read_more(buffer: &mut Vec<u8>, role: Role) {
-    let prompt = if buffer.is_empty() {
+pub fn read_more(role: Role, message_begin: bool) -> Vec<u8> {
+    let prompt = if message_begin {
         match role {
             Role::Client => "C: ",
             Role::Server => "S: ",
@@ -30,7 +30,7 @@ pub fn read_more(buffer: &mut Vec<u8>, role: Role) {
         std::process::exit(0);
     }
 
-    buffer.extend_from_slice(line.as_bytes());
+    line.into_bytes()
 }
 
 fn read_line(prompt: &str, role: Role) -> String {