DevExtDownloader/vscode.py at master · XFY9326/DevExtDownloader · 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
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
import asyncio
import shutil
from pathlib import Path

from dev_ext_downloader.common.models import DownloadOptions
from dev_ext_downloader.vscode import VSCodeExt, VSCodeExtFilterOptions, TargetPlatformType
from dev_ext_downloader.vscode import download_latest_extensions, generate_index_html

# Download dir
DOWNLOAD_DIR: Path = Path("./downloads/VSCode")

# Download temp dir
TEMP_DIR: Path = DOWNLOAD_DIR / ".temp"

# Task spec path
TASK_SPEC_PATH: Path = DOWNLOAD_DIR / "task-spec.json"

# Skip if exists or not
# If exists, skip download
SKIP_IF_EXISTS: bool = True

# Only keep latest version
# Depends on metadata
KEEP_ONLY_LATEST: bool = True

# Download concurrency
DOWNLOAD_CONCURRENCY: int = 8

# No metadata or not
# Generate [ext_id.json] before download
NO_METADATA: bool = False

# Flatten dir or not
# No flatten dir:
# [download_dir]
#     └── [ext_id]
#         ├── [ext_id.vsix]
#         └── [ext_id.json]
# Flatten dir:
# [download_dir]
#     ├── [ext_id.vsix]
#     └── [ext_id.json]
FLATTEN_DIR: bool = False

# Target platform or None
TARGET_PLATFORM: tuple[TargetPlatformType, ...] | None = (
    TargetPlatformType.WIN32_X64,
    TargetPlatformType.LINUX_X64
)

# Target platform fallback or None
TARGET_PLATFORM_FALLBACK: TargetPlatformType | None = TargetPlatformType.UNIVERSAL

# Required VSCode version or None
# It will download the latest version if not set
VSCODE_VERSION: str | None = "1.97.2"

# Include prerelease or not
# If not set, it will download the latest prerelease version
INCLUDE_PRERELEASE: bool = True

# VSIX packages id list
# Example: https://marketplace.visualstudio.com/items?itemName=ms-python.python
# [ext_id] is ms-python.python
VSIX_LIST: set[str | VSCodeExt] = {
    "ms-python.python",
    "ms-python.debugpy"
}

# For local test
# noinspection PyBroadException
try:
    from local_config.vscode import *
except:
    pass


async def main() -> None:
    await download_latest_extensions(
        query_ext=VSIX_LIST,
        target_dir=DOWNLOAD_DIR,
        temp_dir=TEMP_DIR,
        concurrency=DOWNLOAD_CONCURRENCY,
        task_spec_path=TASK_SPEC_PATH,
        default_download_options=DownloadOptions(
            skip_if_exists=SKIP_IF_EXISTS,
            no_metadata=NO_METADATA,
            flatten_dir=FLATTEN_DIR,
            keep_only_latest=KEEP_ONLY_LATEST
        ),
        default_filter_options=VSCodeExtFilterOptions(
            target_platform=TARGET_PLATFORM,
            target_platform_fallback=TARGET_PLATFORM_FALLBACK,
            vscode_version=VSCODE_VERSION,
            include_prerelease=INCLUDE_PRERELEASE,
        ),
    )
    if not NO_METADATA:
        await generate_index_html(download_dir=DOWNLOAD_DIR, is_flatten=FLATTEN_DIR)


if __name__ == "__main__":
    try:
        asyncio.run(main())
    finally:
        shutil.rmtree(TEMP_DIR, ignore_errors=True)