new signatures generation and live system

routes/config.json

@@ -5,5 +5,5 @@
 	"static-path": "/app/static",
 	"previews-path": "/app/static/previews",
 	"signatures-db": "/app/signatures.db",
-	"buildVersion": "1.0 build57"
+	"buildVersion": "1.0 build58"
 }

routes/helpers.py

@@ -41,16 +41,15 @@ def configFile():
     return config
 
 # Signatures check!!! Wow!!!
-def checkSignature(request, signature: str):
+def checkSignature(signature: str):
     config = configFile()
     db = EazySQLite3(config['signatures-db'])
-    query = db.query(f"SELECT * FROM `table` WHERE `key`=\"{signature}\" AND `endtime`>{time.time()}")
+    query = db.query(f"SELECT * FROM `table` WHERE `key` LIKE \"{signature}\" AND `endtime`>{time.time()}")
     if len(query['details']) == 0: return False
-    
-    ip = hashlib.md5(request.remote_addr.encode()).hexdigest()
-    creatorKey = hashlib.md5(f"IP={ip};DATE={time.strftime('%Y-%m-%d')}".encode()).hexdigest()
     for row in query['details']:
-        if creatorKey == row[2]: return True
+        if signature == row[0]:
+            db.query(f"UPDATE `table` SET `endtime`={round(time.time())+3600} WHERE `key`=\"{signature}\"")
+            return True
     return False
 
 # Hook for yt-dlp

routes/osuApi/findSong.py

@@ -7,7 +7,7 @@ def findSong(request):
         if request.method == 'POST': signature = request.form['signature']
         else: signature = request.args['signature']
     except: return {"status": "error", "details": { "error_code": 103, "error_details": "No signature" }}
-    if not helpers.checkSignature(request, signature): return {"status": "error", "details": { "error_code": 105, "error_details": "Invalid signature" }}
+    if not helpers.checkSignature(signature): return {"status": "error", "details": { "error_code": 105, "error_details": "Invalid signature" }}
     
     try:
         if request.method == 'POST': query = request.form['query']

routes/osuApi/getFull.py

@@ -9,7 +9,7 @@ def getFull(request):
         if request.method == 'POST': signature = request.form['signature']
         else: signature = request.args['signature']
     except: return {"status": "error", "details": { "error_code": 103, "error_details": "No signature" }}
-    if not helpers.checkSignature(request, signature): return {"status": "error", "details": { "error_code": 105, "error_details": "Invalid signature" }}
+    if not helpers.checkSignature(signature): return {"status": "error", "details": { "error_code": 105, "error_details": "Invalid signature" }}
     
     try:
         if request.method == 'POST': beatmapId = request.form['beatmapId']

routes/osuApi/getPreview.py

@@ -7,7 +7,7 @@ def getPreview(request):
         if request.method == 'POST': signature = request.form['signature']
         else: signature = request.args['signature']
     except: return {"status": "error", "details": { "error_code": 103, "error_details": "No signature" }}
-    if not helpers.checkSignature(request, signature): return {"status": "error", "details": { "error_code": 105, "error_details": "Invalid signature" }}
+    if not helpers.checkSignature(signature): return {"status": "error", "details": { "error_code": 105, "error_details": "Invalid signature" }}
     
     try:
         if request.method == 'POST': beatmapId = request.form['beatmapId']

routes/siteRoutes/sigGen.py

@@ -13,7 +13,7 @@ def signatureGen(request):
     text = f"--START SIGNATURE-- {userKey}{endtime} --END SIGNATURE--"
     salt = hashlib.md5("funapi-salt".encode()).hexdigest()[0:16]
     
-    creatorKey = hashlib.md5(f"USERKEY={userKey};DATE={time.strftime('%Y-%m-%d')}".encode()).hexdigest()
+    creatorKey = hashlib.md5(f"HOST-IP:{request.remote_addr};USERKEY={userKey};DATE={time.strftime('%Y-%m-%d')};".encode()).hexdigest()
     
     key = hashlib.blake2b(text.encode(), key=userKey.encode(), salt=salt.encode()).hexdigest().upper()
     

routes/witaiApi/recognizeVoice.py

@@ -13,7 +13,7 @@ def recognizeVoice(request):
         if request.method == 'POST': signature = request.form['signature']
         else: signature = request.args['signature']
     except: return {"status": "error", "details": { "error_code": 103, "error_details": "No signature" }}
-    if not helpers.checkSignature(request, signature): return {"status": "error", "details": { "error_code": 105, "error_details": "Invalid signature" }}
+    if not helpers.checkSignature(signature): return {"status": "error", "details": { "error_code": 105, "error_details": "Invalid signature" }}
     try:
         if request.method == 'POST': extendInfo = request.form['extendInfo']
         else: extendInfo = request.args['extendInfo']

routes/ytApi/get.py

@@ -7,7 +7,7 @@ def get(request, check = "huh"):
         if request.method == 'POST': signature = request.form['signature']
         else: signature = request.args['signature']
     except: return {"status": "error", "details": { "error_code": 103, "error_details": "No signature" }}
-    if not helpers.checkSignature(request, signature): return {"status": "error", "details": { "error_code": 105, "error_details": "Invalid signature" }}
+    if not helpers.checkSignature(signature): return {"status": "error", "details": { "error_code": 105, "error_details": "Invalid signature" }}
     
     try:
         if request.method == 'POST': url = request.form['url']

routes/ytApi/search.py

@@ -13,7 +13,7 @@ def search(request):
         if request.method == 'POST': signature = request.form['signature']
         else: signature = request.args['signature']
     except: return {"status": "error", "details": { "error_code": 103, "error_details": "No signature" }}
-    if not helpers.checkSignature(request, signature): return {"status": "error", "details": { "error_code": 105, "error_details": " signature" }}
+    if not helpers.checkSignature(signature): return {"status": "error", "details": { "error_code": 105, "error_details": " signature" }}
     html = urllib.request.urlopen("https://www.youtube.com/results?search_query={}".format(urllib.parse.quote_plus(searchQuery)))
     videoList = re.findall(r"watch\?v=(\S{11})", html.read().decode())
     videoIds = dict()