from manta import Manta
from manta.io_core import IOCore
from manta.memory_core import MemoryCore
from manta.uart import UARTInterface
from amaranth import *
import tempfile
import os
import yaml

def test_io_core_dump():
    # Create some dummy signals to pass to the IO Core
    probe0 = Signal(1)
    probe1 = Signal(2)
    probe2 = Signal(3)
    probe3 = Signal(4, init=13)

    # Create Manta instance
    manta = Manta()
    manta.cores.test_core = IOCore(
        inputs = [probe0, probe1],
        outputs = [probe2, probe3]
    )

    # Create Temporary File
    tf = tempfile.NamedTemporaryFile(delete=False)
    tf.close()

    # Export Manta configuration
    manta.export_config(tf.name)

    # Parse the exported YAML
    with open(tf.name, "r") as f:
        data = yaml.safe_load(f)

    # Verify that exported YAML matches configuration
    expected = {
        "cores": {
            "test_core": {
            "type": "io",
                "inputs": {
                    "probe0": 1,
                    "probe1": 2
                },
                "outputs": {
                    "probe2": {
                        "width": 3,
                        "initial_value": 0
                    },
                    "probe3": {
                        "width": 4,
                        "initial_value": 13
                    }
                }
            }
        }
    }

    if data != expected:
        raise ValueError("Exported YAML does not match configuration!")

def test_memory_core_dump():
    # Create Manta instance
    manta = Manta()
    manta.cores.test_core = MemoryCore(
        mode = "bidirectional",
        width = 32,
        depth = 1024,
    )

    # Create Temporary File
    tf = tempfile.NamedTemporaryFile(delete=False)
    tf.close()

    # Export Manta configuration
    manta.export_config(tf.name)

    # Parse the exported YAML
    with open(tf.name, "r") as f:
        data = yaml.safe_load(f)

    # Verify that exported YAML matches configuration
    expected = {
        "cores": {
            "test_core": {
            "type": "memory",
            "mode":"bidirectional",
            "width":32,
            "depth":1024
            }
        }
    }

    if data != expected:
        raise ValueError("Exported YAML does not match configuration!")

def test_logic_analyzer_core_dump():
    raise ValueError

def test_uart_interface_dump():
    manta = Manta()
    manta.interface = UARTInterface(
        port = "/dev/ttyUSB0",
        baudrate = 115200,
        clock_freq = 100e6
    )

    # Create Temporary File
    tf = tempfile.NamedTemporaryFile(delete=False)
    tf.close()

    # Export Manta configuration
    manta.export_config(tf.name)

    # Parse the exported YAML
    with open(tf.name, "r") as f:
        data = yaml.safe_load(f)

    # Verify that exported YAML matches configuration
    expected = {
        "uart": {
            "port": "/dev/ttyUSB0",
            "baudrate": 115200,

            # Be careful with the float comparison here, copy-pasting from the
            # exported YAML seems to have the best results. Otherwise this test
            # will fail when it shouldn't.
            "clock_freq": 100000000.0,
            "chunk_size": 256
        }
    }

    if data != expected:
        raise ValueError("Exported YAML does not match configuration!")

def test_ethernet_interface_dump():
    raise ValueError