#from __future__ import with_statement
from fabric import Connection
from util import find_data_file
from util import setup_child
from util import fprint
from util import macos
from invoke import exceptions
import sys

c = None
settings = None

def sftp_send_data(config, filename, filetype):
    setup_child()
    if not macos:
        fprint("Connecting over SSH to " + config['sftp']['host'], settings)
        global c
        if c is None:
            c = Connection(host=config['sftp']['host'], user=config['sftp']['user'], port=config['sftp']['port'], connect_kwargs={"key_filename": find_data_file(config['sftp']['keyfile']),})

        fprint("Sending data over SFTP: " + filename, settings)
        fprint(c.put(find_data_file(filename), remote=config['sftp']['filepath'][filetype]), settings)
        fprint("Data sent over SFTP successfully", settings)
    #command = 'ls ' + config['sftp']['filepath'][filetype]
    #fprint(c.run(command))

def check_for_file(config, filename, location):
    setup_child()
    fprint("Connecting over SSH to " + config['sftp']['host'], settings)
    global c
    if c is None:
        c = Connection(host=config['sftp']['host'], user=config['sftp']['user'], port=config['sftp']['port'], connect_kwargs={"key_filename": find_data_file(config['sftp']['keyfile']),})
    fprint("Checking for existence of file " + config['sftp']['filepath'][location] + "/" + filename, settings)
    try:
        res = c.run("ls -l " + config['sftp']['filepath'][location] + "/" + filename, hide=True)
        fprint("File " + filename + " exists!", settings)
        return c.run("cat " + config['sftp']['filepath'][location] + "/" + filename, hide=True)
    except exceptions.UnexpectedExit:
        return False

def run_ssh(config, command, location):
    setup_child()
    fprint("Connecting over SSH to " + config['sftp']['host'], settings)
    global c
    if c is None:
        c = Connection(host=config['sftp']['host'], user=config['sftp']['user'], port=config['sftp']['port'], connect_kwargs={"key_filename": find_data_file(config['sftp']['keyfile']),})
    fprint("cd to " + config['sftp']['filepath'][location], settings)
    with c.cd(config['sftp']['filepath'][location]):
        fprint("Running ssh command: " + command, settings)
        res = c.run(command, hide=True, asynchronous=True)
        return res