from core.experience import Experience, ExperienceParameter


class MpPvAt(object):
	def __init__(self, at, cmd):
		self.at = at
		self.cmd = cmd
		self.delta = 0

	def __str__(self):
		return "Pv... at " + str(self.at) + "(" + str(self.delta) + \
				") will be " + self.cmd


class  ExperienceNCPV(Experience):
	"""
	NC PV : NetCat and Pipe Viewer
	"""
	SERVER_NC_LOG = "netcat_server"
	CLIENT_NC_LOG = "netcat_client"
	NC_BIN = "/usr/local/bin/nc"
	PV_BIN = "/usr/local/bin/pv"
	PING_OUTPUT = "ping.log"

	def __init__(self, xpParamFile, mpTopo, mpConfig):
		Experience.__init__(self, xpParamFile, mpTopo, mpConfig)
		self.loadParam()
		self.ping()
		Experience.classicRun(self)

	def ping(self):
		self.mpTopo.commandTo(self.mpConfig.client, "rm " + \
				ExperienceNCPV.PING_OUTPUT )
		count = self.xpParam.getParam(ExperienceParameter.PINGCOUNT)
		for i in range(0, self.mpConfig.getClientInterfaceCount()):
			 cmd = self.pingCommand(self.mpConfig.getClientIP(i),
				 self.mpConfig.getServerIP(), n = count)
			 self.mpTopo.commandTo(self.mpConfig.client, cmd)

	def pingCommand(self, fromIP, toIP, n=5):
		s = "ping -c " + str(n) + " -I " + fromIP + " " + toIP + \
				  " >> " + ExperienceNCPV.PING_OUTPUT
		print(s)
		return s

	def loadParam(self):
		self.pvg = self.xpParam.getParam(ExperienceParameter.PVG)
		self.pvz = self.xpParam.getParam(ExperienceParameter.PVZ)
		self.pvRateLimit = self.xpParam.getParam(ExperienceParameter.PVRATELIMIT)
		self.ddibs = self.xpParam.getParam(ExperienceParameter.DDIBS)
		self.ddobs = self.xpParam.getParam(ExperienceParameter.DDOBS)
		self.ddcount = self.xpParam.getParam(ExperienceParameter.DDCOUNT)
		self.ncServerPort = self.xpParam.getParam(ExperienceParameter.NCSERVERPORT)
		self.pvRateLimit = self.xpParam.getParam(ExperienceParameter.PVRATELIMIT)
		self.ncClientPort = []
		for k in sorted(self.xpParam.paramDic):
			if k.startswith(ExperienceParameter.NCCLIENTPORT):
				port = self.xpParam.paramDic[k]
				self.ncClientPort.append(port)
		if len(self.ncClientPort) == 0:
			d = self.xpParam.getParam(ExperienceParameter.NCCLIENTPORT)
			self.ncClientPort.append(d)
		self.loadPvAt()

	def loadPvAt(self):
		self.changePvAt = []
		self.changePv = self.xpParam.getParam(ExperienceParameter.CHANGEPV)
		if self.changePv != "yes":
			print("Don't change pv rate...")
			return
		changePvAt = self.xpParam.getParam(ExperienceParameter.CHANGEPVAT)
		if not isinstance(changePvAt, list):
			changePvAt = [changePvAt]
		for p in changePvAt:
			tab = p.split(",")
			if len(tab)==2:
				o = MpPvAt(float(tab[0]), tab[1])
				self.addPvAt(o)
			else:
				print("pv wrong line : " + p)

	def addPvAt(self, p):
		if len(self.changePvAt) == 0 :
			p.delta = p.at
		else:
			if p.at > self.changePvAt[-1].at:
				p.delta = p.at - self.changePvAt[-1].at
			else:
				print("Do not take into account " + p.__str__() + \
						"because ooo !")
				return

		self.changePvAt.append(p)

	def getPvChangeCmd(self):
		cmd = ""
		for p in self.changePvAt:
			cmd = cmd + "sleep " + str(p.delta)
			cmd = cmd + " && "
			cmd = cmd + ExperienceNCPV.PV_BIN + " -R " + self.pvPid
			cmd = cmd + " " + p.cmd + " && "
		cmd = cmd + " true &"
		return cmd

	def prepare(self):
		Experience.prepare(self)
		self.mpTopo.commandTo(self.mpConfig.client, "rm " + \
				ExperienceNCPV.CLIENT_NC_LOG )
		self.mpTopo.commandTo(self.mpConfig.server, "rm " + \
				ExperienceNCPV.SERVER_NC_LOG )

	def getNCServerCmd(self, id):
		s = ExperienceNCPV.NC_BIN + " -d  " + \
				" -l " + self.ncServerPort  + \
				" 1>/dev/null 2>" + ExperienceNCPV.SERVER_NC_LOG + \
				"_" + str(id) + ".log &"
		print(s)
		return s

	def getNCClientCmd(self, id):
		s = "dd if=/dev/urandom ibs=" + self.ddibs + \
				" obs=" + self.ddobs + \
				" count=" + self.ddcount + \
				" | " + ExperienceNCPV.PV_BIN + \
				" -g " + self.pvg + " -z " + self.pvz + \
				" -q --rate-limit " + self.pvRateLimit + \
				" | " + ExperienceNCPV.NC_BIN + " " + \
				"  -p " + self.ncClientPort[id] + " " + \
				self.mpConfig.getServerIP() + " " + \
				self.ncServerPort + " " + \
				"&>" + ExperienceNCPV.CLIENT_NC_LOG + \
				"_" + str(id) + ".log"
		print(s)
		return s
	def getPvPidCmd(self):
		s = "pgrep -n pv"
		return s

	def clean(self):
		Experience.clean(self)
		#todo use cst
		self.mpTopo.commandTo(self.mpConfig.server, "killall netcat")


	def run(self):
		for i in range(0, len(self.ncClientPort)):
			cmd = self.getNCServerCmd(i)
			self.mpTopo.commandTo(self.mpConfig.server, cmd)

			cmd = self.getNCClientCmd(i)
			self.mpConfig.client.sendCmd(cmd)

			cmd = self.getPvPidCmd()
			self.pvPid = None
			while self.pvPid == None or self.pvPid == "": 
				self.pvPid = self.mpTopo.commandTo(self.mpConfig.server, cmd)[:-1]
				print("guessing pv pid ... :" + str(self.pvPid))

			cmd = self.getPvChangeCmd()
			print(cmd)
			self.mpTopo.commandTo(self.mpConfig.server, cmd)


			self.mpConfig.client.waitOutput()

			self.mpTopo.commandTo(self.mpConfig.client, "sleep 1")