Add Pretendo Support

api/networks.py

@@ -0,0 +1,24 @@
+from enum import IntEnum
+from api.public import nintendoBotFC, pretendoBotFC
+
+# Selectable networks
+class NetworkIDsToName(IntEnum):
+	nintendo = 0
+	pretendo = 1
+
+def getBotFriendCodeFromNetworkId(network:int):
+    match network:
+        case 0:
+            return nintendoBotFC
+        case 1:
+            return pretendoBotFC
+
+def nameToNetworkId(network:int):
+    if network == None:
+        network = 0
+    else:
+        try:
+            network = NetworkIDsToName[network].value
+        except:
+            network = 0
+    return network

api/public.py

@@ -1 +1,2 @@
-botFC = str(233790548638).zfill(12) # FC == Friendcode
+pretendoBotFC = str(220707964911).zfill(12) # FC == Friendcode
+nintendoBotFC = str(439944935945).zfill(12) # FC == Friendcode

api/template.private.py

@@ -1,20 +1,24 @@
-# Created by Deltaion Lee (MCMi460) on Github
+# Created by Deltaion Lee (MCMi460) & Preloading on Github
 # This is my private file, but with all of the dangerous stuff removed
 # You can use it as a template for yours
 
 SERIAL_NUMBER:str = "" # Serial number on console minus the last digit
 MAC_ADDRESS:str = "" # Console MAC address (see WiFi settings), all lowercase with no colons
-DEVICE_CERT:bytes = bytes.fromhex("") # Unique console certificate. Get from sniffing traffic
+DEVICE_CERT:bytes = bytes.fromhex("") # Unique console certificate. Get from sniffing traffic or from the LocalSeedFriendCode file
 DEVICE_NAME:str = "" # Doesn't matter
 
 # 3DS does NOT send NEX credentials over NASC
 # They are generated once when the account is created and stored on the device
 # Homebrew like https://github.com/Stary2001/nex-dissector/tree/master/get_3ds_pid_password
 # can be used to dump the PID and password
-PID:int = 0
-PID_HMAC:str = "" # Sniff console traffic or dump from friends title save (bytes 66-84)
+# You must redump these keys for each network.
+NINTENDO_PID:int = 0
+NINTENDO_NEX_PASSWORD:str = ""
 
-NEX_PASSWORD:str = ""
+PRETENDO_PID:int = 0
+PRETENDO_NEX_PASSWORD:str = ""
+
+PID_HMAC:str = "" # Sniff console traffic or dump from friends title save (bytes 66-84) somewhere im the file
 
 REGION:int = 1 # USA
 LANGUAGE:int = 1 # English

api/util.py

@@ -39,12 +39,12 @@ def getPath(path):
 
     return os.path.join(root, path)
 
-def startDBTime(time):
+def startDBTime(time, network):
     with sqlite3.connect('sqlite/fcLibrary.db') as con:
-    	cursor = con.cursor()
-    	cursor.execute('DELETE FROM config')
-    	cursor.execute('INSERT INTO config (BACKEND_UPTIME) VALUES (%s)' % (time,))
-    	con.commit()
+        cursor = con.cursor()
+        cursor.execute('DELETE FROM config WHERE network=' + str(network)) # doing this isn't the most intelegent of ideas but oh well (i hope you don't need to ever add another config :D)
+        cursor.execute('INSERT INTO config (BACKEND_UPTIME, NETWORK) VALUES (%s, %s)' % (time, network,))
+        con.commit()
 
 try:
     terminalSize = os.get_terminal_size(0).columns - 2

client/app.py

@@ -75,7 +75,7 @@ def assignVariables(self):
 
         self.loginButton.clicked.connect(self.changeState)
 
-        self.botFCLabel.setText('-'.join(botFC[i:i+4] for i in range(0, len(botFC), 4)))
+        self.botFCLabel.setText('-'.join(nintendoBotFC[i:i+4] for i in range(0, len(nintendoBotFC), 4)))
 
     def stylize(self):
         self.underLyingButton.click()

client/client.py

@@ -23,7 +23,7 @@
 ## The below contains 3dsrpc.com-specific information
 ## You will have to provide your own 'bot' FC if you are planning
 ## on running your own front and backend.
-convertFriendCodeToPrincipalId(botFC) # A quick verification check
+convertFriendCodeToPrincipalId(nintendoBotFC) # A quick verification check
 
 _REGION = typing.Literal['ALL', 'US', 'JP', 'GB', 'KR', 'TW']
 path = getAppPath()
@@ -132,7 +132,7 @@ def fetch(self):
             APIExcept(r)
         if r['Exception']:
             if 'not recognized' in r['Exception']['Error']:
-                print('%sRemember, the bot\'s friend code is as follows:\n%s%s' % (Color.YELLOW, '-'.join(botFC[i:i+4] for i in range(0, len(botFC), 4)), Color.DEFAULT))
+                print('%sRemember, the bot\'s friend code is as follows:\n%s%s' % (Color.YELLOW, '-'.join(nintendoBotFC[i:i+4] for i in range(0, len(nintendoBotFC), 4)), Color.DEFAULT))
             raise APIException(r['Exception'])
         return r
 
@@ -215,7 +215,7 @@ def main():
         else:
             raise Exception()
     except:
-        print('%sPlease take this time to add the bot\'s FC to your target 3DS\' friends list.\n%sBot FC: %s%s' % (Color.YELLOW, Color.DEFAULT, Color.BLUE, '-'.join(botFC[i:i+4] for i in range(0, 12, 4))))
+        print('%sPlease take this time to add the bot\'s FC to your target 3DS\' friends list.\n%sBot FC: %s%s' % (Color.YELLOW, Color.DEFAULT, Color.BLUE, '-'.join(nintendoBotFC[i:i+4] for i in range(0, 12, 4))))
         input('%s[Press enter to continue]%s' % (Color.GREEN, Color.DEFAULT))
         friendCode = input('Please enter your 3DS\' friend code\n> %s' % Color.PURPLE)
         config = {}

server/backend.py

@@ -4,11 +4,13 @@
 from nintendo import nasc
 from nintendo.nex import backend, friends, settings, streams
 from nintendo.nex import common
-import anyio, time, sqlite3, sys, traceback
+from enum import Enum
+import anyio, time, sqlite3, sys, traceback, argparse
 sys.path.append('../')
-from api.private import SERIAL_NUMBER, MAC_ADDRESS, DEVICE_CERT, DEVICE_NAME, REGION, LANGUAGE, PID, PID_HMAC, NEX_PASSWORD
+from api.private import SERIAL_NUMBER, MAC_ADDRESS, DEVICE_CERT, DEVICE_NAME, REGION, LANGUAGE, NINTENDO_PID, PRETENDO_PID, PID_HMAC, NINTENDO_NEX_PASSWORD, PRETENDO_NEX_PASSWORD
 from api import *
 from api.love2 import *
+from api.networks import NetworkIDsToName
 
 import logging
 logging.basicConfig(level=logging.INFO)
@@ -17,16 +19,16 @@
 since = 0
 quicker = 6
 begun = time.time()
-startDBTime(begun)
+
+network:int = 0
 
 async def main():
 	while True:
 		time.sleep(1)
 		print('Grabbing new friends...')
 		with sqlite3.connect('sqlite/fcLibrary.db') as con:
 			cursor = con.cursor()
-
-			cursor.execute('SELECT friendCode, lastAccessed FROM friends')
+			cursor.execute('SELECT friendCode, lastAccessed FROM ' + NetworkIDsToName(network).name + "_friends")
 			result = cursor.fetchall()
 			if not result:
 				continue
@@ -39,15 +41,34 @@ async def main():
 
 				try:
 					client = nasc.NASCClient()
-					client.set_title(0x0004013000003202, 20)
-					client.set_device(SERIAL_NUMBER, MAC_ADDRESS, DEVICE_CERT, DEVICE_NAME)
+					client.set_title(0x0004013000003202, 20) # to be honest, this should be seperate between networks, so if one eg, gets banned, you still get to keep the friend code for the other network, but i'm lazy.
 					client.set_locale(REGION, LANGUAGE)
-					client.set_user(PID, PID_HMAC)
 
+					if network == 0: # Nintendo Network
+						client.set_url("nasc.nintendowifi.net")
+						PID = NINTENDO_PID
+						NEX_PASSWORD = NINTENDO_NEX_PASSWORD
+
+					elif network == 1:
+						client.set_url("nasc.pretendo.cc")
+						client.context.set_authority(None)
+						PID = PRETENDO_PID
+						NEX_PASSWORD = PRETENDO_NEX_PASSWORD
+
+					else:
+						raise Exception(NetworkIDsToName(network).name + " is not a valid network")
+
+					client.set_device(SERIAL_NUMBER, MAC_ADDRESS, DEVICE_CERT, DEVICE_NAME)
+					client.set_user(PID , PID_HMAC)
+
 					response = await client.login(0x3200)
 
 					s = settings.load('friends')
 					s.configure("ridfebb9", 20000)
+
+					if network == 1: # If the app starts randomly hanging on Pretendo, try removing this, and the next line.
+						s["prudp.ping_timeout"] = 100000000000 # oh my god this is horrifying, but it makes it works, so who am i to care
+
 					async with backend.connect(s, response.host, response.port) as be:
 						async with be.login(str(PID), NEX_PASSWORD) as client:
 							friends_client = friends.FriendsClientV1(client)
@@ -64,50 +85,56 @@ async def main():
 							removables = await friends_client.get_all_friends()
 							for friend in removables:
 								time.sleep(delay / quicker)
-								await friends_client.remove_friend_by_principal_id(friend.unk1)
+								await friends_client.remove_friend_by_principal_id(friend.pid)
 							print('Removed %s friends' % str(len(removables)))
 
 							removeList = []
 							cleanUp = []
-							time.sleep(delay)
-							await friends_client.add_friend_by_principal_ids(0, rotation)
+							if network == 1:
+								for friend_pid in rotation:
+									time.sleep(delay / quicker)
+									await friends_client.add_friend_by_principal_id(0, friend_pid) # the add_friend_by_principal_ids hasn't been implemented yet on pretendo, so this is a fix for now. 
+							else:
+								time.sleep(delay)
+								await friends_client.add_friend_by_principal_ids(0, rotation)
+
 
 							time.sleep(delay)
 							t = await friends_client.get_all_friends()
 							if len(t) < len(rotation):
 								for ID in rotation:
-									if ID not in [ f.unk1 for f in t ]:
+									if ID not in [ f.pid for f in t ]:
 										removeList.append(ID)
 							x = t
 							t = []
 							for t1 in x:
-								if t1.unk1 in rotation:
+								if t1.pid in rotation:
 									t.append(t1)
 								else:
-									cleanUp.append(t1.unk1)
+									cleanUp.append(t1.pid)
 
 							for remover in removeList:
-								cursor.execute('DELETE FROM friends WHERE friendCode = ?', (str(convertPrincipalIdtoFriendCode(remover)).zfill(12),))
-								cursor.execute('DELETE FROM discordFriends WHERE friendCode = ?', (str(convertPrincipalIdtoFriendCode(remover)).zfill(12),))
+								cursor.execute('DELETE FROM ' + NetworkIDsToName(network).name + "_friends" + ' WHERE friendCode = ?', (str(convertPrincipalIdtoFriendCode(remover)).zfill(12),))
+								cursor.execute('DELETE FROM discordFriends WHERE friendCode = ? AND network = ?', (str(convertPrincipalIdtoFriendCode(remover)).zfill(12), str(network)))
 							con.commit()
 
 							if len(t) > 0:
 								time.sleep(delay)
-								f = await friends_client.get_friend_presence([ e.unk1 for e in t ])
+								f = await friends_client.get_friend_presence([ e.pid for e in t ])
 								users = []
 								for game in f:
 									# game.unk == principalId
-									users.append(game.unk)
+									users.append(game.pid)
 									#print(game.__dict__)
 									#print(game.presence.__dict__)
 									#print(game.presence.game_key.__dict__)
 									gameDescription = game.presence.game_mode_description
 									if not gameDescription: gameDescription = ''
 									joinable = bool(game.presence.join_availability_flag)
 
-									cursor.execute('UPDATE friends SET online = ?, titleID = ?, updID = ?, joinable = ?, gameDescription = ?, lastOnline = ? WHERE friendCode = ?', (True, game.presence.game_key.title_id, game.presence.game_key.title_version, joinable, gameDescription, time.time(), str(convertPrincipalIdtoFriendCode(users[-1])).zfill(12)))
+									cursor.execute('UPDATE ' + NetworkIDsToName(network).name + "_friends" + ' SET online = ?, titleID = ?, updID = ?, joinable = ?, gameDescription = ?, lastOnline = ? WHERE friendCode = ?', (True, game.presence.game_key.title_id, game.presence.game_key.title_version, joinable, gameDescription, time.time(), str(convertPrincipalIdtoFriendCode(users[-1])).zfill(12)))
 								for user in [ h for h in rotation if not h in users ]:
-									cursor.execute('UPDATE friends SET online = ?, titleID = ?, updID = ? WHERE friendCode = ?', (False, 0, 0, str(convertPrincipalIdtoFriendCode(user)).zfill(12)))
+									cursor.execute('UPDATE ' + NetworkIDsToName(network).name + "_friends" + ' SET online = ?, titleID = ?, updID = ? WHERE friendCode = ?', (False, 0, 0, str(convertPrincipalIdtoFriendCode(user)).zfill(12)))
 
 								con.commit()
 
@@ -120,16 +147,16 @@ async def main():
 								for ti in t:
 									work = False
 									for l in list_:
-										if l[0] == ti.unk1 and time.time() - l[1] <= 600:
+										if l[0] == ti.pid and time.time() - l[1] <= 600:
 											work = True
 									if not work:
 										continue
 
 									time.sleep(delay)
 
-									ti.unk2 = 0 # A cursed (but operable) 'hack'
+									ti.friend_code = 0 # A cursed (but operable) 'hack'
 									try:
-										j1 = await friends_client.get_friend_persistent_info([ti.unk1,])
+										j1 = await friends_client.get_friend_persistent_info([ti.pid,])
 									except:
 										continue
 									comment = j1[0].message
@@ -139,7 +166,7 @@ async def main():
 									if not comment.endswith(' '):
 										# Get user's mii + username from mii
 										m = await friends_client.get_friend_mii([ti,])
-										username = m[0].mii.unk1
+										username = m[0].mii.name
 										mii_data = m[0].mii.mii_data
 										obj = MiiData()
 										obj.decode(obj.convert(io.BytesIO(mii_data)))
@@ -149,7 +176,7 @@ async def main():
 										jeuFavori = j1[0].game_key.title_id
 									else:
 										comment = ''
-									cursor.execute('UPDATE friends SET username = ?, message = ?, mii = ?, jeuFavori = ? WHERE friendCode = ?', (username, comment, face, jeuFavori, str(convertPrincipalIdtoFriendCode(ti.unk1)).zfill(12)))
+									cursor.execute('UPDATE ' + NetworkIDsToName(network).name + "_friends" + ' SET username = ?, message = ?, mii = ?, jeuFavori = ? WHERE friendCode = ?', (username, comment, face, jeuFavori, str(convertPrincipalIdtoFriendCode(ti.pid)).zfill(12)))
 									con.commit()
 
 							for friend in rotation + cleanUp:
@@ -159,10 +186,21 @@ async def main():
 					print('An error occurred!\n%s' % e)
 					print(traceback.format_exc())
 					time.sleep(2)
-
+	
 if __name__ == '__main__':
 	try:
+		parser = argparse.ArgumentParser()
+		parser.add_argument('-n', '--network', choices=[member.name.lower() for member in NetworkIDsToName], required=True)
+		args = parser.parse_args()
+
+		network = NetworkIDsToName[args.network.lower()].value
+		if network == NetworkIDsToName.pretendo.value:
+			# Pretendo shouldn't care about a delay, like maybe nintendo? Since this was here just to prevent spamming Nintendo, we don't need it for pretendo. It will also make it faster.
+			# Maybe later it should just not have a delay for all networks?
+			delay, quicker = 0, 1
+		startDBTime(begun, network)
 		anyio.run(main)
 	except (KeyboardInterrupt, Exception) as e:
-		startDBTime(0)
-		print(e)
+		if network is not None:
+			startDBTime(0, network)
+		print(e)