From b59de25bce54021d0713c54777bf35b06b60ed59 Mon Sep 17 00:00:00 2001
From: TabulateJarl8 <tabulatejarl8@gmail.com>
Date: Mon, 5 Apr 2021 17:45:33 -0400
Subject: [PATCH] Add Discord RPC to legendary

---
 legendary/__init__.py |  2 +-
 legendary/cli.py      | 45 +++++++++++++++++++++++++++++++++++++++++++
 setup.py              |  4 +++-
 3 files changed, 49 insertions(+), 2 deletions(-)

diff --git a/legendary/__init__.py b/legendary/__init__.py
index 1760b2a..55a621f 100644
--- a/legendary/__init__.py
+++ b/legendary/__init__.py
@@ -1,4 +1,4 @@
 """Legendary!"""
 
-__version__ = '0.20.6'
+__version__ = '0.20.7'
 __codename__ = 'A Red Letter Day'
diff --git a/legendary/cli.py b/legendary/cli.py
index 9c3f613..0bc6b7f 100644
--- a/legendary/cli.py
+++ b/legendary/cli.py
@@ -10,11 +10,17 @@ import shlex
 import subprocess
 import time
 import webbrowser
+from pypresence import Presence
+import platform
+# import threading
+import psutil
+# import asyncio
 
 from distutils.util import strtobool
 from getpass import getuser
 from logging.handlers import QueueListener
 from multiprocessing import freeze_support, Queue as MPQueue
+from multiprocessing import Process
 from sys import exit, stdout
 
 from legendary import __version__, __codename__
@@ -510,8 +516,47 @@ class LegendaryCLI:
             logger.debug(f'Working directory: {cwd}')
             if env:
                 logger.debug('Environment overrides:', env)
+
+            rpc = Process(target=self.startRPC, args=(app_name, params))
+            rpc.start()
+
             subprocess.Popen(params, cwd=cwd, env=env)
 
+    def startRPC(self, app_name, params):
+        try:
+            RPC = Presence('828711025863688192')
+            RPC.connect()
+
+            app_title = self.core.get_game(app_name).app_title
+
+            exe_name = None
+            for i in range(len(params)):
+                if params[i].endswith('.exe'):
+                    exe_name = os.path.basename(params[i])
+                    break
+
+            if exe_name is not None:
+                start = str(time.time()).split(".")[0]
+                RPC.update(large_image="legendarylogofull", large_text=app_title, state="via Legendary on " + platform.system(), details=app_title, start=start)
+                while True:
+                    game_running = False
+                    pids = psutil.pids()
+                    for i in range(len(pids)):
+                        try:
+                            p = psutil.Process(pids[i])
+                            if p.cmdline()[0].find(exe_name) != -1:
+                                # Game is running
+                                game_running = True
+                                break
+                        except Exception:
+                            pass
+                    if game_running is True:
+                        time.sleep(5)
+                    else:
+                        break
+        except Exception as e:
+            print(f'Warning: {e} on Discord RPC start')
+
     def install_game(self, args):
         if self.core.is_installed(args.app_name):
             igame = self.core.get_installed_game(args.app_name)
diff --git a/setup.py b/setup.py
index be343fd..8ebaddc 100644
--- a/setup.py
+++ b/setup.py
@@ -36,7 +36,9 @@ setup(
     install_requires=[
         'requests<3.0',
         'setuptools',
-        'wheel'
+        'wheel',
+        'pypresence',
+        'psutil'
     ],
     url='https://github.com/derrod/legendary',
     description='Free and open-source replacement for the Epic Games Launcher application',