-
Notifications
You must be signed in to change notification settings - Fork 60
/
ViewController.swift
212 lines (188 loc) · 8.04 KB
/
ViewController.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
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
//===----------------------------------------------------------------------===//
//
// This source file is part of the SwiftNIO open source project
//
// Copyright (c) 2017-2018 Apple Inc. and the SwiftNIO project authors
// Licensed under Apache License v2.0
//
// See LICENSE.txt for license information
// See CONTRIBUTORS.txt for the list of SwiftNIO project authors
//
// SPDX-License-Identifier: Apache-2.0
//
//===----------------------------------------------------------------------===//
import UIKit
import NIOCore
import NIOExtras
import NIOPosix
import NIOTransportServices
import NIOTLS
import NIOSSL
#if canImport(Network)
import Network
#endif
func sendEmail(_ email: Email,
group: EventLoopGroup,
communicationHandler: @escaping (String) -> Void,
queue: DispatchQueue,
_ handler: @escaping (Error?) -> Void) {
let emailSentPromise: EventLoopPromise<Void> = group.next().makePromise()
let bootstrap: NIOClientTCPBootstrap
do {
bootstrap = try configureBootstrap(group: group,
email: email,
emailSentPromise: emailSentPromise,
communicationHandler: communicationHandler)
} catch {
queue.async {
handler(error)
}
return
}
let connection = bootstrap.connect(host: Configuration.shared.serverConfig.hostname,
port: Configuration.shared.serverConfig.port)
connection.cascadeFailure(to: emailSentPromise)
emailSentPromise.futureResult.map {
connection.whenSuccess { $0.close(promise: nil) }
queue.async {
handler(nil)
}
}.whenFailure { error in
connection.whenSuccess { $0.close(promise: nil) }
queue.async {
handler(error)
}
}
}
class ViewController: UIViewController {
var group: EventLoopGroup? = nil
@IBOutlet weak var sendEmailButton: UIButton!
@IBOutlet weak var logView: UITextView!
@IBOutlet weak var textField: UITextView!
@IBOutlet weak var subjectField: UITextField!
@IBOutlet weak var toField: UITextField!
@IBOutlet weak var fromField: UITextField!
@IBAction func sendEmailAction(_ sender: UIButton) {
self.logView.text = ""
let email = Email(senderName: nil,
senderEmail: self.fromField.text!,
recipientName: nil,
recipientEmail: self.toField.text!,
subject: self.subjectField.text!,
body: self.textField.text!)
let commHandler: (String) -> Void = { str in
DispatchQueue.main.async {
self.logView.text += str + "\n"
}
}
sendEmail(email, group: self.group!, communicationHandler: commHandler, queue: DispatchQueue.main) { maybeError in
assert(Thread.isMainThread)
if let error = maybeError {
self.sendEmailButton.titleLabel?.text = "❌"
self.logView.text += "ERROR: \(error)\n"
} else {
self.sendEmailButton.titleLabel?.text = "✅"
}
}
}
override func viewWillAppear(_ animated: Bool) {
let group = makeEventLoopGroup(loopCount: 1, implementation: .best)
self.group = group
}
override func viewWillDisappear(_ animated: Bool) {
try! self.group?.syncShutdownGracefully()
self.group = nil
}
override func viewDidLoad() {
super.viewDidLoad()
// Do any additional setup after loading the view, typically from a nib.
}
}
// MARK: - NIO/NIOTS handling
func makeNIOSMTPHandlers(communicationHandler: @escaping (String) -> Void,
email: Email,
emailSentPromise: EventLoopPromise<Void>) -> [ChannelHandler] {
return [
PrintEverythingHandler(handler: communicationHandler),
ByteToMessageHandler(LineBasedFrameDecoder()),
SMTPResponseDecoder(),
MessageToByteHandler(SMTPRequestEncoder()),
SendEmailHandler(configuration: Configuration.shared.serverConfig,
email: email,
allDonePromise: emailSentPromise)
]
}
private let sslContext = try! NIOSSLContext(configuration: TLSConfiguration.makeClientConfiguration())
func configureBootstrap(group: EventLoopGroup,
email: Email,
emailSentPromise: EventLoopPromise<Void>,
communicationHandler: @escaping (String) -> Void) throws -> NIOClientTCPBootstrap {
let hostname = Configuration.shared.serverConfig.hostname
let bootstrap: NIOClientTCPBootstrap
switch (NetworkImplementation.best, Configuration.shared.serverConfig.tlsConfiguration) {
case (.transportServices, .regularTLS), (.transportServices, .insecureNoTLS):
if #available(macOS 10.14, iOS 12, tvOS 12, watchOS 3, *) {
bootstrap = NIOClientTCPBootstrap(NIOTSConnectionBootstrap(group: group),
tls: NIOTSClientTLSProvider())
} else {
fatalError("Network.framework unsupported on this OS yet it was selected as the best option.")
}
case (.transportServices, .startTLS):
if #available(macOS 10.14, iOS 12, tvOS 12, watchOS 3, *) {
bootstrap = try NIOClientTCPBootstrap(NIOTSConnectionBootstrap(group: group),
tls: NIOSSLClientTLSProvider(context: sslContext,
serverHostname: hostname))
} else {
fatalError("Network.framework unsupported on this OS yet it was selected as the best option.")
}
case (.posix, _):
bootstrap = try NIOClientTCPBootstrap(ClientBootstrap(group: group),
tls: NIOSSLClientTLSProvider(context: sslContext,
serverHostname: hostname))
}
switch Configuration.shared.serverConfig.tlsConfiguration {
case .regularTLS:
bootstrap.enableTLS()
case .insecureNoTLS, .startTLS:
() // no TLS to start with
}
return bootstrap.channelInitializer { channel in
channel.pipeline.addHandlers(makeNIOSMTPHandlers(communicationHandler: communicationHandler,
email: email,
emailSentPromise: emailSentPromise))
}
}
/// Network implementation and by extension which version of NIO to use.
enum NetworkImplementation {
/// POSIX sockets and NIO.
case posix
/// NIOTransportServices (and Network.framework).
case transportServices
/// Return the best implementation available for this platform, that is NIOTransportServices
/// when it is available or POSIX and NIO otherwise.
static var best: NetworkImplementation {
if #available(OSX 10.14, iOS 12.0, tvOS 12.0, watchOS 6.0, *) {
return .transportServices
} else {
return .posix
}
}
}
/// Makes an appropriate `EventLoopGroup` based on the given implementation.
///
/// For `.posix` this is a `MultiThreadedEventLoopGroup`, for `.networkFramework` it is a
/// `NIOTSEventLoopGroup`.
///
/// - Parameter implementation: The network implementation to use.
func makeEventLoopGroup(loopCount: Int, implementation: NetworkImplementation) -> EventLoopGroup {
switch implementation {
case .transportServices:
guard #available(OSX 10.14, iOS 12.0, tvOS 12.0, watchOS 6.0, *) else {
// This is gated by the availability of `.networkFramework` so should never happen.
fatalError(".networkFramework is being used on an unsupported platform")
}
return NIOTSEventLoopGroup(loopCount: loopCount, defaultQoS: .utility)
case .posix:
return MultiThreadedEventLoopGroup(numberOfThreads: loopCount)
}
}