import paramiko
import os.path

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()

	def mkdir_p(self,sftp, remote_directory):
		"""Change to this directory, recursively making new folders if needed.
		Returns True if any folders were created. Recursive algorithm."""
		if remote_directory == '/':
			# absolute path so change directory to root
			sftp.chdir('/')
			return
		if remote_directory == '':
			# top-level relative directory must exist
			return
		try:
			sftp.chdir(remote_directory) # sub-directory exists
		except IOError:
			dirname, basename = os.path.split(remote_directory.rstrip('/'))
			self.mkdir_p(sftp, dirname) # make parent directories
			sftp.mkdir(basename) # sub-directory missing, so created it
			sftp.chdir(basename)
		return True
	
	def send_multiple_files_in_directory(self,local_files,directory):
		sftp=self.ssh.open_sftp()
#		try:
#			sftp.chdir(directory)  # Test if remote_path exists
#		except (IOError,FileNotFoundError):
#			sftp.mkdir(directory)  # Create remote_path
#			sftp.chdir(directory)
		self.mkdir_p(sftp, directory)
		for f in set(local_files):	
			sftp.put(f, f)
		sftp.close()