#-----------------------------------------------------------------------------------
import const
import hiraide
import hiraSetting
import tracemalloc
import time
import asyncio
import websockets
import uuid
from hiraThread import hiraThread
class hiraWSSV:
	parent		= None
	cWS			= None
	AR			= []
	def __init__(self):
		self.cWS	= None
		self.AR		= []
		self.myLoop	= asyncio.new_event_loop()  
		asyncio.set_event_loop(self.myLoop)
	def setParent(self,parent):
		self.parent	= parent
	def searchWs(self,ws):
		n = len(self.AR)
		for w in range(n):
			if self.AR[w]["ws"] == ws:
				return	w
		return	-1
	async def async_connect(self,websocket):
		print(f"async_connect:websocket:{websocket}")
		pass
	async def async_message(self,ws,mes):
		print(f"async_message:mes:{mes}")
		await self.async_send(ws,"FROM async_message")
		pass
	async def async_except(self,e,idx):
		print(f"async_except:idx:e:[{idx}]:{e}")
		pass
	async def async_send(self,ws,mes):
		await ws.send(mes)
		pass
	async def async_close(self,idx):
		try:
			ws = self.AR[idx]["ws"]
			await ws.close()
			self.AR.pop(idx)
		except Exception as e:
			pass
	def sendA(self,ws,mes):
		self.myLoop.run_until_complete(self.async_send(ws,mes))
	def send(self,idx,mes):
		if(idx >= 0):
			self.myLoop.run_until_complete(self.async_send(self.AR[idx]["ws"],mes))
		else:
			self.myLoop.run_until_complete(self.async_send(self.cWS,mes))
	def sendC(self,mes):
		self.myLoop.run_until_complete(self.async_send(self.cWS,mes))
		print("type:",type(mes))
	def close(self,idx):
		self.myLoop.run_until_complete(self.async_close(idx))
	async def async_except_after(self):
		print("hiraWSSV:async_except_after")
		pass
	async def connect(self,websocket):
		self.cWS	= websocket
		self.AR.append({"ws":websocket})
		await self.async_connect(websocket)
		while True:
			try:
				message = await websocket.recv()
				await self.async_message(websocket,message)
			except Exception as e:
				idx = self.searchWs(websocket)
				await self.async_except(e,idx)
				if idx >= 0:
					self.AR.pop(idx)
				try:
					await websocket.close()
				except Exception as e:
					pass
				await self.async_except_after()
				break
	async def main(self,host,port):
		async with websockets.serve(self.connect, host, port):
			print(f"WebSocket server is running on ws://{host}:{port}")
			await asyncio.Future()  
	def RunForever(self,host,port):
		print(f"
		print(f"  host; {host}")
		print(f"  port; {port}")
		print("
		try:
			self.myLoop.run_until_complete(self.main(host, port))
		finally:
			self.myLoop.close()