monocoque-rs 0.1.1

//! REP Server for Interop Testing
//!
//! Simple echo server that replies to all requests with "Echo: <message>"

use bytes::Bytes;
use monocoque_zmtp::rep::RepSocket;
use std::env;

#[compio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    // Parse port from args (default 5555)
    let args: Vec<String> = env::args().collect();
    let port = if args.len() > 2 && args[1] == "--port" {
        args[2].parse::<u16>()?
    } else {
        5555
    };

    // Bind REP socket
    let addr = format!("127.0.0.1:{}", port);
    let listener = compio::net::TcpListener::bind(&addr).await?;
    println!("REP server listening on {}", addr);

    // Accept connection
    let (stream, peer) = listener.accept().await?;
    println!("Accepted connection from {}", peer);

    let mut socket = RepSocket::new(stream).await?;

    // Echo loop
    loop {
        match socket.recv().await? {
            Some(msg) => {
                println!("Received {} frames", msg.len());

                // Echo back with "Echo: " prefix on first frame
                let mut reply = Vec::with_capacity(msg.len());

                if !msg.is_empty() {
                    let first_frame = format!("Echo: {}", String::from_utf8_lossy(&msg[0]));
                    reply.push(Bytes::from(first_frame));

                    // Copy remaining frames as-is
                    for frame in &msg[1..] {
                        reply.push(frame.clone());
                    }
                } else {
                    reply.push(Bytes::from("Echo: <empty>"));
                }

                socket.send(reply).await?;
            }
            None => {
                println!("Connection closed");
                break;
            }
        }
    }

    Ok(())
}