import paramiko
import time
import sys
import json
import threading
import logging


login = 'user1'
password = 'pass1234'
searchpass = []
port = 22

class LogPipe(threading.Thread):
	def __init__(self, level):
		threading.Thread.__init__(self)
		self.daemon = False
		self.level = level
		self.fdRead, self.fdWrite = os.pipe()
		self.pipeReader = os.fdopen(self.fdRead)
		self.start()

	def fileno(self):
		return self.fdWrite

	def run(self):
		for line in iter(self.pipeReader.readline, ''):
			logging.log(self.level, line.strip('\n'))
		self.pipeReader.close()

	def close(self):
		os.close(self.fdWrite)

def execute_ssh_command(host, port, username, password, command):
    try:
        # Create the SSH client.
        ssh = paramiko.SSHClient()
        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
 
        # Connect to the host.
        ssh.connect(host, port, username, password, look_for_keys=False)
 
        # Send the command (non-blocking)
        stdin, stdout, stderr = ssh.exec_command(command)
 
        # Wait for the command to terminate
        while not stdout.channel.exit_status_ready() and not stdout.channel.recv_ready():
            time.sleep(1)
 
        stdoutstring = stdout.readlines()
        stderrstring = stderr.readlines()
        return stdoutstring, stderrstring
    finally:
        if ssh is not None:
            # Close client connection.
            ssh.close()

def sendCommand (hostname,command):
	returnvalue = 0
	logging.info('Host '+hostname+', command: '+command)
	Try:
#add .mydomain for FQDN
		(stdoutstring, stderrstring) = execute_ssh_command(hostname+'.mydomain', port, login, password, command+'\n')
		if (len(stdoutstring)>0):
			logging.info(stdoutstring[0])
		if (len(stderrstring)>0):
			logging.info(stderrstring[0])
	except Exception as e:
		return returnvalue
	else:
		returnvalue = stdoutstring[0]
	finally:
		return returnvalue
			
			
def checkswitch(hostname,mac_addr):
	try:
		returnvalue = {}
		returnvalue['host']=hostname
		answer = sendCommand(hostname,'show ethernet-switching table | match '+mac_addr)
		if(answer!=0):
			iface = answer.split()[4]
			returnvalue['iface']=iface
			#cut .0 prefix in the interface name
			answer = sendCommand(hostname,'show interfaces '+iface[:-2]+' descriptions | last 1 | no-more')
			iface = answer.split()
			if(len(iface)>2):
				iface=iface[3]
				returnvalue['description']=iface
			else:
				returnvalue['description']='none'
			searchpass.append(returnvalue)
			if (iface[:3]=='SW>'):
				checkswitch(iface[3:],mac_addr)
		else:
			returnvalue['iface']='none'
			searchpass.append(returnvalue)
	except Exception as e:
		logging.info(e)

			
def main(argv):

	mac_addr = argv[0]
	
	#configure log
	logging.basicConfig(filename='/var/log/findmac.log', level=logging.INFO, format='%(asctime)s %(message)s')
	logging.info('Find MAC: '+mac_addr)
	checkswitch('CORE',mac_addr)
	for switch in searchpass:
		print (json.dumps(switch, ensure_ascii=False))

if __name__ == "__main__":
	main(sys.argv[1:])