swifter/Sources/HttpServer.swift

//
//  HttpServer.swift
//  Swifter
//
//  Copyright (c) 2014-2016 Damian Kołakowski. All rights reserved.
//

#if os(Linux)
    import Glibc
#else
    import Foundation
#endif

public class HttpServer: HttpServerIO {
    
    public static let VERSION = "1.2.6"
    
    private let router = HttpRouter()
    
    public override init() {
        self.DELETE  = MethodRoute(method: "DELETE", router: router)
        self.UPDATE  = MethodRoute(method: "UPDATE", router: router)
        self.HEAD    = MethodRoute(method: "HEAD", router: router)
        self.POST    = MethodRoute(method: "POST", router: router)
        self.GET     = MethodRoute(method: "GET", router: router)
        self.PUT     = MethodRoute(method: "PUT", router: router)
        self.OPTIONS = MethodRoute(method: "OPTIONS", router: router)
        
        self.delete  = MethodRoute(method: "DELETE", router: router)
        self.update  = MethodRoute(method: "UPDATE", router: router)
        self.head    = MethodRoute(method: "HEAD", router: router)
        self.post    = MethodRoute(method: "POST", router: router)
        self.get     = MethodRoute(method: "GET", router: router)
        self.put     = MethodRoute(method: "PUT", router: router)
        self.options = MethodRoute(method: "OPTIONS", router: router)
    }
    
    public var DELETE, UPDATE, HEAD, POST, GET, PUT, OPTIONS : MethodRoute
    public var delete, update, head, post, get, put, options : MethodRoute
    
    public subscript(path: String) -> (HttpRequest -> HttpResponse)? {
        set {
            router.register(nil, path: path, handler: newValue)
        }
        get { return nil }
    }
    
    public var routes: [String] {
        return router.routes();
    }
    
    public var notFoundHandler: (HttpRequest -> HttpResponse)?
    
    public var middleware = Array<(HttpRequest) -> HttpResponse?>()

    override public func dispatch(request: HttpRequest) -> ([String:String], HttpRequest -> HttpResponse) {
        for layer in middleware {
            if let response = layer(request) {
                return ([:], { _ in response })
            }
        }
        if let result = router.route(request.method, path: request.path) {
            return result
        }
        if let notFoundHandler = self.notFoundHandler {
            return ([:], notFoundHandler)
        }
        return super.dispatch(request)
    }
    
    public struct MethodRoute {
        public let method: String
        public let router: HttpRouter
        public subscript(path: String) -> (HttpRequest -> HttpResponse)? {
            set {
                router.register(method, path: path, handler: newValue)
            }
            get { return nil }
        }
    }
}