import URLRouting
import Vapor
import VaporRouting

// Taken from github.com/nevillco/vapor-routing

public extension Application {
  /// Mounts a router to the Vapor application.
  ///
  /// See ``VaporRouting`` for more information on usage.
  ///
  /// - Parameters:
  ///   - router: A parser-printer that works on inputs of `URLRequestData`.
  ///   - middleware: A closure for providing any per-route migrations to be run before processing the request.
  ///   - closure: A closure that takes a `Request` and the router's output as arguments.
  func mount<R: Parser>(
    _ router: R,
    middleware: @escaping @Sendable (R.Output) -> [any Middleware]? = { _ in nil },
    use closure: @escaping @Sendable (Request, R.Output) async throws -> any AsyncResponseEncodable
  ) where R.Input == URLRequestData, R: Sendable, R.Output: Sendable {
    self.middleware.use(AsyncRoutingMiddleware(router: router, middleware: middleware, respond: closure))
  }
}

/// Serves requests using a router and response handler.
///
/// You will not typically need to interact with this type directly. Instead you should use the
/// `mount` method on your Vapor application.
///
/// See ``VaporRouting`` for more information on usage.
public struct AsyncRoutingMiddleware<Router: Parser>: AsyncMiddleware
  where Router.Input == URLRequestData,
  Router: Sendable,
  Router.Output: Sendable
{
  let router: Router
  let middleware: @Sendable (Router.Output) -> [any Middleware]?
  let respond: @Sendable (Request, Router.Output) async throws -> any AsyncResponseEncodable

  public func respond(
    to request: Request,
    chainingTo next: any AsyncResponder
  ) async throws -> Response {
    if request.body.data == nil {
      try await _ = request.body.collect(max: request.application.routes.defaultMaxBodySize.value)
        .get()
    }

    guard let requestData = URLRequestData(request: request)
    else { return try await next.respond(to: request) }

    let route: Router.Output
    do {
      route = try router.parse(requestData)
    } catch let routingError {
      do {
        return try await next.respond(to: request)
      } catch {
        request.logger.info("\(routingError)")

        guard request.application.environment == .development
        else { throw error }

        return Response(status: .notFound, body: .init(string: "Routing \(routingError)"))
      }
    }

    if let middleware = middleware(route) {
      return try await middleware.makeResponder(chainingTo: AsyncBasicResponder { request in
        try await self.respond(request, route).encodeResponse(for: request)
      }).respond(to: request).get()

      // return try await middleware.respond(
      //   to: request,
      //   chainingTo: AsyncBasicResponder { request in
      //     try await self.respond(request, route).encodeResponse(for: request)
      //   }
      // ).get()
    } else {
      return try await respond(request, route).encodeResponse(for: request)
    }
  }
}

// Usage:
// app.mount(
//   router,
//   middleware: { route in
//     case .onboarding: return nil
//     case .signIn: return BasicAuthMiddleware()
//     default: return BearerAuthMiddleware()
//   },
//   use: { request, route in
//     // route handline
//   }
// )