import multiprocessing
import os

from robyn import Response, Robyn, jsonify
from robyn.argument_parser import Config


class SpecialConfig(Config):
    def __init__(self):
        super().__init__()
        self.workers = 2
        self.processes = (os.cpu_count() * 2) + 1
        self.log_level = "WARN"


app = Robyn(__file__, config=SpecialConfig())


@app.get("/plaintext", const=True)
def plaintext() -> str:
    return "Hello, world!"


@app.get("/json", const=True)
def json() -> str:
    return Response(
        status_code=200,
        description=jsonify({"message": "Hello, world!"}),
        headers={"Content-Type": "application/json"}
        )



if __name__ == "__main__":
    app.add_response_header("Server", "Robyn")

    app.start(host="0.0.0.0", port=8080)