From e445989ec73002d9210faf9cf4a881fe5aa49c20 Mon Sep 17 00:00:00 2001
From: Cameron Gutman <aicommander@gmail.com>
Date: Wed, 12 Aug 2020 22:26:56 -0700
Subject: [PATCH] Add direct-to-desktop mode

---
 .../ViewControllers/MainFrameViewController.m | 65 +++++++++++++++++--
 1 file changed, 58 insertions(+), 7 deletions(-)

diff --git a/Limelight/ViewControllers/MainFrameViewController.m b/Limelight/ViewControllers/MainFrameViewController.m
index 8e9e71f..50085fc 100644
--- a/Limelight/ViewControllers/MainFrameViewController.m
+++ b/Limelight/ViewControllers/MainFrameViewController.m
@@ -158,8 +158,15 @@ static NSMutableSet* hostList;
                 return;
             }
             
-            [self updateAppsForHost:host];
-            [self hideLoadingFrame: nil];
+            if ([self isHostDirectToDesktop:host]) {
+                [self hideLoadingFrame: nil];
+                [self prepareToStreamApp:[host.appList anyObject]];
+                [self performSegueWithIdentifier:@"createStreamFrame" sender:nil];
+            }
+            else {
+                [self updateAppsForHost:host];
+                [self hideLoadingFrame: nil];
+            }
         });
     }
     Log(LOG_I, @"Using cached app list: %d", usingCachedAppList);
@@ -199,10 +206,18 @@ static NSMutableSet* hostList;
                     return;
                 }
                 
-                [self updateAppsForHost:host];
-                [self->_appManager stopRetrieving];
-                [self->_appManager retrieveAssetsFromHost:host];
-                [self hideLoadingFrame: nil];
+                // We'll hit this case on the first load after pairing
+                if (!usingCachedAppList && [self isHostDirectToDesktop:host]) {
+                    [self hideLoadingFrame: nil];
+                    [self prepareToStreamApp:[host.appList anyObject]];
+                    [self performSegueWithIdentifier:@"createStreamFrame" sender:nil];
+                }
+                else {
+                    [self updateAppsForHost:host];
+                    [self->_appManager stopRetrieving];
+                    [self->_appManager retrieveAssetsFromHost:host];
+                    [self hideLoadingFrame: nil];
+                }
             });
         }
     });
@@ -866,7 +881,8 @@ static NSMutableSet* hostList;
     [self retrieveSavedHosts];
     _discMan = [[DiscoveryManager alloc] initWithHosts:[hostList allObjects] andCallback:self];
         
-    if ([hostList count] == 1) {
+    if ([hostList count] == 1 && [[hostList anyObject] appList].count > 0 && ![self isHostDirectToDesktop:[hostList anyObject]]) {
+        // Launch directly to the applist if this is the only host (and it's not a direct-to-desktop host)
         [self hostClicked:[hostList anyObject] view:nil];
     }
     else {
@@ -979,6 +995,26 @@ static NSMutableSet* hostList;
 {
     [super viewWillAppear:animated];
     
+    // If we're returning from a direct-to-desktop stream, show the host selection view
+    if (_selectedHost != nil && [self isHostDirectToDesktop:_selectedHost]) {
+        // _selectedHost will get wiped out by showHostSelectionView, so we need to save it
+        TemporaryHost* host = self->_selectedHost;
+        
+        // Quit the remote desktop stream
+        dispatch_async(dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0), ^{
+            HttpManager* hMan = [[HttpManager alloc] initWithHost:host.activeAddress uniqueId:self->_uniqueId serverCert:host.serverCert];
+            HttpResponse* quitResponse = [[HttpResponse alloc] init];
+            HttpRequest* quitRequest = [HttpRequest requestForResponse: quitResponse withUrlRequest:[hMan newQuitAppRequest]];
+            
+            // Exempt this host from discovery while handling the quit operation
+            [self->_discMan pauseDiscoveryForHost:host];
+            [hMan executeRequestSynchronously:quitRequest];
+            [self->_discMan resumeDiscoveryForHost:host];
+        });
+        
+        [self showHostSelectionView];
+    }
+    
     // We can get here on home press while streaming
     // since the stream view segues to us just before
     // entering the background. We can't check the app
@@ -1173,6 +1209,21 @@ static NSMutableSet* hostList;
     }
 }
 
+- (BOOL) isHostDirectToDesktop:(TemporaryHost*)host {
+    // To comply with App Store Guidelines 4.2.7, we will go directly to the stream
+    // like a remote desktop solution if the host is configured to support that.
+    if ([host.appList count] == 1) {
+        TemporaryApp* app = [host.appList anyObject];
+        
+        return [app.name isEqualToString:@"mstsc.exe"] ||
+               [app.name isEqualToString:@"Remote Desktop"] ||
+               [app.name isEqualToString:@"Desktop Stream"] ||
+               [app.name isEqualToString:@"Desktop"];
+    }
+    
+    return false;
+}
+
 - (void) updateAppsForHost:(TemporaryHost*)host {
     if (host != _selectedHost) {
         Log(LOG_W, @"Mismatched host during app update");