session-ios/SessionMessagingKitTests/_TestUtilities/TestOnionRequestAPI.swift

// Copyright © 2022 Rangeproof Pty Ltd. All rights reserved.

import Foundation
import Combine
import SessionSnodeKit
import SessionUtilitiesKit

@testable import SessionMessagingKit

// FIXME: Change 'OnionRequestAPIType' to have instance methods instead of static methods once everything is updated to use 'Dependencies'
class TestOnionRequestAPI: OnionRequestAPIType {
    struct RequestData: Codable {
        let urlString: String?
        let httpMethod: String
        let headers: [String: String]
        let body: Data?
        let destination: OnionRequestAPIDestination
        
        var publicKey: String? {
            switch destination {
                case .snode: return nil
                case .server(_, _, let x25519PublicKey, _, _): return x25519PublicKey
            }
        }
    }

    class ResponseInfo: ResponseInfoType {
        let requestData: RequestData
        let code: Int
        let headers: [String: String]
        
        init(requestData: RequestData, code: Int, headers: [String: String]) {
            self.requestData = requestData
            self.code = code
            self.headers = headers
        }
    }
    
    class var mockResponse: Data? { return nil }
    
    static func sendOnionRequest(_ request: URLRequest, to server: String, with x25519PublicKey: String) -> AnyPublisher<(ResponseInfoType, Data?), Error> {
        let responseInfo: ResponseInfo = ResponseInfo(
            requestData: RequestData(
                urlString: request.url?.absoluteString,
                httpMethod: (request.httpMethod ?? "GET"),
                headers: (request.allHTTPHeaderFields ?? [:]),
                body: request.httpBody,
                server: server,
                version: .v4,
                publicKey: x25519PublicKey
            ),
            code: 200,
            headers: [:]
        )
        
        return Just((responseInfo, mockResponse))
            .setFailureType(to: Error.self)
            .eraseToAnyPublisher()
    }
    
    static func sendOnionRequest(_ payload: Data, to snode: Snode) -> AnyPublisher<(ResponseInfoType, Data?), Error> {
        let responseInfo: ResponseInfo = ResponseInfo(
            requestData: RequestData(
                urlString: nil,
                httpMethod: "POST",
                headers: [:],
                snodeMethod: nil,
                body: payload,
                
                server: "",
                version: .v3,
                publicKey: snode.x25519PublicKey
            ),
            code: 200,
            headers: [:]
        )
        
        return Just((responseInfo, mockResponse))
            .setFailureType(to: Error.self)
            .eraseToAnyPublisher()
    }
}