106 lines
4.8 KiB
Python
106 lines
4.8 KiB
Python
"""Outputs a Factorio Mod to facilitate integration with Archipelago"""
|
|
|
|
import os
|
|
import zipfile
|
|
from typing import Optional
|
|
import threading
|
|
import json
|
|
|
|
import jinja2
|
|
import Utils
|
|
import shutil
|
|
import Options
|
|
from BaseClasses import MultiWorld
|
|
from .Technologies import tech_table
|
|
|
|
template: Optional[jinja2.Template] = None
|
|
locale_template: Optional[jinja2.Template] = None
|
|
|
|
template_load_lock = threading.Lock()
|
|
|
|
base_info = {
|
|
"version": Utils.__version__,
|
|
"title": "Archipelago",
|
|
"author": "Berserker",
|
|
"homepage": "https://archipelago.gg",
|
|
"description": "Integration client for the Archipelago Randomizer",
|
|
"factorio_version": "1.1"
|
|
}
|
|
|
|
# TODO: clean this up, probably as a jinja macro; then add logic for the recipes in completion condition
|
|
rocket_recipes = {
|
|
Options.MaxSciencePack.option_space_science_pack:
|
|
'{{"rocket-control-unit", 10}, {"low-density-structure", 10}, {"rocket-fuel", 10}}',
|
|
Options.MaxSciencePack.option_utility_science_pack:
|
|
'{{"speed-module", 10}, {"steel-plate", 10}, {"solid-fuel", 10}}',
|
|
Options.MaxSciencePack.option_production_science_pack:
|
|
'{{"speed-module", 10}, {"steel-plate", 10}, {"solid-fuel", 10}}',
|
|
Options.MaxSciencePack.option_chemical_science_pack:
|
|
'{{"advanced-circuit", 10}, {"steel-plate", 10}, {"solid-fuel", 10}}',
|
|
Options.MaxSciencePack.option_military_science_pack:
|
|
'{{"defender-capsule", 10}, {"stone-wall", 10}, {"coal", 10}}',
|
|
Options.MaxSciencePack.option_logistic_science_pack:
|
|
'{{"electronic-circuit", 10}, {"stone-brick", 10}, {"coal", 10}}',
|
|
Options.MaxSciencePack.option_automation_science_pack:
|
|
'{{"copper-cable", 10}, {"iron-plate", 10}, {"wood", 10}}'
|
|
}
|
|
|
|
def generate_mod(world: MultiWorld, player: int, seedname: str):
|
|
global template, locale_template
|
|
with template_load_lock:
|
|
if not template:
|
|
mod_template_folder = Utils.local_path("data", "factorio", "mod_template")
|
|
template = jinja2.Template(open(os.path.join(mod_template_folder, "data-final-fixes.lua")).read())
|
|
locale_template = jinja2.Template(open(os.path.join(mod_template_folder, "locale", "en", "locale.cfg")).read())
|
|
control_template = jinja2.Template(open(os.path.join(mod_template_folder, "control.lua")).read())
|
|
# get data for templates
|
|
player_names = {x: world.player_names[x][0] for x in world.player_ids}
|
|
locations = []
|
|
for location in world.get_filled_locations(player):
|
|
locations.append((location.name, location.item.name, location.item.player))
|
|
mod_name = f"AP-{seedname}-P{player}-{world.player_names[player][0]}"
|
|
tech_cost = {0: 0.1,
|
|
1: 0.25,
|
|
2: 0.5,
|
|
3: 1,
|
|
4: 2,
|
|
5: 5,
|
|
6: 10}[world.tech_cost[player].value]
|
|
template_data = {"locations": locations, "player_names" : player_names, "tech_table": tech_table,
|
|
"mod_name": mod_name, "allowed_science_packs": world.max_science_pack[player].get_allowed_packs(),
|
|
"tech_cost_scale": tech_cost, "custom_data": world.custom_data[player],
|
|
"tech_tree_layout_prerequisites": world.tech_tree_layout_prerequisites[player],
|
|
"rocket_recipe" : rocket_recipes[world.max_science_pack[player].value],
|
|
"slot_name": world.player_names[player][0]}
|
|
for factorio_option in Options.factorio_options:
|
|
template_data[factorio_option] = getattr(world, factorio_option)[player].value
|
|
control_code = control_template.render(**template_data)
|
|
data_final_fixes_code = template.render(**template_data)
|
|
|
|
mod_dir = Utils.output_path(mod_name)+"_"+Utils.__version__
|
|
en_locale_dir = os.path.join(mod_dir, "locale", "en")
|
|
os.makedirs(en_locale_dir, exist_ok=True)
|
|
shutil.copytree(Utils.local_path("data", "factorio", "mod"), mod_dir, dirs_exist_ok=True)
|
|
with open(os.path.join(mod_dir, "data-final-fixes.lua"), "wt") as f:
|
|
f.write(data_final_fixes_code)
|
|
with open(os.path.join(mod_dir, "control.lua"), "wt") as f:
|
|
f.write(control_code)
|
|
locale_content = locale_template.render(**template_data)
|
|
with open(os.path.join(en_locale_dir, "locale.cfg"), "wt") as f:
|
|
f.write(locale_content)
|
|
info = base_info.copy()
|
|
info["name"] = mod_name
|
|
with open(os.path.join(mod_dir, "info.json"), "wt") as f:
|
|
json.dump(info, f, indent=4)
|
|
|
|
# zip the result
|
|
zf_path = os.path.join(mod_dir+".zip")
|
|
with zipfile.ZipFile(zf_path, compression=zipfile.ZIP_DEFLATED, mode='w') as zf:
|
|
for root, dirs, files in os.walk(mod_dir):
|
|
for file in files:
|
|
zf.write(os.path.join(root, file),
|
|
os.path.relpath(os.path.join(root, file),
|
|
os.path.join(mod_dir, '..')))
|
|
shutil.rmtree(mod_dir)
|
|
|