client.lua

-- manages client sockets
-- buffers sending and receiving

clients={}
client={}
local buffering={}
function client.new(sk,norec)
	local cl
	cl={
		closed=false,
		sk=sk,
		sbuffer="",
		send=function(txt)
			if txt then
				cl.sbuffer=cl.sbuffer..txt
			end
			local t,err,c=sk:send(cl.sbuffer)
			if not t and err~="timeout" then
				cl.onError("send",err)
				return
			elseif not c then
				print("done sending")
				hook.remrsocket(cl.sk)
				bfrind=1
				cl.sbuffer=""
				buffering[cl.sk]=nil
				if cl.onDoneSending then
					local f=cl.onDoneSending
					cl.onDoneSending=nil
					f(cl)
				end
				return
			elseif not buffering[cl.sk] then
				hook.newrsocket(cl.sk)
				buffering[cl.sk]=cl
			end
			cl.sbuffer=cl.sbuffer:sub(c+1)
		end,
		close=function(...)
			print("cl tryclose")
			if not cl.closed then
				print("cl close")
				-- cleanup
				hook.remsocket(sk)
				hook.remrsocket(sk)
				clients[sk]=nil
				buffering[cl]=nil
				sk:close()
				cl.closed=true
				if cl.onClose then
					cl.onClose(...)
				end
			end
		end,
		onError=function(t,err)
			print("cl error "..err)
			cl.close()
		end,
		rbuffer="",
		receive=function(mode)
			if mode then
				if cl.rbuffer<mode then
					return
				end
				local txt=cl.rbuffer:sub(1,mode)
				cl.rbuffer=cl.rbuffer:sub(5)
				return txt
			end
			local line,rbuffer=cl.rbuffer:match("^(.-)\r?\n(.*)")
			if line then
				cl.rbuffer=rbuffer
				return line
			end
		end,
	}
	clients[sk]=cl
	if not norec then
		receivehead(cl)
	end
	return cl
end

hook.new("select",function(rq,sq)
	for k,v in pairs(sq) do
		if type(v)~="number" then
			if buffering[v] then
				buffering[v].send()
			end
		end
	end
	for k,v in pairs(rq) do
		if clients[v] then
			local cl=clients[v]
			local s2,err,s=v:receive("*a")
			if err=="closed" then
				cl.close()
			else
				cl.rbuffer=cl.rbuffer..(s or s2)
				if cl.onReceive then
					cl.onReceive()
				end
			end
		end
	end
end,"client socket manager")