rawprinter 0.1.3

An IOT server written in rust for connecting via USB to raw printers.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63

mod extra;

use std::error::Error;
use escpos_rs::{Printer, PrinterProfile};
use clap::{Parser};
use clap_num::{maybe_hex};
use tiny_http::{Server, Response};
use extra::escape;

#[derive(Parser)]
#[clap(author, version, about)]
struct Cli {
    ///HTTP port to listen to.
    #[clap(short, long, default_value = "8001")]
    port: u16,

    ///Device vendor ID in lsusb command
    #[clap(short, long, required=true, value_parser=maybe_hex::<u16>)]
    vendor_id: u16,

    ///Device product ID in lsusb command
    #[clap(short, long, required=true, value_parser=maybe_hex::<u16>)]
    device_id: u16,
}

fn main() -> Result<(), Box<dyn Error>> {
    let cli = Cli::parse();

    let printer = Printer::new(PrinterProfile::usb_builder(
        cli.vendor_id,
        cli.device_id
    ).build())?.ok_or("no device found!")?;

    let host = format!("localhost:{}", cli.port);

    let server = match Server::http(&host) {
        Ok(server) => server,
        Err(msg) => {
            return Err(msg.to_string().into());
        }
    };

    println!("Raw Printer server running at: http://{}", host);
    for mut request in server.incoming_requests() {
        let method = request.method().to_string();
        let url = request.url().to_string();
        let mut body = String::new();

        request.as_reader().read_to_string(&mut body)?;
        request.respond(
            if &method == "POST" && &url == "/" {
                match printer.print(&escape(body)) {
                    Ok(()) => Response::from_string("").with_status_code(200),
                    Err(err) => Response::from_string(&err.to_string())
                        .with_status_code(500)
                }
            } else {
                Response::from_string("").with_status_code(404)
            }
        )?;
    }
    Ok(())
}