ipcd/src/debuggingws.cr

require "option_parser"
require "ipc"
require "socket"
require "./colors"
require "./ws"

require "socket"
require "http"
require "http/server"
require "base64"
require "digest"
require "./utils"

class IPC::Connection
	def initialize(fd : Int32)
		external_connection = LibIPC::Connection.new
		external_connection.fd = fd
		initialize(external_connection)
	end
end

class IPC::Message
	def to_buffer
		to_message @type, String.new(@payload)
	end
end


class WrappedTCPFileDescriptor < TCPSocket
	# do not close the connection when garbage collected!!
	def finalize
		# puts "WrappedTCPFileDescriptor garbage collection!!"
		# super
	end
end

service_name = "websocket"
domain = "localhost"
port_to_listen = 1234

OptionParser.parse! do |parser|
	parser.on "-p port", "--port port", "Port to listen on." do |port|
		port_to_listen = port.to_u16
	end

	parser.on "-d domain", "--domain domain", "Domain to listen on (example: localhost)." do |dom|
		domain = dom
	end

	parser.on "-s service-name", "--service-name service-name", "Service name." do |name|
		service_name = name
	end

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


class InstanceStorage
	property service : IPC::SwitchingService
	property fdlist : Hash(Int32,String)
	property socklist : Hash(Int32, TCPSocket)
	property wssocklist : Hash(Int32, WebSocket)
	property switchtable : Hash(Int32, Int32)
	property connectionlist : Hash(Int32, IPC::Connection)

	property alreadyconnected : Hash(Int32, Bool)

	def initialize (@service : IPC::SwitchingService)
		# fdlist_client = [] of TCPSocket
		@fdlist = Hash(Int32,String).new
		@socklist = Hash(Int32, TCPSocket).new
		@wssocklist = Hash(Int32, WebSocket).new
		@switchtable = Hash(Int32, Int32).new
		@connectionlist = Hash(Int32, IPC::Connection).new
		@alreadyconnected = Hash(Int32, Bool).new
	end

	def remove_fd (fdclient : Int32)
		# 1. closing both the client and the service
		fdservice = @switchtable[fdclient]?
		tcpfdc = @socklist[fdclient]

		# 2. closing the TCP connections
		tcpfdc.close unless tcpfdc.closed?

		# 3. removing the client and the service fds from the loop check
		@service.remove_fd (fdclient)

		# 5. removing both the client and the service from the switchtable
		@switchtable = @switchtable.select do |fdc, fds|
			fdc != fdclient && fds != fdclient
		end

		# 6. removing the client and the service from fdlist
		@fdlist = @fdlist.select do |fd,v| fd != fdclient end

		unless fdservice.nil?
			service = @connectionlist[fdservice]
			service.close
			@service.remove_fd (fdservice)
			@connectionlist = @connectionlist.select do |k, v|
				k != fdservice
			end

			@fdlist = @fdlist.select do |fd,v| fd != fdservice end
		end
	end
end

server = TCPServer.new(domain, port_to_listen)
service = IPC::SwitchingService.new service_name
service << server.fd
context = InstanceStorage.new service



def websocket_client_connection(client, context : InstanceStorage)
	request = HTTP::Request.from_io client
	# pp! request

	if request.nil?
		raise "#REQUEST IS NIL"
	end

	if request.is_a? HTTP::Request::BadRequest
		raise "BAD REQUEST DAZE~"
	end

	# FIXME: check they actually wanted to upgrade to websocket

	key = request.headers["Sec-WebSocket-Key"]

	response_key = Digest::SHA1.base64digest key + "258EAFA5-E914-47DA-95CA-C5AB0DC85B11"
	# puts response_key

	# HTTP inside bru
	headers_header = "HTTP/1.1 101 Switching Protocols"
	headers = HTTP::Headers {
		"Upgrade" => "websocket",
		"Connection" => "Upgrade",
		"Sec-WebSocket-Accept" => response_key
	}

	headers = headers.map { |key, value| "#{key}: #{value[0]}\r\n" }.join

	# puts "#{headers_header}\n#{headers.to_s}\r\n"
	client.send "#{headers_header}\n#{headers.to_s}\r\n"

	wsclient = WebSocket.new client
	wsclient.send "are we websocket yet?"

	# the client is still not connected to a service
	context.fdlist[client.fd] = "not connected"

	# listen to the client's file descriptor
	context.service << client.fd
	# puts "#{CBLUE}new client: #{client.fd}#{CRESET}"

	# registering the client into storing structures to avoid being garbage collected
	context.socklist[client.fd] = client
	context.wssocklist[client.fd] = wsclient

	context.alreadyconnected[client.fd] = false
end


def closing_client (fdclient : Int, context : InstanceStorage)
	# puts "closing the client #{fdclient}"
	context.remove_fd fdclient
	context.alreadyconnected[fdclient] = false
end

puts "listening under the service name #{service_name} and url ws://#{domain}:#{port_to_listen}/"

service.loop do |event|
	case event
	when IPC::Event::Connection
		puts "#{CBLUE}IPC::Event::Connection: #{event.connection.fd}#{CRESET}"
	when IPC::Event::Disconnection
		puts "#{CBLUE}IPC::Event::Disconnection: #{event.connection.fd}#{CRESET}"
	when IPC::Event::ExtraSocket
		puts "#{CBLUE}IPC::Event::ExtraSocket#{CRESET}"

		# 1. accept new websocket clients
		if server.fd == event.connection.fd
			client = server.accept
			begin
				websocket_client_connection client, context
				puts "#{CBLUE}new client:#{CRESET} #{client.fd}"
			rescue e
				puts "Exception: #{CRED}#{e}#{CRESET}"
				client.close
			end
			next
		end

		# 2. active fd != server fd
		# is this a service or a client?
		activefd = event.connection.fd

		# since it's an external communication
		# we have to read the message here, it's not handled by default in libipc
		wsclient = context.wssocklist[activefd]
		begin
			message = wsclient.read
		rescue e
			puts "#{CRED}Exception (receiving a message)#{CRESET} #{e}"
		end

		if message.nil?
			puts "#{CBLUE}disconnection of client#{CRESET} #{event.connection.fd}"
			closing_client activefd, context
			next
		end

		# TODO: should send json values soon
		if context.alreadyconnected[activefd]
			print_hexa(String.new(message), "#{CBLUE}Received message hexa#{CRESET}")
			wsclient.send String.new(message)
		else
			print_hexa(String.new(message), "#{CBLUE}Received message hexa#{CRESET}")
			context.alreadyconnected[activefd] = true
			puts "then send OK"
			wsclient.send "OK"
		end


	when IPC::Event::Switch
		puts "\033[36mIPC::Event::Switch#{CRESET}: from fd #{event.connection.fd}"

		raise "Not implemented."

	# IPC::Event::Message has to be the last entry
	# because ExtraSocket and Switch inherit from Message class
	when IPC::Event::Message
		puts "#{CBLUE}IPC::Event::Message#{CRESET}: #{event.connection.fd}"

		raise "Not implemented."
	end
end