#!/usr/bin/env python3

import os
import argparse
import glob

# figure out what the name of the h5 file to put on site is in the given path
def get_H5_filename(path):
    return glob.glob("*.he5")[0]

def build_parser():
    dbHost = "insarmaps.rsmas.miami.edu"
    parser = argparse.ArgumentParser(description='Convert a Unavco format H5 file for ingestion into insarmaps.')
    required = parser.add_argument_group("required arguments")
    required.add_argument("-f", "--file", help="unavco file to ingest", required=True)
    required.add_argument("-u", "--user", help="username for the insarmaps database", required=True)
    required.add_argument("-p", "--password", help="password for the insarmaps database", required=True)
    required.add_argument("--host", default=dbHost, help="postgres DB URL for insarmaps database", required=True)
    required.add_argument("-U", "--server_user", help="username for the insarmaps server (the machine where the tileserver and http server reside)", required=True)
    parser.add_argument("-P", "--server_password", help="password for the insarmaps server (the machine where the tileserver and http server reside)", required=True)

    return parser

def main():
    parser = build_parser()
    parseArgs = parser.parse_args()

    dbUsername = parseArgs.user
    dbPassword = parseArgs.password
    dbHost = parseArgs.host
    serverUser = parseArgs.server_user
    serverPassword = parseArgs.server_password

    bjobScriptFilename = "run_pysar2insarmaps.py"

    path = parseArgs.file

    h5FileFullName = parseArgs.file

    curProjName = h5FileFullName.split(".")[0]

    jsonFolder = "json/"
    mbtilesFile = jsonFolder + curProjName + ".mbtiles"

# create working directory in scratch and copy relevant files over
    scratch_dir = os.environ["SCRATCHDIR"] + "/" + curProjName
    print("making directory " + scratch_dir)
    os.system("mkdir " + scratch_dir)
    command = "cp " + h5FileFullName + " " + scratch_dir + "/"
    print("copying files to scratch with command " + command)
    os.system(command)

# go to scratch dir, and run the bjob command
    unavcoToJsonMbtilesCommand = "unavco2json_mbtiles.py " + h5FileFullName + " " + jsonFolder

    jsonMbtilesToInsarmapsCommand = "json_mbtiles2insarmaps.py -u " + dbUsername + " -p " + dbPassword + " -U " + serverUser + " -P " + serverPassword + " --mbtiles_file " + mbtilesFile + " --json_folder " + jsonFolder + " --host " + dbHost

    command = "echo '" + unavcoToJsonMbtilesCommand + " && " + jsonMbtilesToInsarmapsCommand + "' > " + bjobScriptFilename

    os.chdir(scratch_dir)
    os.system(command)

    os.system("createBatch.pl " + bjobScriptFilename)
    print("bjob finished")

if __name__ == '__main__':
    main()