import os
import signal
import subprocess

import pytest
from multiaddr import Multiaddr
import trio
from trio.lowlevel import open_process

from libp2p.crypto.secp256k1 import create_new_key_pair
from libp2p.custom_types import TProtocol
from libp2p.host.basic_host import BasicHost
from libp2p.network.swarm import Swarm
from libp2p.peer.id import ID
from libp2p.peer.peerinfo import PeerInfo
from libp2p.peer.peerstore import PeerStore
from libp2p.security.insecure.transport import InsecureTransport
from libp2p.stream_muxer.mplex.mplex import MPLEX_PROTOCOL_ID, Mplex
from libp2p.transport.upgrader import TransportUpgrader
from libp2p.transport.websocket.transport import WebsocketTransport

PLAINTEXT_PROTOCOL_ID = "/plaintext/1.0.0"


@pytest.mark.trio
async def test_ping_with_js_node():
    # Path to the JS node script
    js_node_dir = os.path.join(os.path.dirname(__file__), "js_libp2p", "js_node", "src")
    script_name = "ws_ping_node.mjs"

    # Launch the JS libp2p node (long-running)
    proc = await open_process(
        ["node", script_name],
        stdout=subprocess.PIPE,
        stderr=subprocess.DEVNULL,
        cwd=js_node_dir,
    )
    try:
        # Read first two lines (PeerID and multiaddr)
        buffer = b""
        with trio.fail_after(10):
            while buffer.count(b"\n") < 2:
                chunk = await proc.stdout.receive_some(1024)  # type: ignore
                if not chunk:
                    break
                buffer += chunk

        lines = buffer.decode().strip().split("\n")
        peer_id_line, addr_line = lines[0], lines[1]
        peer_id = ID.from_base58(peer_id_line)
        maddr = Multiaddr(addr_line)

        # Set up Python host
        key_pair = create_new_key_pair()
        py_peer_id = ID.from_pubkey(key_pair.public_key)
        peer_store = PeerStore()
        peer_store.add_key_pair(py_peer_id, key_pair)

        upgrader = TransportUpgrader(
            secure_transports_by_protocol={
                TProtocol(PLAINTEXT_PROTOCOL_ID): InsecureTransport(key_pair)
            },
            muxer_transports_by_protocol={TProtocol(MPLEX_PROTOCOL_ID): Mplex},
        )
        transport = WebsocketTransport()
        swarm = Swarm(py_peer_id, peer_store, upgrader, transport)
        host = BasicHost(swarm)

        # Connect to JS node
        peer_info = PeerInfo(peer_id, [maddr])
        await host.connect(peer_info)
        assert host.get_network().connections.get(peer_id) is not None
        await trio.sleep(0.1)

        # Ping protocol
        stream = await host.new_stream(peer_id, [TProtocol("/ipfs/ping/1.0.0")])
        await stream.write(b"ping")
        data = await stream.read(4)
        assert data == b"pong"

        # Cleanup
        await host.close()
    finally:
        proc.send_signal(signal.SIGTERM)
        await trio.sleep(0)