//
//  PacketTunnelTransport.swift
//  MullvadVPN
//
//  Created by Sajad Vishkai on 2022-10-03.
//  Copyright © 2025 Mullvad VPN AB. All rights reserved.
//

import Foundation
import MullvadREST
import MullvadRustRuntime
import MullvadTypes
import Operations
import PacketTunnelCore

struct PacketTunnelTransport: RESTTransport {
    var name: String {
        "packet-tunnel"
    }

    let tunnel: any TunnelProtocol

    init(tunnel: any TunnelProtocol) {
        self.tunnel = tunnel
    }

    func sendRequest(
        _ request: URLRequest,
        completion: @escaping @Sendable (Data?, URLResponse?, Error?) -> Void
    ) -> Cancellable {
        let proxyRequest = ProxyURLRequest(
            id: UUID(),
            urlRequest: request
        )

        // If the URL provided to the proxy request was invalid, indicate failure via `.badURL` and return a no-op.
        guard let proxyRequest else {
            completion(nil, nil, URLError(.badURL))
            return AnyCancellable {}
        }

        return tunnel.sendRequest(proxyRequest) { result in
            switch result {
            case let .success(reply):
                completion(
                    reply.data,
                    reply.response?.originalResponse,
                    reply.error?.originalError
                )

            case let .failure(error):
                let returnError = error.isOperationCancellationError ? URLError(.cancelled) : error

                completion(nil, nil, returnError)
            }
        }
    }
}

final class PacketTunnelAPITransport: APITransportProtocol {
    var name: String {
        "packet-tunnel-transport"
    }

    let tunnel: any TunnelProtocol

    init(tunnel: any TunnelProtocol) {
        self.tunnel = tunnel
    }

    func sendRequest(
        _ request: APIRequest,
        completion: @escaping @Sendable (ProxyAPIResponse) -> Void
    ) -> Cancellable {
        let proxyRequest = ProxyAPIRequest(
            id: UUID(),
            request: request
        )

        return tunnel.sendAPIRequest(proxyRequest) { result in
            switch result {
            case let .success(reply):
                completion(reply)

            case let .failure(error):
                let error = error.isOperationCancellationError ? URLError(.cancelled) : error
                completion(
                    ProxyAPIResponse(
                        data: nil,
                        error: APIError(
                            statusCode: 0,
                            errorDescription: error.localizedDescription,
                            serverResponseCode: nil
                        )))
            }
        }
    }
}