aboutsummarylogtreecommitdiffstats
path: root/main.py
blob: 54dceb09b8fe6b79f494a530d73dabebbde8492c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
from ftplib import FTP
from pypresence import Presence
from hashlib import sha1 
import hmac
import requests
import re
import json
import time
AppID = "1474289681125605376"






def CreateConfig():
    import subprocess
    from pathlib import Path
    config_dir = Path.home() / ".config" / "ps4-discord"
    config_file = config_dir / "Config.json"

    config_dir.mkdir(parents=True, exist_ok=True)

    if config_file.exists():
        print("Config exists")
        return

    print(f"Config will be created in {config_file}")

    print("""
Turn on PS4
Enable FTP / GoldHEN

Enter PS4 IP (example 192.168.0.13)
""")

    while True:
        ftp = FTP()
        ip = input("IP: ")

        try:
            print("Testing PS4 IP ...")
            ftp.connect(host=ip, port=2121)
            ftp.quit()
            print("PS4 was found")
            break
        except:
            subprocess.run("clear")
            print("Wrong IP or PS4 not reachable")

    config_path = Path.home() / ".config" / "ps4-discord" / "Config.json"
    config_path.parent.mkdir(parents=True, exist_ok=True)

    with open(config_path, "w") as file:
        json.dump({"IP": ip}, file)

    print(f"Config saved in {config_file}")


def ReadConfig():
    import json
    from pathlib import Path
    config_path = Path.home() / ".config" / "ps4-discord" / "Config.json"

    try:
        with open(config_path, "r") as file:
            data = json.load(file)
        return data["IP"]

    except FileNotFoundError:
        print(f"Config not found at {config_path}")
        quit()





#Get Game Inforamtion Such As Cusa ID like CUSA17419 - Persona 5 Royal Using FTP
def get_Game_Info(type,ip):
    #Initialize FTP
    ftp = FTP()
    respond = []
    #Trying Connect to PS4 using FTP
    try:
        #Connecting 
        ftp.connect(host=ip, port=2121)
        #Going to mnt/sandbox
        ftp.cwd("mnt/sandbox")
        #Check a mnt/sandbox Because this Directory Store Temp File of Currently Running apps
        ftp.dir(respond.append)
    except:
        # If IP of console is wrong or it is offline we return NoN means Not Online
        print(f"PS4 Not Found by ip {ip}")
        return("NoN")
    #check For all CUSA ID's
    respond = re.findall("CUSA.*?_000",str(respond))
    #Close Connection with PS4
    ftp.quit()
    #If Sandbox folder Don't Have Any CUSA Folders means player in menu 
    if respond == "[]":
        respond = "Menu"
        return "Menu"
    else:
        respond = str(respond).replace("_000']","")
        respond = str(respond).replace("['","")
    id = f"{respond}_00"
    id = str(id)
    if id == "[]_00":
        return "Menu"
    #using Sony server for checking with game name corespond to CUSA id also we get a Cover for a game 
    #creating key for access Sony API
    tmdb_key = bytearray.fromhex('F5DE66D2680E255B2DF79E74F890EBF349262F618BCAE2A9ACCDEE5156CE8DF2CDF2D48C71173CDC2594465B87405D197CF1AED3B7E9671EEB56CA6753C2E6B0')
    id_hash = hmac.new(tmdb_key, bytes(id, "utf-8"), sha1)
    id_hash = id_hash.hexdigest().upper()
    url = f"http://tmdb.np.dl.playstation.net/tmdb2/{id}_{id_hash}/{id}.json"
   
    #making a request to website with key 
    rq = requests.get(url)
    rq = str(rq.json())
    icon = re.findall("http://gs2-sec.ww.prod.dl.playstation.net/gs2-sec/appkgo/prod/.*?_00/.*?/i_.*?/.*?/icon0.png",rq)
    icon = str(icon).replace("']","")
    icon = icon.replace("['","")
    name = re.findall("'name': '.*?'",rq)
    name = str(name[0]).replace("'name': '","")
    name = name.replace("'","")
    #Return Options 
    #Return Cover
    if type == "icon":
        print(icon)
        return str(icon)
    #Return Name
    elif type == "name":
        return str(name)
    #Return Menu if any of this opeions is not selected
    else:
        return "Menu"
stat = True

#check Does discord is currently running
def discord_detect():
    import subprocess
    return bool(subprocess.call(
        ["pgrep", "-x", "Discord"],
        stdout=subprocess.DEVNULL,
        stderr=subprocess.DEVNULL
    ) == 0)
    
    

#initialize Discord Presence
Discord = Presence(AppID)

CreateConfig()
ip = ReadConfig()

while True:
    #if Discord is Running we try to show a Game
    if discord_detect() == True:
        #if Game info not NoN and Stat is True we connect to Discord and Display a Game
        if stat == True and get_Game_Info("name",ip) != "NoN":
            try:
                #connecting
                print("Connecting to Discord")
                Discord.connect()
                #Updating
                print("Updating Game")
                Discord.update(pid=0,small_text="TEST",large_image=f"{get_Game_Info("icon",ip)}",details=f"Playing {get_Game_Info("name",ip)}",state="on PlayStation 4",name=f"{get_Game_Info("name",ip)}",large_text=f"{get_Game_Info("name",ip)}")
                print("Game Updated")
                #puting stat to false means we connected to discord and we dont need connect to it again
                stat = False
            except:
                time.sleep(10)    
        #if we alredy connect to discord and gameinfo not Non, Change it     
        elif stat == False and get_Game_Info("name",ip) != "NoN":
            try:
                #clar old one
                print("Cleaing Old Game")
                Discord.clear(pid=0)
                #update with new one
                print("Updating game")
                Discord.update(pid=0,small_text="TEST",large_image=f"{get_Game_Info("icon",ip)}",details=f"Playing {get_Game_Info("name",ip)}",state="on PlayStation 4",name=f"{get_Game_Info("name",ip)}",large_text=f"{get_Game_Info("name",ip)}")
                print("Game Updated")
                #time oout for 60 sec
                time.sleep(60)
            except:
                time.sleep(10) 
        #if get_Game_Info is NonN we try ti disconnect from Discord and clear game name
        elif get_Game_Info("name",ip) == "NoN":
            print("PlayStation was not found Trying again in 30s")
            try:
                #clear game name 
                Discord.clear(0)
                #disconnect from discord
                Discord.close()
                if stat != True:stat = True
            except:
                if stat != True:stat = True
            #time out for 30sec
            time.sleep(30)
    #if Discord is not found we put app into sleep mode 
    else:
        print("Discord is Not Found trying again after 20s")
        if stat != True:stat = True
        time.sleep(20)