# SPDX-FileCopyrightText: 2022 Espressif Systems (Shanghai) CO LTD
# SPDX-License-Identifier: Unlicense OR CC0-1.0
import os
import re
import typing

import pexpect
import pytest
from pytest_embedded_idf import IdfDut

if typing.TYPE_CHECKING:
    from conftest import OpenOCD


def _test_idf_gdb(openocd_dut: 'OpenOCD', dut: IdfDut) -> None:
    # Need to wait a moment to connect via OpenOCD after the hard reset happened.
    # Along with this check that app runs ok
    dut.expect('Hello world!')

    # Don't need to have output from UART anymore
    dut.serial.stop_redirect_thread()

    gdb_env = os.environ.copy()
    gdb_env['ESP_IDF_GDB_TESTING'] = '1'

    with openocd_dut.run(), open(os.path.join(dut.logdir, 'gdb.txt'), 'w') as gdb_log, pexpect.spawn(
        f'idf.py -B {dut.app.binary_path} gdb --batch',
        env=gdb_env,
        timeout=60,
        logfile=gdb_log,
        encoding='utf-8',
        codec_errors='ignore',
    ) as p:
        p.expect(re.compile(r'add symbol table from file.*bootloader.elf'))
        p.expect(re.compile(r'add symbol table from file.*rom.elf'))
        p.expect_exact('hit Temporary breakpoint 1, app_main ()')


@pytest.mark.jtag
@pytest.mark.esp32
@pytest.mark.esp32s2
@pytest.mark.esp32c2
def test_idf_gdb(openocd_dut: 'OpenOCD', dut: IdfDut) -> None:
    _test_idf_gdb(openocd_dut, dut)


@pytest.mark.usb_serial_jtag
@pytest.mark.esp32c3
@pytest.mark.esp32s3
@pytest.mark.esp32c6
@pytest.mark.esp32h2
def test_idf_gdb_usj(openocd_dut: 'OpenOCD', dut: IdfDut) -> None:
    _test_idf_gdb(openocd_dut, dut)