44 lines
2.1 KiB
Python
44 lines
2.1 KiB
Python
#!/usr/bin/env python3
|
|
"""Install the version-checked Classic adapter into an isolated CMaNGOS checkout."""
|
|
import argparse
|
|
import pathlib
|
|
import shutil
|
|
|
|
def patch(path, anchor, replacement):
|
|
text = path.read_text()
|
|
if replacement in text:
|
|
return
|
|
if text.count(anchor) != 1:
|
|
raise RuntimeError(f"Upstream source changed: {path.name}")
|
|
path.write_text(text.replace(anchor, replacement))
|
|
|
|
def install(source):
|
|
here = pathlib.Path(__file__).resolve().parent
|
|
for path in (here / "native").glob("AzerothAI.*"):
|
|
shutil.copy2(path, source / "src/game/World" / path.name)
|
|
for relative, anchor, invocation in [
|
|
("src/game/World/World.cpp", " sMapMgr.Update(diff);", "AzerothAIUpdate(diff);"),
|
|
("src/game/Entities/Unit.cpp", " uint32 health = victim->GetHealth();", "damage = AzerothAIBoundDamage(victim, damage);")
|
|
]:
|
|
path = source / relative
|
|
include = '#ifdef ENABLE_PLAYERBOTS\n#include "World/AzerothAI.h"\n#endif\n'
|
|
if include not in path.read_text():
|
|
path.write_text(include + path.read_text())
|
|
patch(path, anchor, anchor + "\n#ifdef ENABLE_PLAYERBOTS\n " + invocation + "\n#endif")
|
|
# Upstream treats unknown gear modes as generated equipment. Explicitly
|
|
# preserve ordinary character-creation equipment for our level-one roster.
|
|
path = source / "src/modules/PlayerBots/playerbot/PlayerbotMgr.cpp"
|
|
anchor = ' sRandomPlayerbotMgr.SetValue(botGuid, "create gear", 1, gear);'
|
|
patch(path, anchor, ' if (gear != "none")\n ' + anchor)
|
|
path = source / "src/game/Entities/CharacterHandler.cpp"
|
|
include = '#ifdef ENABLE_PLAYERBOTS\n#include "World/AzerothAI.h"\n#endif\n'
|
|
if include not in path.read_text():
|
|
path.write_text(include + path.read_text())
|
|
anchor = 'void PlayerbotHolder::AddPlayerBot(uint32 playerGuid, uint32 masterAccount)\n{'
|
|
patch(path, anchor, anchor + '\n if (!AzerothAIAllowBotLogin(playerGuid)) return;')
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("source", type=pathlib.Path)
|
|
install(parser.parse_args().source.resolve())
|