-
Notifications
You must be signed in to change notification settings - Fork 146
/
RequestBehaviour.swift
193 lines (156 loc) · 6.14 KB
/
RequestBehaviour.swift
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
//
// Generated by SwagGen
// https://github.com/yonaskolb/SwagGen
//
import Foundation
public protocol RequestBehaviour {
/// runs first and allows the requests to be modified. If modifying asynchronously use validate
func modifyRequest(request: AnyRequest, urlRequest: URLRequest) -> URLRequest
/// validates and modifies the request. complete must be called with either .success or .fail
func validate(request: AnyRequest, urlRequest: URLRequest, complete: @escaping (RequestValidationResult) -> Void)
/// called before request is sent
func beforeSend(request: AnyRequest)
/// called when request successfuly returns a 200 range response
func onSuccess(request: AnyRequest, result: Any)
/// called when request fails with an error. This will not be called if the request returns a known response even if the a status code is out of the 200 range
func onFailure(request: AnyRequest, error: APIClientError)
/// called if the request recieves a network response. This is not called if request fails validation or encoding
func onResponse(request: AnyRequest, response: AnyResponse)
}
public enum RequestValidationResult {
case success(URLRequest)
case failure(String)
}
// Provides empty defaults so that each function becomes optional
public extension RequestBehaviour {
func modifyRequest(request: AnyRequest, urlRequest: URLRequest) -> URLRequest { return urlRequest }
func validate(request: AnyRequest, urlRequest: URLRequest, complete: @escaping (RequestValidationResult) -> Void) {
complete(.success(urlRequest))
}
func beforeSend(request: AnyRequest) {}
func onSuccess(request: AnyRequest, result: Any) {}
func onFailure(request: AnyRequest, error: APIClientError) {}
func onResponse(request: AnyRequest, response: AnyResponse) {}
}
// Group different RequestBehaviours together
struct RequestBehaviourGroup {
let request: AnyRequest
let behaviours: [RequestBehaviour]
init<T>(request: APIRequest<T>, behaviours: [RequestBehaviour]) {
self.request = request.asAny()
self.behaviours = behaviours
}
func beforeSend() {
behaviours.forEach {
$0.beforeSend(request: request)
}
}
func validate(_ urlRequest: URLRequest, complete: @escaping (RequestValidationResult) -> Void) {
if behaviours.isEmpty {
complete(.success(urlRequest))
return
}
var count = 0
var modifiedRequest = urlRequest
func validateNext() {
let behaviour = behaviours[count]
behaviour.validate(request: request, urlRequest: modifiedRequest) { result in
count += 1
switch result {
case .success(let urlRequest):
modifiedRequest = urlRequest
if count == self.behaviours.count {
complete(.success(modifiedRequest))
} else {
validateNext()
}
case .failure(let error):
complete(.failure(error))
}
}
}
validateNext()
}
func onSuccess(result: Any) {
behaviours.forEach {
$0.onSuccess(request: request, result: result)
}
}
func onFailure(error: APIClientError) {
behaviours.forEach {
$0.onFailure(request: request, error: error)
}
}
func onResponse(response: AnyResponse) {
behaviours.forEach {
$0.onResponse(request: request, response: response)
}
}
func modifyRequest(_ urlRequest: URLRequest) -> URLRequest {
var urlRequest = urlRequest
behaviours.forEach {
urlRequest = $0.modifyRequest(request: request, urlRequest: urlRequest)
}
return urlRequest
}
}
//MARK: Type erased Requests and Responses
public typealias AnyResponse = APIResponse<AnyResponseValue>
public class AnyRequest: APIRequest<AnyResponseValue> {
private let requestPath: String
override public var path: String {
return requestPath
}
init<T>(request: APIRequest<T>) {
requestPath = request.path
super.init(service: request.service.asAny(), parameters: request.parameters, headers: request.headers, encodeBody: request.encodeBody)
}
}
public struct AnyResponseValue: APIResponseValue, CustomDebugStringConvertible, CustomStringConvertible {
public typealias SuccessType = Any
public let statusCode: Int
public let successful: Bool
public let response: Any
public let responseEnum: Any
public let success: Any?
public init(statusCode: Int, successful: Bool, response: Any, responseEnum: Any, success: Any?) {
self.statusCode = statusCode
self.successful = successful
self.response = response
self.responseEnum = responseEnum
self.success = success
}
public init(statusCode: Int, data: Data, decoder: ResponseDecoder) throws {
fatalError()
}
public var description:String {
return "\(responseEnum)"
}
public var debugDescription: String {
if let debugDescription = responseEnum as? CustomDebugStringConvertible {
return debugDescription.debugDescription
} else {
return "\(responseEnum)"
}
}
}
extension APIResponseValue {
public func asAny() -> AnyResponseValue {
return AnyResponseValue(statusCode: statusCode, successful: successful, response: response, responseEnum: self, success: success)
}
}
extension APIResponse {
public func asAny() -> APIResponse<AnyResponseValue> {
return APIResponse<AnyResponseValue>(request: request.asAny(), result: result.map{ $0.asAny() }, urlRequest: urlRequest, urlResponse: urlResponse, data: data, timeline: timeline)
}
}
extension APIRequest {
public func asAny() -> AnyRequest {
return AnyRequest(request: self)
}
}
extension APIService {
public func asAny() -> APIService<AnyResponseValue> {
return APIService<AnyResponseValue>(id: id, tag: tag, method: method, path: path, hasBody: hasBody, securityRequirement: securityRequirement)
}
}