sorcery/main.py at main · konstructmc/sorcery · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
import json
import os
import shutil
import zipfile
from pathlib import Path

DATA_PATH = "data.json"


def load_data():
    try:
        with open(DATA_PATH, "r") as file:
            data = json.load(file)
            return data
    except FileNotFoundError:
        return {}
    except json.JSONDecodeError:
        print("Error: Data file is corrupted.")
        return {}


DATA = load_data()


def create_pack():
    id = DATA.get("id", "test")
    pack_dir = Path(id)
    if pack_dir.exists():
        print(f"Error: Directory '{id}' already exists.")
        return

    os.mkdir(pack_dir)
    os.mkdir(pack_dir / "assets")
    os.mkdir(pack_dir / "assets" / id)
    os.mkdir(pack_dir / "assets" / id / "textures")
    os.mkdir(pack_dir / "assets" / id / "textures" / "item")
    os.mkdir(pack_dir / "assets" / id / "models")
    os.mkdir(pack_dir / "assets" / id / "models" / "item")
    os.mkdir(pack_dir / "assets" / "minecraft")
    os.mkdir(pack_dir / "assets" / "minecraft" / "items")

    with open(pack_dir / "pack.mcmeta", "w") as pack_file:
        pack_file.write(
            json.dumps(
                {
                    "pack": {
                        "pack_format": 64,
                        "description": DATA.get(
                            "description", "A custom resource pack"
                        ),
                    }
                },
                indent=4,
            )
        )

    for item in DATA.get("items", []):
        item_id = item.get("id")
        parent = item.get("parent")
        texture_path = item.get("texture_path")

        if not item_id or not parent or not texture_path:
            continue

        model_data = {
            "parent": "item/generated",
            "textures": {"layer0": f"{id}:item/{item_id}"},
        }

        item_data = {
            "model": {
                "type": "select",
                "property": "custom_model_data",
                "fallback": {"type": "model", "model": f"minecraft:item/{parent}"},
                "cases": [
                    {
                        "when": item_id,
                        "model": {"type": "model", "model": f"{id}:item/{item_id}"},
                    }
                ],
            }
        }

        with open(
            pack_dir / "assets" / id / "models" / "item" / f"{item_id}.json", "w"
        ) as model_file:
            model_file.write(json.dumps(model_data, indent=4))

        with open(
            pack_dir / "assets" / "minecraft" / "items" / f"{parent}.json", "w"
        ) as item_file:
            item_file.write(json.dumps(item_data, indent=4))

        shutil.copyfile(
            texture_path,
            pack_dir / "assets" / id / "textures" / "item" / f"{item_id}.png",
        )

    zip_filename = f"{id}_resource_pack.zip"
    with zipfile.ZipFile(zip_filename, "w", zipfile.ZIP_DEFLATED) as zipf:
        for root, dirs, files in os.walk(pack_dir):
            for file in files:
                file_path = Path(root) / file
                zipf.write(file_path, file_path.relative_to(pack_dir.parent))

    print(f"Resource pack '{zip_filename}' created successfully.")


if __name__ == "__main__":
    create_pack()