await server.use(PrismTracingMiddleware())

await server.use(PrismTracingMiddleware(
    headerName: "X-Request-ID",
    correlationHeader: "X-Correlation-ID",
    generateIfMissing: true
))

await server.get("/orders/:id") { request in
    if let trace = request.traceContext {
        print("Request ID: \(trace.requestID)")
        print("Correlation: \(trace.correlationID ?? "none")")
        print("Elapsed: \(trace.elapsed)")
    }

    let order = try await fetchOrder(request.parameters["id"]!)
    return .json(order)
}

await server.post("/payments") { request in
    guard let trace = request.traceContext else {
        return .json(["error": "Missing trace"], status: .internalServerError)
    }

    let logger = PrismTracingLogger(context: trace)

    print(logger.log("Processing payment"))
    // [a1b2c3d4-...] Processing payment

    print(logger.log("Calling payment gateway"))
    // [a1b2c3d4-...] [corr-id-xyz] Calling payment gateway

    return .json(["status": "processed"])
}

await server.post("/orders") { request in
    let trace = request.traceContext
    let client = PrismHTTPClient()

    // Forward trace headers to downstream service
    var headers: [String: String] = [:]
    if let rid = trace?.requestID {
        headers["X-Request-ID"] = UUID().uuidString
        headers["X-Correlation-ID"] = trace?.correlationID ?? rid
        headers["X-Parent-ID"] = rid
    }

    let paymentResult = try await client.post(
        "http://payment-service/charge",
        body: chargeData,
        headers: headers
    )

    return .json(["status": "created"], status: .created)
}

let logger = PrismTracingLogger(requestID: "manual-123", correlationID: "batch-456")
print(logger.log("Starting batch job"))
// [manual-123] [batch-456] Starting batch job