import json
from fastapi import FastAPI, Response
from pydantic import BaseModel
from .formats import print_bill, print_customer_order

app = FastAPI(
    title="Print Server FastAPI",
    description="Server that receive request for printing",
    version="0.0.1"
)


class Info(BaseModel):
    content: str
    ip_printer: str
    user_name: str


@app.post("/print_bill")
def print_ticket_bill(info: Info):
    info = dict(info)
    data = info["content"]
    address = info["ip_printer"]
    waiter = info["user_name"]
    data = json.loads(data.replace("'", "\""))
    print_bill(data, address, waiter)

    message = "!Impresion Realizada!"

    return Response(content=message, status_code=200)


@app.post("/order_kitchen")
def print_ticket_file_kitchen(info: Info):
    info = dict(info)
    data = info["content"]
    address = info["ip_printer"]
    waiter = info["user_name"]
    data = json.loads(data.replace("'", "\""))
    print_customer_order(data, address, waiter)

    message = "!Impresion Realizada!"

    return Response(content=message, status_code=200)


@app.post("/order_bar")
def print_ticket_file_bar(info: Info):
    info = dict(info)
    data = info["content"]
    address = info["ip_printer"]
    waiter = info["user_name"]
    data = json.loads(data.replace("'", "\""))
    print_customer_order(data, address, waiter)

    message = "!Impresion Realizada!"

    return Response(content=message, status_code=200)