import sys

import collections

import socket

import ssl

import logging as log

from datetime import datetime

import config as conf

import stats

from util import Progress

from hashtree import HashTree,hashBlock

from netnode import BaseConnection,NetNode

class Connection(BaseConnection):

		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:

		except ConnectionRefusedError:

			print("Couldn't connect to {0}:{1}".format(conf.hosts[0],conf.port))

			sys.exit(1)

		self.createNetworkers()

class Client(NetNode):

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

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

		super().__init__(filename,treeFile)

	## Asks server for node hashes to determine which are to be transferred.

	#

	# Uses a binary HashTree, where item at k is hash of items at 2k+1, 2k+2.

	#

	# Requests nodes in order of a batch DFS. Needs stack of size O(treeDepth*batchSize). Nodes in each tree level are accessed in order.

	def negotiate(self):

		localTree=self._tree

		blocksToTransfer=[]

		nodeStack=collections.deque([0]) # root

		# initialize session

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

import sys

import os.path

import logging as log

from argparse import ArgumentParser

import config as conf

import stats

from hashtree import HashTree

from client import Client, Connection as ClientConnection

from server import Miniserver

def _checkFile(f):

	if not os.path.isfile(f):

		print("invalid file specified:",f,file=sys.stderr)

		sys.exit(1)

def buildTree(args):

	_checkFile(args.datafile)

	if os.path.isfile(args.treefile):

		treeMod=os.stat(args.treefile).st_mtime

		dataMod=os.stat(args.datafile).st_mtime

		if dataMod<treeMod and not args.force:

			print("tree file is up to date")

			return

	tree=HashTree.fromFile(args.datafile)

	tree.save(args.treefile)

def push(args):

	_checkFile(args.datafile)

	if args.tree:

		_checkFile(args.tree)

	if args.port: conf.port=args.port

	c=Client(args.datafile,args.tree)

def pull(args):

	_checkFile(args.datafile)

	if args.tree:

		_checkFile(args.tree)

	if args.port: conf.port=args.port

	c=Client(args.datafile,args.tree)

		c.setConnection(con)

		blocksToTransfer=c.negotiate()

		c.pullData(blocksToTransfer)

	print()

	print(stats.report())

def serve(args):

	_checkFile(args.datafile)

	if args.tree:

		_checkFile(args.tree)

	if args.host: conf.hosts.insert(0,args.host)

	if args.port: conf.port=args.port

	s=Miniserver(args.datafile,args.tree)

	try:

		spawnDaemon(s.serve)

	except Exception as e:

		log.exception("exception: %s",e)

parser=ArgumentParser()

subparsers=parser.add_subparsers()

pBuild=subparsers.add_parser("build")

from hashtree import HashTree

class BaseConnection: # abstract

	def __init__(self):

		self._socket=None

		self.incoming=None

		self.outcoming=None

	def createNetworkers(self):

		fr=self._socket.makefile(mode="rb")

		fw=self._socket.makefile(mode="wb")

		self.incoming=NetworkReader(fr)

		self.outcoming=NetworkWriter(fw)

	def __enter__(self):

		return self.incoming,self.outcoming

	def __exit__(self, exc_type, exc_val, exc_tb):

		try:

			self._socket.shutdown(socket.SHUT_RDWR)

			self._socket.close()

		except OSError:

class NetNode:

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

		self._incoming=None

		self._outcoming=None

		self._filename=filename

		self._treeFile=treeFile

		if treeFile:

			self._tree=HashTree.load(treeFile)

		else:

			self._tree=HashTree.fromFile(filename)

		self._newLeaves=dict()

	def _updateTree(self):

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

		for (k,v) in self._newLeaves.items():

			self._tree.updateLeaf(k, v)

		self._tree.save(self._treeFile)

class Server(NetNode):

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

		super().__init__(filename,treeFile)

		(self._incoming,self._outcoming)=connection

		self.BLOCK_SIZE=self._tree.BLOCK_SIZE

		self._lastIndex=-1

		self._dataFileHandle=None

	@staticmethod

	def run(*args):

		s=Server(*args)

		s.serve()

	@property

	def _dataFile(self):

		if not self._dataFileHandle:

			self._dataFileHandle=open(self._filename, mode="rb+")

		return self._dataFileHandle

	def serve(self):

		try:

			while self._serveOne(): pass

			log.warning(e)

	def _serveOne(self):

		jsonData,binData=self._incoming.readMsg()

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

			assert jsonData["blockSize"]==self.BLOCK_SIZE

			assert jsonData["blockCount"]==self._tree.leafCount

			self._outcoming.writeMsg({"command": "ack"})

		elif jsonData["command"]=="req":

			if jsonData["dataType"]=="data":

				self._outcoming.writeMsg(*self._requestData(jsonData["index"]))

			else:

				self._outcoming.writeMsg(*self._requestHash(jsonData["index"]))

		elif jsonData["command"]=="send" and jsonData["dataType"]=="data":

			self._outcoming.writeMsg(*self._receiveData(jsonData,binData))

		elif jsonData["command"]=="end":

			self._finalize()

			self._locked=False

			return False

	def tearDownClass(cls):

		os.remove(filename)

		sys.stdout.close()

		sys.stdout=cls._stdout

	def test_build(self):

		treeFile=os.path.join(dataDir,"test.bin")

		refFile=os.path.join(dataDir,"test1.bin")

		tree=HashTree.fromFile(os.path.join(dataDir,"test1.img"))

		tree.save(treeFile)

		self.assertEqual(*compareFiles(refFile,treeFile))

		os.remove(treeFile)

	def test_push(self):

		ms=Miniserver(filename)

		p=multiprocessing.Process(target=ms.serve)

		p.start()

		for clientFile in ("test2.img","test3.img","test4.img"):

			clientFile=os.path.join(dataDir,clientFile)

			c=Client(clientFile)

				c.setConnection(con)

				blocksToTransfer=c.negotiate()

				c.sendData(blocksToTransfer)

			self.assertEqual(*compareFiles(clientFile,filename))

		p.terminate()

		p.join()

	def test_pull(self):

		serverFile=os.path.join(dataDir,"test3.img")

		ms=Miniserver(serverFile)

		p=multiprocessing.Process(target=ms.serve)

		p.start()

		c=Client(filename)

			c.setConnection(con)

			blocksToTransfer=c.negotiate()

			c.pullData(blocksToTransfer)

		self.assertEqual(*compareFiles(serverFile,filename))

		p.terminate()

		p.join()

			return

	except OSError as e:

		print("fork #1 failed: {0} ({1})".format(e.errno,e.strerror),file=sys.stderr)

		sys.exit(1)

	os.setsid()

	# do second fork

	try:

		pid = os.fork()

		if pid > 0:

			# exit from second parent

			print("[{0}] server running".format(pid))

			sys.exit(0)

	except OSError as e:

		print("fork #2 failed: {0} ({1})".format(e.errno,e.strerror),file=sys.stderr)

		sys.exit(1)

	fun()

	# all done

	os._exit(os.EX_OK)

class Progress:

	def __init__(self,n,i0=0):

		self._n=n

		self._i0=i0

		self._i=i0

		self._last=""

	def p(self,i):

		i0=self._i0

		n=self._n

		assert i0<=i<n or n<i<=i0, (i0,i,n)

		percentage=Progress._p(i,n,i0)

		res="{0}%".format(percentage)

		if res!=self._last:

			print(res,end="\r")

			self._last=res

	def done(self):

		print("100%")

	@staticmethod

	def _p(i,n,i0):

		_1=1 if n>=i0 else -1