|
|
@@ -7,34 +7,60 @@
|
|
|
|
|
|
import Foundation
|
|
|
|
|
|
-enum ResponseStatus {
|
|
|
- case OK(String)
|
|
|
- case NotFound
|
|
|
+enum Response {
|
|
|
+
|
|
|
+ case OK(String), Created, Accepted
|
|
|
+ case MovedPermanently(String)
|
|
|
+ case BadRequest, Unauthorized, Forbidden, NotFound
|
|
|
+ case InternalServerError
|
|
|
|
|
|
- func numericValue() -> Int {
|
|
|
+ func statusCode() -> Int {
|
|
|
switch self {
|
|
|
- case .OK(_):
|
|
|
- return 200
|
|
|
- case .NotFound:
|
|
|
- return 404
|
|
|
+ case .OK(_) : return 200
|
|
|
+ case .Created : return 201
|
|
|
+ case .Accepted : return 202
|
|
|
+ case .MovedPermanently : return 301
|
|
|
+ case .BadRequest : return 400
|
|
|
+ case .Unauthorized : return 401
|
|
|
+ case .Forbidden : return 403
|
|
|
+ case .NotFound : return 404
|
|
|
+ case .InternalServerError : return 500
|
|
|
}
|
|
|
}
|
|
|
|
|
|
- func textValue() -> String {
|
|
|
+ func reasonPhrase() -> String {
|
|
|
+ switch self {
|
|
|
+ case .OK(_) : return "OK"
|
|
|
+ case .Created : return "Created"
|
|
|
+ case .Accepted : return "Accepted"
|
|
|
+ case .MovedPermanently : return "Moved Permanently"
|
|
|
+ case .BadRequest : return "Bad Request"
|
|
|
+ case .Unauthorized : return "Unauthorized"
|
|
|
+ case .Forbidden : return "Forbidden"
|
|
|
+ case .NotFound : return "Not Found"
|
|
|
+ case .InternalServerError : return "Internal Server Error"
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ func headers() -> Dictionary<String, String> {
|
|
|
+ switch self {
|
|
|
+ case .MovedPermanently(let location) : return [ "Location" : location ]
|
|
|
+ default: return Dictionary()
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ func body() -> String? {
|
|
|
switch self {
|
|
|
- case .OK(let text):
|
|
|
- return text
|
|
|
- case .NotFound:
|
|
|
- return "Not found"
|
|
|
+ case .OK(let text) : return text
|
|
|
+ default : return nil
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
|
|
|
-typealias Handler = Void -> ResponseStatus
|
|
|
-
|
|
|
-/* HTTP server */
|
|
|
class HttpServer
|
|
|
{
|
|
|
+ typealias Handler = Void -> Response
|
|
|
+
|
|
|
var handlers = Dictionary<String, Handler>()
|
|
|
var acceptSocket: CInt = -1
|
|
|
|
|
|
@@ -47,57 +73,63 @@ class HttpServer
|
|
|
}
|
|
|
}
|
|
|
|
|
|
- func start(listenPort: in_port_t) -> (Bool, String?) {
|
|
|
- releaseAcceptSocket()
|
|
|
- let (socket, error) = Socket.tcpForListen(listenPort)
|
|
|
- if ( socket == -1 ) {
|
|
|
- return (false, error)
|
|
|
+ func routes() -> Array<String> {
|
|
|
+ var results = Array<String>()
|
|
|
+ for (key,_) in handlers {
|
|
|
+ results.append(key)
|
|
|
}
|
|
|
- acceptSocket = socket
|
|
|
- dispatch_async(dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_BACKGROUND, 0), { () -> Void in
|
|
|
- while ( self.acceptSocket != -1 ) {
|
|
|
- var addr = sockaddr(sa_len: 0, sa_family: 0, sa_data: (0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0)), len: socklen_t = 0
|
|
|
- let socket = accept(self.acceptSocket, &addr, &len)
|
|
|
- if ( socket == -1 ) {
|
|
|
- self.releaseAcceptSocket();
|
|
|
- return
|
|
|
- }
|
|
|
- Socket.nosigpipe(socket)
|
|
|
- dispatch_async(dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_BACKGROUND, 0), { () -> Void in
|
|
|
- let parser = HttpParser()
|
|
|
- while let (path, headers) = parser.parseHttpHeader(socket) {
|
|
|
- let keepAlive = parser.supportsKeepAlive(headers)
|
|
|
-
|
|
|
- if let handler = self.handlers[path] {
|
|
|
- let responseStatus = handler()
|
|
|
- let responseText = responseStatus.textValue()
|
|
|
- let nsdata =
|
|
|
- responseText
|
|
|
- .bridgeToObjectiveC()
|
|
|
- .dataUsingEncoding(NSUTF8StringEncoding)
|
|
|
-
|
|
|
- Socket.writeStringUTF8(socket, string: "HTTP/1.1 \(responseStatus.numericValue())\r\n")
|
|
|
- Socket.writeStringUTF8(socket, string: "Content-Length: \(nsdata.length)\r\n")
|
|
|
- if keepAlive {
|
|
|
- Socket.writeStringUTF8(socket, string: "Connection: keep-alive\r\n")
|
|
|
- }
|
|
|
- Socket.writeStringUTF8(socket, string: "\r\n")
|
|
|
- Socket.writeStringUTF8(socket, string: responseText)
|
|
|
- } else {
|
|
|
- Socket.writeStringUTF8(socket, string: "HTTP/1.1 \(ResponseStatus.NotFound.numericValue())\r\n")
|
|
|
- Socket.writeStringUTF8(socket, string: "Content-Length: 0\r\n")
|
|
|
- if keepAlive {
|
|
|
- Socket.writeStringUTF8(socket, string: "Connection: keep-alive\r\n")
|
|
|
+ return results
|
|
|
+ }
|
|
|
+
|
|
|
+ func start(listenPort: in_port_t = 8080, error:NSErrorPointer = nil) -> Bool {
|
|
|
+ releaseAcceptSocket()
|
|
|
+ if let socket = Socket.tcpForListen(port: listenPort, error: error) {
|
|
|
+ acceptSocket = socket
|
|
|
+ dispatch_async(dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_BACKGROUND, 0), {
|
|
|
+ while ( self.acceptSocket != -1 ) {
|
|
|
+ if let socket = Socket.acceptClientSocket(self.acceptSocket) {
|
|
|
+ dispatch_async(dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_BACKGROUND, 0), {
|
|
|
+ let parser = HttpParser()
|
|
|
+ while let (path, headers) = parser.nextHttpRequest(socket) {
|
|
|
+ let keepAlive = parser.supportsKeepAlive(headers)
|
|
|
+ if let handler = self.handlers[path] {
|
|
|
+ HttpServer.writeResponse(socket, response: handler(), keepAlive: keepAlive)
|
|
|
+ } else {
|
|
|
+ HttpServer.writeResponse(socket, response: Response.NotFound, keepAlive: keepAlive)
|
|
|
+ }
|
|
|
+ if !keepAlive { break }
|
|
|
}
|
|
|
- Socket.writeStringUTF8(socket, string: "\r\n")
|
|
|
- }
|
|
|
- if !keepAlive { break }
|
|
|
+ Socket.release(socket)
|
|
|
+ });
|
|
|
+ } else {
|
|
|
+ self.releaseAcceptSocket()
|
|
|
}
|
|
|
- Socket.release(socket)
|
|
|
- });
|
|
|
- }
|
|
|
- });
|
|
|
- return (true, nil)
|
|
|
+ }
|
|
|
+ });
|
|
|
+ return true
|
|
|
+ }
|
|
|
+ return false
|
|
|
+ }
|
|
|
+
|
|
|
+ class func writeResponse(socket: CInt, response: Response, keepAlive: Bool) {
|
|
|
+ Socket.writeStringUTF8(socket, string: "HTTP/1.1 \(response.statusCode()) \(response.reasonPhrase())\r\n")
|
|
|
+ let messageBody = response.body()
|
|
|
+ if let body = messageBody {
|
|
|
+ let nsdata = body.bridgeToObjectiveC().dataUsingEncoding(NSUTF8StringEncoding)
|
|
|
+ Socket.writeStringUTF8(socket, string: "Content-Length: \(nsdata.length)\r\n")
|
|
|
+ } else {
|
|
|
+ Socket.writeStringUTF8(socket, string: "Content-Length: 0\r\n")
|
|
|
+ }
|
|
|
+ if keepAlive {
|
|
|
+ Socket.writeStringUTF8(socket, string: "Connection: keep-alive\r\n")
|
|
|
+ }
|
|
|
+ for (name, value) in response.headers() {
|
|
|
+ Socket.writeStringUTF8(socket, string: "\(name): \(value)\r\n")
|
|
|
+ }
|
|
|
+ Socket.writeStringUTF8(socket, string: "\r\n")
|
|
|
+ if let body = messageBody {
|
|
|
+ Socket.writeStringUTF8(socket, string: body)
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
func stop() {
|
Damian Kołakowski
