mod_shhd.py

import copy
import json
import os
import argparse
import sys
import ctypes

from assets.data_shhd import (get_building_cost_address, get_building_health_address, get_building_population_address,
                              get_unit_health_address, get_unit_arrow_dmg_address, get_unit_xbow_dmg_address,
                              get_unit_stone_dmg_address, get_resource_buy_address, get_resource_sell_address,
                              get_scenario_pgr_address, get_scenario_pgr_crowded_address,
                              get_skirmish_pgr_address, get_unit_melee_dmg_address, read, write, unit_names)


def write_with_uninst_info(shc, address, value, size):
    if str(address) not in uninstall[str(size)]:
        uninstall[str(size)][str(address)] = read(shc, address, size)
    write(shc, address, value, size)


def apply_aob_as_patch(address, array):
    with open(exe_path, "r+b") as shc:
        shc.seek(0)
        shc.seek(address)
        for elem in array:
            if type(elem) == tuple:
                value = elem[0]
                size = elem[1]
                shc.write(int(value).to_bytes(size, byteorder='little'))
            else:
                shc.write(int(str(elem)).to_bytes(1, byteorder='little'))


"""
def modify_religion_rules(religion_rules):
    with open(exe_path, "r+b") as shc:
        for key in religion_rules:
            if key == "thresholds":
                thresholds = religion_rules[key]
                write_with_uninst_info(shc, 0x5BD62, thresholds[0], 1)
                write_with_uninst_info(shc, 0x5BD6B, thresholds[1], 1)
                write_with_uninst_info(shc, 0x5BD77, thresholds[2], 1)
                write_with_uninst_info(shc, 0x5BD85, thresholds[3], 1)

                write_with_uninst_info(shc, 0x401DF, thresholds[0], 1)
                write_with_uninst_info(shc, 0x401E8, thresholds[1], 1)
                write_with_uninst_info(shc, 0x401F4, thresholds[2], 1)
                write_with_uninst_info(shc, 0x40202, thresholds[3], 1)

                write_with_uninst_info(shc, 0x40241, thresholds[0] - 1, 1)
                write_with_uninst_info(shc, 0x40250, thresholds[1] - 1, 1)
                write_with_uninst_info(shc, 0x4025F, thresholds[2] - 1, 1)
                write_with_uninst_info(shc, 0x4026E, thresholds[3] - 1, 1)

                write_with_uninst_info(shc, 0x40248, thresholds[0], 4)
                write_with_uninst_info(shc, 0x40257, thresholds[1], 4)
                write_with_uninst_info(shc, 0x40266, thresholds[2], 4)
                write_with_uninst_info(shc, 0x40275, thresholds[3], 4)

                write_with_uninst_info(shc, 0x3EE52, thresholds[0] - 1, 1)
                write_with_uninst_info(shc, 0x3EE5B, thresholds[1] - 1, 1)
                write_with_uninst_info(shc, 0x3EE67, thresholds[2] - 1, 1)
                write_with_uninst_info(shc, 0x3EE75, thresholds[3] - 1, 1)
            elif key == "bonuses":
                bonuses = religion_rules[key]
                write_with_uninst_info(shc, 0x401EC, bonuses[0], 4)
                write_with_uninst_info(shc, 0x401F8, bonuses[1], 1)
                write_with_uninst_info(shc, 0x4020B, bonuses[2] - bonuses[3], 1)
                write_with_uninst_info(shc, 0x4020E, bonuses[3], 4)

                write_with_uninst_info(shc, 0x3EE5F, bonuses[0], 4)
                write_with_uninst_info(shc, 0x3EE6B, bonuses[1], 4)
                write_with_uninst_info(shc, 0x3EE7E, bonuses[2] - bonuses[3], 1)
                write_with_uninst_info(shc, 0x3EE81, bonuses[3], 1)

                write_with_uninst_info(shc, 0x5BD6F, bonuses[0], 4)
                write_with_uninst_info(shc, 0x5BD7B, bonuses[1], 4)
                write_with_uninst_info(shc, 0x5BD8E, bonuses[2] - bonuses[3], 1)
                write_with_uninst_info(shc, 0x5BD91, bonuses[3], 4)
            elif key == "church_bonus":
                church_bonus = religion_rules[key]
                write_with_uninst_info(shc, 0x5BDA0, church_bonus, 1)
                write_with_uninst_info(shc, 0x4039D, church_bonus, 1)
                write_with_uninst_info(shc, 0x3EE92, church_bonus, 1)
            elif key == "cathedral_bonus":
                cathedral_bonus = religion_rules[key]
                write_with_uninst_info(shc, 0x5BDAC, cathedral_bonus, 1)
                write_with_uninst_info(shc, 0x403AC, cathedral_bonus, 1)
                write_with_uninst_info(shc, 0x3EE9E, cathedral_bonus, 1)


def modify_beer_rules(beer_rules):
    with open(exe_path, "r+b") as shc:
        for key in beer_rules:
            if key == "thresholds":
                thresholds = beer_rules[key]
                write_with_uninst_info(shc, 0X3B481, thresholds[0], 1)
                write_with_uninst_info(shc, 0X3B48A, thresholds[1], 1)
                write_with_uninst_info(shc, 0X3B496, thresholds[2], 1)
                write_with_uninst_info(shc, 0X3B4A4, thresholds[3], 1)

                write_with_uninst_info(shc, 0x3EF1B, thresholds[0], 1)
                write_with_uninst_info(shc, 0x3EF24, thresholds[1], 1)
                write_with_uninst_info(shc, 0x3EF30, thresholds[2], 1)
                write_with_uninst_info(shc, 0x3EF3E, thresholds[3], 1)

                write_with_uninst_info(shc, 0x5BDF8, thresholds[0], 1)
                write_with_uninst_info(shc, 0x5BE08, thresholds[1], 1)
                write_with_uninst_info(shc, 0x5BE14, thresholds[2], 1)
                write_with_uninst_info(shc, 0x5BE22, thresholds[3], 1)
            elif key == "bonuses":
                bonuses = beer_rules[key]
                write_with_uninst_info(shc, 0x3B48E, bonuses[0], 1)
                write_with_uninst_info(shc, 0x3B49A, bonuses[1], 1)
                write_with_uninst_info(shc, 0x3B4AD, bonuses[2] - bonuses[3], 1)
                write_with_uninst_info(shc, 0x3B4AF, bonuses[3], 4)

                write_with_uninst_info(shc, 0x3EF28, bonuses[0], 1)
                write_with_uninst_info(shc, 0x3EF34, bonuses[1], 1)
                write_with_uninst_info(shc, 0x3EF47, bonuses[2] - bonuses[3], 1)
                write_with_uninst_info(shc, 0x3EF4A, bonuses[3], 1)

                write_with_uninst_info(shc, 0x5BE0C, bonuses[0], 1)
                write_with_uninst_info(shc, 0x5BE18, bonuses[1], 1)
                write_with_uninst_info(shc, 0x5BE2B, bonuses[2] - bonuses[3], 1)
                write_with_uninst_info(shc, 0x5BE2E, bonuses[3], 1)
            elif key == "coverage_per_inn":
                coverage_per_inn = beer_rules[key]
                write_with_uninst_info(shc, 0x59077, coverage_per_inn * 100, 4)


def modify_food_rules(food_rules):
    with open(exe_path, "r+b") as shc:
        for key in food_rules:
            if key == "ration_bonuses":
                ration_bonuses = food_rules[key]
                write_with_uninst_info(shc, 0x3B90C, ration_bonuses[0], 4)
                write_with_uninst_info(shc, 0x3B951, ration_bonuses[0], 4)
                write_with_uninst_info(shc, 0x3B948, ration_bonuses[1], 4)
                write_with_uninst_info(shc, 0x3B937, ration_bonuses[2], 1)
                write_with_uninst_info(shc, 0x3B92A, ration_bonuses[3], 4)

                write_with_uninst_info(shc, 0x3EAC3, ration_bonuses[0], 4)
                write_with_uninst_info(shc, 0x3EAFB, ration_bonuses[0], 4)
                write_with_uninst_info(shc, 0x3EAF5, ration_bonuses[1] - 1, 1)
                write_with_uninst_info(shc, 0x3EAE2, ration_bonuses[2] - 3, 1)
                write_with_uninst_info(shc, 0x3EAD5, ration_bonuses[3], 4)

                write_with_uninst_info(shc, 0x5BB66, ration_bonuses[0], 4)
                write_with_uninst_info(shc, 0x5BB78, ration_bonuses[0], 4)
                write_with_uninst_info(shc, 0x5BB85, ration_bonuses[1] - 1, 1)
                write_with_uninst_info(shc, 0x5BBA1, ration_bonuses[2], 4)
                write_with_uninst_info(shc, 0x5BB97, ration_bonuses[3], 4)
            elif key == "variety_bonuses":
                variety_bonuses = food_rules[key]
                write_with_uninst_info(shc, 0x3BADD, variety_bonuses[0] - 2, 1)
                write_with_uninst_info(shc, 0x3BAE7, variety_bonuses[1] - 3, 1)
                write_with_uninst_info(shc, 0x3BAF0, variety_bonuses[2], 4)

                write_with_uninst_info(shc, 0x3EB1B, variety_bonuses[0], 1)
                write_with_uninst_info(shc, 0x3EB25, variety_bonuses[1], 1)
                write_with_uninst_info(shc, 0x3EB2F, variety_bonuses[2], 1)

                write_with_uninst_info(shc, 0x5BBB9, variety_bonuses[0], 1)
                write_with_uninst_info(shc, 0x5BBC3, variety_bonuses[1], 1)
                write_with_uninst_info(shc, 0x5BB4D, variety_bonuses[2], 1)


def modify_taxation_rules(taxation_rules):
    with open(exe_path, "r+b") as shc:
        for key in taxation_rules:
            if key == "popularity":
                popularity = taxation_rules[key]
                write_with_uninst_info(shc, 0x3B0AE, popularity[0], 4)
                write_with_uninst_info(shc, 0x3B0BD, popularity[1], 4)
                write_with_uninst_info(shc, 0x3B0C9, popularity[2], 4)
                write_with_uninst_info(shc, 0x3B0D5, popularity[3], 4)
                write_with_uninst_info(shc, 0x3B0E1, popularity[4], 4)
                write_with_uninst_info(shc, 0x3B0Ed, popularity[5], 4)
                write_with_uninst_info(shc, 0x3B0F9, popularity[6], 4)
                write_with_uninst_info(shc, 0x3B105, popularity[7], 4)
                write_with_uninst_info(shc, 0x3B111, popularity[8], 4)
                write_with_uninst_info(shc, 0x3B11D, popularity[9], 4)
                write_with_uninst_info(shc, 0x3B129, popularity[10], 4)
                write_with_uninst_info(shc, 0x3B133, popularity[11], 4)

                write_with_uninst_info(shc, 0x3EBE3, popularity[0], 4)
                write_with_uninst_info(shc, 0x3EBF0, max(-127, min(popularity[1]-1, 127)), 1)
                write_with_uninst_info(shc, 0x3EBFA, max(-127, min(popularity[2]-2, 127)), 1)
                write_with_uninst_info(shc, 0x3EC04, max(-127, min(popularity[3]-3, 127)), 1)
                write_with_uninst_info(shc, 0x3EC0E, max(-127, min(popularity[4]-4, 127)), 1)
                write_with_uninst_info(shc, 0x3EC18, max(-127, min(popularity[5]-5, 127)), 1)
                write_with_uninst_info(shc, 0x3EC21, popularity[6], 4)
                write_with_uninst_info(shc, 0x3EC2D, popularity[7], 4)
                write_with_uninst_info(shc, 0x3EC39, popularity[8], 4)
                write_with_uninst_info(shc, 0x3EC45, popularity[9], 4)
                write_with_uninst_info(shc, 0x3EC56, popularity[10], 4)
                write_with_uninst_info(shc, 0x3EC54, popularity[11]-popularity[10], 1)

                write_with_uninst_info(shc, 0x5BC73, popularity[0], 4)
                write_with_uninst_info(shc, 0x5BC7F, popularity[1], 4)
                write_with_uninst_info(shc, 0x5BC8B, popularity[2], 4)
                write_with_uninst_info(shc, 0x5BC97, popularity[3], 4)
                write_with_uninst_info(shc, 0x5BCA3, popularity[4], 4)
                write_with_uninst_info(shc, 0x5BCAF, popularity[5], 4)
                write_with_uninst_info(shc, 0x5BCBB, popularity[6], 4)
                write_with_uninst_info(shc, 0x5BCC7, popularity[7], 4)
                write_with_uninst_info(shc, 0x5BCD3, popularity[8], 4)
                write_with_uninst_info(shc, 0x5BCDF, popularity[9], 4)
                write_with_uninst_info(shc, 0x5BCF0, popularity[10], 4)
                write_with_uninst_info(shc, 0x5BCEE, popularity[11]-popularity[10], 1)
            elif key == "neutral_tax_level":
                neutral_level = taxation_rules[key]
                write_with_uninst_info(shc, 0x3B09F, neutral_level, 1)
                write_with_uninst_info(shc, 0x593A8, neutral_level, 1)
                write_with_uninst_info(shc, 0x593E8, neutral_level, 1)
                write_with_uninst_info(shc, 0x3ADCE, neutral_level, 1)
                write_with_uninst_info(shc, 0x3EBC4, neutral_level, 1)
                write_with_uninst_info(shc, 0x5BC5D, neutral_level, 1)
                write_with_uninst_info(shc, 0x5C1AF, neutral_level, 1)
                write_with_uninst_info(shc, 0x560C2, neutral_level, 4)
                jump_addr_list = [0x3B0AE, 0x3B0BD, 0x3B0C9, 0x3B0D5, 0x3B0E1, 0x3B0ED,
                                  0x3B0F9, 0x3B105, 0x3B111, 0x3B11D, 0x3B129, 0x3B133]
                jump_distance = jump_addr_list[neutral_level] - 0x3b0A8 - 2
                neutral_happiness = read(shc, jump_addr_list[neutral_level], 4)
                write_with_uninst_info(shc, 0x3EBCF, neutral_happiness, 4)
                write_with_uninst_info(shc, 0x5BC65, neutral_happiness, 4)
                write_with_uninst_info(shc, 0x3b0A8, jump_distance, 1)
            elif key == "advantage_multiplier":
                advantage_multipliers = taxation_rules[key]
                human_big_ai_medium = advantage_multipliers["human_big_ai_medium"]
                ai_big = advantage_multipliers["ai_big"]
                write_with_uninst_info(shc, 0x59326, human_big_ai_medium, 4)
                write_with_uninst_info(shc, 0x592FA, ai_big, 4)


def modify_fear_factor_rules(fear_factor_rules):
    with open(exe_path, "r+b") as shc:
        for key in fear_factor_rules:
            if key == "popularity_per_good_level":
                write_with_uninst_info(shc, 0x3F664, fear_factor_rules[key], 1)
                write_with_uninst_info(shc, 0x3EDBC, fear_factor_rules[key], 1)
                write_with_uninst_info(shc, 0x5BE5B, fear_factor_rules[key], 1)
            elif key == "popularity_per_bad_level":
                write_with_uninst_info(shc, 0x3F66E, fear_factor_rules[key], 1)
                write_with_uninst_info(shc, 0x3EDC8, fear_factor_rules[key], 1)
                write_with_uninst_info(shc, 0x5BE69, fear_factor_rules[key], 1)
            elif key == "productivity":
                productivity = key["productivity"]
                write_with_uninst_info(shc, 0x590FF, productivity[0], 4)
                write_with_uninst_info(shc, 0x5910C, productivity[1], 4)
                write_with_uninst_info(shc, 0x59119, productivity[2], 4)
                write_with_uninst_info(shc, 0x59126, productivity[3], 4)
                write_with_uninst_info(shc, 0x59133, productivity[4], 4)
                write_with_uninst_info(shc, 0x5913F, productivity[5], 4)
                write_with_uninst_info(shc, 0x5914C, productivity[6], 4)
                write_with_uninst_info(shc, 0x59159, productivity[7], 4)
                write_with_uninst_info(shc, 0x59166, productivity[8], 4)
                write_with_uninst_info(shc, 0x59179, productivity[9], 1)
                write_with_uninst_info(shc, 0x5917C, productivity[10], 1)
            elif key == "coverage":
                write_with_uninst_info(shc, 0xB430, fear_factor_rules[key], 1)


def enable_custom_taxation(tax_table):
    tax_jumpout_instructions = [
        0xE9, 0xDF, 0xB2, 0xFA, 0xFF,             # jmp Stronghold_Crusader_Extreme.exe+459B
        0x90, 0x90,                               # nop nop
        0xC1, 0xF8, 0x03,                         # sar eax,02 { 2 }
        0x83, 0x3D, 0xF0, 0x4D, 0x35, 0x02, 0x00  # cmp dword ptr [Stronghold_Crusader_Extreme.exe+1F54DF0],00
    ]
    apply_aob_as_patch(0x592B7, tax_jumpout_instructions)

    custom_tax_instructions = [
        0x8A, 0x80, 0xBB, 0x45, 0x40, 0x00,  # mov al,[eax+Stronghold_Crusader_Extreme.exe+45BB]
        0x0F, 0xAF, 0x44, 0x24, 0x0C,        # imul eax,[esp+0C]
        0xE9, 0x13, 0x4D, 0x05, 0x00,        # jmp Stronghold_Crusader_Extreme.exe+5937C
    ]
    apply_aob_as_patch(0x459B, custom_tax_instructions)

    bribe_jumpout_instructions = [
        0x8B, 0x44, 0x24, 0x08,        # mov eax,[esp+08]
        0xE9, 0x32, 0xB2, 0xFA, 0xFF,  # jmp Stronghold_Crusader_Extreme.exe+45AB
        0x90, 0x90,                    # nop nop
        0xC1, 0xF8, 0x02               # sar eax,01
    ]
    apply_aob_as_patch(0x59370, bribe_jumpout_instructions)

    custom_bribe_instructions = [
        0x8A, 0x80, 0xBB, 0x45, 0x40, 0x00,  # mov al,[eax+Stronghold_Crusader_Extreme.exe+45BB]
        0x0F, 0xAF, 0x44, 0x24, 0x0C,        # imul eax,[esp+0C]
        0xE9, 0xC0, 0x4D, 0x05, 0x00         # jmp Stronghold_Crusader_Extreme.exe+5937C
    ]
    apply_aob_as_patch(0x45AB, custom_bribe_instructions)

    apply_aob_as_patch(0x45BB, [int(float(a)*20) for a in tax_table if a != "0.00"])


def enable_custom_combat_bonus(damage_table):
    combat_jumpout_instructions = [
        0x31, 0xC9,                    # xor ecx, ecx
        0xE9, 0xA1, 0x2F, 0xED, 0xFF,  # jmp 4045C8 (45C8)
        0x90, 0x90, 0x90, 0x90         # nop nop nop nop
    ]
    apply_aob_as_patch(0x131620, combat_jumpout_instructions)

    custom_combat_instructions = [
        0x8A, 0x88, 0xDD, 0x45, 0x40, 0x00,  # mov cl,[eax+004045DD]
        0x0F, 0xAF, 0x4C, 0x24, 0x04,        # imul ecx,[esp+04]
        0xE9, 0x53, 0xD0, 0x12, 0x00         # jmp 0053162B
    ]
    apply_aob_as_patch(0x45C8, custom_combat_instructions)

    apply_aob_as_patch(0x45D8, damage_table)
"""


def uninstall_custom_combat_bonus():
    original_combat_bonus_instructions = [
        0x83, 0xC0, 0x14,
        0x0F, 0xAF, 0x44, 0x24, 0x04,
        0x8D, 0x0C, 0x80
    ]
    apply_aob_as_patch(0x531620, original_combat_bonus_instructions)

    codecave_cleanup = [0x03] * 27
    apply_aob_as_patch(0x4045C8, codecave_cleanup)


def uninstall_custom_taxation():
    original_tax_instructions = [0x0F, 0xAF, 0x44, 0x24, 0x0C, 0x99, 0x2B, 0xC2, 0xD1, 0xF8]
    apply_aob_as_patch(0x592B7, original_tax_instructions)

    original_bribe_instructions = [0xB8, 0x05, 0x00, 0x00, 0x00, 0x2B, 0x44, 0x24, 0x08, 0x0F, 0xAF, 0x44, 0x24, 0x0C]
    apply_aob_as_patch(0x59370, original_bribe_instructions)

    codecave_cleanup = [0x03] * 43  # original content in the code cave
    apply_aob_as_patch(0x459B, codecave_cleanup)


def modify_building_stats(buildings):
    with open(exe_path, "r+b") as shc:
        size = 4
        for key in buildings:
            current_building = buildings[key]

            if "cost" in current_building.keys():
                address = get_building_cost_address(key)

                for cost in current_building["cost"]:
                    write_with_uninst_info(shc, address, cost, size)
                    address += size

            if "health" in current_building.keys():
                address = get_building_health_address(key)
                write_with_uninst_info(shc, address, current_building["health"], size)

            if "housing" in current_building.keys():
                address = get_building_population_address(key)
                write_with_uninst_info(shc, address, current_building["housing"], size)


def modify_unit_stats(units):
    with open(exe_path, "r+b") as shc:
        size = 4
        for key in units:
            current_unit = units[key]

            if "health" in current_unit.keys():
                address = get_unit_health_address(key)
                write_with_uninst_info(shc, address, current_unit["health"], size)

            if "arrowDamage" in current_unit.keys():
                address = get_unit_arrow_dmg_address(key)
                write_with_uninst_info(shc, address, current_unit["arrowDamage"], size)

            if "xbowDamage" in current_unit.keys():
                address = get_unit_xbow_dmg_address(key)
                write_with_uninst_info(shc, address, current_unit["xbowDamage"], size)

            if "stoneDamage" in current_unit.keys():
                address = get_unit_stone_dmg_address(key)
                write_with_uninst_info(shc, address, current_unit["stoneDamage"], size)

            if "baseMeleeDamage" in current_unit.keys():
                for defender in unit_names:
                    address = get_unit_melee_dmg_address(key, defender)
                    write_with_uninst_info(shc, address, current_unit["baseMeleeDamage"], size)

            if "meleeDamageVs" in current_unit.keys():
                for defender in current_unit["meleeDamageVs"]:
                    address = get_unit_melee_dmg_address(key, defender)
                    write_with_uninst_info(shc, address, current_unit["meleeDamageVs"][defender], size)


def modify_trade_costs(resources):
    with open(exe_path, "r+b") as shc:
        size = 4
        for key in resources:
            current_resource = resources[key]

            if "buy" in current_resource.keys():
                address = get_resource_buy_address(key)
                write_with_uninst_info(shc, address, current_resource["buy"], size)

            if "sell" in current_resource.keys():
                address = get_resource_sell_address(key)
                write_with_uninst_info(shc, address, current_resource["sell"], size)


def modify_population_gathering_rate(gathering_rates):
    with open(exe_path, "r+b") as shc:
        size = 4
        for pgr in gathering_rates:
            current_pgr = gathering_rates[pgr]

            if pgr == "Skirmish":
                for threshold in current_pgr:
                    address = get_skirmish_pgr_address(threshold)
                    write_with_uninst_info(shc, address, current_pgr[threshold], size)

            if pgr == "Scenario_lt_100":
                for threshold in current_pgr:
                    address = get_scenario_pgr_address(threshold)
                    write_with_uninst_info(shc, address, current_pgr[threshold], size)

            if pgr == "Scenario_gt_100":
                for threshold in current_pgr:
                    address = get_scenario_pgr_crowded_address(threshold)
                    write_with_uninst_info(shc, address, current_pgr[threshold], size)


def install_mod():
    if os.path.exists(configpath):
        with open(configpath, "r") as f:
            config = json.load(f)
    else:
        print("Cannot find config.json (must be in same directory as installer)!", file=sys.stderr)

    if os.path.isfile(uninst_path):
        with open(uninst_path, "r") as f:
            try:
                uninstall.update(json.load(f))
            except json.JSONDecodeError:
                pass

    if "4" not in uninstall:
        uninstall["4"] = {}

    for cfg, val in config.items():
        if cfg == "buildings":
            modify_building_stats(val)
        elif cfg == "units":
            modify_unit_stats(val)
        elif cfg == "resources":
            modify_trade_costs(val)
        elif cfg == "population_gathering_rate":
            modify_population_gathering_rate(val)
        # elif cfg == "religion":
        #     modify_religion_rules(val)
        # elif cfg == "beer":
        #     modify_beer_rules(val)
        # elif cfg == "food":
        #     modify_food_rules(val)
        # elif cfg == "fear_factor":
        #     modify_fear_factor_rules(val)
        # elif cfg == "taxation":
        #     modify_taxation_rules(val)
        # elif cfg == "special":
        #     for key, change in val.items():
        #         if not ("special" in uninstall):
        #             uninstall["special"] = {}
        #
        #         if key == "custom_taxation":
        #             uninstall["special"]["custom_taxation"] = copy.deepcopy(change)
        #             tax_table = change["table"]
        #             enable_custom_taxation(tax_table)
        #         elif key == "custom_combat_bonus":
        #             uninstall["special"]["custom_combat_bonus"] = copy.deepcopy(change)
        #             damage_table = change["table"]
        #             enable_custom_combat_bonus(damage_table)

        elif cfg == "other":
            with open(exe_path, "r+b") as shc:
                for change in val:
                    address = int(change["address"], 16)
                    size = change["size"]

                    if str(size) not in uninstall:
                        uninstall[str(size)] = {}

                    if type(change["value"]) is not list:
                        write_with_uninst_info(shc, address, change["value"], size)

                    elif type(change["value"]) is list:
                        for item in change["value"]:
                            write_with_uninst_info(shc, address, item, size)
                            address += size

    if settings.create_uninst:
        with open("uninstall.json", "w") as f:
            json.dump(uninstall, f, indent=4)


def uninstall_mod():
    if os.path.isfile(uninst_path):
        with open(uninst_path, "r") as f:
            try:
                uninstall.update(json.load(f))
            except json.JSONDecodeError:
                return

        with open(exe_path, "r+b") as shc:
            for size in uninstall:
                for key in uninstall[size]:
                    try:
                        write(shc, int(key), uninstall[size][key], int(size))
                    except:
                        if "custom_taxation" in uninstall[size]:
                            uninstall_custom_taxation()
                        if "custom_combat_bonus" in uninstall[size]:
                            uninstall_custom_combat_bonus()


if __name__ == "__main__":
    argp = argparse.ArgumentParser()
    argp.add_argument("-gamepath", required=False,
                      help="Folder of SH HD, defaults to parent directory")
    argp.add_argument("-create_uninst", action="store_true", required=False,
                      help="Create uninstall.json to be able to revert changes later.")

    settings = argp.parse_args()
    if settings.gamepath:
        gamedir = os.path.abspath(settings.gamepath)
    else:
        gamedir = os.path.abspath(os.path.join(".", ".."))
    if getattr(sys, "frozen", False):
        my_dir = os.path.dirname(os.path.abspath(sys.executable))
    else:
        my_dir = os.path.dirname(os.path.abspath(__file__))

    uninst_path = os.path.join(my_dir, "uninstall.json")
    configpath = os.path.join(my_dir, "config.json")
    uninstall = {}
    retries = 0
    if "Stronghold.exe" not in os.listdir(gamedir):
        ctypes.windll.user32.MessageBoxW(0, f"StrongholdDE.dll not found in {gamedir}", "Game not found", 0)
        sys.exit(1)

    exe_path = os.path.join(gamedir, "Stronghold.exe")

    if len(sys.argv) > 1 and sys.argv[1] == "uninstall":
        uninstall_mod()
    else:
        install_mod()