ToJSON.swift

//
//  ToJSON.swift
//  ObjectMapper
//
//  Created by Tristan Himmelman on 2014-10-13.
//  Copyright (c) 2014 hearst. All rights reserved.
//

import Foundation

class ToJSON {
	
	func addObject(object: AnyObject, inout toDictionary dictionary: [String : AnyObject]){
		
	}
	
    func baseType<N>(field: N, key: String, inout dictionary: [String : AnyObject]) {
		
		var temp = dictionary
		var currentKey = key

        switch N.self {
		// basic Types
        case is Bool.Type:
            dictionary[currentKey] = field as Bool
        case is Int.Type:
            dictionary[currentKey] = field as Int
        case is Double.Type:
            dictionary[currentKey] = field as Double
        case is Float.Type:
            dictionary[currentKey] = field as Float
        case is String.Type:
            dictionary[currentKey] = field as String

		// Arrays with basic types
		case is Array<Bool>.Type:
			dictionary[currentKey] = field as Array<Bool>
		case is Array<Int>.Type:
			dictionary[currentKey] = field as Array<Int>
		case is Array<Double>.Type:
			dictionary[currentKey] = field as Array<Double>
		case is Array<Float>.Type:
			dictionary[currentKey] = field as Array<Float>
		case is Array<String>.Type:
			dictionary[currentKey] = field as Array<String>
			
		// Dictionaries with basic types
		case is Dictionary<String, Bool>.Type:
			dictionary[currentKey] = field as Dictionary<String,Bool>
		case is Dictionary<String, Bool>.Type:
			dictionary[currentKey] = field as Dictionary<String,Int>
		case is Dictionary<String, Bool>.Type:
			dictionary[currentKey] = field as Dictionary<String,Double>
		case is Dictionary<String, Bool>.Type:
			dictionary[currentKey] = field as Dictionary<String,Float>
		case is Dictionary<String, String>.Type:
			dictionary[currentKey] = field as Dictionary<String,String>
        default:
            //println("Default")
            return
        }
    }
    
    func optionalBaseType<N>(field: N?, key: String, inout dictionary: [String : AnyObject]) {
        if let field: N = field {
            baseType(field, key: key, dictionary: &dictionary)
        }
    }
    
    func baseArray(field: Array<AnyObject>, key: String, inout dictionary: [String : AnyObject]){
        dictionary[key] = NSArray(array: field)
    }
    
    func optionalBaseArray(field: Array<AnyObject>?, key: String, inout dictionary: [String : AnyObject]){
        if let value = field {
            baseArray(value, key: key, dictionary: &dictionary)
        }
    }
    
    func baseDictionary(field: Dictionary<String, AnyObject>, key: String, inout dictionary: [String : AnyObject]){
        dictionary[key] = NSDictionary(dictionary: field)
    }
    
    func optionalBaseDictionary(field: Dictionary<String, AnyObject>?, key: String, inout dictionary: [String : AnyObject]){
        if let value = field {
            baseDictionary(value, key: key, dictionary: &dictionary)
        }
    }
    
    func object<N: Mappable>(field: N, key: String, inout dictionary: [String : AnyObject]) {
        let mapper = Mapper<N>()
        
        dictionary[key] = NSDictionary(dictionary: mapper.toJSON(field))
    }
    
    func optionalObject<N: Mappable>(field: N?, key: String, inout dictionary: [String : AnyObject]) {
        if let field = field {
            object(field, key: key, dictionary: &dictionary)
        }
    }
    
    func objectArray<N: Mappable>(field: Array<N>, key: String, inout dictionary: [String : AnyObject]) {
        var JSONObjects = NSMutableArray()
        
        for object in field {
            let mapper = Mapper<N>()
            JSONObjects.addObject(mapper.toJSON(object))
        }

        if JSONObjects.count > 0 {
            dictionary[key] = JSONObjects
        }
    }
    
    func optionalObjectArray<N: Mappable>(field: Array<N>?, key: String, inout dictionary: [String : AnyObject]) {
        if let field = field {
            objectArray(field, key: key, dictionary: &dictionary)
        }
    }
    
    func objectDictionary<N: Mappable>(field: Dictionary<String, N>, key: String, inout dictionary: [String : AnyObject]) {
        var JSONObjects = NSMutableDictionary()
        
        for (k, object) in field {
            let mapper = Mapper<N>()
            JSONObjects.setObject(mapper.toJSON(object), forKey: k)

        }
        
        if JSONObjects.count > 0 {
            dictionary[key] = JSONObjects
        }
    }
    
    func optionalObjectDictionary<N: Mappable>(field: Dictionary<String, N>?, key: String, inout dictionary: [String : AnyObject]) {
        if let field = field {
            objectDictionary(field, key: key, dictionary: &dictionary)
        }
    }
}