#!/usr/bin/env python3
import os
import gzip
import argparse

parser = argparse.ArgumentParser()
parser.add_argument('-o', '--output', help='Output file name', default='stdout')
parser.add_argument('-W', '--webroot', help='Output file name', default='webdir/')
parser.add_argument('--gzip', dest='gzip', action='store_true')
parser.add_argument('--no-gzip', dest='gzip', action='store_false')
parser.set_defaults(gzip=False)
parser.add_argument('--header', dest='header', action='store_true')
parser.add_argument('--no-header', dest='header', action='store_false')
parser.set_defaults(header=True)
parser.add_argument('input', nargs='+', default=os.getcwd())
args = parser.parse_args()


def dumpBin2CHex(f, b):
    oStr = "\t"
    n = 0
    for val in b:
        oStr += hex(val) + ", "
        n += 1
        if n % 8 == 0:
            oStr += "\n\t"
    oStr += "\n"
    f.write(oStr)


f_fsdata_c = open(args.output, 'w')
f_fsdata_c.write('#include "httpd/fsdata.h"\n\n')

httpFiles = [file for file in args.input if (args.webroot in file)]

lastFileStruct = "NULL"

for file in httpFiles:
    response = b''

    webPath = ("/" + file.removeprefix(args.webroot)).replace("//", "/")
    print("{} > {}".format(file, webPath))

    if args.header:
        if ("404" in file):
            response = b'HTTP/1.0 404 File not found\r\n'
        else:
            response = b'HTTP/1.0 200 OK\r\n'
        response += b"lwIP/1.4.1 (http://savannah.nongnu.org/projects/lwip)\r\n"
        fext = file.split('.')[-1]
        ctype = b'Content-type: text/plain\r\n'
        if (fext.endswith("html") or fext.endswith("htm") or fext.endswith("shtml") or fext.endswith(
                "shtm") or fext.endswith("ssi")):
            ctype = b'Content-type: text/html\r\n'
        if (fext.endswith("js")):
            ctype = b'Content-type: application/x-javascript\r\n'
        if (fext.endswith("css")):
            ctype = b'Content-type: text/css\r\n'
        if (fext.endswith("ico")):
            ctype = b'Content-type: image/x-icon\r\n'
        if (fext.endswith("gif")):
            ctype = b'Content-type: image/gif\r\n'
        if (fext.endswith("png")):
            ctype = b'Content-type: image/png\r\n'
        if(fext.endswith("jpg")):
            ctype = b'Content-type: image/jpeg\r\n'
        if(fext.endswith("bmp")):
            ctype = b'Content-type: image/bmp\r\n'
        if(fext.endswith("class")):
            ctype = b'Content-type: application/octet-stream\r\n'
        if(fext.endswith("ram")):
            ctype = b'Content-type: audio/x-pn-realaudio\r\n'
        response += ctype

        binFile = open(file, 'rb')
        binData = binFile.read()
        compEff = False
        if args.gzip:
            compData = gzip.compress(binData, 9)
            if len(compData) < len(binData):
                compEff = True
                print("- Compressed from {} to {}".format(len(binData), len(compData)))
                binData = compData
            else:
                print("- Compression skipped Orig: {} Comp: {}".format(len(binData), len(compData)))
        binFile.close()

        if compEff:
            response += b'Content-Encoding: gzip\r\n'
        response += b"\r\n"
        response += binData
        binFile.close()
        escFile = file.replace("/", "_").replace(".", "_")
        escFileData = "data_" + escFile
        escFileFile = "file_" + escFile

        f_fsdata_c.write('static const unsigned char {}[] = {{\n'.format(escFileData))
        f_fsdata_c.write('\t/* LOCAL:{} */\n'.format(file))
        f_fsdata_c.write('\t/* WEB:  {} */\n'.format(webPath))
        fnameBin = webPath.encode("ascii") + b'\0'
        dumpBin2CHex(f_fsdata_c, fnameBin)
        dumpBin2CHex(f_fsdata_c, response)
        f_fsdata_c.write("};\n\n")

        f_fsdata_c.write("const struct fsdata_file {}[] = {{{{\n  {},\n  {}, {} + {}, sizeof({}) - {}, 1 }}}};\n\n"
                .format(escFileFile, lastFileStruct, escFileData, escFileData, len(fnameBin), escFileData, len(fnameBin)))
        # TODO: The last value is 1 if args.header == True
        lastFileStruct = escFileFile

f_fsdata_c.write("\n")
f_fsdata_c.write("#define FS_ROOT {}\n\n".format(lastFileStruct))
f_fsdata_c.write("#define FS_NUMFILES {}\n\n".format(len(httpFiles)))