Lumia Test 7

-- color_control.lua

-- 1. Define color-to-relay mapping
local colorToRelay = {
  red =    { relayName = "redstone_relay_0", side = "front" },
  blue =   { relayName = "redstone_relay_1", side = "front" },
  green =  { relayName = "redstone_relay_2", side = "front" },
  yellow = { relayName = "redstone_relay_3", side = "front" },
  purple = { relayName = "redstone_relay_4", side = "front" },
  orange = { relayName = "redstone_relay_5", side = "front" },
}

-- 2. Helper: turn off all relays
local function turnOffAll()
  for _, info in pairs(colorToRelay) do
    local relay = peripheral.wrap(info.relayName)
    if relay then
      relay.setOutput(info.side, false)
    end
  end
end

turnOffAll()  -- ensure everything is off at start

-- Keep track of what's currently active
local currentColor = nil

local function activateColor(newColor)
  -- If our newColor is a boolean, convert it to a string just so we don't get a concat error
  if type(newColor) == "boolean" then
    newColor = tostring(newColor)  -- "true" or "false"
  end

  -- 3A. Turn off the old color's relay
  if currentColor and colorToRelay[currentColor] then
    local oldInfo = colorToRelay[currentColor]
    local oldRelay = peripheral.wrap(oldInfo.relayName)
    if oldRelay then
      oldRelay.setOutput(oldInfo.side, false)
    end
  end

  -- 3B. Turn on the new color's relay
  if colorToRelay[newColor] then
    local info = colorToRelay[newColor]
    local relay = peripheral.wrap(info.relayName)
    if relay then
      relay.setOutput(info.side, true)
      currentColor = newColor
      print("Activated color: "..newColor)
    else
      print("Relay peripheral not found for color: "..newColor)
    end
  else
    print("Unknown color: "..tostring(newColor))
  end
end

-- 4. Connect to your Node.js WebSocket server
local url = "ws://kev.unoriginal.studio:8080"
print("Connecting to "..url)
http.websocketAsync(url)

-- 5. Event loop to handle WebSocket messages
while true do
  local event, p1, p2, p3 = os.pullEvent()
  if event == "websocket_success" then
    print("WebSocket connected!")
    -- p2 is the websocket handle
    -- Optionally send a greeting
    p2.send("Hello from CC: Tweaked")

  elseif event == "websocket_failure" then
    print("Failed to connect:", p2)
    break  -- exit the program, or handle reconnection logic

  elseif event == "websocket_message" then
    -- p2 = the websocket handle, p3 = the message content (could be boolean or string)
    local colorMsg = p3
    print("Received color:", tostring(colorMsg))
    activateColor(colorMsg)

  elseif event == "websocket_closed" then
    print("WebSocket closed.")
    break
  end
end

print("Exiting program.")