#!/usr/bin/env python3
import argparse
import typing as t
import time
from pathlib import Path

import serial


def iter_frames(raw_data: bytes) -> t.Iterator[bytes]:
    FRAME_START = 0x02
    next_frame = b""
    # Discard first bytes
    raw_data = raw_data[raw_data.find(FRAME_START) :]

    for ch in raw_data:
        # Discard first bytes
        if ch == FRAME_START and next_frame:
            yield next_frame
            next_frame = b""
        next_frame += bytes([ch])


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("-p", "--port", default="/dev/ttyUSB1")
    parser.add_argument("-b", "--baud", type=int, default=1200)
    parser.add_argument(
        "--frame-gap", type=int, default=1500, help="Time between two frames (in ms)"
    )
    parser.add_argument(
        "recorded_data",
        help="File containing real recorded data to replay",
        type=Path,
    )
    args = parser.parse_args()

    data: bytes = b""
    with args.recorded_data.open("rb") as h:
        data = h.read()

    with serial.Serial(port=args.port, baudrate=args.baud) as ser:
        for frame in iter_frames(data):
            print(f"Feeding {len(frame)}b frame")
            start_time = time.monotonic()
            ser.write(frame)
            elapsed_time = time.monotonic() - start_time
            sleep_time = args.frame_gap / 1000 - elapsed_time
            if sleep_time > 0:
                time.sleep(sleep_time)


if __name__ == "__main__":
    main()