require("oct_utils")
require("termbox_defs")
json = require "json"

-- begin message type constants
OCT_LOBBY_MSG_CLIENTREG = 0; -- Client registration message
OCT_LOBBY_MSG_CLIENTUNREG = 1; -- Client unreg message
OCT_LOBBY_MSG_CLIENTLIST = 2; -- Client list broadcast message
OCT_LOBBY_MSG_START = 3; -- Start the game message
OCT_LOBBY_MSG_PING = 4; -- Querying connection
OCT_LOBBY_MSG_ACK = 5; -- "yes", used to respond to ping and other stuff
OCT_LOBBY_MSG_NAK = 6; -- "no"
-- end message type constants

oct_lobby_clientlist = {};
oct_lobby_timers = {};
oct_lobby_timers["clientlist"] = 1000; -- Every 1000 calls to lobby_client broadcast client list

function lobby_server(maxplayers)
   msg,addr,port = oct_recv();
   if (msg ~= "") then
	  msg_obj = json.decode(msg)[1] -- dumps to 1-elem array, get first elem

	  if (msg_obj == nil) then
		 OCT_LOG_WARNING("Malformed json received")
	  end
	  
	  -- The main state machine
	  if (msg_obj["msg_type"] == OCT_LOBBY_MSG_CLIENTREG) then
		 -- Client registration
		 req_name = msg_obj["name"]
		 req_port = msg_obj["port"]
		 req_addr = convert_known_host_to_ip(addr)
		 
		 -- First, check if clientlist contains a conflicting username
		 if (oct_lobby_clientlist[req_name] ~= nil) then
			response = json.encode({
				  { msg_type = OCT_LOBBY_MSG_NAK }
			})
			OCT_LOG_WARNING("Rejected attempt to register from client: " .. req_name .. " @ " .. req_addr .. ":" .. req_port)
			
			oct_send(response, req_addr, req_port)
		 else
			response = json.encode({
				  { msg_type = OCT_LOBBY_MSG_ACK }
			})

			oct_lobby_clientlist[req_name] = {}
			oct_lobby_clientlist[req_name]["addr"] = req_addr
			oct_lobby_clientlist[req_name]["port"] = req_port
			OCT_LOG_INFO("Registered new client: " .. req_name .. " @ " .. req_addr .. ":" .. req_port)
			
			oct_send(response, req_addr, req_port)
		 end
	  end
   end
end

client_connected = false;
oct_started = false;
client_wait_for_reg_response = false;

function lobby_client(ip, port, name, my_port)
   -- If not already connected, connect sending info
   if (not client_connected and client_wait_for_reg_response == false) then
	  msg_to_server = json.encode({
			{ name = name, msg_type=OCT_LOBBY_MSG_CLIENTREG, port=my_port }
	  })
	  oct_send(msg_to_server, ip, port)
	  client_wait_for_reg_response = true -- prevent this from being run again
   end

   msg,addr,port = oct_recv()
   if (msg ~= "") then
	  msg_obj = json.decode(msg)[1]
	  
	  -- State machine code here
	  -- Registration was successful
	  if (client_wait_for_reg_response == true and
		  msg_obj["msg_type"] == OCT_LOBBY_MSG_ACK) then
		 OCT_LOG_INFO("Server accepted registration request!")
		 client_connected = 1
		 client_wait_for_reg_response = false

		 -- Registration was unsuccessful
	  elseif (client_wait_for_reg_response == true and
			  msg_obj["msg_type"] == OCT_LOBBY_MSG_NAK) then
		 OCT_LOG_ERROR("Server rejected registration request, try using a different name")
		 client_connected = 0
		 client_wait_for_reg_response = false
	  end
   end
end