From a3682e4ce5533e0d6c60a1b8ab64af90c8bbe61c Mon Sep 17 00:00:00 2001
From: Alan Maynard <alan.maynard@uconn.edu>
Date: Thu, 28 Mar 2019 14:47:55 -0400
Subject: [PATCH] SWF-21 basic paradigm for passing account data to root
 interface controller

---
 .../Base.lproj/Interface.storyboard           |  4 +-
 .../ExtensionDelegate.swift                   | 18 ++++++++-
 .../MainMenuInterfaceController.swift         | 39 ++++++++++++-------
 3 files changed, 43 insertions(+), 18 deletions(-)

diff --git a/SynchronyFinancial/SynchronyFinancial WatchKit App/Base.lproj/Interface.storyboard b/SynchronyFinancial/SynchronyFinancial WatchKit App/Base.lproj/Interface.storyboard
index 9ca3cbb..0dfc536 100644
--- a/SynchronyFinancial/SynchronyFinancial WatchKit App/Base.lproj/Interface.storyboard	
+++ b/SynchronyFinancial/SynchronyFinancial WatchKit App/Base.lproj/Interface.storyboard	
@@ -9,10 +9,10 @@
         <plugIn identifier="com.apple.InterfaceBuilder.IBWatchKitPlugin" version="14490.20"/>
     </dependencies>
     <scenes>
-        <!--Main Menu Interface Controller-->
+        <!--root-->
         <scene sceneID="grn-eT-Mhz">
             <objects>
-                <controller fullScreen="YES" id="Qge-pO-HLc" customClass="MainMenuInterfaceController" customModule="SynchronyFinancial_WatchKit_App" customModuleProvider="target">
+                <controller identifier="root" fullScreen="YES" id="Qge-pO-HLc" customClass="MainMenuInterfaceController" customModule="SynchronyFinancial_WatchKit_App" customModuleProvider="target">
                     <items>
                         <button alignment="center" id="8yj-m6-1L9">
                             <group key="contentGroup" width="1" height="0.5" alignment="left" contentMode="scaleAspectFit" layout="vertical" id="SBG-wc-YDJ">
diff --git a/SynchronyFinancial/SynchronyFinancial WatchKit Extension/ExtensionDelegate.swift b/SynchronyFinancial/SynchronyFinancial WatchKit Extension/ExtensionDelegate.swift
index 9d9c78e..599043c 100644
--- a/SynchronyFinancial/SynchronyFinancial WatchKit Extension/ExtensionDelegate.swift	
+++ b/SynchronyFinancial/SynchronyFinancial WatchKit Extension/ExtensionDelegate.swift	
@@ -9,9 +9,26 @@
 import WatchKit
 
 class ExtensionDelegate: NSObject, WKExtensionDelegate {
+    var initialDict: [String: [Account]] = [:]
 
     func applicationDidFinishLaunching() {
         // Perform any final initialization of your application.
+        FetchData.attemptLogin { result, error in
+            switch result {
+            case true:
+                FetchData.getAccountInfo { accounts, error in
+                    guard error == nil else {
+                        NSLog("Account Retrieval Error: %s", error?.localizedDescription ?? "")
+                        return
+                    }
+
+                    self.initialDict.updateValue(accounts, forKey: "accts")
+                    WKInterfaceController.reloadRootControllers(withNames: ["root"], contexts: [self.initialDict])
+                }
+            case false:
+                NSLog("Login Error: %s", error?.localizedDescription ?? "Unknown Error")
+            }
+        }
     }
 
     func applicationDidBecomeActive() {
@@ -56,5 +73,4 @@ class ExtensionDelegate: NSObject, WKExtensionDelegate {
             }
         }
     }
-
 }
diff --git a/SynchronyFinancial/SynchronyFinancial WatchKit Extension/MainMenuInterfaceController.swift b/SynchronyFinancial/SynchronyFinancial WatchKit Extension/MainMenuInterfaceController.swift
index 28133a0..646e1af 100644
--- a/SynchronyFinancial/SynchronyFinancial WatchKit Extension/MainMenuInterfaceController.swift	
+++ b/SynchronyFinancial/SynchronyFinancial WatchKit Extension/MainMenuInterfaceController.swift	
@@ -20,22 +20,31 @@ class MainMenuInterfaceController: WKInterfaceController {
     override func awake(withContext context: Any?) {
         super.awake(withContext: context)
 
-        // We should find a better place for this call, just for testing purposes
-        FetchData.attemptLogin { result, error in
-            switch result {
-            case true:
-                FetchData.getAccountInfo { accounts, error in
-                    guard error == nil else {
-                        NSLog("Account Retrieval Error: %s", error?.localizedDescription ?? "")
-                        return
-                    }
-
-                    self.acctDict["accts"] = accounts
-                }
-            case false:
-                NSLog("Login Error: %s", error?.localizedDescription ?? "Unknown Error")
-            }
+        guard let initial = context as? [String: [Account]], let accts = initial["accts"] else {
+            presentAlert(withTitle: "Loading Accounts", message: nil, preferredStyle: .alert, actions: [WKAlertAction(title: "Cancel", style: .cancel, handler: {
+                    self.dismiss()
+                })])
+            NSLog("Error getting context from WKExtensionDelegate")
+            return
         }
+
+        acctDict.updateValue(accts, forKey: "accts")
+        // We should find a better place for this call, just for testing purposes
+//        FetchData.attemptLogin { result, error in
+//            switch result {
+//            case true:
+//                FetchData.getAccountInfo { accounts, error in
+//                    guard error == nil else {
+//                        NSLog("Account Retrieval Error: %s", error?.localizedDescription ?? "")
+//                        return
+//                    }
+//
+//                    self.acctDict["accts"] = accounts
+//                }
+//            case false:
+//                NSLog("Login Error: %s", error?.localizedDescription ?? "Unknown Error")
+//            }
+//        }
         //just here for testing. Should be moved into the accounts controller when that branch is merged
 //        FetchData.getAccountInfo()