import os
import sys
import getopt

from difflib import unified_diff
from string import Template

thp_template = Template("""format_version = 1
package_name = "$thp_name"
package_version = "$version"
package_version_tuple = [$version_list]
files = [ 
$files
]

additional_files = [ ]
install_order = 50
options = { "cycle-install" : False }
platform = { "os" : "$os",
             "arch" : "$arch" }
require_features = [ "pythonscripts" ]
require_packages = []
scripts = { "python2" : 
              [ $scripts ]
          }
""")

def usage():
    print "Known commands:"
    print "  thpconfig --thp_name=AppName"
    print "            --version_list=1,2,3"
    print "            --scan=path/to/folder/"
    print "            --os=linux"
    print "            --arch=x86"
    print "            --scripts=\"['script1.py', ['preinst', 'postinst']],['script2.py', ['postinst']]\""
    print "            --old_file_list=path/to/oldfiles"
    print "            --generate_file_list=0|1"
    print "            --config_file_list=path/to/configfiles"
    sys.exit(1)

def get_files(top, configs):
    ready = []
    raw = []
    for root, dirs, files in os.walk(top):
        for f in files:
            is_config = "False"
            f_value = "/".join([root, f]).replace(top, "")
            if f_value in configs:
                is_config = "True"
            ready.append("(\"%s\", %s)," % (f_value, is_config))
            raw.append("/".join([root, f]).replace(top, ""))
    return ready, raw

def thpconfig(args):
    optlist, args = getopt.getopt(args, '', ["thp_name=",
                                             "version_list=",
                                             "scan=",
                                             "os=",
                                             "arch=",
                                             "scripts=",
                                             "old_file_list=",
                                             "generate_file_list=",
                                             "config_file_list="])

    mapping = {}
    scan_dir = ""
    old_file_list = ""
    config_file_list = ""
    generate_file_list = True
    for key, val in optlist:
        if key == "--scan":
            scan_dir = val
            continue
        if key == "--generate_file_list":
            generate_file_list = (val == "1")
            continue
        if key == "--old_file_list":
            old_file_list = val
            continue
        if key == "--config_file_list":
            config_file_list = val
            continue
        mapping[key[2:]] = val

    mapping["version"] = ".".join(mapping["version_list"].split(","))

    configs = []
    if len(config_file_list) != 0:
        configs = open(config_file_list, "r").read().split("\n")
        
    files, raw = get_files(scan_dir, configs)
    mapping["files"] = "\n".join(files)

    out = open("%s-%s_thp.cfg" % (mapping["thp_name"], mapping["version"]), "w")
    try:
        out.write(thp_template.substitute(mapping))
    except KeyError, e:
        print "You are missing the following parameter:", e
        sys.exit(1)
    out.close()

    if generate_file_list:
        file_list = open("%s-%s_thp.filelist" % (mapping["thp_name"], mapping["version"]), "w")
        file_list.write("\n".join(raw))
        file_list.close()

    if len(old_file_list) != 0:
        old = open(old_file_list, "r").read().split("\n")
        new = raw
        for line in unified_diff(old, new):
            print line

def main():
    if len(sys.argv) < 2:
        usage()
    cmd = sys.argv[1]
    args = sys.argv[2:]
    if cmd in [ "thpconfig" ]:
        globals()[cmd](args)
    else:
        usage()


if __name__ == "__main__":
    main()