158 lines
5.4 KiB
Python
158 lines
5.4 KiB
Python
# Copyright (c) 2020-2021 The reone project contributors
|
|
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
"""Script to bootstrap module from existing modules."""
|
|
|
|
import os
|
|
import shutil
|
|
|
|
extract_dir = r"D:\OpenKotOR\Extract\KotOR"
|
|
if not os.path.exists(extract_dir):
|
|
raise RuntimeError("Extraction directory not found")
|
|
|
|
data_dir = os.path.join(extract_dir, "data")
|
|
if not os.path.exists(data_dir):
|
|
raise RuntimeError("Data directory not found")
|
|
|
|
modules_dir = os.path.join(extract_dir, "modules")
|
|
if not os.path.exists(modules_dir):
|
|
raise RuntimeError("Modules directory not found")
|
|
|
|
bootstrap_dir = r"D:\OpenKotOR\Mods"
|
|
if not os.path.exists(modules_dir):
|
|
raise RuntimeError("Bootstrap directory not found")
|
|
|
|
tools_dir = r"D:\Source\reone\build\bin\RelWithDebInfo"
|
|
if not os.path.exists(modules_dir):
|
|
raise RuntimeError("Tools directory not found")
|
|
|
|
|
|
def fetch_module_names():
|
|
global modules_dir
|
|
names = []
|
|
for entry in os.listdir(modules_dir):
|
|
module_dir = os.path.join(modules_dir, entry)
|
|
if os.path.isdir(module_dir):
|
|
names.append(entry.lower())
|
|
|
|
return names
|
|
|
|
|
|
def create_build_script(module_dir, mod_name):
|
|
bat_path = os.path.join(module_dir, "build.ps1")
|
|
with open(bat_path, "w") as f:
|
|
lines = [
|
|
"$env:PATH+=\";{}\"\n".format(tools_dir),
|
|
"reone-tools --to-gff {}\\module.ifo.json\n".format(mod_name),
|
|
"reone-tools --to-gff {0}\\{0}.are.json\n".format(mod_name),
|
|
"reone-tools --to-gff {0}\\{0}.git.json\n".format(mod_name),
|
|
"reone-tools --to-mod {}".format(mod_name)
|
|
]
|
|
f.writelines(lines)
|
|
|
|
|
|
def determine_actual_module_name(name):
|
|
global modules_dir
|
|
|
|
module_dir = os.path.join(modules_dir, name)
|
|
for entry in os.listdir(module_dir):
|
|
filename, ext = os.path.splitext(entry)
|
|
if ext == ".are":
|
|
return filename
|
|
|
|
return name
|
|
|
|
|
|
def copy_module_files(mod_name_from, mod_name_from_actual, mod_name_to, module_dir_out, unpack_dir):
|
|
global extract_dir
|
|
global data_dir
|
|
|
|
# Copy IFO, ARE and GIT
|
|
module_dir = os.path.join(modules_dir, mod_name_from)
|
|
ext_to_copy = [".ifo", ".are", ".git"]
|
|
for entry in os.listdir(module_dir):
|
|
path = os.path.join(module_dir, entry)
|
|
filename, ext = os.path.splitext(entry)
|
|
if ext == ".json" and list(filter(lambda x: x in filename, ext_to_copy)):
|
|
filename = filename.replace(mod_name_from_actual, mod_name_to)
|
|
dest = os.path.join(unpack_dir, filename + ext)
|
|
print("Copying {} to {}...".format(path, dest))
|
|
try:
|
|
shutil.copy(path, dest)
|
|
except Exception as e:
|
|
print(e)
|
|
|
|
# Copy PTH
|
|
module_tpl_dir = os.path.join(modules_dir, mod_name_from + "_s")
|
|
for entry in os.listdir(module_tpl_dir):
|
|
path = os.path.join(module_tpl_dir, entry)
|
|
filename, ext = os.path.splitext(entry)
|
|
if ext == ".pth" and not "-ascii" in filename:
|
|
filename = filename.replace(mod_name_from_actual, mod_name_to)
|
|
dest = os.path.join(unpack_dir, filename + ext)
|
|
print("Copying {} to {}...".format(path, dest))
|
|
try:
|
|
shutil.copy(path, dest)
|
|
except Exception as e:
|
|
print(e)
|
|
|
|
# Copy LYT and VIS
|
|
for entry in os.listdir(data_dir):
|
|
path = os.path.join(data_dir, entry)
|
|
filename, ext = os.path.splitext(entry)
|
|
if filename == mod_name_from_actual and (ext == ".lyt" or ext == ".vis"):
|
|
filename = filename.replace(mod_name_from_actual, mod_name_to)
|
|
dest = os.path.join(module_dir_out, filename + ext)
|
|
print("Copying {} to {}...".format(path, dest))
|
|
try:
|
|
shutil.copy(path, dest)
|
|
except Exception as e:
|
|
print(e)
|
|
|
|
|
|
module_names = fetch_module_names()
|
|
|
|
while True:
|
|
mod_name_from = input("Existing module name: ")
|
|
mod_found = mod_name_from.lower() in module_names
|
|
if mod_found:
|
|
break
|
|
else:
|
|
print("Variants:")
|
|
for name in module_names:
|
|
print(name)
|
|
|
|
while True:
|
|
mod_name_to = input("New module name: ")
|
|
if mod_name_to.lower() in module_names:
|
|
print("Duplicate module name!")
|
|
continue
|
|
elif mod_name_to:
|
|
break
|
|
|
|
print("Bootstrapping module {} from {}...".format(mod_name_to, mod_name_from))
|
|
|
|
module_dir_out = os.path.join(bootstrap_dir, mod_name_to)
|
|
if not os.path.exists(module_dir_out):
|
|
os.mkdir(module_dir_out)
|
|
|
|
create_build_script(module_dir_out, mod_name_to)
|
|
|
|
unpack_dir = os.path.join(module_dir_out, mod_name_to)
|
|
if not os.path.exists(unpack_dir):
|
|
os.mkdir(unpack_dir)
|
|
|
|
mod_name_from_actual = determine_actual_module_name(mod_name_from)
|
|
copy_module_files(mod_name_from, mod_name_from_actual, mod_name_to, module_dir_out, unpack_dir)
|