# Test removal of targets with flags == 0
import logging

logging.basicConfig(
    level=logging.DEBUG, format="%(asctime)s [%(levelname)s] %(name)s: %(message)s"
)

from target_simulator.gui.payload_router import DebugPayloadRouter
from target_simulator.analysis.simulation_state_hub import SimulationStateHub
import target_simulator.gui.payload_router as pr_mod


class FakeRisTarget:
    def __init__(self, flags, x, y, z, heading):
        self.flags = flags
        self.x = x
        self.y = y
        self.z = z
        self.heading = heading


class FakeScenario:
    def __init__(self, timetag=123456789):
        self.timetag = timetag


class FakeParsed:
    def __init__(self, tgt_list):
        self.tgt = type("T", (), {"tgt": tgt_list})()
        self.scenario = FakeScenario()

    @staticmethod
    def from_buffer_copy(payload):
        raise RuntimeError("Should be monkeypatched")


# first, create initial payload with 2 active targets
rads = [0.0, 1.5707964897155762]
fake_targets = []
for i, h in enumerate(rads):
    fake_targets.append(
        FakeRisTarget(flags=1, x=1000 + i, y=2000 + i, z=100.0, heading=h)
    )

original_parser = pr_mod.SfpRisStatusPayload


def make_parser(targets):
    return type(
        "P", (), {"from_buffer_copy": staticmethod(lambda payload: FakeParsed(targets))}
    )


# inject initial
pr_mod.SfpRisStatusPayload = make_parser(fake_targets)
hub = SimulationStateHub()
router = DebugPayloadRouter(simulation_hub=hub, update_queue=None)
router._handle_ris_status(b"INJ")
print("After first injection, hub ids:", hub.get_all_target_ids())

# now create a payload where target 1 is inactive (flags=0)
fake_targets2 = [
    FakeRisTarget(flags=1, x=1000, y=2000, z=100.0, heading=0.0),
    FakeRisTarget(flags=0, x=0, y=0, z=0, heading=0.0),
]
pr_mod.SfpRisStatusPayload = make_parser(fake_targets2)
router._handle_ris_status(b"INJ2")
print("After second injection (one inactive), hub ids:", hub.get_all_target_ids())

# restore
pr_mod.SfpRisStatusPayload = original_parser
print("Done")