websocketd/tests/test-ws.cr

require "http/web_socket"

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

require "ipc"

require "../src/lib_modifications.cr"

require "json"

class TestIPC
	property ipcc : IPC::Client
	property is_json : Bool

	def initialize(service_name : String)
		@is_json = uri.ends_with? ".JSON"
		@ipcc = IPC::Client.new service_name
	end

	# TODO
	#def run
	#	yield @ipcc
	#	ipcc.close
	#end
end

class TestWS
	property ws : WebSocket
	property is_json : Bool

	def initialize(uri : String)
		@ws = WebSocket.new(URI.parse(uri))
		@is_json = uri.ends_with? ".JSON"
		@ws.on_close do |socket|
			raise "socket is closing"
		end
	end

	def read
		m = nil
		loop do
			m = @ws.run_once
			if m.nil?
				raise "empty message"
			end

			# remove ping messages, they are not application-relevent
			unless m.is_a?(HTTP::WebSocket::Ping)
				break
			end
			puts "received a ping message, skipping"
		end

		m
	end

	def send(type : Int32, data : String)

		m : String | Bytes

		if @is_json
			m = IPC::Message.new(0, 1.to_u8, type.to_u8, data).to_json
		else
			m = IPC::Message.new(0, 1.to_u8, type.to_u8, data).to_packet
		end
	
		@ws.send m
	end

	def close
		@ws.close
	end
end