#!/usr/bin/env python3
import os
import sys
if hasattr(sys, '_MEIPASS'): # we're running in a bundle, go to where we have our bundled assets
os.chdir(sys._MEIPASS)
import argparse
import shutil
import requests
import pathlib
import hashlib
from configparser import RawConfigParser
import util
from util import config
# Apply updates to the actual mod pack
def install():
mods_location = os.path.join(config["pack"]["location"], "mods")
whitelist = config["pack"]["whitelist"]
blacklist = config["pack"]["blacklist"]
# Actual download links are stored in pack-lock.ini, so we need to read it in here
pack_lock = RawConfigParser()
pack_lock.read('pack-lock.ini')
print(f"Updating pack with version {pack_lock['global']['pack_version']}...")
# create the mods folder if it doesn't already exist
pathlib.Path(mods_location).mkdir(parents=True, exist_ok=True)
names = set(f'{mod}.jar' for mod in pack_lock['mod_versions'].keys())
# whitelist client mods (e.g. optifine)
names.update(whitelist)
i = 0
for entry in pack_lock['mod_versions'].items():
name = f'{entry[0]}.jar'
checksum, url = entry[1].split(',')
mod_path = os.path.join(mods_location, name)
i += 1
if (os.path.exists(mod_path) and os.path.isfile(mod_path) and \
hashlib.sha1(open(mod_path, 'rb').read()).hexdigest() == checksum) or \
name in blacklist:
print(f"Skipping {name}, already up to date")
else:
print(f'Installing {name} from {url}...')
print(f' ({i} of {len(pack_lock["mod_versions"])})', end='\r')
util.download_file(url, mod_path)
print("Done!" + " " * 8)
# Remove any old mods that might be stuck in the mods folder
print()
print("Removing old mods...")
for jar in os.listdir(mods_location):
if jar not in names and os.path.splitext(jar)[1] == ".jar":
os.remove(os.path.join(mods_location, jar))
print(f"Removing '{jar}'")
print('\nInstalling configs...')
# For config files, we don't need to remove any extras, as they
# will simply be ignored by Forge
if not os.path.exists('config'):
raise RuntimeError("Error: config folder must exist!")
mc_config_folder = os.path.join(config['pack']['location'], 'config')
if not os.path.exists(mc_config_folder):
os.mkdir(mc_config_folder)
shutil.copytree('config/', mc_config_folder, dirs_exist_ok=True)
print()
print("Finished installing mods!")
def apply_updates():
"""
Using the URLs defined in pack.ini, update all mods to the latest
compatible version and write them out to pack-lock.ini
"""
print("Populating version file...")
print("Getting new versions of all mods...")
mod_urls = util.find_updated_urls([x for x in config['mods'].values()], config['pack']['game_version'], threads=16)
print("Downloading and checksumming all mods...")
if None in mod_urls:
print("[!] Checksum generation aborted due to invalid URLs. Please fix them and try again.")
exit(1)
checksums = util.find_checksums(mod_urls)
# Write the updated mods list out to pack-lock.ini
pack_lock = RawConfigParser()
pack_lock.read('pack-lock.ini')
pack_lock['global']['pack_version'] = str(int(pack_lock['global']['pack_version']) + 1)
# This is only needed for the self-update feature
pack_lock['global']['config_files'] = ','.join(os.listdir('config'))
pack_lock['mod_versions'] = {name: f'{checksum},{url}' for name, checksum, url in zip(config['mods'].keys(), checksums, mod_urls)}
with open('pack-lock.ini', 'w') as f:
pack_lock.write(f)
print()
print("Done!")
print("Updates applied to pack-lock.ini")
print(f"New pack version is {pack_lock['global']['pack_version']}")
print("[!] No mods were installed. To update your mods folder, run 'update.py install'")
# Find if any updates are available
def check_updates(mods, version_file, version=(2, 0, 0)):
pack_version = util.get_version_from_file(version_file)
print("Checking for updates to version " + str(pack_version) + "...")
pack_lock = RawConfigParser()
pack_lock.read('pack-lock.ini')
latest = [(k, mods[k]) for k in mods.keys()]
old_urls = []
for mod in pack_lock['mod_versions'].keys():
old_urls.append(pack_lock['mod_versions'][mod].split(',')[1])
num_updates = 0
print("Checking updates...")
ffx = util.firefox()
for mod in latest:
print("Checking for updates to {mod[0]}...".format(mod=mod), end="")
sys.stdout.flush() # takes care of line-buffered terminals
if 'curseforge' in mod[1]:
url = util.find_cdn(ffx, mod[1], version)
else:
url = requests.get(mod[1]).url
if url in old_urls:
print(" No updates")
else:
print(" Found update: " + url.split('/')[-1])
num_updates += 1
ffx.close()
print("Finished checking for updates. {num} mods can be updated".format(num=num_updates))
if num_updates >= 0:
print("Run 'python update.py apply_updates' to create a new version with these updates applied.")
parser = argparse.ArgumentParser(
description="A Simple Git-Based Modpack Manager",
formatter_class=argparse.RawDescriptionHelpFormatter,
epilog='''\
Available commands:
install : Downloads mods listed in version.txt and populates the mods folder specified in pack-location.txt
apply_updates : Using the urls in mods.txt, repopulates version.txt to reflect the most recent mod versions
check_updates : Compares version.txt and mods.txt to see if any mods can be updated
''')
parser.add_argument('command',
nargs='?',
default='install',
help="The action to perform (default: install)")
if __name__ == "__main__":
args = parser.parse_args()
mods = config['mods']
pack = config['pack']
# run the command
if args.command == "install":
install()
elif args.command == "apply_updates":
apply_updates()
elif args.command == "check_updates":
check_updates(config['mods'], "pack-lock.ini", config['pack']["game_version"])
else:
print("Error: command \"" + args.command + "\" does not exist")
parser.print_help()
sys.exit(1)
