mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-10-28 00:53:45 +08:00
firestar5683
b6c81875b0
Own Connect credentials, registration and upload preferences through shared access providers. Co-authored-by: Adeeb Shihadeh <adeebshihadeh@gmail.com> Co-authored-by: dirwin31 <83434411+dirwin31@users.noreply.github.com>
75 lines
2.7 KiB
Python
75 lines
2.7 KiB
Python
import jwt
|
|
import os
|
|
import requests
|
|
from datetime import datetime, timedelta, UTC
|
|
from openpilot.common.hardware.hw import Paths
|
|
from openpilot.common.version import get_version
|
|
|
|
from openpilot.starpilot.connect.provider import active_provider, konik_key_pair
|
|
|
|
API_HOST = os.getenv('API_HOST', 'https://api.commadotai.com')
|
|
|
|
# name: jwt signature algorithm
|
|
KEYS = {"id_rsa": "RS256",
|
|
"id_ecdsa": "ES256"}
|
|
|
|
|
|
class Api:
|
|
def __init__(self, dongle_id):
|
|
self.dongle_id = dongle_id
|
|
self.jwt_algorithm, self.private_key, _ = get_key_pair()
|
|
|
|
def get(self, *args, **kwargs):
|
|
return self.request('GET', *args, **kwargs)
|
|
|
|
def post(self, *args, **kwargs):
|
|
return self.request('POST', *args, **kwargs)
|
|
|
|
def request(self, method, endpoint, timeout=None, access_token=None, **params):
|
|
return api_get(endpoint, method=method, timeout=timeout, access_token=access_token, **params)
|
|
|
|
def get_token(self, payload_extra=None, expiry_hours=1):
|
|
if self.private_key is None:
|
|
raise RuntimeError("private key is not configured")
|
|
now = datetime.now(UTC).replace(tzinfo=None)
|
|
payload = {
|
|
'identity': self.dongle_id,
|
|
'nbf': now,
|
|
'iat': now,
|
|
'exp': now + timedelta(hours=expiry_hours)
|
|
}
|
|
if payload_extra is not None:
|
|
payload.update(payload_extra)
|
|
token = jwt.encode(payload, self.private_key, algorithm=self.jwt_algorithm)
|
|
if isinstance(token, bytes):
|
|
token = token.decode('utf8')
|
|
return token
|
|
|
|
|
|
def api_get(endpoint, method='GET', timeout=None, access_token=None, session=None, **params):
|
|
provider = active_provider()
|
|
if provider.name == 'offline' or os.getenv('STARPILOT_CLOUD_DISABLED') == '1':
|
|
raise RuntimeError('Cloud provider is unavailable')
|
|
headers = {}
|
|
if access_token is not None:
|
|
headers['Authorization'] = "JWT " + access_token
|
|
|
|
headers['User-Agent'] = "openpilot-" + get_version()
|
|
|
|
# TODO: add session to Api
|
|
req = requests if session is None else session
|
|
return req.request(method, (API_HOST if provider.name == "comma" else provider.api) + "/" + endpoint, timeout=timeout, headers=headers, params=params)
|
|
|
|
|
|
def get_key_pair() -> tuple[str, str, str] | tuple[None, None, None]:
|
|
provider = active_provider()
|
|
if provider.name == 'offline' or os.getenv('STARPILOT_CLOUD_DISABLED') == '1':
|
|
return None, None, None
|
|
if provider.name == 'konik':
|
|
return konik_key_pair()
|
|
for key in KEYS:
|
|
if os.path.isfile(Paths.persist_root() + f'/comma/{key}') and os.path.isfile(Paths.persist_root() + f'/comma/{key}.pub'):
|
|
with open(Paths.persist_root() + f'/comma/{key}') as private, open(Paths.persist_root() + f'/comma/{key}.pub') as public:
|
|
return KEYS[key], private.read(), public.read()
|
|
return None, None, None
|