swifter/Swifter/HttpParser.swift

//
//  HttpParser.swift
//
//  Created by Damian Kolakowski on 05/06/14.
//  Copyright (c) 2014 Damian Kołakowski. All rights reserved.
//

import Foundation

class HttpParser {
    
    class func err(reason:String) -> NSError {
        return NSError.errorWithDomain("HttpParser", code: 0, userInfo:[NSLocalizedFailureReasonErrorKey : reason])
    }
    
    func nextHttpRequest(socket: CInt, error:NSErrorPointer = nil) -> (String, Dictionary<String, String>)? {
        if let statusLine = nextLine(socket, error: error) {
            let statusTokens = split(statusLine, { $0 == " " })
            if ( statusTokens.count < 3 ) {
                if error { error.memory = HttpParser.err("Invalid status line: \(statusLine)") }
                return nil
            }
            let path = statusTokens[1]
            if let headers = nextHeaders(socket, error: error) {
                return (path, headers)
            }
        }
        return nil
    }
    
    func nextHeaders(socket: CInt, error:NSErrorPointer) -> Dictionary<String, String>? {
        var headers = Dictionary<String, String>()
        while let headerLine = nextLine(socket, error: error) {
            if ( headerLine.isEmpty ) {
                return headers
            }
            let headerTokens = split(headerLine, { $0 == ":" })
            if ( headerTokens.count >= 2 ) {
                // RFC 2616 - "Hypertext Transfer Protocol -- HTTP/1.1", paragraph 4.2, "Message Headers":
                // "Each header field consists of a name followed by a colon (":") and the field value. Field names are case-insensitive."
                // We can keep lower case version.
                let headerName = headerTokens[0].lowercaseString
                let headerValue = headerTokens[1]
                if ( !headerName.isEmpty && !headerValue.isEmpty ) {
                    headers.updateValue(headerValue, forKey: headerName)
                }
            }
        }
        return nil
    }
    
    func nextLine(socket: CInt, error:NSErrorPointer) -> String? {
        // TODO - read more bytes than one. It makes the server very slow.
        // TODO - check if there is a nicer way to manipulate bytes with Swift ( recv(...) -> String )
        var characters: String = ""
        var buff: UInt8[] = UInt8[](count: 1, repeatedValue: 0), n: Int = 1
        do {
            n = recv(socket, &buff, 1, 0);
            if ( n > 0 && buff[0] > 13 /* CR */ ) {
                characters += Character(UnicodeScalar(UInt32(buff[0])))
            }
        } while ( n > 0 && buff[0] != 10 /* NL */ )
        if ( n == -1 ) {
            if error { error.memory = Socket.socketRecentError("recv(...) failed.") }
            return nil
        }
        return characters
    }
    
    func supportsKeepAlive(headers: Dictionary<String, String>) -> Bool {
        if let value = headers["connection"] {
            return "keep-alive" == value.stringByTrimmingCharactersInSet(NSCharacterSet.whitespaceAndNewlineCharacterSet()).lowercaseString
        }
        return false
    }
}