import asyncio
import json
import math
import random
import time
import websockets

# ---------------------------------------------------------
# Vibrational Substrate Simulation Engine — V-VIB-05
# ---------------------------------------------------------

class VibrationalEntity:
    def __init__(self, id, kind):
        self.id = id
        self.kind = kind
        self.state = {
            "depth": random.uniform(0.2, 0.9),
            "phase": random.uniform(0, math.pi * 2),
            "coupling": random.uniform(0.3, 0.9),
            "load": random.uniform(0.1, 0.8),
            "stress": "normal",
            "pulse": False
        }

    def update(self, t):
        # Wells drift slowly
        if "well" in self.kind.lower():
            self.state["depth"] = 0.5 + 0.4 * math.sin(t / 7 + hash(self.id) % 10)
            self.state["phase"] = (self.state["phase"] + 0.01) % (math.pi * 2)
            self.state["pulse"] = self.state["depth"] > 0.85

        # Corridors fluctuate with load
        if "corridor" in self.kind.lower():
            self.state["load"] = 0.5 + 0.4 * math.sin(t / 5 + hash(self.id) % 10)
            self.state["pulse"] = self.state["load"] > 0.75

        # Anchors respond to random stress spikes
        if "anchor" in self.kind.lower():
            if random.random() < 0.01:
                self.state["stress"] = random.choice(["normal", "warning", "critical"])
            self.state["pulse"] = self.state["stress"] == "critical"

    def serialize(self):
        return {
            "entity": self.id,
            "kind": self.kind,
            "state": self.state
        }


# ---------------------------------------------------------
# Registry
# ---------------------------------------------------------

entities = [
    VibrationalEntity("RW-ALPHA", "Resonance well — Deep fulcrum"),
    VibrationalEntity("RW-BETA", "Resonance well — Subharmonic basin"),
    VibrationalEntity("RW-GAMMA", "Resonance well — Shallow ridge"),
    VibrationalEntity("PC-01", "Pulse corridor — North band"),
    VibrationalEntity("PC-02", "Pulse corridor — Southern drift"),
    VibrationalEntity("PC-03", "Pulse corridor — Vertical spine"),
    VibrationalEntity("IA-NORTH", "Infrastructure anchor — Structural beam"),
    VibrationalEntity("IA-CORE", "Infrastructure anchor — Signal trunk"),
    VibrationalEntity("IA-SOUTH", "Infrastructure anchor — Conduit terminus"),
]


# ---------------------------------------------------------
# WebSocket Broadcast Loop
# ---------------------------------------------------------

async def telemetry_stream(websocket):
    print("Client connected.")
    t = 0

    try:
        while True:
            t += 0.1

            # Update all entities
            for e in entities:
                e.update(t)

            # Broadcast full state
            payload = {
                "timestamp": time.time(),
                "entities": [e.serialize() for e in entities]
            }

            await websocket.send(json.dumps(payload))
            await asyncio.sleep(0.25)

    except websockets.exceptions.ConnectionClosed:
        print("Client disconnected.")


async def main():
    print("Starting V-VIB-05 Telemetry Server on ws://localhost:8765")
    async with websockets.serve(telemetry_stream, "localhost", 8765):
        await asyncio.Future()  # run forever


if __name__ == "__main__":
    asyncio.run(main())