mirror of
https://github.com/norohind/FDEV-CAPI-Handler.git
synced 2025-06-01 16:11:22 +03:00
norohind
b6392f8d20
Because user can don't have a game so no nickname, but his FDEV tokens still suits for not CAPI requests (i.e. squadrons api)
85 lines
2.3 KiB
Python
85 lines
2.3 KiB
Python
import hashlib
|
|
import base64
|
|
from typing import Union
|
|
|
|
from EDMCLogging import get_main_logger
|
|
import requests
|
|
import config
|
|
|
|
logger = get_main_logger()
|
|
|
|
|
|
def get_tokens_request(code, code_verifier) -> requests.Response:
|
|
"""
|
|
Performs initial requesting access and refresh tokens
|
|
|
|
:param code:
|
|
:param code_verifier:
|
|
:return:
|
|
"""
|
|
token_request: requests.Response = requests.post(
|
|
url=config.TOKEN_URL,
|
|
headers=
|
|
{
|
|
'Content-Type': 'application/x-www-form-urlencoded',
|
|
'User-Agent': config.PROPER_USER_AGENT
|
|
},
|
|
data=f'redirect_uri={config.REDIRECT_URL}&'
|
|
f'code={code}&'
|
|
f'grant_type=authorization_code&'
|
|
f'code_verifier={code_verifier}&'
|
|
f'client_id={config.CLIENT_ID}')
|
|
|
|
return token_request
|
|
|
|
|
|
def get_nickname(access_token: str) -> Union[str, None]:
|
|
nickname = None
|
|
try:
|
|
nickname = requests.get(
|
|
url='https://companion.orerve.net/profile',
|
|
headers=
|
|
{
|
|
'Authorization': f'Bearer {access_token}',
|
|
'User-Agent': config.PROPER_USER_AGENT
|
|
}
|
|
).json()["commander"]["name"]
|
|
|
|
except Exception as e:
|
|
logger.warning(f"Couldn't get or set nickname", exc_info=e)
|
|
|
|
return nickname
|
|
|
|
|
|
def get_fid(access_token: str) -> str:
|
|
fid = None
|
|
try:
|
|
fid = requests.get(
|
|
url='https://auth.frontierstore.net/me',
|
|
headers={
|
|
'Authorization': f'Bearer {access_token}',
|
|
'User-Agent': config.PROPER_USER_AGENT
|
|
}
|
|
).json()["customer_id"]
|
|
|
|
except Exception as e:
|
|
logger.exception(f"Couldn't get FID", exc_info=e)
|
|
|
|
return fid
|
|
|
|
|
|
def refresh_request(refresh_token: str) -> requests.Response:
|
|
return requests.post(
|
|
url=config.TOKEN_URL,
|
|
headers={'Content-Type': 'application/x-www-form-urlencoded'},
|
|
data=f'grant_type=refresh_token&client_id={config.CLIENT_ID}&refresh_token={refresh_token}')
|
|
|
|
|
|
def generate_challenge(_verifier: bytes) -> str:
|
|
"""
|
|
It takes code verifier and return code challenge
|
|
:param _verifier:
|
|
:return:
|
|
"""
|
|
return base64.urlsafe_b64encode(hashlib.sha256(_verifier).digest())[:-1].decode('utf-8')
|