APIService

import Alamofire

protocol APIServiceProtocol {
    func fetchLaunchesWithQuery(startDate: Date, endDate: Date, completion: @escaping (LaunchResponse?, Error?, AFDataResponse<Any>) -> Void) -> DataRequestProtocol
    func fetchRocket(rocketName: String, completion: @escaping (Rocket?, Error?, AFDataResponse<Any>) -> Void) -> DataRequestProtocol
}

extension APIServiceProtocol {
    @discardableResult func fetchLaunchesWithQuery(startDate: Date = Calendar.current.date(byAdding: .year, value: -3, to: Date()) ?? Date(), endDate: Date = Date(), completion: @escaping (LaunchResponse?, Error?, AFDataResponse<Any>) -> Void) -> DataRequestProtocol {
        return fetchLaunchesWithQuery(startDate: startDate, endDate: endDate, completion: completion)
    }
    
    @discardableResult func fetchRocket(rocketName: String, completion: @escaping (Rocket?, Error?, AFDataResponse<Any>) -> Void) -> DataRequestProtocol {
        return fetchRocket(rocketName: rocketName, completion: completion)
    }
}

class APIService: APIServiceProtocol {
    let baseUrlString: String
    var sessionManager: SessionProtocol
    
    init(baseUrlString: String = "https://api.spacexdata.com/v4", sessionManager: SessionProtocol = AF) {
        self.baseUrlString = baseUrlString
        self.sessionManager = sessionManager
    }
    
    @discardableResult func fetchLaunchesWithQuery(startDate: Date, endDate: Date, completion: @escaping (LaunchResponse?, Error?, AFDataResponse<Any>) -> Void) -> DataRequestProtocol {
        let parameters = [
            "query": [
                "date_utc": [
                    "$gte": DateFormatter.iso8601Full.string(from: startDate),
                    "$lte": DateFormatter.iso8601Full.string(from: endDate)
                ]
            ],
            "options": [
                "sort": [
                    "utc_date": "asc"
                ],
                "pagination": false
            ]
        ]
        
        let postLaunchesQueryEndpoint = "\(baseUrlString)/launches/query"
        let method: HTTPMethod = .post
        let headers: HTTPHeaders? = nil
        
        return sessionManager.request(postLaunchesQueryEndpoint, method: method, parameters: parameters, encoding: method == .get ? URLEncoding.default : JSONEncoding.default, headers: headers)
            .responseJSON { dataResponse in
                guard let data = dataResponse.data else { return }
                
                let decoder = JSONDecoder()
                do {
                    let launches = try decoder.decode(LaunchResponse.self, from: data)
                    completion(launches, nil, dataResponse)
                } catch {
                    completion(nil, error, dataResponse)
                }
            }
    }
    
    @discardableResult func fetchRocket(rocketName: String, completion: @escaping (Rocket?, Error?, AFDataResponse<Any>) -> Void) -> DataRequestProtocol {
        
        let postLaunchesQueryEndpoint = "\(baseUrlString)/rockets/\(rocketName)"
        
        return sessionManager.request(postLaunchesQueryEndpoint, method: .get, parameters: nil, encoding: URLEncoding.default, headers: nil)
            .responseJSON { dataResponse in
                guard let data = dataResponse.data else { return }
                
                let decoder = JSONDecoder()
                do {
                    let rocket = try decoder.decode(Rocket.self, from: data)
                    completion(rocket, nil, dataResponse)
                } catch {
                    completion(nil, error, dataResponse)
                }
            }
    }
}