Plaid.swift

//
//  Plaid.swift
//  Plaid Swift Wrapper
//
//  Created by Cameron Smith on 4/20/15.
//  Copyright (c) 2015 Cameron Smith. All rights reserved.
//

import Foundation
//        import AVFoundation

struct Plaid {
    static var baseURL:String!
    static var clientId:String!
    static var secret:String!
    
    static func initializePlaid(clientId: String, secret: String, appStatus: BaseURL) {
        Plaid.clientId = clientId
        Plaid.secret = secret
        switch appStatus {
        case .Production:
            baseURL = "https://api.plaid.com/"
        case .Testing:
            baseURL = "https://tartan.plaid.com/"
        }
    }
}

let session = NSURLSession.sharedSession()

enum BaseURL {
    case Production
    case Testing
}

public enum ProductType {
    case Auth
    case Connect
    case Balance
}

public enum Institution {
    case amex
    case bofa
    case capone360
    case schwab
    case chase
    case citi
    case fidelity
    case pnc
    case us
    case usaa
    case wells
}

public struct Account {
    let institutionName: String
    let id: String
    let user: String
    let balance: Double
    let productName: String
    let lastFourDigits: String
    let limit: NSNumber?
    
    public init (account: [String:AnyObject]) {
        let meta = account["meta"] as! [String:AnyObject]
        let accountBalance = account["balance"] as! [String:AnyObject]
        
        institutionName = account["institution_type"] as! String
        id = account["_id"] as! String
        user = account["_user"] as! String
        balance = accountBalance["current"] as! Double
        productName = meta["name"] as! String
        lastFourDigits = meta["number"] as! String
        limit = meta["limit"] as? NSNumber
    }
}

public struct Transaction {
    let account: String
    let id: String
    let amount: Double
    let date: String
    let name: String
    let pending: Bool
    
    let address: String?
    let city: String?
    let state: String?
    let zip: String?
    let storeNumber: String?
    let latitude: Double?
    let longitude: Double?
    
    let trxnType: String?
    let locationScoreAddress: Double?
    let locationScoreCity: Double?
    let locationScoreState: Double?
    let locationScoreZip: Double?
    let nameScore: Double?
    
    let category:NSArray?
    
    public init(transaction: [String:AnyObject]) {
        let meta = transaction["meta"] as! [String:AnyObject]
        let location = meta["location"] as? [String:AnyObject]
        let coordinates = location?["coordinates"] as? [String:AnyObject]
        let score = transaction["score"] as? [String:AnyObject]
        let locationScore = score?["location"] as? [String:AnyObject]
        let type = transaction["type"] as? [String:AnyObject]
        
        account = transaction["_account"] as! String
        id = transaction["_id"] as! String
        amount = transaction["amount"] as! Double
        date = transaction["date"] as! String
        name = transaction["name"] as! String
        pending = transaction["pending"] as! Bool
        
        address = location?["address"] as? String
        city = location?["city"] as? String
        state = location?["state"] as? String
        zip = location?["zip"] as? String
        storeNumber = location?["store_number"] as? String
        latitude = coordinates?["lat"] as? Double
        longitude = coordinates?["lon"] as? Double
        
        trxnType = type?["primary"] as? String
        locationScoreAddress = locationScore?["address"] as? Double
        locationScoreCity = locationScore?["city"] as? Double
        locationScoreState = locationScore?["state"] as? Double
        locationScoreZip = locationScore?["zip"] as? Double
        nameScore = score?["name"] as? Double
        
        category = transaction["category"] as? NSArray
    }
    
}

//MARK: Add Connect or Auth User

func PS_addUser(userType: ProductType, username: String, password: String, pin: String?, institution: Institution, completion: (response: NSURLResponse?, accessToken:String, mfaType:String?, mfa:[[String:AnyObject]]?, accounts: [Account]?, transactions: [Transaction]?, error:NSError?) -> ()) {
    let baseURL = Plaid.baseURL!
    let clientId = Plaid.clientId!
    let secret = Plaid.secret!
    
    let institutionStr: String = institutionToString(institution: institution)
    
    
    if userType == .Auth {
        //Fill in for Auth call
        
    } else if userType == .Connect {
        
        let optionsDict: [String:AnyObject] =
        [
            "list":true
        ]
        
        let optionsDictStr = dictToString(optionsDict)
        
        var urlString:String?
        if pin != nil {
            urlString = "\(baseURL)connect?client_id=\(clientId)&secret=\(secret)&username=\(username)&password=\(password.encodValue)&pin=\(pin!)&type=\(institutionStr)&\(optionsDictStr.encodValue)"
        }
        else {
            urlString = "\(baseURL)connect?client_id=\(clientId)&secret=\(secret)&username=\(username)&password=\(password.encodValue)&type=\(institutionStr)&options=\(optionsDictStr.encodValue)"
            
        }
        
        print("urlString: \(urlString!)")
        
        let url:NSURL! = NSURL(string: urlString!)
        let request = NSMutableURLRequest(URL: url)
        request.HTTPMethod = "POST"
        
        let task = session.dataTaskWithRequest(request, completionHandler: {
            data, response, error in
//            var error:NSError?
            var mfaDict:[[String:AnyObject]]?
            var type:String?
            
            
            
            do {
                if let jsonResult:NSDictionary? = try NSJSONSerialization.JSONObjectWithData(data!, options: NSJSONReadingOptions.MutableContainers) as? NSDictionary {
                    

                //println("jsonResult: \(jsonResult!)")
                    
                    if let token:String = jsonResult?.valueForKey("access_token") as? String {
                        if let mfaResponse = jsonResult!.valueForKey("mfa") as? [[String:AnyObject]] {
    //                        let mfaTwo = mfaResponse[0]
                            mfaDict = mfaResponse
                            if let typeMfa = jsonResult!.valueForKey("type") as? String {
                                type = typeMfa
                            }
                            completion(response: response, accessToken: token, mfaType: type, mfa: mfaDict, accounts: nil, transactions: nil, error: error)
                        } else {
                            let acctsArray:[[String:AnyObject]] = jsonResult?.valueForKey("accounts") as! [[String:AnyObject]]
                            let accts = acctsArray.map{Account(account: $0)}
                            let trxnArray:[[String:AnyObject]] = jsonResult?.valueForKey("transactions") as! [[String:AnyObject]]
                            let trxns = trxnArray.map{Transaction(transaction: $0)}
                            
//                            print(token)
                            
                            completion(response: response, accessToken: token, mfaType: nil, mfa: nil, accounts: accts, transactions: trxns, error: error)
                        }
                    }
                
                } else {
                    //Handle invalid cred login
                }
            } catch{
                print(error)
            }

        })
        task.resume()
    }
}

//MARK: MFA funcs

//func PS_submitMFAResponse(accessToken: String, response: String, completion: (response: NSURLResponse?, accounts: [Account]?, transactions: [Transaction]?, error: NSError?) -> ()) {
//    let baseURL = Plaid.baseURL!
//    let clientId = Plaid.clientId!
//    let secret = Plaid.secret!
//    
//    
//    let urlString:String = "\(baseURL)connect/step?client_id=\(clientId)&secret=\(secret)&access_token=\(accessToken)&mfa=\(response.encodValue)"
//    print("urlString: \(urlString)")
//    let url:NSURL! = NSURL(string: urlString)
//    var request = NSMutableURLRequest(URL: url)
//    request.HTTPMethod = "POST"
//    print("MFA request: \(request)")
//    
//    do {
//        let task = try session.dataTaskWithRequest(request, completionHandler: { data, response  in
//        print("mfa response: \(response)")
//        print("mfa data: \(data)")
////        print(error)
////        var error:NSError?
//        
//        
//        
//        
//            let jsonResult:NSDictionary? = try NSJSONSerialization.JSONObjectWithData(data!, options: NSJSONReadingOptions.MutableContainers) as? NSDictionary
//
//        if jsonResult?.valueForKey("accounts") != nil {
//            let acctsArray:[[String:AnyObject]] = jsonResult?.valueForKey("accounts") as! [[String:AnyObject]]
//            let accts = acctsArray.map{Account(account: $0)}
//            let trxnArray:[[String:AnyObject]] = jsonResult?.valueForKey("transactions") as! [[String:AnyObject]]
//            let trxns = trxnArray.map{Transaction(transaction: $0)}
//            
//            completion(response: response, accounts: accts, transactions: trxns, error: error)
//        }
//        
//        print("jsonResult: \(jsonResult!)")
//    })
//        
//    task.resume()
//    } catch {
//        print(error)
//    }
//    
//    
//}


//MARK: Get balance

func PS_getUserBalance(accessToken: String, completion: (response: NSURLResponse?, accounts:[Account], error:NSError?) -> ()) {
    let baseURL = Plaid.baseURL!
    let clientId = Plaid.clientId!
    let secret = Plaid.secret!
    
    let urlString:String = "\(baseURL)balance?client_id=\(clientId)&secret=\(secret)&access_token=\(accessToken)"
    let url:NSURL? = NSURL(string: urlString)
    
    let task = session.dataTaskWithURL(url!, completionHandler: { (data, response, error) -> Void in
        
        do {
            if let jsonResult:NSDictionary? = try NSJSONSerialization.JSONObjectWithData(data!, options: NSJSONReadingOptions.MutableContainers) as? NSDictionary {
                
                if let dataArray:[[String:AnyObject]] = jsonResult?.valueForKey("accounts") as! [[String : AnyObject]] {
                    let userAccounts = dataArray.map{Account(account: $0)}
                    completion(response: response, accounts: userAccounts, error: error)
                
                }
//                print(jsonResult)

            }
        
//                            return jsonResult
        
        } catch {
            print(error)
        }

    })
    task.resume()
}

//MARK: Get transactions (Connect)

func PS_getUserTransactions(accessToken: String, showPending: Bool, beginDate: String?, endDate: String?, completion: (response: NSURLResponse?, transactions:[Transaction], error:NSError?) -> ()) {
    let baseURL = Plaid.baseURL!
    let clientId = Plaid.clientId!
    let secret = Plaid.secret!
    
    var optionsDict: [String:AnyObject] =
    [
        "pending": true
    ]
    
    if let beginDate = beginDate {
        optionsDict["gte"] = beginDate
    }
    
    if let endDate = endDate {
        optionsDict["lte"] = endDate
    }
    
    let optionsDictStr = dictToString(optionsDict)
    let urlString:String = "\(baseURL)connect?client_id=\(clientId)&secret=\(secret)&access_token=\(accessToken)&\(optionsDictStr.encodValue)"
    let url:NSURL = NSURL(string: urlString)!
    let request = NSMutableURLRequest(URL: url)
    request.HTTPMethod = "POST"
    
    let task = NSURLSession.sharedSession().dataTaskWithURL(url) {
        data, response, error in
//        var error: NSError?
        
        let jsonResult:NSDictionary?
        
        do {
            jsonResult = try NSJSONSerialization.JSONObjectWithData(data!, options: NSJSONReadingOptions.MutableContainers) as? NSDictionary

        
        let dataArray:[[String:AnyObject]] = jsonResult?.valueForKey("transactions") as! [[String:AnyObject]]
        let userTransactions = dataArray.map{Transaction(transaction: $0)}
        completion(response: response, transactions: userTransactions, error: error)
            
        } catch let error as NSError {
            print(error)
        }
    }
    task.resume()
    
}


//MARK: Helper funcs

func plaidDateFormatter(date: NSDate) -> String {
    let dateFormatter = NSDateFormatter()
    dateFormatter.dateFormat = "yyyy-MM-dd"
    let dateStr = dateFormatter.stringFromDate(date)
    return dateStr
}

func dictToString(dictionary: AnyObject) -> NSString {
//    if NSJSONSerialization.isValidJSONObject(value) {
    
//        let theJSONData = NSJSONSerialization.dataWithJSONObject(
//            dictionary ,
//            options: NSJSONWritingOptions(0))
//        let theJSONText = NSString(data: theJSONData!,
//            encoding: NSASCIIStringEncoding)
//        
        
//        do {
//            if let data = try NSJSONSerialization.dataWithJSONObject(value, options: NSJSONWritingOptions(0)) {
//                
//                
//                if let string = NSString(data: data, encoding: NSUTF8StringEncoding) {
//                    return string
//                
//            }
//        } catch error {
//            print(error)
//        }
        
//    do {
//        do {
//        let data = try NSJSONSerialization.dataWithJSONObject(value, options: NSJSONWritingOptions(rawValue: 0))
//            let string = NSString(data: data, encoding: NSUTF8StringEncoding)
//        } catch {
//            print(error
//        }
        
        return "[]"//dictionary.description
//    } else {
//        print(error)
//    }
    
}

func institutionToString(institution institution: Institution) -> String {
    var institutionStr: String {
        switch institution {
        case .amex:
            return "amex"
        case .bofa:
            return "bofa"
        case .capone360:
            return "capone360"
        case .chase:
            return "chase"
        case .citi:
            return "citi"
        case .fidelity:
            return "fidelity"
        case .pnc:
            return "pnc"
        case .schwab:
            return "schwab"
        case .us:
            return "us"
        case .usaa:
            return "usaa"
        case .wells:
            return "wells"
        }
    }
    return institutionStr
}

extension String {
    var encodValue:String {
        return self.stringByAddingPercentEscapesUsingEncoding(NSUTF8StringEncoding)!
//        return self.stringByAddingPercentEncodingWithAllowedCharacters(NSUTF8StringEncoding)
//        return self.stringByAddingPercentEscapesUsingEncodingWithAllowedCharacters(NSUTF8StringEncoding)!
    }
}

extension NSString {
    var encodValue:String {
        return self.stringByAddingPercentEscapesUsingEncoding(NSUTF8StringEncoding)!
    }
}