import os, struct, sys

def dump_fps4(name, name2, base_path):

    with open(name, 'rb') as f:
    
        head, tail = os.path.split(name)
        file_name = tail.split(".")[0]
        
        fps4 = f.read(4)
        if fps4 != b'FPS4':
            #print("Wrong file.")
            return
            
        file_number = struct.unpack('<L', f.read(4))[0]
        header_size = struct.unpack('<L', f.read(4))[0]
        offset = struct.unpack('<L', f.read(4))[0]
        block_size = struct.unpack('<H', f.read(2))[0]
        
        if block_size == 0x2C:
            f.seek(header_size, 0)
            files_offsets = []
            files_sizes = []
            files_names = []
            
            for i in range(file_number-1):
                files_offsets.append(struct.unpack('<L', f.read(4))[0])
                files_sizes.append(struct.unpack('<L', f.read(4))[0])
                f.read(4) # File size
                fname = f.read(block_size-0xC).decode("ASCII").strip('\x00')
                files_names.append(fname)
                
            try:
                
                os.mkdir(os.path.join(base_path,file_name.upper()))
            except:
                pass
    
            if offset != 0x00:
                for i in range(file_number-1):
                    o = open(os.path.join(base_path, file_name.upper(),files_names[i]), 'wb')
                    f.seek(files_offsets[i], 0)
                    o.write(f.read(files_sizes[i]))
                    o.close()
            else:
                with open(name2, 'rb') as f2:
                    for i in range(file_number-1):
                        o = open(os.path.join(base_path, file_name.upper(), files_names[i]), 'wb')
                        f2.seek(files_offsets[i], 0)
                        o.write(f2.read(files_sizes[i]))
                        o.close()
         

def dump_folder(folder):
    
    os.chdir(folder)

    # Only for m.b & m.dat
    for f in os.listdir(os.getcwd()):
        if '.B' in f:
            dump_fps4(f, f.split('.')[0]+'.MAPBIN')
        print ("Extracting %s" % f)

def pack_folder(folder, dat='.dat'):
    
    buffer = 0
    files_sizes = []
    files_names = []
    b_file = open(folder + '.b', 'wb')
    dat_file = open(folder + dat, 'wb')
    files = [ele for ele in os.listdir(folder) if os.path.isfile(folder + '/' + ele)]
    
    for n in files:
        f = open(os.path.join(folder, n), 'rb')
        files_names.append(n)
        data = f.read()
        dat_file.write(data)
        files_sizes.append(len(data))
        f.close()

    dat_file.close()

    b_file.write(b'\x46\x50\x53\x34') # FPS4
    b_file.write(struct.pack('<L', len(files) + 1))
    b_file.write(struct.pack('<L', 0x1C))
    b_file.write(b'\x00' * 4 + b'\x2C\x00\x0F\x00\x01\x01\x00\x00' +  b'\x00' * 4)

    for i in range(len(files)):
        b_file.write(struct.pack('<L', buffer))
        b_file.write(struct.pack('<L', files_sizes[i]))
        b_file.write(struct.pack('<L', files_sizes[i]))
        b_file.write(files_names[i].encode())
        b_file.write(b'\x00' * (32 - (len(files_names[i]) % 32)))
        buffer += files_sizes[i]

    b_file.write(struct.pack('<L', buffer) + b'\x00'*12)
    b_file.close()

def pack_all(folder):
    
    os.chdir(folder)
    
    for d in os.listdir(os.getcwd()):
        if os.path.isdir(d):
            pack_folder(d)
            print ("Packing %s" % d)

def pack_m(folder):
    
    os.chdir(folder)
    
    for d in os.listdir(os.getcwd()):
        if os.path.isdir(d):
            pack_folder(d, '.MAPBIN')
            print ("Packing %s" % d)



if __name__ == '__main__':
    print("allo")
    if sys.argv[1] == '-d':
        dump_fps4(sys.argv[2], sys.argv[3], sys.argv[4])
    elif sys.argv[1] == '-dm':
        dump_folder(sys.argv[2])
    elif sys.argv[1] == '-i':
        pack_folder(sys.argv[2])
    elif sys.argv[1] == '-ia':
        pack_all(sys.argv[2])
    elif sys.argv[1] == '-im':
        pack_m(sys.argv[2])
        print ("Done!")

        
    sys.exit(1)