Techmino/parts/net.lua

local data=love.data
local rem=table.remove

local WS,TIME=WS,TIME
local yield=YIELD

local NET={
	connected=false,
	allow_online=false,
	accessToken=false,
	roomList={},
	roomInfo={
		-- rid=false,
		name=false,
		-- type=false,
		private=false,
		-- count=false,
		capacity=false,
		start=false,
	},
	allReady=false,
	connectingStream=false,
	waitingStream=false,
	serverGaming=false,
	streamRoomID=false,

	UserCount="_",
	PlayCount="_",
	StreamCount="_",
}

local mesType={
	Connect=true,
	Self=true,
	Broadcast=true,
	Private=true,
	Server=true,
}

--Lock & Unlock submodule
local locks do
	local rawset=rawset
	locks=setmetatable({},{
		__index=function(self,k)rawset(self,k,-1e99)return -1e99 end,
		__newindex=function(self,k)rawset(self,k,-1e99)end,
	})
end
function NET.lock(name,T)
	if TIME()>=locks[name]then
		locks[name]=TIME()+(T or 1e99)
		return true
	else
		return false
	end
end
function NET.unlock(name)
	locks[name]=-1e99
end
function NET.getlock(name)
	return TIME()<locks[name]
end

--Parse json message
local function _parse(res)
	res=JSON.decode(res)
	if res then
		if mesType[res.type]then
			return res
		else
			LOG.print(
				"WS error:"..(
					res.type and(
						res.reason and res.type..": "..res.reason or
						res.type
					)or
					"[NO Message]"
				),
			'warn')
		end
	end
end

--WS close message
local function wsCloseMessage(message)
	local mes=JSON.decode(message:sub(3))
	if mes then
		LOG.print(("%s [%s] %s"):format(text.wsClose,mes.type or"unknown type",mes.reason or""),'error')
	else
		LOG.print(text.wsClose,'error')
	end
end

--Remove player when leave
local function removePlayer(L,sid)
	for i=1,#L do
		if L[i].sid==sid then
			rem(L,i)
			break
		end
	end
end

--Connect
function NET.wsconn_app()
	WS.connect('app','/app')
end
function NET.wsconn_user_pswd(email,password)
	if NET.lock('wsc_user',5)then
		WS.connect('user','/user',JSON.encode{
			email=email,
			password=password,
		})
	end
end
function NET.wsconn_user_token(uid,authToken)
	if NET.lock('wsc_user',5)then
		WS.connect('user','/user',JSON.encode{
			uid=uid,
			authToken=authToken,
		})
	end
end
function NET.wsconn_play()
	if NET.lock('wsc_play',5)then
		WS.connect('play','/play',JSON.encode{
			uid=USER.uid,
			accessToken=NET.accessToken,
		})
	end
end
function NET.wsconn_stream()
	if NET.lock('wsc_stream',5)then
		NET.serverGaming=true
		WS.connect('stream','/stream',JSON.encode{
			uid=USER.uid,
			accessToken=NET.accessToken,
			rid=NET.streamRoomID,
		})
		TASK.new(NET.updateWS_stream)
	end
end

--Disconnect
function NET.wsclose_app()WS.close('app')end
function NET.wsclose_user()WS.close('user')end
function NET.wsclose_play()WS.close('play')end
function NET.wsclose_stream()
	NET.serverGaming=false
	WS.close('stream')
end

--Account & User
function NET.register(username,email,password)
	if NET.lock('register')then
		WS.send('app',JSON.encode{
			action=2,
			data={
				username=username,
				email=email,
				password=password,
			}
		})
	end
end
function NET.pong(wsName,message)
	WS.send(wsName,type(message)=='string'and message or"",'pong')
end
function NET.getAccessToken()
	if NET.lock('access_and_login',10)then
		WS.send('user',JSON.encode{action=0})
	end
end
function NET.getUserInfo(uid)
	local hash=(not SETTING.dataSaving or nil)and USERS.getHash(uid)
	WS.send('user',JSON.encode{
		action=1,
		data={
			uid=uid,
			hash=hash,
		},
	})
end
function NET.freshPlayerCount()
	while true do
		for _=1,260 do yield()end
		if WS.status('app')=='running'then
			WS.send('app',JSON.encode{action=3})
		end
	end
end

--Room
function NET.fetchRoom()
	if NET.lock('fetchRoom',3)then
		WS.send('play',JSON.encode{
			action=0,
			data={
				type=nil,
				begin=0,
				count=10,
			}
		})
	end
end
function NET.createRoom(roomName,capacity,roomType,password)
	if NET.lock('enterRoom',1.26)then
		NET.roomInfo.name=roomName
		NET.roomInfo.type=roomType
		NET.roomInfo.private=not not password
		NET.roomInfo.capacity=capacity
		WS.send('play',JSON.encode{
			action=1,
			data={
				name=roomName,
				capacity=capacity,
				roomData={type=roomType},
				password=password,
				config=dumpBasicConfig(),
			}
		})
	end
end
function NET.enterRoom(room,password)
	if NET.lock('enterRoom',1.26)then
		SFX.play('reach',.6)
		NET.roomInfo.name=room.name
		NET.roomInfo.type=room.type
		NET.roomInfo.private=not not password
		NET.roomInfo.capacity=room.capacity
		NET.roomInfo.start=room.start
		WS.send('play',JSON.encode{
			action=2,
			data={
				rid=room.rid,
				config=dumpBasicConfig(),
				password=password,
			}
		})
	end
end

--Play
function NET.checkPlayDisconn()
	return WS.status('play')~='running'
end
function NET.signal_quit()
	if NET.lock('quit',3)then
		WS.send('play','{"action":3}')
	end
end
function NET.sendMessage(mes)
	WS.send('play','{"action":4,"data":'..JSON.encode{message=mes}..'}')
end
function NET.changeConfig()
	WS.send('play','{"action":5,"data":'..JSON.encode({config=dumpBasicConfig()})..'}')
end
function NET.signal_ready(ready)
	if NET.lock('ready',3)and not NET.serverGaming then
		WS.send('play','{"action":6,"data":{"ready":'..tostring(ready)..'}}')
	end
end
function NET.signal_die()
	WS.send('stream','{"action":4,"data":{"score":0,"survivalTime":0}}')
end
function NET.uploadRecStream(stream)
	WS.send('stream','{"action":5,"data":{"stream":"'..data.encode('string','base64',stream)..'"}}')
end

--Chat
function NET.sendChatMes(mes)
	WS.send('chat',"T"..data.encode('string','base64',mes))
end
function NET.quitChat()
	WS.send('chat','q')
end

--WS tick funcs
function NET.updateWS_app()
	while true do
		yield()
		if WS.status('app')=='running'then
			local message,op=WS.read('app')
			if message then
				if op=='ping'then
					NET.pong('app',message)
				elseif op=='pong'then
				elseif op=='close'then
					wsCloseMessage(message)
					return
				else
					local res=_parse(message)
					if res then
						if res.type=='Connect'then
							NET.connected=true
							if VERSION.code>=res.lowest then
								NET.allow_online=true
								if USER.authToken then
									NET.wsconn_user_token(USER.uid,USER.authToken)
								end
							end
							if VERSION.code<res.newestCode then
								LOG.print(text.oldVersion:gsub("$1",res.newestName),180,'message')
							end
							LOG.print(res.notice,300,'message')
						elseif res.action==0 then--Get new version info
							--?
						elseif res.action==1 then--Get notice
							--?
						elseif res.action==2 then--Register
							if res.type=='Self'or res.type=='Server'then
								LOG.print(res.data.message,300,'message')
								if SCN.cur=='register'then
									SCN.back()
								end
							else
								LOG.print(res.reason or"Registration failed",300,'message')
							end
							NET.unlock('register')
						elseif res.action==3 then--Get player counts
							NET.UserCount=res.data.User
							NET.PlayCount=res.data.Play
							NET.StreamCount=res.data.Stream
							--res.data.Chat
						end
					else
						WS.alert('app')
					end
				end
			end
		end
	end
end
function NET.updateWS_user()
	while true do
		yield()
		if WS.status('user')=='running'then
			local message,op=WS.read('user')
			if message then
				if op=='ping'then
					NET.pong('user',message)
				elseif op=='pong'then
				elseif op=='close'then
					wsCloseMessage(message)
					return
				else
					local res=_parse(message)
					if res then
						if res.type=='Connect'then
							if res.uid then
								USER.uid=res.uid
								USER.authToken=res.authToken
								FILE.save(USER,'conf/user','q')
								if SCN.cur=='login'then SCN.back()end
							end
							LOG.print(text.loginSuccessed,'message')

							--Get self infos
							NET.getUserInfo(USER.uid)
							NET.unlock('wsc_user')
						elseif res.action==0 then--Get accessToken
							NET.accessToken=res.accessToken
							LOG.print(text.accessSuccessed,'message')
							NET.wsconn_play()
						elseif res.action==1 then--Get userInfo
							USERS.updateUserData(res.data)
						end
					else
						WS.alert('user')
					end
				end
			end
		end
	end
end
function NET.updateWS_play()
	while true do
		yield()
		if WS.status('play')=='running'then
			local message,op=WS.read('play')
			if message then
				if op=='ping'then
					NET.pong('play',message)
				elseif op=='pong'then
				elseif op=='close'then
					wsCloseMessage(message)
					return
				else
					local res=_parse(message)
					if res then
						local d=res.data
						if res.type=='Connect'then
							SCN.go('net_menu')
							NET.unlock('wsc_play')
							NET.unlock('access_and_login')
						elseif res.action==0 then--Fetch rooms
							NET.roomList=res.roomList
							NET.unlock('fetchRoom')
						elseif res.action==1 then--Create room (not used)
							--?
						elseif res.action==2 then--Player join
							if res.type=='Self'then
								--Enter new room
								netPLY.clear()
								if d.players then
									for _,p in next,d.players do
										netPLY.add{
											uid=p.uid,
											username=p.username,
											sid=p.sid,
											ready=p.ready,
											config=p.config,
										}
									end
								end
								--TODO: d.roomData (json)
								loadGame('netBattle',true,true)
							else
								--Load other players
								netPLY.add{
									uid=d.uid,
									username=d.username,
									sid=d.sid,
									ready=d.ready,
									config=d.config,
								}
								if SCN.socketRead then SCN.socketRead('join',d)end
								NET.allReady=false
							end
						elseif res.action==3 then--Player leave
							if not d.uid then
								NET.wsclose_stream()
								NET.unlock('quit')
								SCN.back()
							else
								removePlayer(netPLY.list,d.sid)
								netPLY.freshPos()
								removePlayer(PLAYERS,d.sid)
								removePlayer(PLY_ALIVE,d.sid)
								if SCN.socketRead then SCN.socketRead('leave',d)end
							end
						elseif res.action==4 then--Player talk
							if SCN.socketRead then SCN.socketRead('talk',d)end
						elseif res.action==5 then--Player change settings
							netPLY.setConf(d.uid,d.config)
						elseif res.action==6 then--One ready
							netPLY.setReady(d.uid,d.ready)
						elseif res.action==7 then--All Ready
							SFX.play('reach',.6)
							NET.allReady=true
						elseif res.action==8 then--Set
							NET.streamRoomID=d.rid
							NET.allReady=false
							NET.connectingStream=true
							NET.wsconn_stream()
						elseif res.action==9 then--Game finished
							NET.wsclose_stream()
							if SCN.socketRead then SCN.socketRead('finish',d)end
							NET.roomInfo.start=false
						end
					else
						WS.alert('play')
					end
				end
			end
		end
	end
end
function NET.updateWS_stream()
	while true do
		yield()
		if WS.status('stream')=='running'then
			local message,op=WS.read('stream')
			if message then
				if op=='ping'then
					NET.pong('stream',message)
				elseif op=='pong'then
				elseif op=='close'then
					wsCloseMessage(message)
					return
				else
					local res=_parse(message)
					if res then
						local d=res.data
						if res.type=='Connect'then
							NET.unlock('wsc_stream')
							NET.connectingStream=false
							NET.waitingStream=true
							netPLY.setConnect(USER.uid)
							netPLY.freshStreamConn(res.data.connected)
						elseif res.action==0 then--Game start
							NET.waitingStream=false
							if SCN.socketRead then SCN.socketRead('go',d)end
							NET.roomInfo.start=true
						elseif res.action==1 then--Game finished
							--?
						elseif res.action==2 then--Player join
							netPLY.setConnect(d.uid)
						elseif res.action==3 then--Player leave
							--?
						elseif res.action==4 then--Player died
							for _,P in next,PLY_ALIVE do
								if P.uid==d.uid then
									P:lose(true)
									break
								end
							end
						elseif res.action==5 then--Receive stream
							SCN.socketRead('stream',d)
						end
					else
						WS.alert('stream')
					end
				end
			end
		end
	end
end
function NET.updateWS_chat()
	while true do
		yield()
		if WS.status('chat')=='running'then
			local message,op=WS.read('chat')
			if message then
				if op=='ping'then
					NET.pong('chat',message)
				elseif op=='pong'then
				elseif op=='close'then
					wsCloseMessage(message)
					return
				else
					local res=_parse(message)
					if res then
						--TODO
					else
						WS.alert('chat')
					end
				end
			end
		end
	end
end

return NET