fiatlux/firmware/mkwebfs.py
7m9 8a9e1b6708
All checks were successful
continuous-integration/drone/push Build is passing
Added gzip compression
2021-09-09 02:35:15 +02:00

102 lines
3.7 KiB
Python
Executable file

#!/usr/bin/env python3
import os
import gzip
incHttpHeader = True
useCompression = True
fileDir = "fsdata/fs/"
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('fsdata/fsdata.c', 'w')
f_fsdata_c.write('#include "httpd/fsdata.h"\n\n')
httpFiles = []
for root, dirs, files in os.walk(fileDir):
for file in files:
httpFiles.append("{}/{}".format(root, file).replace("//", "/"))
lastFileStruct = "NULL"
for file in httpFiles:
response = b''
if incHttpHeader:
print(file)
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 useCompression:
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
webPath = ("/" + file.replace(fileDir, "")).replace("//", "/")
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 incHttpHeader == 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)))