swifter/Sources/HttpServer.swift

//
//  HttpServer2.swift
//  Swifter
//
//  Created by Damian Kolakowski on 17/12/15.
//  Copyright © 2015 Damian Kołakowski. All rights reserved.
//

import Foundation

public class HttpServer: HttpServerIO {
    
    public static let VERSION = "1.0.6"
    
    public override init() { }
    
    private let router = HttpRouter()
    
    public var routes: [(method: String?, path: String)] {
        return router.routes();
    }
    
    public subscript(path: String) -> (HttpRequest -> HttpResponse)? {
        set {
            if let handler = newValue {
                self.router.register(nil, path: path, handler: handler)
            } else {
                self.router.unregister(nil, path: path)
            }
        }
        get { return nil }
    }
    
    public lazy var DELETE : Route = self.newRoute("DELETE")
    public lazy var UPDATE : Route = self.newRoute("UPDATE")
    public lazy var HEAD   : Route = self.newRoute("HEAD")
    public lazy var POST   : Route = self.newRoute("POST")
    public lazy var GET    : Route = self.newRoute("GET")
    public lazy var PUT    : Route = self.newRoute("PUT")
    
    public struct Route {
        public let method: String
        public let router: HttpRouter
        public subscript(path: String) -> (HttpRequest -> HttpResponse)? {
            set {
                if let handler = newValue {
                    router.register(method, path: path, handler: handler)
                } else {
                    router.unregister(method, path: path)
                }
            }
            get { return nil }
        }
    }
    
    private func newRoute(method: String) -> Route {
        return Route(method: method, router: self.router)
    }
    
    override public func dispatch(method: String, url: String) -> ([String:String], HttpRequest -> HttpResponse) {
        if let handler = router.select(method, url: url) {
            return handler
        }
        return super.dispatch(method, url: url)
    }
}