'''
    Handling of compressed MacOS resources.
    Author: Max Poliakovski 2018
'''
import struct
import sys

from GreggDecompress import GreggDecompress, GreggCompress


def DecompressResource(inf):
    # get the extended resource header first
    hdrFields = struct.unpack(">IHBBI", inf.read(12))
    if hdrFields[0] != 0xA89F6572:
        print("Invalid extended resource header sig: 0x%X" % hdrFields[0])
    if hdrFields[1] != 18:
        print("Suspicious extended resource header length: %d" % hdrFields[1])
    if hdrFields[2] != 8 and hdrFields[2] != 9:
        print("Unknown ext res header format: %d" % hdrFields[2])
    if (hdrFields[3] & 1) == 0:
        print("extAttributes,bit0 isn't set. Treat this res as uncompressed.")

    print("Uncompressed length: %d" % hdrFields[4])

    if hdrFields[2] == 8:
        DonnSpecific = struct.unpack(">BBHH", inf.read(6))
        print("DonnDecompress isn't supported yet.")
        exit()
    else:
        GreggSpecific = struct.unpack(">HHBB", inf.read(6))

    fsize = inf.seek(0, 2)
    print("Compressed size: %d" % fsize)
    inf.seek(hdrFields[1], 0) # rewind to the start of compressed data

    dstBuf = bytearray()
    srcBuf = inf.read(fsize - hdrFields[1])

    # invoke GreggyBits decompressor and pass over required header data
    GreggDecompress(srcBuf, dstBuf, hdrFields[4], GreggSpecific[2], GreggSpecific[3])

    with open("Dump", 'wb') as outstream:
        outstream.write(dstBuf)

    # re-compress
    recompBuf = bytearray()

    # re-create extended resource header
    recompBuf.extend([0xA8, 0x9F, 0x65, 0x72, 0x00, 0x12, 0x09, 0x01])
    recompBuf.extend(hdrFields[4].to_bytes(4, 'big'))
    recompBuf.extend([0x00, 0x02, 0x00, 0x00])
    recompBuf.append(GreggSpecific[2])
    recompBuf.append(GreggSpecific[3])

    GreggCompress(dstBuf, recompBuf, hdrFields[4], customTab=True, isBitmapped=True)

    with open("RecompDump", 'wb') as outstream:
        outstream.write(recompBuf)


if __name__ == "__main__":
    if len(sys.argv) < 2:
        file = "Compressed"
    else:
        file = sys.argv[1]

    with open(file, 'rb') as instream:
        DecompressResource(instream)