from justapi import JSONResponse
from justapi import JustAPIApp

app = JustAPIApp()

@app.get("/set-cookie")
def set_cookie():
    response = JSONResponse(content={"message": "cookie set"})
    response.set_cookie("session_id", "abc123", httponly=True, secure=True)
    return response

@app.get("/clear-cookie")
def clear_cookie():
    response = JSONResponse(content={"message": "cookie cleared"})
    response.delete_cookie("session_id")
    return response

@app.get("/with-headers")
def with_headers():
    response = JSONResponse(content={"data": "value"})
    response.headers["X-Request-Id"] = "req-123"
    response.headers["X-Response-Time"] = "42ms"
    return response