from hashtree import HashTree

import collections

import socket

import logging as log

import config as conf

from networkers import NetworkReader,NetworkWriter

import hashlib

import os

class HashTree:

	HASH_LEN=16 # bytes

	BLOCK_SIZE=4096 # bytes

	## Prepares a tree containing leafCount leaves.

	def fromFile(cls,filename):

		with open(filename,"rb") as f:

			stat=os.fstat(f.fileno())

			size=stat.st_size # !! symlinks

			leafCount=(size-1)//HashTree.BLOCK_SIZE+1 # number of leaf blocks

			res=cls(leafCount)

			for i in range(leafCount):

				data=f.read(HashTree.BLOCK_SIZE)

				res.insertLeaf(hashlib.sha256(data).digest()[HashTree.HASH_LEN:])

		res.buildTree()

		return res

	@classmethod

	def load(cls,filename):

		while index>=0:

			self.store[index]=hashlib.sha256(self.store[index*2+1]+self.store[index*2+2]).digest()[HashTree.HASH_LEN:]

			index=(index-1)//2

	## Fast construction of the tree over the leaves. O(n).

	def buildTree(self):

		for i in range(self.leafStart-1,-1,-1):

			self.store[i]=hashlib.sha256(self.store[i*2+1]+self.store[i*2+2]).digest()[HashTree.HASH_LEN:]

if __name__=="__main__":

	f1=HashTree.fromFile(open("serverFile.txt",mode='rb'))

	f2=HashTree.fromFile(open("clientFile.txt",mode='rb'))

	s.serve()

parser=ArgumentParser()

subparsers=parser.add_subparsers()

pUpdate=subparsers.add_parser("update")

pUpdate.add_argument("-p","--port",type=int)

pUpdate.add_argument("--host",default="127.0.0.1")

pUpdate.add_argument("-t","--tree",help="stored hash tree location")

pUpdate.add_argument("datafile")