aboutsummaryrefslogtreecommitdiff
path: root/codegen/lib/download.py
blob: d9e2e63f61cbee9f700aeaf780c43bf1efb1e961 (plain)
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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
from lib.utils import get_dir_location
import xml.etree.ElementTree as ET
from .mappings import Mappings
import requests
import json
import os

# make sure the downloads directory exists
print('Making downloads')
if not os.path.exists(get_dir_location('downloads')):
    print('Made downloads directory.', get_dir_location('downloads'))
    os.mkdir(get_dir_location('downloads'))


def get_burger():
    if not os.path.exists(get_dir_location('downloads/Burger')):
        print('\033[92mDownloading Burger...\033[m')
        os.system(
            f'cd {get_dir_location("downloads")} && git clone https://github.com/pokechu22/Burger && cd Burger && git pull')

        print('\033[92mInstalling dependencies...\033[m')
        os.system('cd downloads/Burger && pip install six jawa')


def get_generator_mod():
    if not os.path.exists(get_dir_location('downloads/minecraft-data-generator-server')):
        print('\033[92mDownloading u9g/minecraft-data-generator-server...\033[m')
        os.system(
            f'cd {get_dir_location("downloads")} && git clone https://github.com/u9g/minecraft-data-generator-server && cd minecraft-data-generator-server && git pull')
    return get_dir_location('downloads/minecraft-data-generator-server')


def get_version_manifest():
    if not os.path.exists(get_dir_location(f'downloads/version_manifest.json')):
        print(
            f'\033[92mDownloading version manifest...\033[m')
        version_manifest_data = requests.get(
            'https://piston-meta.mojang.com/mc/game/version_manifest_v2.json').json()
        with open(get_dir_location(f'downloads/version_manifest.json'), 'w') as f:
            json.dump(version_manifest_data, f)
    else:
        with open(get_dir_location(f'downloads/version_manifest.json'), 'r') as f:
            version_manifest_data = json.load(f)
    return version_manifest_data


def get_version_data(version_id: str):
    if not os.path.exists(get_dir_location(f'downloads/{version_id}.json')):
        version_manifest_data = get_version_manifest()

        print(
            f'\033[92mGetting data for \033[1m{version_id}..\033[m')
        try:
            package_url = next(
                filter(lambda v: v['id'] == version_id, version_manifest_data['versions']))['url']
        except StopIteration:
            raise ValueError(
                f'No version with id {version_id} found. Maybe delete downloads/version_manifest.json and try again?')
        package_data = requests.get(package_url).json()
        with open(get_dir_location(f'downloads/{version_id}.json'), 'w') as f:
            json.dump(package_data, f)
    else:
        with open(get_dir_location(f'downloads/{version_id}.json'), 'r') as f:
            package_data = json.load(f)
    return package_data


def get_client_jar(version_id: str):
    if not os.path.exists(get_dir_location(f'downloads/client-{version_id}.jar')):
        package_data = get_version_data(version_id)
        print('\033[92mDownloading client jar...\033[m')
        client_jar_url = package_data['downloads']['client']['url']
        with open(get_dir_location(f'downloads/client-{version_id}.jar'), 'wb') as f:
            f.write(requests.get(client_jar_url).content)


def get_server_jar(version_id: str):
    if not os.path.exists(get_dir_location(f'downloads/server-{version_id}.jar')):
        package_data = get_version_data(version_id)
        print('\033[92mDownloading server jar...\033[m')
        server_jar_url = package_data['downloads']['server']['url']
        with open(get_dir_location(f'downloads/server-{version_id}.jar'), 'wb') as f:
            f.write(requests.get(server_jar_url).content)


def get_mappings_for_version(version_id: str):
    if not os.path.exists(get_dir_location(f'downloads/mappings-{version_id}.txt')):
        package_data = get_version_data(version_id)

        client_mappings_url = package_data['downloads']['client_mappings']['url']

        mappings_text = requests.get(client_mappings_url).text

        with open(get_dir_location(f'downloads/mappings-{version_id}.txt'), 'w') as f:
            f.write(mappings_text)
    else:
        with open(get_dir_location(f'downloads/mappings-{version_id}.txt'), 'r') as f:
            mappings_text = f.read()
    return Mappings.parse(mappings_text)


def get_yarn_versions():
    # https://meta.fabricmc.net/v2/versions/yarn
    if not os.path.exists(get_dir_location('downloads/yarn_versions.json')):
        print('\033[92mDownloading yarn versions...\033[m')
        yarn_versions_data = requests.get(
            'https://meta.fabricmc.net/v2/versions/yarn').json()
        with open(get_dir_location('downloads/yarn_versions.json'), 'w') as f:
            json.dump(yarn_versions_data, f)
    else:
        with open(get_dir_location('downloads/yarn_versions.json'), 'r') as f:
            yarn_versions_data = json.load(f)
    return yarn_versions_data


def get_yarn_data(version_id: str):
    for version in get_yarn_versions():
        if version['gameVersion'] == version_id:
            return version


def get_fabric_api_versions():
    # https://maven.fabricmc.net/net/fabricmc/fabric-api/fabric-api/maven-metadata.xml
    if not os.path.exists(get_dir_location('downloads/fabric_api_versions.json')):
        print('\033[92mDownloading Fabric API versions...\033[m')
        fabric_api_versions_xml_text = requests.get(
            'https://maven.fabricmc.net/net/fabricmc/fabric-api/fabric-api/maven-metadata.xml').text
        # parse xml
        fabric_api_versions_data_xml = ET.fromstring(
            fabric_api_versions_xml_text)
        fabric_api_versions = []

        versioning_el = fabric_api_versions_data_xml.find('versioning')
        assert versioning_el
        versions_el = versioning_el.find('versions')
        assert versions_el

        for version_el in versions_el.findall('version'):
            fabric_api_versions.append(version_el.text)

        with open(get_dir_location('downloads/fabric_api_versions.json'), 'w') as f:
            f.write(json.dumps(fabric_api_versions))
    else:
        with open(get_dir_location('downloads/fabric_api_versions.json'), 'r') as f:
            fabric_api_versions = json.loads(f.read())
    return fabric_api_versions


def clear_version_cache():
    print('\033[92mClearing version cache...\033[m')
    files = [
        'version_manifest.json',
        'yarn_versions.json',
        'fabric_api_versions.json'
    ]
    for file in files:
        if os.path.exists(get_dir_location(f'downloads/{file}')):
            os.remove(get_dir_location(f'downloads/{file}'))

    os.system(
        f'cd {get_dir_location("downloads/Burger")} && git pull')
    os.system(
        f'cd {get_dir_location("downloads/minecraft-data-generator-server")} && git pull')