From 551229c6b012d29224a4459ad46bcd97bbaa11a5 Mon Sep 17 00:00:00 2001
From: Erfan Abdi <erfangplus@gmail.com>
Date: Sat, 4 Sep 2021 07:15:10 +0430
Subject: [PATCH] actions: Start session by lunching apps

---
 tools/actions/app_manager.py     | 6 ++++--
 tools/actions/session_manager.py | 6 ++++--
 tools/services/user_manager.py   | 4 +++-
 3 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/tools/actions/app_manager.py b/tools/actions/app_manager.py
index fc5c224..46d0818 100644
--- a/tools/actions/app_manager.py
+++ b/tools/actions/app_manager.py
@@ -77,7 +77,8 @@ def launch(args):
             logging.error("WayDroid container is {}".format(
                 session_cfg["session"]["state"]))
     else:
-        logging.error("WayDroid session is stopped")
+        logging.error("Starting waydroid session")
+        tools.actions.session_manager.start(args, launch)
 
 def list(args):
     if os.path.exists(tools.config.session_defaults["config_path"]):
@@ -128,4 +129,5 @@ def showFullUI(args):
             logging.error("WayDroid container is {}".format(
                 session_cfg["session"]["state"]))
     else:
-        logging.error("WayDroid session is stopped")
+        logging.error("Starting waydroid session")
+        tools.actions.session_manager.start(args, showFullUI)
diff --git a/tools/actions/session_manager.py b/tools/actions/session_manager.py
index 6d24ca9..0262c1a 100644
--- a/tools/actions/session_manager.py
+++ b/tools/actions/session_manager.py
@@ -9,7 +9,7 @@ import tools.config
 from tools import services
 
 
-def start(args):
+def start(args, unlocked_cb=None):
     def signal_handler(sig, frame):
         stop(args)
         sys.exit(0)
@@ -38,8 +38,10 @@ def start(args):
         session_cfg = tools.config.load_session()
         if container_state != session_cfg["session"]["state"]:
             if session_cfg["session"]["state"] == "RUNNING":
-                services.user_manager.start(args)
+                services.user_manager.start(args, unlocked_cb)
                 services.clipboard_manager.start(args)
+                if unlocked_cb:
+                    unlocked_cb = None
             elif session_cfg["session"]["state"] == "STOPPED":
                 services.user_manager.stop(args)
                 services.clipboard_manager.stop(args)
diff --git a/tools/services/user_manager.py b/tools/services/user_manager.py
index 4e7455b..98b7d28 100644
--- a/tools/services/user_manager.py
+++ b/tools/services/user_manager.py
@@ -8,7 +8,7 @@ from tools.interfaces import IUserMonitor
 from tools.interfaces import IPlatform
 
 
-def start(args):
+def start(args, unlocked_cb=None):
     def makeDesktopFile(appInfo):
         showApp = False
         for cat in appInfo["categories"]:
@@ -66,6 +66,8 @@ def start(args):
                 makeWaydroidDesktopFile(False)
             else:
                 makeWaydroidDesktopFile(True)
+        if unlocked_cb:
+            unlocked_cb(args)
 
     def packageStateChanged(mode, packageName, uid):
         platformService = IPlatform.get_service(args)
-- 
2.47.3