use tonic::{Request, Response, Status};

use crate::escpos_printer::EscposPrinter;
use crate::proto::print_request::Content;
use crate::proto::receipt_printer_server::ReceiptPrinter;
use crate::proto::{PrintRequest, PrintResponse, TypstContent};
use crate::typst_renderer::render_typst_receipt;

#[derive(Debug)]
pub struct ReceiptPrinterService {
    template_path: String,
    printer_host: String,
    printer_port: u16,
}

impl ReceiptPrinterService {
    pub fn new(template_path: String, printer_host: String, printer_port: u16) -> Self {
        Self {
            template_path,
            printer_host,
            printer_port,
        }
    }
}

#[tonic::async_trait]
impl ReceiptPrinter for ReceiptPrinterService {
    async fn print(
        &self,
        request: Request<PrintRequest>,
    ) -> Result<Response<PrintResponse>, Status> {
        let Some(Content::Typst(TypstContent {
            content: Some(content),
        })) = &request.get_ref().content
        else {
            println!("No Typst content received");
            return Ok(Response::new(PrintResponse::default()));
        };

        let image = match render_typst_receipt(self.template_path.as_ref(), content) {
            Ok(image) => image,
            Err(error) => {
                println!("Error rendering receipt: {error}");
                return Ok(Response::new(PrintResponse::default()));
            }
        };

        println!("Connecting to printer...");
        let mut printer = match EscposPrinter::open(self.printer_host.as_ref(), self.printer_port) {
            Ok(printer) => printer,
            Err(error) => {
                println!("Error connecting to printer: {error}");
                return Ok(Response::new(PrintResponse::default()));
            }
        };
        println!("Printing...");
        if let Err(error) = printer.print_image(&image) {
            println!("Error when printing: {error}");
            return Ok(Response::new(PrintResponse::default()));
        }
        println!("Done!");

        Ok(Response::new(PrintResponse::default()))
    }
}