import requests
from requests.adapters import HTTPAdapter, Retry
from datetime import datetime, timedelta

from rich.table import Table
from rich.console import Console

console = Console()

table = Table(title="VSCode Extensions")


def get_vscode_extensions(
    id,
    max_page=10000,
    page_size=100,
    include_versions=True,
    include_files=True,
    include_category_and_tags=True,
    include_shared_accounts=True,
    include_version_properties=True,
    exclude_non_validated=False,
    include_installation_targets=True,
    include_asset_uri=True,
    include_statistics=True,
    include_latest_version_only=False,
    unpublished=False,
    include_name_conflict_info=True,
    api_version="7.2-preview.1",
    session=None,
):
    if not session:
        session = requests.session()

    headers = {"Accept": f"application/json; charset=utf-8; api-version={api_version}"}

    flags = 0
    if include_versions:
        flags |= 0x1

    if include_files:
        flags |= 0x2

    if include_category_and_tags:
        flags |= 0x4

    if include_shared_accounts:
        flags |= 0x8

    if include_shared_accounts:
        flags |= 0x8

    if include_version_properties:
        flags |= 0x10

    if exclude_non_validated:
        flags |= 0x20

    if include_installation_targets:
        flags |= 0x40

    if include_asset_uri:
        flags |= 0x80

    if include_statistics:
        flags |= 0x100

    if include_latest_version_only:
        flags |= 0x200

    if unpublished:
        flags |= 0x1000

    if include_name_conflict_info:
        flags |= 0x8000

    for page in range(1, max_page + 1):
        body = {
            "filters": [
                {
                    "criteria": [{"filterType": 7, "value": id}],
                    "pageNumber": page,
                    "pageSize": page_size,
                    "sortBy": 0,
                    "sortOrder": 0,
                }
            ],
            "assetTypes": [],
            "flags": flags,
        }

        r = session.post(
            "https://marketplace.visualstudio.com/_apis/public/gallery/extensionquery",
            json=body,
            headers=headers,
        )
        r.raise_for_status()
        response = r.json()

        extensions = response["results"][0]["extensions"]
        for extension in extensions:
            yield extension

        if len(extensions) != page_size:
            break


def main():
    retry_strategy = Retry(
        total=3,
        backoff_factor=1,
        status_forcelist=[429, 500, 502, 503, 504],
        allowed_methods=["HEAD", "GET", "OPTIONS"],
    )
    adapter = HTTPAdapter(max_retries=retry_strategy)
    session = requests.Session()
    session.mount("https://", adapter)
    session.mount("http://", adapter)

    details = []
    categories = {}
    used = set()

    table.add_column("Name", style="cyan", no_wrap=True)
    table.add_column("Version")
    table.add_column("Last Update")
    table.add_column("Install Count", justify="right")
    table.add_column("Description")

    for id in open("vscode-extensions.txt").readlines():
        for extension in get_vscode_extensions(id.strip(), session=session):
            extension_name = extension["extensionName"]
            extension_description = extension.get("shortDescription", "")
            extensions_versions = extension["versions"]
            extension_last_update = extension["lastUpdated"]
            last_update_dt = datetime.strptime(
                extension_last_update, "%Y-%m-%dT%H:%M:%S.%fZ"
            )
            last_update = last_update_dt.strftime("%Y-%m-%d")
            extension_categories = extension["categories"]

            extensions_statistics = dict(
                {
                    (item["statisticName"], item["value"])
                    for item in extension["statistics"]
                }
            )
            extension_version_info = extensions_versions.pop(0)
            extension_version = extension_version_info["version"]

            table.add_row(
                extension_name,
                extension_version,
                last_update,
                str(round(extensions_statistics.get("install", 0))),
                extension_description,
            )
            info = f"{id.strip():50} {last_update:10} {str(round(extensions_statistics.get("install", 0))):>12} {extension_description}"

            for cat in extension_categories:
                if info in used:
                    continue
                used.add(cat)
                categories.setdefault(cat, []).append(info)

            details.append(info)
            break

    console.print(table)

    with open("vscode-extensions-with-desc.txt", "w") as f:
        f.write("\n".join(details))

    with open("vscode-extensions-categories.txt", "w") as f:
        for cat, extensions in sorted(categories.items()):
            f.write(f"{cat}\n")
            f.write("\n".join(extensions))
            f.write("\n\n---\n\n")


if __name__ == "__main__":
    main()