153 lines
6.9 KiB
Python
153 lines
6.9 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
import shutil
|
|
import zipfile
|
|
from pathlib import Path
|
|
|
|
from warium_source import MODID, ROOT, clean_dir, download_original, safe_java_identifier, write_json
|
|
|
|
GENERATED_JAVA = ROOT / "src" / "generated" / "java"
|
|
GENERATED_RESOURCES = ROOT / "src" / "generated" / "resources"
|
|
INVENTORY_DIR = ROOT / "docs" / "inventory"
|
|
|
|
|
|
def main() -> None:
|
|
jar = download_original()
|
|
clean_dir(GENERATED_JAVA)
|
|
clean_dir(GENERATED_RESOURCES)
|
|
INVENTORY_DIR.mkdir(parents=True, exist_ok=True)
|
|
|
|
with zipfile.ZipFile(jar) as archive:
|
|
names = archive.namelist()
|
|
block_names = sorted(stem(n) for n in names if n.startswith(f"assets/{MODID}/blockstates/") and n.endswith(".json"))
|
|
item_model_names = sorted(stem(n) for n in names if n.startswith(f"assets/{MODID}/models/item/") and n.endswith(".json"))
|
|
standalone_items = sorted(name for name in item_model_names if name not in set(block_names))
|
|
|
|
extract_resources(archive)
|
|
merge_language_defaults()
|
|
|
|
inventory = {
|
|
"source": "Warium 1.2.7",
|
|
"source_sha1": "528d81630a23fb4004e3abdd99b16bd225cd1e92",
|
|
"modid": MODID,
|
|
"blocks_from_blockstates": block_names,
|
|
"standalone_items_from_item_models": standalone_items,
|
|
"counts": {
|
|
"blockstates": len(block_names),
|
|
"item_models": len(item_model_names),
|
|
"standalone_items": len(standalone_items),
|
|
"classes": len([n for n in names if n.endswith(".class")]),
|
|
"procedures": len([n for n in names if n.startswith("net/mcreator/crustychunks/procedures/") and n.endswith(".class")]),
|
|
"entities": len([n for n in names if n.startswith("net/mcreator/crustychunks/entity/") and n.endswith(".class")]),
|
|
"recipes": len([n for n in names if n.startswith(f"data/{MODID}/recipes/") and n.endswith(".json")]),
|
|
"loot_tables": len([n for n in names if n.startswith(f"data/{MODID}/loot_tables/") and n.endswith(".json")]),
|
|
},
|
|
}
|
|
write_json(INVENTORY_DIR / "original-inventory.json", inventory)
|
|
write_generated_registries(block_names, standalone_items)
|
|
|
|
|
|
def stem(path: str) -> str:
|
|
return Path(path).name.removesuffix(".json")
|
|
|
|
|
|
def extract_resources(archive: zipfile.ZipFile) -> None:
|
|
for info in archive.infolist():
|
|
if info.is_dir():
|
|
continue
|
|
name = info.filename
|
|
if name.startswith(f"assets/{MODID}/"):
|
|
target = GENERATED_RESOURCES / name
|
|
elif name.startswith(f"data/{MODID}/"):
|
|
target = GENERATED_RESOURCES / migrate_data_path(name)
|
|
elif name == "pack.mcmeta":
|
|
continue
|
|
else:
|
|
continue
|
|
target.parent.mkdir(parents=True, exist_ok=True)
|
|
with archive.open(info) as src, target.open("wb") as dst:
|
|
shutil.copyfileobj(src, dst)
|
|
|
|
|
|
def migrate_data_path(path: str) -> str:
|
|
path = path.replace(f"data/{MODID}/loot_tables/", f"data/{MODID}/loot_table/")
|
|
path = path.replace(f"data/{MODID}/tags/items/", f"data/{MODID}/tags/item/")
|
|
path = path.replace(f"data/{MODID}/tags/blocks/", f"data/{MODID}/tags/block/")
|
|
path = path.replace(f"data/{MODID}/tags/entity_types/", f"data/{MODID}/tags/entity_type/")
|
|
path = path.replace(f"data/{MODID}/tags/fluids/", f"data/{MODID}/tags/fluid/")
|
|
return path
|
|
|
|
|
|
def write_generated_registries(blocks: list[str], items: list[str]) -> None:
|
|
package_dir = GENERATED_JAVA / "net" / "mcreator" / "crustychunks" / "init"
|
|
package_dir.mkdir(parents=True, exist_ok=True)
|
|
used: set[str] = set()
|
|
lines: list[str] = [
|
|
"package net.mcreator.crustychunks.init;",
|
|
"",
|
|
"import net.mcreator.crustychunks.CrustyChunksMod;",
|
|
"import net.minecraft.core.registries.Registries;",
|
|
"import net.minecraft.network.chat.Component;",
|
|
"import net.minecraft.world.item.BlockItem;",
|
|
"import net.minecraft.world.item.CreativeModeTab;",
|
|
"import net.minecraft.world.item.Item;",
|
|
"import net.minecraft.world.item.ItemStack;",
|
|
"import net.minecraft.world.item.Items;",
|
|
"import net.minecraft.world.level.block.Block;",
|
|
"import net.minecraft.world.level.block.state.BlockBehaviour;",
|
|
"import net.neoforged.bus.api.IEventBus;",
|
|
"import net.neoforged.neoforge.registries.DeferredHolder;",
|
|
"import net.neoforged.neoforge.registries.DeferredRegister;",
|
|
"",
|
|
"public final class GeneratedRegistries {",
|
|
" public static final DeferredRegister<Block> BLOCKS = DeferredRegister.create(Registries.BLOCK, CrustyChunksMod.MODID);",
|
|
" public static final DeferredRegister<Item> ITEMS = DeferredRegister.create(Registries.ITEM, CrustyChunksMod.MODID);",
|
|
" public static final DeferredRegister<CreativeModeTab> CREATIVE_TABS = DeferredRegister.create(Registries.CREATIVE_MODE_TAB, CrustyChunksMod.MODID);",
|
|
"",
|
|
]
|
|
|
|
for name in blocks:
|
|
ident = safe_java_identifier(name, used)
|
|
lines.append(f' public static final DeferredHolder<Block, Block> {ident} = BLOCKS.register("{name}", () -> new Block(BlockBehaviour.Properties.of().strength(2.0F, 6.0F)));')
|
|
lines.append(f' public static final DeferredHolder<Item, BlockItem> {ident}_ITEM = ITEMS.register("{name}", () -> new BlockItem({ident}.get(), new Item.Properties()));')
|
|
|
|
for name in items:
|
|
ident = safe_java_identifier(name, used)
|
|
lines.append(f' public static final DeferredHolder<Item, Item> {ident} = ITEMS.register("{name}", () -> new Item(new Item.Properties()));')
|
|
|
|
lines.extend([
|
|
"",
|
|
' public static final DeferredHolder<CreativeModeTab, CreativeModeTab> WARIUM_TAB = CREATIVE_TABS.register("warium", () -> CreativeModeTab.builder()',
|
|
' .title(Component.translatable("itemGroup.crusty_chunks.warium"))',
|
|
" .icon(() -> new ItemStack(Items.IRON_INGOT))",
|
|
" .displayItems((parameters, output) -> ITEMS.getEntries().forEach(entry -> output.accept(entry.get())))",
|
|
" .build());",
|
|
"",
|
|
" private GeneratedRegistries() {",
|
|
" }",
|
|
"",
|
|
" public static void register(IEventBus modBus) {",
|
|
" BLOCKS.register(modBus);",
|
|
" ITEMS.register(modBus);",
|
|
" CREATIVE_TABS.register(modBus);",
|
|
" }",
|
|
"}",
|
|
"",
|
|
])
|
|
(package_dir / "GeneratedRegistries.java").write_text("\n".join(lines), encoding="utf-8")
|
|
|
|
|
|
def merge_language_defaults() -> None:
|
|
lang_path = GENERATED_RESOURCES / "assets" / MODID / "lang" / "en_us.json"
|
|
if lang_path.exists():
|
|
data = json.loads(lang_path.read_text(encoding="utf-8"))
|
|
else:
|
|
data = {}
|
|
data.setdefault("itemGroup.crusty_chunks.warium", "Warium")
|
|
write_json(lang_path, data)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|