mcbe_getzip.py

#!/usr/bin/env python3
"""
If the ZIP of the current version of Minecraft Bedrock Edition server isn't in ~,
download it, and remove outdated ZIPs in ~.
"""

import argparse
import pathlib
import re
import sys

import bs4
import requests
import toml

CLOBBER = True
CONFIG_FILE = pathlib.Path("/etc/MCscripts/mcbe-getzip.toml")
VERSIONS = ("current", "preview")
ZIPS_DIR = pathlib.Path(pathlib.Path.home(), "bedrock_zips")

PARSER = argparse.ArgumentParser(
    description=(
        "If the ZIP of the current version of Minecraft Bedrock Edition server isn't "
        + "in ~, download it, and remove outdated ZIPs in ~."
    )
)
CLOBBER_GROUP = PARSER.add_mutually_exclusive_group()
CLOBBER_GROUP.add_argument(
    "--clobber", action="store_true", help="remove outdated ZIPs in ~ (default)"
)
CLOBBER_GROUP.add_argument(
    "-n", "--no-clobber", action="store_true", help="don't remove outdated ZIPs in ~"
)
VERSIONS_GROUP = PARSER.add_mutually_exclusive_group()
VERSIONS_GROUP.add_argument(
    "-b",
    "--both",
    action="store_true",
    help="download current and preview versions (default)",
)
VERSIONS_GROUP.add_argument(
    "-c", "--current", action="store_true", help="download current version"
)
VERSIONS_GROUP.add_argument(
    "-p", "--preview", action="store_true", help="download preview version"
)
ARGS = PARSER.parse_args()

if CONFIG_FILE.is_file():
    CONFIG = toml.load(CONFIG_FILE)
    if "clobber" in CONFIG:
        if not isinstance(CONFIG["clobber"], bool):
            sys.exit(f"clobber must be TOML boolean, check {CONFIG_FILE}")
        CLOBBER = CONFIG["clobber"]
    if "versions" in CONFIG:
        if CONFIG["versions"] == "both":
            VERSIONS = ("current", "preview")
        elif CONFIG["versions"] == "current":
            VERSIONS = ("current",)
        elif CONFIG["versions"] == "preview":
            VERSIONS = ("preview",)
        else:
            sys.exit(f"No versions {CONFIG['versions']}, check {CONFIG_FILE}")

if ARGS.clobber:
    CLOBBER = True
elif ARGS.no_clobber:
    CLOBBER = False

if ARGS.both:
    VERSIONS = ("current", "preview")
elif ARGS.current:
    VERSIONS = ("current",)
elif ARGS.preview:
    VERSIONS = ("preview",)

ZIPS_DIR.mkdir(parents=True, exist_ok=True)

webpage_res = requests.get(
    "https://www.minecraft.net/en-us/download/server/bedrock",
    headers={
        "User-Agent": "Mozilla/5.0 (X11; Linux x86_64)",
        "Accept-Language": "en-US",
    },
    timeout=60,
)
webpage_res.raise_for_status()
webpage = bs4.BeautifulSoup(webpage_res.text, "html.parser")
urls = [link.get("href") for link in webpage.find_all("a")]
urls = [url for url in urls if url]

print(
    "Enter Y if you agree to the Minecraft End User License Agreement and Privacy",
    "Policy",
)
# Does prompting the EULA seem so official that it violates the EULA?
print("Minecraft End User License Agreement: https://minecraft.net/eula")
print("Privacy Policy: https://go.microsoft.com/fwlink/?LinkId=521839")
if input().lower() != "y":
    sys.exit("input != y")
for version in VERSIONS:
    if version == "current":
        for urlx in urls:
            urlx = re.match(r"^https://[^ ]+bin-linux/bedrock-server-[^ ]+\.zip$", urlx)
            if urlx:
                url = urlx.string
                break
    elif version == "preview":
        for urlx in urls:
            urlx = re.match(
                r"^https://[^ ]+bin-linux-preview/bedrock-server-[^ ]+\.zip$", urlx
            )
            if urlx:
                url = urlx.string
                break
    else:
        continue
    current_ver = pathlib.Path(url).name
    # Symlink to current/preview zip
    if pathlib.Path(ZIPS_DIR, version).is_symlink():
        INSTALLED_VER = pathlib.Path(ZIPS_DIR, version).resolve().name
    else:
        INSTALLED_VER = None

    if not pathlib.Path(ZIPS_DIR, current_ver).is_file():
        zip_res = requests.get(
            url,
            headers={
                "User-Agent": "Mozilla/5.0 (X11; Linux x86_64)",
                "Accept-Language": "en-US",
            },
            timeout=60,
            stream=True,
        )
        zip_res.raise_for_status()
        if pathlib.Path(ZIPS_DIR, current_ver + ".part").is_file():
            pathlib.Path(ZIPS_DIR, current_ver + ".part").unlink()
        for chunk in zip_res.iter_content(chunk_size=8192):
            pathlib.Path(ZIPS_DIR, current_ver + ".part").open(mode="ab").write(chunk)
        pathlib.Path(ZIPS_DIR, current_ver + ".part").rename(
            pathlib.Path(ZIPS_DIR, current_ver)
        )
    if INSTALLED_VER != current_ver:
        if pathlib.Path(ZIPS_DIR, version).is_symlink():
            pathlib.Path(ZIPS_DIR, version).unlink()
        pathlib.Path(ZIPS_DIR, version).symlink_to(pathlib.Path(ZIPS_DIR, current_ver))
if CLOBBER:
    for zipfile in ZIPS_DIR.glob("bedrock-server-*.zip"):
        try:
            if not zipfile.samefile(
                pathlib.Path(ZIPS_DIR, "current")
            ) and not zipfile.samefile(pathlib.Path(ZIPS_DIR, "preview")):
                zipfile.unlink()
        except FileNotFoundError:
            pass