mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-11-06 20:43:46 +08:00
firestar5683
8f132013d2
Expose map operations and status through Galaxy presentation endpoints. Co-authored-by: Danny <dannyirwin31@yahoo.com> Co-authored-by: Prabhaav Pillai <prabhaav@msu.edu> Co-authored-by: dirwin31 <83434411+dirwin31@users.noreply.github.com>
75 lines
3.3 KiB
Python
75 lines
3.3 KiB
Python
"""Authenticated HTTP projection of the single parked map-operation owner.
|
|
|
|
This adapter owns no processes or map files. Download authority and lifecycle
|
|
stay in the manager child, shared by native UI and Galaxy.
|
|
"""
|
|
|
|
import re
|
|
|
|
|
|
class MapOperationError(Exception):
|
|
def __init__(self, status: int, code: str):
|
|
super().__init__(code)
|
|
self.status = status
|
|
self.code = code
|
|
|
|
|
|
def validate_action(operation: str, payload: object) -> dict:
|
|
if type(payload) is not dict:
|
|
raise ValueError('Invalid map operation')
|
|
if any(type(key) is not str for key in payload):
|
|
raise ValueError('Invalid map operation')
|
|
fields = {key: value for key, value in payload.items() if type(key) is str}
|
|
if operation == 'start':
|
|
region = fields.get('regionToken')
|
|
generation = fields.get('expectedCurrentGeneration')
|
|
transfer = fields.get('maxTransferBytes')
|
|
disk = fields.get('maxNewDiskBytes')
|
|
if (set(fields) != {'regionToken', 'maxTransferBytes', 'maxNewDiskBytes', 'expectedCurrentGeneration'} or
|
|
type(region) is not str or not 1 <= len(region) <= 128 or
|
|
type(generation) is not str or re.fullmatch(r'(?:[0-9a-f]{64})?', generation) is None or
|
|
type(transfer) is not int or not 0 < transfer <= 8 << 30 or
|
|
type(disk) is not int or not 0 < disk <= 16 << 30):
|
|
raise ValueError('Invalid map start')
|
|
elif operation == 'cancel':
|
|
operation_id = fields.get('operationId')
|
|
if (set(fields) != {'operationId'} or type(operation_id) is not str or
|
|
re.fullmatch(r'[a-zA-Z0-9_:-]{1,80}', operation_id) is None):
|
|
raise ValueError('Invalid map cancellation')
|
|
else:
|
|
raise ValueError('Invalid map action')
|
|
return {'version': 1, 'op': operation, **fields}
|
|
|
|
|
|
class MapOperations:
|
|
def __init__(self, request=None):
|
|
self._request = request
|
|
|
|
def request(self, operation: str, payload: dict | None = None) -> dict:
|
|
if operation in ('catalog', 'status', 'setup') and payload is None:
|
|
command = {'version': 1, 'op': operation}
|
|
else:
|
|
command = validate_action(operation, payload)
|
|
try:
|
|
if self._request is None:
|
|
from openpilot.starpilot.maps.operation_owner import request_operation
|
|
self._request = request_operation
|
|
response = self._request(command)
|
|
if (type(response) is not dict or type(response.get('version')) is not int or response['version'] != 1 or
|
|
type(response.get('ok')) is not bool):
|
|
raise ValueError('Invalid owner response')
|
|
if response['ok']:
|
|
if set(response) != {'version', 'ok', 'result'} or type(response['result']) is not dict:
|
|
raise ValueError('Invalid owner result')
|
|
return response['result']
|
|
owner_error = response.get('error')
|
|
if (set(response) != {'version', 'ok', 'error'} or type(owner_error) is not dict or
|
|
type(owner_error.get('code')) is not str or re.fullmatch(r'[a-z_]{1,64}', owner_error['code']) is None):
|
|
raise ValueError('Invalid owner error')
|
|
except (OSError, RuntimeError, ValueError) as error:
|
|
raise MapOperationError(503, 'owner_unavailable') from error
|
|
code = owner_error['code']
|
|
status = 400 if code in ('invalid_request', 'invalid_region', 'invalid_budget') else (
|
|
409 if code in ('busy', 'selection_changed', 'not_parked', 'operation_changed') else 503)
|
|
raise MapOperationError(status, code)
|