#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Author: Volkan Şahin <volkansah.in> <bm.volkansahin@gmail.com>

import paramiko

from base.Scope import Scope


class FileTransfer(object):
    def __init__(self, hostname, port, username, password, pkey=None):

        scope = Scope().getInstance()
        self.logger = scope.getLogger()
        self.configuration_manager = scope.getConfigurationManager()

        self.target_hostname = hostname
        self.port = port
        self.target_username = username
        self.target_password = password
        self.p_key = pkey
        self.connection = None

        self.logger.debug('[FileTransfer] Parameters set up')

    def send_file(self, local_path, remote_path):
        self.logger.debug('[FileTransfer]  Sending file ...')

        try:
            sftp = paramiko.SFTPClient.from_transport(self.connection)
            sftp.put(local_path, remote_path)
            self.logger.debug('[FileTransfer] File was sent to %s from %s'.format(local_path,remote_path))
        except Exception as e:
            self.logger.error('[FileTransfer] A problem occurred while sending file. Exception message: %s'.format(str(e)))
        finally:
            self.connection.close()
            self.logger.debug('[FileTransfer] Connection is closed successfully')

    def get_file(self, local_path, remote_path):
        self.logger.debug('[FileTransfer] Getting file ...')
        try:
            sftp = paramiko.SFTPClient.from_transport(self.connection)
            sftp.get(remote_path, local_path)
            self.logger.debug('[FileTransfer] File was downloaded to %s from %s'.format(local_path,remote_path))
        except Exception as e:
            self.logger.error('[FileTransfer] A problem occurred while downloading file. Exception message: %s'.format(str(e)))
        finally:
            self.connection.close()
            self.logger.debug('[FileTransfer] Connection is closed successfully')

    def connect(self):
        self.logger.debug('[FileTransfer]  Connecting to %s via %d'.format(self.target_hostname,self.port))
        try:
            connection = paramiko.Transport((self.target_hostname, self.port))
            connection.connect(username=self.target_username, password=self.target_password, pkey=self.p_key)
            self.connection = connection
            self.logger.debug('[FileTransfer] Connected.')
        except Exception as e:
            self.logger.error('[FileTransfer] A problem occurred while connecting to %s . Exception message: %s'.format(self.target_hostname, str(e)))