require "http/web_socket"
require "option_parser"

require "./colors"
require "./utils"

require "./ws"

uri = "ws://localhost:1234/pong"

OptionParser.parse do |parser|
	parser.on "-u uri", "--uri uri", "URI" do |opturi|
		uri = opturi
	end

	parser.on "-h", "--help", "Show this help" do
		puts parser
		exit 0
	end
end

def read_then_print(ws : WebSocket)
	m = read ws
	puts "message: #{String.new(m)}"
end

def read_then_print_hexa(ws : WebSocket)
	m = read ws
	print_hexa(String.new(m), "#{CBLUE}Received message hexa#{CRESET}")
end

def read(ws : WebSocket)
	m = ws.read
	if m.nil?
		raise "empty message"
	end

	m
end

def send_with_announce(ws : WebSocket, m : String)
	puts "sending #{m}"
	send ws, m
end

def send(ws : WebSocket, m : String | Slice)
	ws.send m
end

begin
	ws = WebSocket.new(URI.parse(uri))

	puts "connection done: sending pong"

	ws.on_close do |socket|
		puts "socket is closing"
		exit 0
	end

	if uri.ends_with? ".JSON"
		json_message = STDIN.gets_to_end
		json_message = json_message.chomp
		puts "json_message: #{json_message}"
		final_json_message = "{ \"mtype\": 2, \"payload\": #{json_message} }"
		puts "final json message: #{final_json_message}"
		send ws, final_json_message
		# send ws, "{ \"mtype\": 2, \"payload\": \"coucou\" }"
		read ws
	else
		send ws, to_message(2, STDIN.gets_to_end)
		read ws
	end

	ws.close
	ws.read
rescue e
	puts "Exception: #{e}"
end