import paramiko



class Connection:
	def __init__(self, hostname, port=22, username=None, password=None):
		self.hostname=hostname
		self.port=port
		if(username!=None):
			self.username=username
		else:
			self.username=''
		if(password!=None):
			self.password=password
		else:
			self.password=''
		self.ssh=paramiko.SSHClient()
		self.ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
		self.connected=False
		return
	
	def connect(self, Timeout=5):
		if(not self.connected):
			try:
				print("Trying to connect to: "+self.username+"@"+self.hostname+":"+str(self.port)+".")
				self.ssh.connect(self.hostname, username=self.username, password=self.password, port=self.port, timeout=Timeout)
				self.connected=True
			except:
				print("Error establishing connection with "+self.username+"@"+self.hostname+":"+str(self.port)+".")
				exit(1)
		else:
			print("Already connected!")
		return

	def disconnect(self):
		if(self.connected):
			try:
				self.ssh.close()
			except:
				print("Cannot disconect. Unknown error.")
		else:
			print("Cannot disconect. Already disconnected.")
		self.connected=False

	def execute(self,command):
		if(self.connected):
			try:
				stdin,stdout,stderr=self.ssh.exec_command(command)
				output=stdout.readlines()
				errors=stderr.readlines()
			#	print(errors)
				return(output)
			except:
				print("Cannot execute remote commands")
		else:
			print("Cannot execute remote commands. Connect first.")

	def send_file(self, local, remote):
		sftp=self.ssh.open_sftp()
		sftp.put(local,remote)
		sftp.close()

	def receive_file(self,remote,local):
		sftp=self.ssh.open_sftp()
		sftp.get(remote,local)
		sftp.close()

	def mkdir_remote(self,directory):
		sftp=self.ssh.open_sftp()
		sftp.mkdir(directory)
		sftp.close()