#!/usr/bin/python
#
# This file is part of MagiskOnWSALocal.
#
# MagiskOnWSALocal is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# MagiskOnWSALocal is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with MagiskOnWSALocal.  If not, see <https://www.gnu.org/licenses/>.
#
# Copyright (C) 2022 LSPosed Contributors
#

import sys

import requests
import json
import re
from pathlib import Path

arch = sys.argv[1]
brand = sys.argv[2]
variant = sys.argv[3]
download_dir = Path.cwd().parent / \
    "download" if sys.argv[4] == "" else Path(sys.argv[4]).resolve()
tempScript = sys.argv[5]
print(
    f"Generating {brand} download link: arch={arch} variant={variant}", flush=True)
abi_map = {"x64": "x86_64", "arm64": "arm64"}
if brand == "OpenGApps":
    # Use Android 12.1 build of OpenGApps (also supports more variants like Full)
    if arch == "x64" and variant == "pico":
        link = "http://peternjeim.ddns.net:8081/ipfs/QmPDiAyqUvZHo9QU7WfoEE9XMqC8ppGyUsSwKQY7chfwHX"
    elif arch == "x64" and variant == "full":
        link = "http://peternjeim.ddns.net:8081/ipfs/QmULfSMwWuukQR7r9KEvwD2XzsChHTvpswmNqJyEU64jwM"
    # Use official Android 11.0 build of OpenGApps (since I didn't build all variants)
    else:
        # TODO: keep it 11.0 since official opengapps does not support 12+ yet
        # As soon as opengapps is available for 12+, we need to get the sdk/release from build.prop and
        # download the corresponding version
        release = "11.0"
        try:
            res = requests.get(f"https://api.opengapps.org/list")
            j = json.loads(res.content)
            link = {i["name"]: i for i in j["archs"][abi_map[arch]]
                    ["apis"][release]["variants"]}[variant]["zip"]
            # print(f"JSON={j}", flush=True)
            DATE=j["archs"][abi_map[arch]]["date"]
            print(f"DATE={DATE}", flush=True)
        except Exception:
            print("Failed to fetch from OpenGApps API, fallbacking to SourceForge RSS...")
            res = requests.get(
                f'https://sourceforge.net/projects/opengapps/rss?path=/{abi_map[arch]}&limit=100')
            link = re.search(f'https://.*{abi_map[arch]}/.*{release}.*{variant}.*\.zip/download', res.text).group().replace(
                '.zip/download', '.zip').replace('sourceforge.net/projects/opengapps/files', 'downloads.sourceforge.net/project/opengapps')
elif brand == "MindTheGapps":
    res = requests.get(
        f'https://sourceforge.net/projects/wsa-mtg/rss?path=/{abi_map[arch]}&limit=100')
    link = re.search(f'https://.*{abi_map[arch]}/.*\.zip/download', res.text).group().replace(
        '.zip/download', '.zip').replace('sourceforge.net/projects/wsa-mtg/files', 'downloads.sourceforge.net/project/wsa-mtg')

print(f"download link: {link}", flush=True)

with open(download_dir/tempScript, 'a') as f:
    f.writelines(f'{link}\n')
    f.writelines(f'  dir={download_dir}\n')
    if brand == "OpenGApps":
        f.writelines(f'  out={brand}-{arch}-{variant}.zip\n')
    elif brand == "MindTheGapps":
        f.writelines(f'  out={brand}-{arch}.zip\n')
    f.close