patcher.py

from zipfile import ZipFile
import os.path
from os import remove, mkdir, rename
from shutil import copy, copytree, rmtree, move
import glob
import subprocess

version = '15_2'

output = 'am2r_' + version

# Thanks, stackoverflow
def _find_getch():
    try:
        import termios
    except ImportError:
        # Non-POSIX. Return msvcrt's (Windows') getch.
        import msvcrt
        return msvcrt.getch

    # POSIX system. Create and return a getch that manipulates the tty.
    import sys, tty
    def _getch():
        fd = sys.stdin.fileno()
        old_settings = termios.tcgetattr(fd)
        try:
            tty.setraw(fd)
            ch = sys.stdin.read(1)
        finally:
            termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
        return ch

    return _getch

getch = _find_getch()
# End stackoverflow code


# Residual file cleanup to prevent accidents
if os.path.isdir('utilities/android/assets'):
    rmtree('utilities/android/assets')

if os.path.isdir(output):
    rmtree(output)

print("-------------------------------------------\n\nAM2R 1.5.1 Python Autopatching Utility\nScripted by Lojemiru\n\n-------------------------------------------\n")

# Check for AM2R_11.zip...
if os.path.isfile('AM2R_11.zip'):
    print("AM2R_11.zip found! Extracting to " + output)
    # Unzip AM2R_11.zip
    try:
        with ZipFile('AM2R_11.zip', 'r') as zipObj:
           # Extract all the contents of zip file in current directory
           zipObj.extractall(output)
    except:
        print("AM2R_11.zip failed to extract. Try unzipping the folder and run this utility again.")
        quit()
else:
    if os.path.isdir('AM2R_11'):
        print("AM2R_11 found! Copying to " + output)
    else:
        print("AM2R_11 not found. Place AM2R_11.zip in this folder and try again.")
        quit()

print("\nSelect your patch type:\n\n1 - Linux\n2 - Android\n\nAwaiting input:\n")

type = getch()

# Determine type
if (type == '1'):
    print("Linux selected.\nApplying AM2R patch...")
    # apply AM2R.bps
    # subprocess.call(['utilities/floating/./flips-linux', '-a', 'patch_data/AM2R.bps', output+'/AM2R.exe', output+'/AM2R'])
    subprocess.call(['xdelta3', '-dfs', output+'/AM2R.exe', 'patch_data/AM2R.xdelta', output+'/AM2R'])

    print("\nApplying data patch...")
    # apply game.unx patch
    # subprocess.call(['utilities/floating/./flips-linux', '-a', 'patch_data/game.bps', output+'/data.win', output+'/game.unx'])
    subprocess.call(['xdelta3', '-dfs', output+'/data.win', 'patch_data/game.xdelta', output+'/game.unx'])

    # clean up Windows files
    print("\nCleaning up residual AM2R 1.1 files...")
    remove(output+'/AM2R.exe')
    remove(output+'/data.win')

    # structure for Linux

    print("\nFormatting game directory...")
    os.mkdir(output+'/assets')

    for z in glob.glob(output+'/*.*'):
        move(z, output+'/assets')

    # install new datafiles...
    print("\nInstalling new datafiles...")
    # music
    for file2 in glob.glob("patch_data/files_to_copy/*.ogg"):
        copy(file2, output+'/assets')

    print("\nInstall high quality in-game music? Increases filesize by 194 MB!\n\n[y/n]\n")

    hq = getch()

    # install HQ music
    if (hq == 'y'):
        print("Copying HQ music...")
        for file in glob.glob("patch_data/HDR_HQ_in-game_music/*.ogg"):
            copy(file, output+'/assets')

    # format music for linux
    print("Formatting music files for Linux application recognition...")

    for music in glob.glob(output+'/assets/*.ogg'):
        rename(music, music.lower())

    # remove old lang
    rmtree(output+'/lang')
    # install new lang
    copytree('patch_data/files_to_copy/lang', output+'/assets/lang')
    # install mods,
    copytree('patch_data/files_to_copy/mods', output+'/assets/mods')
    # text
    for file3 in glob.glob("patch_data/files_to_copy/*.txt"):
        copy(file3, output+'/assets')
    # modifiers
    copy('patch_data/files_to_copy/modifiers.ini', output+'/assets')
    # icons
    copy('patch_data/files_to_copy/icon.png', output+'/assets')
    copy('patch_data/files_to_copy/splash.png', output+'/assets')


elif (type == '2'):
    print("Android selected.\nApplying data patch...")
    # To be completely honest I don't know why I made the Android patch work like this.
    # I'm going to revert it until I run into problems.

    # subprocess.call(['utilities/floating/./flips-linux', '-a', 'patch_data/game.bps', output+'/data.win', output+'/game.unx'])
    # subprocess.call(['xdelta3', '-dfs', output+'/data.win', 'patch_data/game.bps', output+'/game.unx'])

    print("\nApplying Android patch...")
    # subprocess.call(['utilities/floating/./flips-linux', '-a', 'patch_data/droid.bps', output+'/game.unx', output+'/game.droid'])
    subprocess.call(['xdelta3', '-dfs', output+'/data.win', 'patch_data/droid.xdelta', output+'/game.droid'])

    copy("patch_data/android/AM2RWrapper.apk", "utilities/android/")

    remove(output+"/D3DX9_43.dll")
    remove(output+"/AM2R.exe")
    remove(output+"/data.win")
    # remove(output+"/game.unx")

    copytree(output, "utilities/android/assets")
    copy("patch_data/android/AM2R.ini", "utilities/android/assets")

    # Install new datafiles...
    print("\nInstalling new datafiles...")
    # Music
    for file2 in glob.glob("patch_data/files_to_copy/*.ogg"):
        copy(file2, 'utilities/android/assets')

    print("\nInstall high quality in-game music? Increases filesize by 194 MB!\n\n[y/n]\n")

    hq = getch()

    # Install HQ music
    if (hq == 'y'):
        print("Copying HQ music...")
        for file in glob.glob("patch_data/HDR_HQ_in-game_music/*.ogg"):
            copy(file, 'utilities/android/assets')

    # Remove old lang
    rmtree('utilities/android/assets/lang')
    # Install new lang
    copytree('patch_data/files_to_copy/lang', 'utilities/android/assets/lang')
    # Text
    for file3 in glob.glob("patch_data/files_to_copy/*.txt"):
        copy(file3, 'utilities/android/assets')
    # Modifiers
    copy('patch_data/files_to_copy/modifiers.ini', 'utilities/android/assets')

    os.chdir('utilities/android')

    print("\nPackaging APK.")

    # General assets
    for file4 in glob.glob("assets/*.*"):
        subprocess.call(['aapt', 'add', '-f', '-v', 'AM2RWrapper.apk', file4])

    # Language files
    for file5 in glob.glob("assets/lang/*.*"):
        subprocess.call(['aapt', 'add', '-f', '-v', 'AM2RWrapper.apk', file5])

    # Headers
    for file6 in glob.glob("assets/lang/headers/*.*"):
        subprocess.call(['aapt', 'add', '-f', '-v', 'AM2RWrapper.apk', file6])

    # Fonts
    for file7 in glob.glob("assets/lang/fonts/*.*"):
        subprocess.call(['aapt', 'add', '-f', '-v', 'AM2RWrapper.apk', file7])

    print("\nSigning APK... If Java JDK 8 or newer is not present, this will fail!")
    subprocess.call(['java', '-jar', 'uber-apk-signer.jar', '-a', 'AM2RWrapper.apk'])
    # Cleanup
    remove('AM2RWrapper.apk')
    rmtree('assets')
    rmtree('../../'+output)
    # Move APK
    move('AM2RWrapper-aligned-debugSigned.apk', '../../AndroidM2R_'+version+'-signed.apk')

else:
    print("Invalid input. Exiting.")
    quit()

print("\nThe operation was completed successfully. See you next mission!")