from fastapi import FastAPI, Response
from pydantic import BaseModel

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

class Info(BaseModel):
    content : str

def write_to_file(file_path, content):
    with open(file_path, "w") as file:
        file.write(content)


@app.post("/print_ticket")
def print_ticket_file(info : Info):
    write_to_file('/tmp/ticket_prueba.txt', info.content)
    message = "!Impresión Realizada!"

    return Response(content=message, status_code=200)

@app.post("/order_kitchen")
def print_ticket_file_kitchen(info : Info):
    write_to_file('/tmp/ticket_kitchen.txt', info.content)
    message = "!Impresión Realizada!"

    return Response(content=message, status_code=200)

@app.post("/order_bar")
def print_ticket_file_bar(info : Info):
    write_to_file('/tmp/ticket_bar.txt', info.content)
    message = "!Impresión Realizada!"

    return Response(content=message, status_code=200)