import config as conf

import stats

from util import Progress

from hashtree import HashTree,hashBlock

from netnode import BaseConnection,NetNode,FailedConnection

class DeniedConnection(Exception): pass

class Connection(BaseConnection):

	def __init__(self,host,port):

		super().__init__()

		sock=socket.socket(socket.AF_INET, socket.SOCK_STREAM)

		sslContext=ssl.create_default_context(cafile=conf.peers)

		sslContext.check_hostname=False

		sslContext.load_cert_chain(conf.certfile,conf.keyfile)

		self._socket=sslContext.wrap_socket(sock)

		try:

			self._socket.connect((host,port))

			log.exception(e)

			print("Couldn't connect to {0}:{1}".format(host,port))

			raise FailedConnection()

		except ssl.SSLError as e:

			log.exception(e)

			print("Error creating SSL connection to {0}:{1}".format(host,port))

			raise FailedConnection()

		self.createNetworkers()

		print("Connected to {0}".format(host))

class Client(NetNode):

	def __init__(self,filename,treeFile=""):

		print(datetime.now(), "initializing...")

		super().__init__(filename,treeFile)

	def init(self,action):

		jsonData={"command":"init", "blockSize":self._tree.BLOCK_SIZE, "blockCount":self._tree.leafCount, "version":conf.version, "action":action}

		self._outcoming.writeMsg(jsonData)

		jsonData,binData=self._incoming.readMsg()

		if jsonData["command"]=="deny":

			raise DeniedConnection()

		assert jsonData["command"]=="init"

		if treeFile:

			self._tree=HashTree.load(treeFile)

		else:

			self._tree=HashTree.fromFile(filename)

		self._newLeaves=dict()

	def isLocked(self):

		return os.path.isfile(lockFile)

	def _lock(self):

		try:

			f=open(lockFile,"x")

			f.close()

		except FileExistsError:

			raise LockedException()

	def _unlock(self):

		os.remove(lockFile)

	def _updateTree(self):

		log.info("updating hash tree...")

		self._tree.batchUpdate(self._newLeaves.items())

		self._tree.save(self._treeFile)