Preserve app art when updating the app list

Limelight/Database/DataManager.h

@@ -18,6 +18,8 @@
 
 - (NSArray*) getHosts;
 - (void) updateHost:(TemporaryHost*)host;
+- (void) updateAppsForExistingHost:(TemporaryHost *)host;
+- (void) updateIconForExistingApp:(TemporaryApp*)app;
 - (void) removeHost:(TemporaryHost*)host;
 
 - (TemporarySettings*) getSettings;
@@ -25,7 +27,4 @@
 - (void) updateUniqueId:(NSString*)uniqueId;
 - (NSString*) getUniqueId;
 
-// Don't call this outside of database code
-- (App*) getAppForTemporaryApp:(TemporaryApp*)tempApp;
-
 @end

Limelight/Database/DataManager.m

@@ -68,7 +68,49 @@
         }
         
         // Push changes from the temp host to the persistent one
-        [host propagateChangesToParent:parent withDm:self];
+        [host propagateChangesToParent:parent];
+        
+        [self saveData];
+    }];
+}
+
+- (void) updateAppsForExistingHost:(TemporaryHost *)host {
+    [_managedObjectContext performBlockAndWait:^{
+        Host* parent = [self getHostForTemporaryHost:host];
+        if (parent == nil) {
+            // The host must exist to be updated
+            return;
+        }
+        
+        NSMutableSet *applist = [[NSMutableSet alloc] init];
+        for (TemporaryApp* app in host.appList) {
+            // Add a new persistent managed object if one doesn't exist
+            App* parentApp = [self getAppForTemporaryApp:app];
+            if (parentApp == nil) {
+                NSEntityDescription* entity = [NSEntityDescription entityForName:@"App" inManagedObjectContext:_managedObjectContext];
+                parentApp = [[App alloc] initWithEntity:entity insertIntoManagedObjectContext:_managedObjectContext];
+            }
+            
+            [app propagateChangesToParent:parentApp withHost:parent];
+            
+            [applist addObject:parentApp];
+        }
+        
+        parent.appList = applist;
+        
+        [self saveData];
+    }];
+}
+
+- (void) updateIconForExistingApp:(TemporaryApp*)app {
+    [_managedObjectContext performBlockAndWait:^{
+        App* parentApp = [self getAppForTemporaryApp:app];
+        if (parentApp == nil) {
+            // The app must exist to be updated
+            return;
+        }
+        
+        parentApp.image = app.image;
         
         [self saveData];
     }];

Limelight/Database/TemporaryApp.m

@@ -24,7 +24,6 @@
 
 - (void) propagateChangesToParent:(App*)parent withHost:(Host*)host {
     parent.id = self.id;
-    parent.image = self.image;
     parent.name = self.name;
     parent.host = host;
 }

Limelight/Database/TemporaryHost.h

@@ -10,8 +10,6 @@
 #import "Utils.h"
 #import "Host.h"
 
-@class DataManager;
-
 @interface TemporaryHost : NSObject
 
 @property (nonatomic) BOOL online;
@@ -26,13 +24,13 @@ NS_ASSUME_NONNULL_BEGIN
 @property (nonatomic, retain) NSString *mac;
 @property (nonatomic, retain) NSString *name;
 @property (nonatomic, retain) NSString *uuid;
-@property (nonatomic, retain) NSSet *appList;
+@property (nonatomic, retain) NSMutableSet *appList;
 
 - (id) initFromHost:(Host*)host;
 
 - (NSComparisonResult)compareName:(TemporaryHost *)other;
 
-- (void) propagateChangesToParent:(Host*)host withDm:(DataManager*)dm;
+- (void) propagateChangesToParent:(Host*)host;
 
 NS_ASSUME_NONNULL_END
 

Limelight/Database/TemporaryHost.m

@@ -37,7 +37,7 @@
     return self;
 }
 
-- (void) propagateChangesToParent:(Host*)parentHost withDm:(DataManager*)dm {
+- (void) propagateChangesToParent:(Host*)parentHost {
     parentHost.address = self.address;
     parentHost.externalAddress = self.externalAddress;
     parentHost.localAddress = self.localAddress;
@@ -45,22 +45,6 @@
     parentHost.name = self.name;
     parentHost.uuid = self.uuid;
     parentHost.pairState = [NSNumber numberWithInt:self.pairState];
-    
-    NSMutableSet *applist = [[NSMutableSet alloc] init];
-    for (TemporaryApp* app in self.appList) {
-        // Add a new persistent managed object if one doesn't exist
-        App* parentApp = [dm getAppForTemporaryApp:app];
-        if (parentApp == nil) {
-            NSEntityDescription* entity = [NSEntityDescription entityForName:@"App" inManagedObjectContext:parentHost.managedObjectContext];
-            parentApp = [[App alloc] initWithEntity:entity insertIntoManagedObjectContext:parentHost.managedObjectContext];
-        }
-        
-        [app propagateChangesToParent:parentApp withHost:parentHost];
-        
-        [applist addObject:parentApp];
-    }
-    
-    parentHost.appList = applist;
 }
 
 - (NSComparisonResult)compareName:(TemporaryHost *)other {