SplatNet-Backend/spfn-main.py

77 lines
2.2 KiB
Python

from flask import Flask, request, redirect
import requests
import base64
app = Flask(__name__)
SPFN_API_URL = "https://account.spfn.net/api/v2/oauth2/generate_token"
def generate_spfn_token(encoded_creds):
headers = {
"Authorization": f"Basic {encoded_creds}",
"User-Agent": "saturday & allison should lwk kiss..."
}
response = requests.get(SPFN_API_URL, headers=headers, timeout=10)
if not response.ok:
return None
data = response.json()
return data.get("token")
@app.route('/api/v2/sso/spfn/generate_token', methods=['POST'])
def handle_login():
username = request.form.get('username')
password = request.form.get('password')
# fuck you login_host
login_host = request.referrer or "/"
if login_host.endswith('/'):
login_host = login_host[:-1]
creds = f"{username} {password}"
encoded_creds = base64.b64encode(creds.encode()).decode()
try:
token = generate_spfn_token(encoded_creds)
#if onika eat burgers then does oomi appear?
if token:
print("=== TOKEN ===")
print(token)
print("=============")
profile_headers = {
"Authorization": f"Bearer {token}",
"User-Agent": "maple still awake"
}
profile_response = requests.get(
"https://account.spfn.net/api/v2/users/@me/profile",
headers=profile_headers,
timeout=10
)
if profile_response.ok:
print("=== PROFILE ===")
print(profile_response.text)
print("===============")
else:
print("Profile request failed:", profile_response.status_code)
redirect_to = request.args.get('redirect', f"{login_host}/friend_list/")
return redirect(redirect_to)
else:
sep = "&" if "?" in login_host else "?"
return redirect(
f"{login_host}/users/auth/splatfestival/{sep}error=auth&username={username}"
)
except Exception as e:
return f"Proxy Error: {str(e)}", 500
if __name__ == '__main__':
app.run(port=5000)