#!/usr/bin/env python3

# Bekaert David


import os
import glob
import argparse


def createParser():
    '''
    Create command line parser.
    '''

    parser = argparse.ArgumentParser(description='Script that attempts to recognize the sensor automatically and then call the correspodning unzips/unpacks command.')
    parser.add_argument('-i', '--input', dest='input', type=str, required=True,
            help='directory which has all the slc data.')
    parser.add_argument('-rmfile', '--rmfile', dest='rmfile',action='store_true', default=False,
            help='Optional: remove zip/tar/compressed files after unpacking into date structure (default is to keep in archive folder)')
    parser.add_argument('-o', '--output', dest='output', type=str, required=False,
            help='Optional: output directory which will be used for unpacking into isce format (run file generation only).')
    parser.add_argument('-t', '--text_cmd', dest='text_cmd', type=str, default='source ~/.bash_profile;'
            , help='Optional: text command to be added to the beginning of each line of the run files. Default: source ~/.bash_profile;')

    return parser

def cmdLineParse(iargs=None):
    '''
    Command line parser.
    '''

    parser = createParser()
    return parser.parse_args(args = iargs)

def main(iargs=None):
    '''
    The main driver.
    '''

    inps = cmdLineParse(iargs)
    ## parsing of the required input arguments
    inputDir = os.path.abspath(inps.input)
    ## parsing of the optional input arguments
    # output dir for generating runfile
    if inps.output:
        outputDir = os.path.abspath(inps.output)
        outputDir_str = ' -o ' + outputDir
    else:
        outputDir_str = ''
    # textcommand to be added to the runfile start
    text_str = ' -t "' + inps.text_cmd + '"'
    # delete zip file option
    if inps.rmfile: 
        rmfile_str = ' -rmfile'
    else:
        rmfile_str = ''

    # search criteria for the different sensors
    ENV_str = 'ASA*'		    # Envisat
    ALOS1_str = 'ALPSRP*'	    # ALOS-1/ALOS-2 Palsar, zip files and extracted files
    CSK_str = 'EL*'		    # CSK, zip files
    CSK_str2 = 'CSK*.h5'            # CSK, extracted files
    RSAT2_str = 'RS2*SLC*'          # RSAT2 zip files
    RSAT2_str2 = 'imagery_HH.tif'   # RSAT2 extracted files
    TSX_TDX_str = 'dims_op*'        # TSX zip files
    TSX_TDX_str2 = 'T*X*.xml'       # TSX extracted files
    GF3_str = 'GF3*'                # GF3
    GF3_str2 = 'GF3*.meta.xml'      # GF3 meta xml
    HJ2_str = 'HJ2*'                # HJ2
    HJ2_str2 = 'HJ2*.xml'           # HJ2 meta xml

    # combine together
    sensor_str_list = (ENV_str,ALOS1_str,CSK_str,CSK_str2,RSAT2_str,RSAT2_str2,TSX_TDX_str,TSX_TDX_str2,GF3_str,GF3_str2,HJ2_str,HJ2_str2)
    sensor_list = ('Envisat','ALOS1','CSK','CSK','RSAT2','RSAT2','TSX/TDX','TSX/TDX','GF3','GF3','HJ2','HJ2')
    sensor_unpackcommand = ('TODO','TODO','TODO','TODO','prepSlcRSAT2.py','prepSlcRSAT2.py','TODO','TODO','prepSlcGF3.py','prepSlcGF3.py','prepSlcHJ2.py','prepSlcHJ2.py')
    Sensors = dict(zip(sensor_str_list,sensor_list))
    Sensors_unpack = dict(zip(sensor_str_list,sensor_unpackcommand))

    # Loop over the different sensor strings and try to find them
    sensor_found = False
    for sensor_str in Sensors:
        files = glob.iglob(os.path.join(inputDir,'**',sensor_str),recursive=True)
        for file in files:
            sensor_found = True
            print(file)
            sensor_str_keep = sensor_str
            break

    # report back to user
    if sensor_found:
        print("Looks like " + Sensors[sensor_str_keep])
        cmd  = Sensors_unpack[sensor_str_keep] + ' -i ' + inputDir + rmfile_str + outputDir_str + text_str
        print(cmd)
        os.system(cmd)

    else:
        print("Did not find the sensor automatically, unzip and run unpack routines manual")


if __name__ == '__main__':

    main()