From 9303329f977423f234896e4ae5e10ef93720b823 Mon Sep 17 00:00:00 2001
From: Jonathan Duarte <>
Date: Thu, 11 Apr 2019 08:23:30 -0400
Subject: [PATCH] - Functions for utilization and account (temporary) metrics.
 - Added identifiers for wellness score controllers - Added "transaction_code"
 values to respective API call and transaction class

---
 .../Base.lproj/Interface.storyboard           |  18 +--
 .../AccountNumInterfaceController.swift       |  10 +-
 .../ConsistencyInterfaceController.swift      |   4 +-
 .../FetchData.swift                           |   6 +-
 .../MainMenuInterfaceController.swift         |   1 +
 .../UtilizationfaceController.swift           |   6 +-
 .../WellnessInterfaceController.swift         | 107 +++++++++++++++++-
 .../SynchronyFinancial/Transaction.swift      |   4 +-
 8 files changed, 137 insertions(+), 19 deletions(-)

diff --git a/SynchronyFinancial/SynchronyFinancial WatchKit App/Base.lproj/Interface.storyboard b/SynchronyFinancial/SynchronyFinancial WatchKit App/Base.lproj/Interface.storyboard
index 875abcd..0edbf64 100644
--- a/SynchronyFinancial/SynchronyFinancial WatchKit App/Base.lproj/Interface.storyboard	
+++ b/SynchronyFinancial/SynchronyFinancial WatchKit App/Base.lproj/Interface.storyboard	
@@ -1,12 +1,12 @@
 <?xml version="1.0" encoding="UTF-8"?>
-<document type="com.apple.InterfaceBuilder.WatchKit.Storyboard" version="3.0" toolsVersion="14490.70" targetRuntime="watchKit" propertyAccessControl="none" useAutolayout="YES" useTraitCollections="YES" colorMatched="YES" initialViewController="Qge-pO-HLc">
+<document type="com.apple.InterfaceBuilder.WatchKit.Storyboard" version="3.0" toolsVersion="14460.31" targetRuntime="watchKit" propertyAccessControl="none" useAutolayout="YES" useTraitCollections="YES" colorMatched="YES" initialViewController="Qge-pO-HLc">
     <device id="watch38" orientation="portrait">
         <adaptation id="fullscreen"/>
     </device>
     <dependencies>
         <deployment identifier="watchOS"/>
-        <plugIn identifier="com.apple.InterfaceBuilder.IBCocoaTouchPlugin" version="14490.49"/>
-        <plugIn identifier="com.apple.InterfaceBuilder.IBWatchKitPlugin" version="14490.21"/>
+        <plugIn identifier="com.apple.InterfaceBuilder.IBCocoaTouchPlugin" version="14460.20"/>
+        <plugIn identifier="com.apple.InterfaceBuilder.IBWatchKitPlugin" version="14460.16"/>
     </dependencies>
     <scenes>
         <!--root-->
@@ -70,10 +70,10 @@
             </objects>
             <point key="canvasLocation" x="23" y="602"/>
         </scene>
-        <!--Account Num Interface Controller-->
+        <!--AccountScore-->
         <scene sceneID="jxS-9m-lB7">
             <objects>
-                <controller id="Mki-ad-UCC" customClass="AccountNumInterfaceController" customModule="SynchronyFinancial_WatchKit_Extension">
+                <controller identifier="AccountScore" id="Mki-ad-UCC" customClass="AccountNumInterfaceController" customModule="SynchronyFinancial_WatchKit_Extension">
                     <items>
                         <label width="1" alignment="left" text="Account Score" textAlignment="center" id="0ed-0W-LKq"/>
                         <imageView width="1" height="120" alignment="center" verticalAlignment="center" image="scores" contentMode="scaleAspectFit" initiallyAnimated="NO" id="RlX-Hj-mTY"/>
@@ -309,10 +309,10 @@
             </objects>
             <point key="canvasLocation" x="1420" y="-229"/>
         </scene>
-        <!--Utilization Interface Controller-->
+        <!--UtilizationScore-->
         <scene sceneID="nMW-YM-Pwh">
             <objects>
-                <controller id="RNe-Ul-1jm" customClass="UtilizationInterfaceController" customModule="SynchronyFinancial_WatchKit_Extension">
+                <controller identifier="UtilizationScore" id="RNe-Ul-1jm" customClass="UtilizationInterfaceController" customModule="SynchronyFinancial_WatchKit_Extension">
                     <items>
                         <label width="1" alignment="left" text="Utilization Score" textAlignment="center" id="Rzj-RR-iat"/>
                         <imageView width="1" height="120" alignment="center" verticalAlignment="center" image="scores" contentMode="scaleAspectFit" initiallyAnimated="NO" id="MkL-48-J8l"/>
@@ -325,10 +325,10 @@
             </objects>
             <point key="canvasLocation" x="613" y="602"/>
         </scene>
-        <!--Consistency Interface Controller-->
+        <!--PaymentScore-->
         <scene sceneID="do4-GS-cad">
             <objects>
-                <controller id="k4M-kn-ZzM" customClass="ConsistencyInterfaceController" customModule="SynchronyFinancial_WatchKit_Extension">
+                <controller identifier="PaymentScore" id="k4M-kn-ZzM" customClass="ConsistencyInterfaceController" customModule="SynchronyFinancial_WatchKit_Extension">
                     <items>
                         <label width="1" height="20" alignment="left" text="Payment Score" textAlignment="center" id="5cf-d7-ssN"/>
                         <imageView width="1" height="120" alignment="center" verticalAlignment="center" image="scores" contentMode="scaleAspectFit" initiallyAnimated="NO" id="XsN-KK-JY4"/>
diff --git a/SynchronyFinancial/SynchronyFinancial WatchKit Extension/AccountNumInterfaceController.swift b/SynchronyFinancial/SynchronyFinancial WatchKit Extension/AccountNumInterfaceController.swift
index 8f0e948..c208caf 100644
--- a/SynchronyFinancial/SynchronyFinancial WatchKit Extension/AccountNumInterfaceController.swift	
+++ b/SynchronyFinancial/SynchronyFinancial WatchKit Extension/AccountNumInterfaceController.swift	
@@ -20,8 +20,14 @@ class AccountNumInterfaceController: WKInterfaceController {
     override func awake(withContext context: Any?) {
         super.awake(withContext: context)
         
+        if let val: Int = context as? Int {
+            score = val
+        }else {
+            score = 0
+        }
+        
         //Assigns score based on # of accounts
-        switch accNum {
+        /*switch accNum {
         case 2...3:
             score = 50
         case 3...4:
@@ -30,7 +36,7 @@ class AccountNumInterfaceController: WKInterfaceController {
             score = 100
         default:
             score = 25
-        }
+        }*/
     }
 
     override func didAppear() {
diff --git a/SynchronyFinancial/SynchronyFinancial WatchKit Extension/ConsistencyInterfaceController.swift b/SynchronyFinancial/SynchronyFinancial WatchKit Extension/ConsistencyInterfaceController.swift
index 09cfe29..ed2a330 100644
--- a/SynchronyFinancial/SynchronyFinancial WatchKit Extension/ConsistencyInterfaceController.swift	
+++ b/SynchronyFinancial/SynchronyFinancial WatchKit Extension/ConsistencyInterfaceController.swift	
@@ -20,7 +20,7 @@ class ConsistencyInterfaceController: WKInterfaceController {
         super.awake(withContext: context)
         
         //Assigns a score based on # of late payments
-        switch latePayments {
+        /*switch latePayments {
         case 1:
             score = 80
         case 2:
@@ -29,7 +29,7 @@ class ConsistencyInterfaceController: WKInterfaceController {
             score = 25
         default:
             score = 100
-        }
+        }*/
     }
     
     override func didAppear() {
diff --git a/SynchronyFinancial/SynchronyFinancial WatchKit Extension/FetchData.swift b/SynchronyFinancial/SynchronyFinancial WatchKit Extension/FetchData.swift
index 5426ea7..33483b2 100644
--- a/SynchronyFinancial/SynchronyFinancial WatchKit Extension/FetchData.swift	
+++ b/SynchronyFinancial/SynchronyFinancial WatchKit Extension/FetchData.swift	
@@ -92,13 +92,14 @@ class FetchData {
                         let amountString = $0["transaction_amount"].string,
                         let amount = Double(amountString),
                         let confirmationNum = $0["payment_confirmation_number"].string,
+                        let transactionCode = $0["transaction_code"].string,
                         let paymentIdString = $0["payment_id"].string,
                         let paymentId = Int(paymentIdString),
                         let modifiable = $0["is_payment_modifiable"].string {
                         let type: TransactionType = $0["payment_amount_type"].stringValue == "" ? .purchase : .reimbursement
                         let isModifiable: Bool = modifiable == "Y" ? true : false
 
-                        transactions.append(Transaction(type: type, amount: amount, merchantID: desc, date: date, confirmationNum: confirmationNum, paymentId: paymentId, isPending: true, isModifiable: isModifiable))
+                        transactions.append(Transaction(type: type, amount: amount, merchantID: desc, date: date, confirmationNum: confirmationNum, paymentId: paymentId, isPending: true, isModifiable: isModifiable, transactionCode: transactionCode))
                     }
                 }
 
@@ -107,9 +108,10 @@ class FetchData {
                     if let desc = $0["description"].string,
                         let date = Defaults.careCreditDateFormatter.date(from: $0["transaction_date"].stringValue),
                         let amountString = $0["transaction_amount"].string,
+                        let transactionCode = $0["transaction_code"].string,
                         let amount = Double(amountString) {
                         let type: TransactionType = $0["payment_amount_type"].stringValue == "" ? .purchase : .reimbursement
-                        transactions.append(Transaction(type: type, amount: amount, merchantID: desc, date: date, isPending: false))
+                        transactions.append(Transaction(type: type, amount: amount, merchantID: desc, date: date, isPending: false, transactionCode: transactionCode))
                     }
                 }
                 completion(transactions, nil)
diff --git a/SynchronyFinancial/SynchronyFinancial WatchKit Extension/MainMenuInterfaceController.swift b/SynchronyFinancial/SynchronyFinancial WatchKit Extension/MainMenuInterfaceController.swift
index f3cbec5..61a2177 100644
--- a/SynchronyFinancial/SynchronyFinancial WatchKit Extension/MainMenuInterfaceController.swift	
+++ b/SynchronyFinancial/SynchronyFinancial WatchKit Extension/MainMenuInterfaceController.swift	
@@ -52,6 +52,7 @@ class MainMenuInterfaceController: WKInterfaceController {
             return acctDict
         } else if segueIdentifier == "toWellness" {
             // we will probably want to do some setup prior to presenting Wellness
+            return acctDict
         }
 
         return nil
diff --git a/SynchronyFinancial/SynchronyFinancial WatchKit Extension/UtilizationfaceController.swift b/SynchronyFinancial/SynchronyFinancial WatchKit Extension/UtilizationfaceController.swift
index e1c783c..d0c289b 100644
--- a/SynchronyFinancial/SynchronyFinancial WatchKit Extension/UtilizationfaceController.swift	
+++ b/SynchronyFinancial/SynchronyFinancial WatchKit Extension/UtilizationfaceController.swift	
@@ -17,9 +17,11 @@ class UtilizationInterfaceController: WKInterfaceController {
     @IBOutlet weak var image3: WKInterfaceImage!
     override func awake(withContext context: Any?) {
         super.awake(withContext: context)
+        
+        score = context as? Int
 
         // get plist data to generate utilization metric
-        var plistValues = PlistUtil.readPlist(for: .balance)
+        /*var plistValues = PlistUtil.readPlist(for: .balance)
         let balance = plistValues[0][0]
         let credit = plistValues[0][1] > 0 ? plistValues[0][1] : 1
         util = round((Double(balance)/Double(credit)) * 100)
@@ -37,7 +39,7 @@ class UtilizationInterfaceController: WKInterfaceController {
             score = Int(round((100 - util) * 0.6))
         default:
             score = 1
-        }
+        }*/
     }
     
     override func didAppear() {
diff --git a/SynchronyFinancial/SynchronyFinancial WatchKit Extension/WellnessInterfaceController.swift b/SynchronyFinancial/SynchronyFinancial WatchKit Extension/WellnessInterfaceController.swift
index b89001c..3be7420 100644
--- a/SynchronyFinancial/SynchronyFinancial WatchKit Extension/WellnessInterfaceController.swift	
+++ b/SynchronyFinancial/SynchronyFinancial WatchKit Extension/WellnessInterfaceController.swift	
@@ -11,13 +11,40 @@ import Foundation
 
 
 class WellnessInterfaceController: WKInterfaceController {
+    var accounts: [Account] = []
+    var transactions: [Transaction] = []
+    var transactionDict: [String: [Transaction]] = [:]
+    var wellnessScore: Int?
+    var accountScore: Int = 0
+    var utilizationScore: Int = 0
+    var paymentScore: Int = 0
 
     @IBOutlet weak var image: WKInterfaceImage!
+    
     override func awake(withContext context: Any?) {
         super.awake(withContext: context)
         
+        if let context = context as? [String: [Account]], let accts = context["accts"] {
+            accounts = accts
+        }
+        /*for index in 0..<accounts.count{
+            FetchData.fetchTransactionsFor(accountAlias: accounts[index].accountAlias) { transactions, error in
+                guard error == nil else {
+                    NSLog("Error retrieving transactions for account.")
+                    return
+                }
+                self.transactionDict.updateValue(self.transactions, forKey: "transactions")
+            }
+        }*/
+        
+        calculateUtilization()
+        countAccounts()
+        calculateWellness()
+        
         image.setImageNamed("scores")
-        image.startAnimatingWithImages(in: NSRange(location: 1, length: 65), duration: 1.5, repeatCount: 1)
+        image.startAnimatingWithImages(in: NSRange(location: 1, length: wellnessScore ?? 0), duration: 1.5, repeatCount: 1)
+        
+        self.pushController(withName: "UtilizationScore", context: utilizationScore)
     }
 
     override func willActivate() {
@@ -29,4 +56,82 @@ class WellnessInterfaceController: WKInterfaceController {
         // This method is called when watch view controller is no longer visible
         super.didDeactivate()
     }
+    
+    private func calculateUtilization() {
+        var totalBalance: Double = 0
+        var totalCredit: Double = 0
+        
+        for index in 0..<accounts.count {
+            totalBalance += accounts[index].curBalance
+            totalCredit += accounts[index].creditLimit
+        }
+        
+        let util = round(((totalBalance)/(totalCredit)) * 100)
+        
+        switch util {
+        case 0...10:
+            utilizationScore = 100
+        case 11...20:
+            utilizationScore = Int(round((100 - util) * 0.9))
+        case 21...30:
+            utilizationScore = Int(round((100 - util) * 0.85))
+        case 31...60:
+            utilizationScore = Int(round((100 - util) * 0.725))
+        case 61...90:
+            utilizationScore = Int(round((100 - util) * 0.6))
+        default:
+            utilizationScore = 1
+        }
+    }
+    
+    private func countAccounts() {
+        var count: Int = 0
+        
+        for _ in 0..<accounts.count {
+            count += 1
+        }
+        
+        switch count {
+        case 2...3:
+            accountScore = 50
+        case 3...4:
+            accountScore = 75
+        case 5...10:
+            accountScore = 100
+        default:
+            accountScore = 25
+        }
+    }
+    
+    private func calculateWellness() {
+        let m1 = Double(utilizationScore) * 0.50
+        let m2 = Double(accountScore) * 0.20
+        let m3 = Double(paymentScore) * 0.30
+        wellnessScore = Int(round(m1 + m2 + m3))
+    }
+    
+    /*private func calculatePaymentsMissed() {
+        var lateCount: Int = 0
+        for index in 0..<accounts.count {
+            if let alias = selectedAccount?.accountAlias {
+                FetchData.fetchTransactionsFor(accountAlias: alias) { transactions, error in
+                    guard error == nil else {
+                        NSLog("Error retrieving transactions for account.")
+                        return
+                    }
+                
+                    self.transactionDict.updateValue(transactions, forKey: "transactions")
+                    self.pushController(withName: "Transactions", context: self.transactionDict)
+                }
+            }
+        }
+    }*/
+    
+    override func contextForSegue(withIdentifier segueIdentifier: String) -> Any? {
+        if segueIdentifier == "UtilizationScore" {
+            return utilizationScore
+        }
+        return nil
+    }
+    
 }
diff --git a/SynchronyFinancial/SynchronyFinancial/Transaction.swift b/SynchronyFinancial/SynchronyFinancial/Transaction.swift
index e24c2b3..975c3c7 100644
--- a/SynchronyFinancial/SynchronyFinancial/Transaction.swift
+++ b/SynchronyFinancial/SynchronyFinancial/Transaction.swift
@@ -26,9 +26,10 @@ class Transaction: NSObject {
     var paymentId: Int
     var isPending: Bool
     var isModifiable: Bool
+    var transactionCode: String
 
     init(type: TransactionType, amount: Double, merchantID: String, date: Date,
-         confirmationNum: String? = nil, paymentId: Int? = nil, isPending: Bool? = nil, isModifiable: Bool? = nil) {
+         confirmationNum: String? = nil, paymentId: Int? = nil, isPending: Bool? = nil, isModifiable: Bool? = nil, transactionCode: String) {
         self.type = type
         self.amount = amount
         self.merchantID = merchantID
@@ -37,5 +38,6 @@ class Transaction: NSObject {
         self.paymentId = paymentId ?? 0
         self.isPending = isPending ?? false
         self.isModifiable = isModifiable ?? false
+        self.transactionCode = transactionCode
     }
 }