tools/services/user_manager.py

   1 # Copyright 2021 Erfan Abdi
   2 # SPDX-License-Identifier: GPL-3.0-or-later
   3 import logging
   4 import os
   5 import threading
   6 import tools.config
   7 from tools.interfaces import IUserMonitor
   8 from tools.interfaces import IPlatform
   9
  10
  11 def start(args, unlocked_cb=None):
  12     def makeDesktopFile(appInfo):
  13         showApp = False
  14         for cat in appInfo["categories"]:
  15             if cat.strip() == "android.intent.category.LAUNCHER":
  16                 showApp = True
  17         if not showApp:
  18             return -1
  19
  20         packageName = appInfo["packageName"]
  21
  22         desktop_file_path = args.host_user + \
  23             "/.local/share/applications/waydroid." + packageName + ".desktop"
  24         if not os.path.exists(desktop_file_path):
  25             lines = ["[Desktop Entry]", "Type=Application"]
  26             lines.append("Name=" + appInfo["name"])
  27             lines.append("Exec=waydroid app launch " + packageName)
  28             lines.append("Icon=" + args.waydroid_data + "/icons/" + packageName + ".png")
  29             desktop_file = open(desktop_file_path, "w")
  30             for line in lines:
  31                 desktop_file.write(line + "\n")
  32             desktop_file.close()
  33             os.chmod(desktop_file_path, 0o755)
  34             return 0
  35
  36     def makeWaydroidDesktopFile(hide):
  37         desktop_file_path = args.host_user + \
  38             "/.local/share/applications/Waydroid.desktop"
  39         if os.path.isfile(desktop_file_path):
  40             os.remove(desktop_file_path)
  41         lines = ["[Desktop Entry]", "Type=Application"]
  42         lines.append("Name=Waydroid")
  43         lines.append("Exec=waydroid show-full-ui")
  44         if hide:
  45             lines.append("NoDisplay=true")
  46         lines.append("Icon=" + tools.config.tools_src + "/data/AppIcon.png")
  47         desktop_file = open(desktop_file_path, "w")
  48         for line in lines:
  49             desktop_file.write(line + "\n")
  50         desktop_file.close()
  51         os.chmod(desktop_file_path, 0o755)
  52
  53     def userUnlocked(uid):
  54         logging.info("Android with user {} is ready".format(uid))
  55         session_cfg = tools.config.load_session()
  56         args.waydroid_data = session_cfg["session"]["waydroid_data"]
  57         args.host_user = session_cfg["session"]["host_user"]
  58
  59         platformService = IPlatform.get_service(args)
  60         if platformService:
  61             apps_dir = "/.local/share/applications"
  62             if not os.path.exists(apps_dir):
  63                 os.mkdir(apps_dir)
  64                 os.chmod(apps_dir, 0o700)
  65             appsList = platformService.getAppsInfo()
  66             for app in appsList:
  67                 makeDesktopFile(app)
  68             multiwin = platformService.getprop("persist.waydroid.multi_windows", "false")
  69             if multiwin == "false":
  70                 makeWaydroidDesktopFile(False)
  71             else:
  72                 makeWaydroidDesktopFile(True)
  73         if unlocked_cb:
  74             unlocked_cb(args)
  75
  76     def packageStateChanged(mode, packageName, uid):
  77         platformService = IPlatform.get_service(args)
  78         if platformService:
  79             appInfo = platformService.getAppInfo(packageName)
  80             desktop_file_path = args.host_user + \
  81                 "/.local/share/applications/" + packageName + ".desktop"
  82             if mode == 0:
  83                 # Package added
  84                 makeDesktopFile(appInfo)
  85             elif mode == 1:
  86                 if os.path.isfile(desktop_file_path):
  87                     os.remove(desktop_file_path)
  88             else:
  89                 if os.path.isfile(desktop_file_path):
  90                     if makeDesktopFile(appInfo) == -1:
  91                         os.remove(desktop_file_path)
  92
  93     def service_thread():
  94         IUserMonitor.add_service(args, userUnlocked, packageStateChanged)
  95
  96     args.user_manager = threading.Thread(target=service_thread)
  97     args.user_manager.start()
  98
  99 def stop(args):
 100     try:
 101         if args.userMonitorLoop:
 102             args.userMonitorLoop.quit()
 103     except AttributeError:
 104         logging.debug("UserMonitor service is not even started")