import click
import schainpy
import subprocess
from multiprocessing import cpu_count
from schaincli import templates
from schainpy import controller_api
import os
import sys
import glob

def print_version(ctx, param, value):
    if not value or ctx.resilient_parsing:
        return
    click.echo(schainpy.__version__)
    ctx.exit()


@click.command()
@click.option('--version', '-v', is_flag=True, callback=print_version, help='SChain version', type=str)
@click.option('--xml', '-x', default=None, help='run an XML file', type=click.Path(exists=True, resolve_path=True))
@click.argument('command', default='run', required=True)
@click.argument('nextcommand', default=None, required=False, type=click.Path(exists=True, resolve_path=True))
def main(command, nextcommand, version, xml):
    """COMMAND LINE INTERFACE FOR SIGNAL CHAIN - JICAMARCA RADIO OBSERVATORY"""
    if xml is not None:
        runFromXML(xml)
    elif command == 'generate':
        generate()
    elif command == 'test':
        test()
    elif command == 'run':
        if nextcommand is None:
            currentfiles = glob.glob('./*.py')
            numberfiles = len(currentfiles)
            print currentfiles
            if numberfiles > 1:
                click.echo('\x1b[6;37;41m[ERROR] - There is more than one file to run\x1b[0m')
            elif numberfiles == 1:
                subprocess.call(['python ' + currentfiles[0]], shell=True)
            else:
                click.echo('\x1b[6;37;41m[ERROR] - There is no file to run.\x1b[0m')
        else:
            subprocess.call(['python ' + nextcommand], shell=True)
    else:
        click.echo('\x1b[6;37;41m[ERROR] - Command is not defined.\x1b[0m')


def basicInputs():
    inputs = {}
    inputs['desc'] = click.prompt('Enter a description', default="A schain project", type=str)
    inputs['name'] = click.prompt('Name of the project', default="project", type=str)
    inputs['path'] = click.prompt('Data path', default=os.getcwd(), type=click.Path(exists=True, resolve_path=True))
    inputs['startDate'] = click.prompt('Start date', default='1970/01/01', type=str)
    inputs['endDate'] = click.prompt('End date', default='2017/12/31', type=str)
    inputs['startHour'] = click.prompt('Start hour', default='00:00:00', type=str)
    inputs['endHour'] = click.prompt('End hour', default='23:59:59', type=str)
    inputs['figpath'] = inputs['path'] + '/figs'
    return inputs


def generate():
    inputs = basicInputs()
    inputs['multiprocess'] = click.confirm('Is this a multiprocess script?')
    if inputs['multiprocess']:
        inputs['nProcess'] = click.prompt('How many process?', default=cpu_count(), type=int)
        current = templates.multiprocess.format(**inputs)
    else:
        current = templates.basic.format(**inputs)
    scriptname = inputs['name'] + ".py"
    script = open(scriptname, 'w')
    try:
        script.write(current)
        click.echo('\x1b[6;37;42m[SUCCESS] Script {file} generated\x1b[0m'.format(file=scriptname))
    except Exception as e:
        click.echo('\x1b[6;37;41m[ERROR] I cannot create the file. Do you have writing permissions?\x1b[0m')


def test():
    print templates.basic.format(name='hola', desc='desc', path='path', startDate='0', endDate='0')
    click.echo('testing')


def runFromXML(filename):
    controller = controller_api.ControllerThread()
    if not controller.readXml(filename):
        return

    plotterObj = controller.useExternalPlotter()

    controller.start()
    plotterObj.start()

    print "Finishing all processes ..."

    controller.join(5)

    print "End of script"
    return