import time
import json
import asyncio

from .. import Client, BfxWebsocket, Socket

def get_now():
	return int(round(time.time() * 1000))

def ev_worker_override():
		return asyncio.get_event_loop()

class StubbedWebsocket(BfxWebsocket):

	def __init__(self, *args, **kwargs):
		self.sent_items = []
		self.published_items = []
		super().__init__(create_event_emitter=ev_worker_override, *args, **kwargs)

	async def _main(self, host):
		print ("Faking wesocket connection to {}".format(host))

	def _start_new_socket(self):
		socket = Socket(len(self.sockets))
		socket.set_connected()
		socket.ws = self
		self.sockets[socket.id] = socket
		return socket.id

	def _wait_for_socket(self, socketId):
		return

	async def publish(self, data, is_json=True):
		self.published_items += [{
		'time': get_now(),
		'data': data
		}]
		# convert to string and push through the websocket
		data = json.dumps(data) if is_json else data
		return await self.on_message(0, data)
  
	async def publish_auth_confirmation(self):
		return self.publish({"event":"auth","status":"OK","chanId":0,"userId":269499,"auth_id":"58aa0472-b1a9-4690-8ab8-300d68e66aaf","caps":{"orders":{"read":1,"write":1},"account":{"read":1,"write":0},"funding":{"read":1,"write":1},"history":{"read":1,"write":0},"wallets":{"read":1,"write":1},"withdraw":{"read":0,"write":1},"positions":{"read":1,"write":1}}})

	async def send(self, data_string):
		self.sent_items += [{
		'time': get_now(),
		'data': data_string
		}]

	def get_published_items(self):
		return self.published_items

	def get_sent_items(self):
		return self.sent_items

	def get_last_sent_item(self):
		return self.sent_items[-1:][0]

	def get_sent_items_count(self):
		return len(self.sent_items)

class EventWatcher():

	def __init__(self, ws, event):
		self.value = None
		self.event = event
		ws.once(event, self._finish)

	def _finish(self, value):
		self.value = value or {}

	@classmethod
	def watch(cls, ws, event):
		return EventWatcher(ws, event)

	def wait_until_complete(self, max_wait_time=5):
		counter = 0
		while self.value == None:
			if counter > 5:
				raise Exception('Wait time limit exceeded for event {}'.format(self.event))
			time.sleep(1)
			counter += 1
		return self.value

class StubClient():
	ws = None
	res = None

def create_stubbed_client(*args, **kwargs):
	client = StubClient()
	# no support for rest stubbing yet
	client.rest = None
	wsStub = StubbedWebsocket(*args, **kwargs)
	# stub client.ws so tests can use publish
	client.ws = wsStub
	client.ws.API_KEY = "test key"
	client.ws.API_SECRET = "secret key"
	# stub socket so we can track socket send requests
	socket = Socket(0)
	socket.set_connected()
	socket.ws = wsStub
	client.ws.sockets = { 0: socket }
	return client

async def ws_publish_auth_accepted(ws):
	  return await ws.publish({"event":"auth","status":"OK","chanId":0,"userId":269499,"auth_id":"58aa0472-b1a9-4690-8ab8-300d68e66aaf","caps":{"orders":{"read":1,"write":1},"account":{"read":1,"write":0},"funding":{"read":1,"write":1},"history":{"read":1,"write":0},"wallets":{"read":1,"write":1},"withdraw":{"read":0,"write":1},"positions":{"read":1,"write":1}}})

async def ws_publish_connection_init(ws):
	  return await ws.publish({"event":"info","version":2,"serverId":"748c00f2-250b-46bb-8519-ce1d7d68e4f0","platform":{"status":1}})

async def ws_publish_conf_accepted(ws, flags_code):
	  return await ws.publish({"event":"conf","status":"OK","flags":flags_code})