summaryrefslogtreecommitdiffstats
path: root/taskcluster/gecko_taskgraph/util/platforms.py
blob: 3010b327922b5617f04118a81f9d4933c17010a1 (plain)
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
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.


import re

from taskgraph.util.attributes import keymatch

# platform family is extracted from build platform by taking the alphabetic prefix
# and then translating win -> windows
_platform_re = re.compile(r"^[a-z]*")
_renames = {"win": "windows"}


_archive_formats = {
    "linux": ".tar.bz2",
    "macosx": ".tar.gz",
    "windows": ".zip",
}

_executable_extension = {
    "linux": "",
    "macosx": "",
    "windows": ".exe",
}

_architectures = {
    r"linux\b.*": "x86",
    r"linux64\b(?!-aarch64).*": "x86_64",
    r"linux64-aarch64\b.*": "aarch64",
    r"macosx64\b.*": "macos-x86_64-aarch64",
    r"win32\b.*": "x86",
    r"win64\b(?!-aarch64).*": "x86_64",
    r"win64-aarch64\b.*": "aarch64",
}


def platform_family(build_platform):
    """Given a build platform, return the platform family (linux, macosx, etc.)"""
    family = _platform_re.match(build_platform).group(0)
    return _renames.get(family, family)


def archive_format(build_platform):
    """Given a build platform, return the archive format used on the platform."""
    return _archive_formats[platform_family(build_platform)]


def executable_extension(build_platform):
    """Given a build platform, return the executable extension used on the platform."""
    return _executable_extension[platform_family(build_platform)]


def architecture(build_platform):
    matches = keymatch(_architectures, build_platform)
    if len(matches) == 1:
        return matches[0]
    raise Exception(f"Could not determine architecture of platform `{build_platform}`.")