cc-update-server/main.py at main · OpenCentauri/cc-update-server · 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
import swu_packer, gh_extract, tempfile, os, time, hashlib
from flask import Flask, redirect, url_for, request, render_template, send_from_directory

temp_dir = tempfile.TemporaryDirectory()
latest_version = gh_extract.fetch_latest_release("OpenCentauri", "cc-fw-tools")
print(f"Latest version: {latest_version.version}")
swu_path = os.path.join(temp_dir.name, "latest.swu")
bin_path = os.path.join(temp_dir.name, "latest.bin")
latest_version.download_swu(swu_path)
print(f"Downloaded latest SWU version {latest_version.version} to {swu_path}")
swu_packer.pack_swu(swu_path, bin_path)
print(f"Packed SWU to BIN at {bin_path}")

with open(bin_path, "rb") as f:
    data = f.read()
    md5_hash = hashlib.md5(data).hexdigest()

app = Flask(__name__)

@app.route("/latest")
def latest():
    return send_from_directory(temp_dir.name, "latest.bin")

@app.route("/mainboardVersionUpdate/getInfo.do7")
def update():
    version : str = request.args.get("version")

    if not version:
        return "Version parameter is required", 400

    if f"{latest_version.version}-" in version:
        return {
            "code": "000000",
            "messages": None,
            "data": {
                "update": False,
                "version": None,
                "packageUrl": None,
                "firmwareType": 0,
                "packageHash": None,
                "updateStrategy": 0,
                "log": None,
                "timeMS": 0,
                "dataInfoId": None
            },
            "success": True
        }
    else:
        return {
            "code": "000000",
            "messages": None,
            "data": {
                "update": True,
                "version": latest_version.version,
                "packageUrl": os.getenv("HOST", request.url_root.rstrip("/")) + url_for("latest"),
                "firmwareType": 1,
                "packageHash": md5_hash,
                "updateStrategy": 1,
                "log": latest_version.changelog,
                "timeMS": int(time.time() * 1000),
                "dataInfoId": "770b3a5993c04011bcb1c3a23df1fa5a"
            },
            "success": True
        }

if __name__ == '__main__':
    app.run(host="0.0.0.0", port=5000)