pavel
/
swifter
зеркало из https://github.com/httpswift/swifter


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202
							//
//  HttpResponse.swift
//  Swifter
//  Copyright (c) 2015 Damian Kołakowski. All rights reserved.
//

import Foundation

public enum SerializationError: ErrorType {
    case InvalidObject
    case NotSupported
    case EncodingError
}

public protocol Serializer {
    func serialize(object: Any) throws -> String
}

public class JSONSerializer : Serializer {
    public func serialize(object: Any) throws -> String {
        guard let obj = object as? AnyObject where NSJSONSerialization.isValidJSONObject(obj) else {
            throw SerializationError.InvalidObject
        }
        
        let json = try NSJSONSerialization.dataWithJSONObject(obj, options: NSJSONWritingOptions.PrettyPrinted)
        
        guard let string = String(data: json, encoding: NSUTF8StringEncoding) else {
            throw SerializationError.EncodingError
        }
        
        return string
    }
    
    private static func serialize(object: Any) throws -> String {
        let serializer = JSONSerializer()
        return try serializer.serialize(object)
    }
}

public class XMLSerializer: Serializer {
    public func serialize(object: Any) throws -> String {
        throw SerializationError.NotSupported
    }
    
    private static func serialize(object: Any) throws -> String {
        let serializer = XMLSerializer()
        return try serializer.serialize(object)
    }
}

public class PLISTSerializer: Serializer {
    public func serialize(object: Any) throws -> String {
        let format = NSPropertyListFormat.XMLFormat_v1_0
        
        guard let obj = object as? AnyObject where NSPropertyListSerialization.propertyList(obj, isValidForFormat: format) else {
            throw SerializationError.InvalidObject
        }
        
        let plist = try NSPropertyListSerialization.dataWithPropertyList(obj, format: format, options: 0)
        
        guard let string = String(data: plist, encoding: NSUTF8StringEncoding) else {
            throw SerializationError.EncodingError
        }
        
        return string
    }
    
    private static func serialize(object: Any) throws -> String {
        let serializer = PLISTSerializer()
        return try serializer.serialize(object)
    }
}

public enum HttpResponseBody {
    
    case Json(AnyObject)
    case Xml(AnyObject)
    case Plist(AnyObject)
    case Html(String)
    case Text(String)
    case Custom(Serializer, Any)
    
    func data() -> String? {
        do {
            switch self {
                
            case .Json(let object):
                return try JSONSerializer.serialize(object)
                
            case .Xml(let object):
                return try XMLSerializer.serialize(object)
                
            case .Plist(let object):
                return try PLISTSerializer.serialize(object)
                
            case .Text(let body):
                return body
                
            case .Html(let body):
                return "<html><meta charset=\"UTF-8\"><body>\(body)</body></html>"
                
            case .Custom(let serializer, let object):
                return try serializer.serialize(object)
            }
        } catch {
            return "Serialisation error: \(error)"
        }
    }
}

public enum HttpResponse {
    
    case OK(HttpResponseBody), Created, Accepted
    case MovedPermanently(String)
    case BadRequest, Unauthorized, Forbidden, NotFound
    case InternalServerError
    case RAW(Int, String, [String:String]?, NSData)
    
    func statusCode() -> Int {
        switch self {
        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
        case .RAW(let code,_,_,_)   : return code
        }
    }
    
    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"
        case .RAW(_,let pharse,_,_) : return pharse
        }
    }
    
    func headers() -> [String: String] {
        var headers = [String:String]()
        headers["Server"] = "Swifter \(HttpServer.VERSION)"
        switch self {
		case .OK(let body):
            switch body {
                case .Json(_)   : headers["Content-Type"] = "application/json"
                case .Plist(_)  : headers["Content-Type"] = "application/xml"
                case .Xml(_)    : headers["Content-Type"] = "application/xml"
                // 'application/xml' or 'text/xml' ?
                // From RFC: http://www.rfc-editor.org/rfc/rfc3023.txt - "If an XML document -- that is, the unprocessed, 
                // source XML document -- is readable by casual users, text/xml is preferable to application/xml. 
                // MIME user agents (and web user agents) that do not have explicit support for text/xml will treat it as text/plain, 
                // for example, by displaying the XML MIME entity as plain text.
                case .Html(_)   : headers["Content-Type"] = "text/html"
                default:break
            }
        case .MovedPermanently(let location): headers["Location"] = location
        case .RAW(_,_, let rawHeaders,_):
            if let rawHeaders = rawHeaders {
                for (k, v) in rawHeaders {
                    headers.updateValue(v, forKey: k)
                }
            }
        default:break
        }
        return headers
    }
    
    func body() -> NSData? {
        switch self {
        case .OK(let body)          : return body.data()?.dataUsingEncoding(NSUTF8StringEncoding, allowLossyConversion: false)
        case .RAW(_,_,_, let data)  : return data
        default                     : return nil
        }
    }
}

/**
	Makes it possible to compare handler responses with '==', but
	ignores any associated values. This should generally be what
	you want. E.g.:
	
		let resp = handler(updatedRequest)
		if resp == .NotFound {
			print("Client requested not found: \(request.url)")
		}
*/

func ==(inLeft: HttpResponse, inRight: HttpResponse) -> Bool {
	return inLeft.statusCode() == inRight.statusCode()
}