"""Developer placement reuses exact guarded signed Panda firmware choices."""

from dataclasses import replace
import json
import pytest
from openpilot.starpilot.galaxy.developer_settings import DeveloperSettingsOwner
from openpilot.starpilot.galaxy.settings import AuthorityContext, SettingsGateway, SettingsChanged
from openpilot.starpilot.galaxy.tests.test_panda_wake_settings import setup as setup, vehicle
from openpilot.starpilot.panda_wake import FirmwareChoice, snapshot, resolve, startup_choice
from openpilot.starpilot.schema_cache import put_cache
from openpilot.starpilot.ui.feature_settings_state import row_request
from openpilot.starpilot.vehicle_selection import encode


def developer(params, state, owner):
  return DeveloperSettingsOwner(params, lambda: state.parked, panda_wake=owner.panda_wake)


def ignition_row(params, state, owner):
  return next(row for row in developer(params, state, owner).snapshot().rows if row.key == 'IgnoreIgnitionLine')


def test_absent_choice_is_off_and_does_not_write_or_force_ignition(setup):
  params, state, owner, _ = setup
  row = ignition_row(params, state, owner)
  assert row.available and row.value == row.default_value == 'Off' and row.source is None
  assert 'restart' in row.reason.lower()
  assert resolve(state.cp, snapshot(params)) == FirmwareChoice()
  assert not params.get_bool('IgnoreIgnitionLine')


def test_explicit_on_off_is_existing_source_bound_panda_transaction(setup):
  params, state, owner, _ = setup
  for value in ('On', 'Off'):
    row = ignition_row(params, state, owner)
    request = replace(row_request(row, value), confirmation=True, vehicle_fingerprint=state.cp.carFingerprint)
    assert developer(params, state, owner).apply(request)
    assert params.get_bool('IgnoreIgnitionLine') == (value == 'On')
    assert resolve(state.cp, snapshot(params)) == FirmwareChoice(can_only=value == 'On')
    assert not developer(params, state, owner).apply(request)


@pytest.mark.parametrize('revocation', ('onroad', 'unconfirmed', 'firmware', 'selection', 'dependencies'))
def test_developer_ignition_preserves_existing_final_commit_guards(setup, revocation):
  params, state, owner, firmware = setup
  request = replace(row_request(ignition_row(params, state, owner), 'On'), confirmation=True, vehicle_fingerprint=state.cp.carFingerprint)
  if revocation == 'onroad':
    state.parked = False
  elif revocation == 'unconfirmed':
    request = replace(request, confirmation=False)
  elif revocation == 'firmware':
    (firmware / 'panda_h7_can_ignition_only.bin.signed').unlink()
  elif revocation == 'selection':
    params.put('VehicleSelection', json.loads(encode(None)), block=True)
  else:
    params.put_bool('RemoteStartBootsComma', True, block=True)
  assert not developer(params, state, owner).apply(request)
  assert not params.get_bool('IgnoreIgnitionLine')


@pytest.mark.parametrize('brand', ('hyundai', 'tesla'))
def test_developer_does_not_invent_unsupported_can_ignition_vehicle_contract(setup, brand):
  params, state, owner, _ = setup
  state.cp = vehicle(brand)
  params.put('VehicleSelection', json.loads(encode(state.cp.carFingerprint)), block=True)
  assert all(row.key != 'IgnoreIgnitionLine' for row in developer(params, state, owner).snapshot().rows)


def test_actual_gateway_developer_confirm_and_vehicle_page_move(setup, monkeypatch):
  params, state, _, firmware = setup
  from openpilot.starpilot.ui.panda_wake_feature import PandaWakeFeature

  original = PandaWakeFeature.__init__

  def init(feature, owner, *, galaxy):
    original(feature, owner, galaxy=galaxy, firmware_directory=firmware, device_type=lambda: 'tizi')

  monkeypatch.setattr(PandaWakeFeature, '__init__', init)

  class Context:
    def sample(self):
      return AuthorityContext(state.parked, state.cp, state.cp.as_reader().as_builder().to_bytes())

  gateway = SettingsGateway(params, Context(), clock=lambda: 100)
  page = gateway.page('developer', 'session', b'generation')
  index = next(i for i, row in enumerate(page['rows']) if row['label'] == 'Use CAN Ignition Only')
  intent = gateway.preview(page['view'], index, 0, 'session', b'generation', value='On')
  assert 'Panda firmware' in intent['question'] and 'restart' in intent['question']
  with pytest.raises(SettingsChanged):
    gateway.confirm(intent['intent'], 'session', b'generation', session_valid=lambda: False)
  assert not params.get_bool('IgnoreIgnitionLine')
  page = gateway.page('developer', 'session', b'generation')
  index = next(i for i, row in enumerate(page['rows']) if row['label'] == 'Use CAN Ignition Only')
  intent = gateway.preview(page['view'], index, 0, 'session', b'generation', value='On')
  assert gateway.confirm(intent['intent'], 'session', b'generation')
  assert params.get_bool('IgnoreIgnitionLine')
  assert all(row['label'] != 'Use CAN Ignition Only' for row in gateway.page('vehicle', 'session', b'generation')['rows'])


def test_startup_uses_exact_manual_registered_cache_and_auto_remains_stock(setup):
  params, state, _, _ = setup
  params.put_bool('IgnoreIgnitionLine', True, block=True)
  put_cache(params, 'CarParamsPersistent', state.cp.as_reader(), block=True)
  assert startup_choice(params) == FirmwareChoice(can_only=True)
  params.put('VehicleSelection', json.loads(encode(None)), block=True)
  assert startup_choice(params) == FirmwareChoice()
  assert params.get_bool('IgnoreIgnitionLine')