215 lines
No EOL
8 KiB
Python
215 lines
No EOL
8 KiB
Python
import os, os.path, sys
|
|
import urllib, zipfile
|
|
import shutil, glob, fnmatch
|
|
import subprocess, logging, re
|
|
import pprint
|
|
|
|
forge_dir = os.path.dirname(os.path.abspath(__file__))
|
|
mcp_dir = os.path.abspath('..')
|
|
src_dir = os.path.join(mcp_dir, 'src')
|
|
sys.path.append(mcp_dir)
|
|
|
|
def reset_logger():
|
|
log = logging.getLogger()
|
|
while len(log.handlers) > 0:
|
|
log.removeHandler(log.handlers[0])
|
|
|
|
version_reg = re.compile(r'(([a-z]+)Version[\s]+=[\s]+)(\d+);')
|
|
def load_version(build=0):
|
|
info = {'major' : -1,
|
|
'minor' : -1,
|
|
'revision' : -1,
|
|
'build' : -1
|
|
}
|
|
hook_file = os.path.join(forge_dir, 'forge_common/net/minecraft/src/forge/ForgeHooks.java'.replace('/', os.sep))
|
|
with open(hook_file, 'r') as fh:
|
|
buf = fh.read()
|
|
|
|
def proc(match):
|
|
try:
|
|
info[match.group(2)] = int(match.group(3))
|
|
except:
|
|
pass
|
|
return match.group(0)
|
|
|
|
buf = version_reg.sub(proc, buf)
|
|
info['build'] = build
|
|
return info
|
|
|
|
def inject_version(src_file, build=0):
|
|
version = load_version(build)
|
|
|
|
tmp_file = src_file + '.tmp'
|
|
with open(src_file, 'r') as fh:
|
|
buf = fh.read()
|
|
|
|
def mapname(match):
|
|
try:
|
|
return '%s%s;' % (match.group(1), version[match.group(2)])
|
|
except KeyError:
|
|
pass
|
|
return match.group(0)
|
|
|
|
buf = version_reg.sub(mapname, buf).replace('\r\n', '\n')
|
|
|
|
with open(tmp_file, 'wb') as fh:
|
|
fh.write(buf)
|
|
shutil.move(tmp_file, src_file)
|
|
|
|
def zip_folder(path, key, zip):
|
|
files = os.listdir(path)
|
|
for file in files:
|
|
file_path = os.path.join(path, file)
|
|
file_key = os.path.join(key, file)
|
|
if os.path.isdir(file_path):
|
|
zip_folder(file_path, file_key, zip)
|
|
else:
|
|
print file_key
|
|
zip.write(file_path, file_key)
|
|
|
|
def zip_create(path, key, zip_name):
|
|
zip = zipfile.ZipFile(zip_name, 'w', zipfile.ZIP_DEFLATED)
|
|
if os.path.isdir(path):
|
|
zip_folder(path, key, zip)
|
|
else:
|
|
zip.write(path, key)
|
|
zip.close()
|
|
|
|
|
|
def apply_forge_patches(fml_dir, mcp_dir, forge_dir, src_dir, copy_files=True):
|
|
sys.path.append(fml_dir)
|
|
from fml import copytree, apply_patches
|
|
|
|
has_client = os.path.isdir(os.path.join(src_dir, 'minecraft'))
|
|
has_server = os.path.isdir(os.path.join(src_dir, 'minecraft_server'))
|
|
|
|
#patch files
|
|
print 'Applying Minecraft Forge patches'
|
|
sys.stdout.flush()
|
|
if has_client:
|
|
if os.path.isdir(os.path.join(forge_dir, 'patches', 'minecraft')):
|
|
apply_patches(mcp_dir, os.path.join(forge_dir, 'patches', 'minecraft'), src_dir)
|
|
if os.path.isdir(os.path.join(forge_dir, 'patches', 'common')):
|
|
apply_patches(mcp_dir, os.path.join(forge_dir, 'patches', 'common'), src_dir, '/common/', '/minecraft/')
|
|
if copy_files and os.path.isdir(os.path.join(forge_dir, 'client')):
|
|
copytree(os.path.join(forge_dir, 'client'), os.path.join(src_dir, 'minecraft'))
|
|
|
|
if has_server:
|
|
if os.path.isdir(os.path.join(forge_dir, 'patches', 'minecraft_server')):
|
|
apply_patches(mcp_dir, os.path.join(forge_dir, 'patches', 'minecraft_server'), src_dir)
|
|
if os.path.isdir(os.path.join(forge_dir, 'patches', 'common')):
|
|
apply_patches(mcp_dir, os.path.join(forge_dir, 'patches', 'common'), src_dir, '/common/', '/minecraft_server/')
|
|
if copy_files and os.path.isdir(os.path.join(forge_dir, 'server')):
|
|
copytree(os.path.join(forge_dir, 'server'), os.path.join(src_dir, 'minecraft_server'))
|
|
|
|
if os.path.isdir(os.path.join(forge_dir, 'patches', 'common')):
|
|
apply_patches(mcp_dir, os.path.join(forge_dir, 'patches', 'common'), src_dir)
|
|
if copy_files and os.path.isdir(os.path.join(forge_dir, 'common')):
|
|
copytree(os.path.join(forge_dir, 'common'), os.path.join(src_dir, 'common'))
|
|
|
|
def get_conf_copy(mcp_dir, forge_dir):
|
|
#Lets grab the files we dont work on
|
|
for file in ['astyle.cfg', 'version.cfg', 'patches/minecraft_ff.patch', 'patches/minecraft_server_ff.patch']:
|
|
dst_file = os.path.normpath(os.path.join(forge_dir, 'conf', file))
|
|
src_file = os.path.normpath(os.path.join(mcp_dir, 'conf', file))
|
|
if os.path.exists(dst_file):
|
|
os.remove(dst_file)
|
|
shutil.copy(src_file, dst_file)
|
|
print 'Grabbing: ' + src_file
|
|
|
|
gen_merged_srg(mcp_dir, forge_dir)
|
|
gen_merged_exc(mcp_dir, forge_dir)
|
|
#gen_merged_fields(os.path.join(mcp_dir, 'conf', 'client.srg'), os.path.join(mcp_dir, 'conf', 'fields.csv'))
|
|
|
|
|
|
def gen_merged_srg(mcp_dir, forge_dir):
|
|
print 'Generating merged Retroguard data'
|
|
srg_client = os.path.join(mcp_dir, 'conf', 'client.srg')
|
|
srg_server = os.path.join(mcp_dir, 'conf', 'server.srg')
|
|
|
|
if not os.path.isfile(srg_client) or not os.path.isfile(srg_server):
|
|
print 'Could not find client and server srg files in "%s"' % mcp_dir
|
|
return False
|
|
|
|
client = {'PK:': {}, 'CL:': {}, 'FD:': {}, 'MD:': {}}
|
|
with open(srg_client, 'r') as fh:
|
|
for line in fh:
|
|
pts = line.rstrip('\r\n').split(' ')
|
|
if pts[0] == 'MD:':
|
|
client[pts[0]][pts[1] + ' ' + pts[2]] = pts[3] + ' ' + pts[4]
|
|
else:
|
|
client[pts[0]][pts[1]] = pts[2]
|
|
|
|
server = {'PK:': {}, 'CL:': {}, 'FD:': {}, 'MD:': {}}
|
|
with open(srg_server, 'r') as fh:
|
|
for line in fh:
|
|
pts = line.rstrip('\r\n').split(' ')
|
|
if pts[0] == 'MD:':
|
|
server[pts[0]][pts[1] + ' ' + pts[2]] = pts[3] + ' ' + pts[4]
|
|
else:
|
|
server[pts[0]][pts[1]] = pts[2]
|
|
|
|
common = {'PK:': {}, 'CL:': {}, 'FD:': {}, 'MD:': {}}
|
|
for type in common:
|
|
for key, value in client[type].items():
|
|
if key in server[type]:
|
|
if value == server[type][key]:
|
|
client[type].pop(key)
|
|
server[type].pop(key)
|
|
common[type][key] = value
|
|
|
|
for type in common:
|
|
for key, value in client[type].items():
|
|
common[type][key] = value + ' #C'
|
|
|
|
for type in common:
|
|
for key, value in server[type].items():
|
|
common[type][key] = value + ' #S'
|
|
|
|
#Print joined retroguard files
|
|
with open(os.path.join(forge_dir, 'conf', 'joined.srg'), 'w') as f:
|
|
for type in ['PK:', 'CL:', 'FD:', 'MD:']:
|
|
for key in sorted(common[type]):
|
|
f.write('%s %s %s\n' % (type, key, common[type][key]))
|
|
|
|
def gen_merged_exc(mcp_dir, forge_dir):
|
|
print 'Generating merged MCInjector config'
|
|
exc_client = os.path.join(mcp_dir, 'conf', 'client.exc')
|
|
exc_server = os.path.join(mcp_dir, 'conf', 'server.exc')
|
|
|
|
client = {}
|
|
with open(exc_client, 'r') as fh:
|
|
for line in fh:
|
|
if not line.startswith('#'):
|
|
pts = line.rstrip('\r\n').split('=')
|
|
client[pts[0]] = pts[1]
|
|
|
|
server = {}
|
|
with open(exc_server, 'r') as fh:
|
|
for line in fh:
|
|
if not line.startswith('#'):
|
|
pts = line.rstrip('\r\n').split('=')
|
|
server[pts[0]] = pts[1]
|
|
|
|
common = {}
|
|
for key, value in client.items():
|
|
if key in server:
|
|
if value != server[key]:
|
|
print 'Error: Exec for shared function does not match client and server:'
|
|
print 'Function: ' + key
|
|
print 'Client: ' + value
|
|
print 'Server: ' + server[value]
|
|
if value != '|':
|
|
common[key] = value
|
|
client.pop(key)
|
|
server.pop(key)
|
|
else:
|
|
if value != '|':
|
|
common[key] = value
|
|
|
|
common = dict(common.items() + server.items())
|
|
|
|
#Print joined mcinjector files
|
|
with open(os.path.join(forge_dir, 'conf', 'joined.exc'), 'w') as f:
|
|
for key in sorted(common):
|
|
f.write('%s=%s\n' % (key, common[key])) |