summaryrefslogtreecommitdiffstats
path: root/util/dependency_updater/tools/config.py
blob: 93b0cb5b89f0134d2643982d3427fe6d54a39d8b (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
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
107
108
109
110
111
112
113
114
115
116
117
############################################################################
##
## Copyright (C) 2021 The Qt Company Ltd.
## Contact: https://www.qt.io/licensing/
##
## This file is part of the qtqa module of the Qt Toolkit.
##
## $QT_BEGIN_LICENSE:LGPL$
## Commercial License Usage
## Licensees holding valid commercial Qt licenses may use this file in
## accordance with the commercial license agreement provided with the
## Software or, alternatively, in accordance with the terms contained in
## a written agreement between you and The Qt Company. For licensing terms
## and conditions see https://www.qt.io/terms-conditions. For further
## information use the contact form at https://www.qt.io/contact-us.
##
## GNU Lesser General Public License Usage
## Alternatively, this file may be used under the terms of the GNU Lesser
## General Public License version 3 as published by the Free Software
## Foundation and appearing in the file LICENSE.LGPL3 included in the
## packaging of this file. Please review the following information to
## ensure the GNU Lesser General Public License version 3 requirements
## will be met: https://www.gnu.org/licenses/lgpl-3.0.html.
##
## GNU General Public License Usage
## Alternatively, this file may be used under the terms of the GNU
## General Public License version 2.0 or (at your option) the GNU General
## Public license version 3 or any later version approved by the KDE Free
## Qt Foundation. The licenses are as published by the Free Software
## Foundation and appearing in the file LICENSE.GPL2 and LICENSE.GPL3
## included in the packaging of this file. Please review the following
## information to ensure the GNU General Public License requirements will
## be met: https://www.gnu.org/licenses/gpl-2.0.html and
## https://www.gnu.org/licenses/gpl-3.0.html.
##
## $QT_END_LICENSE$
##
############################################################################

import json
import os
from pathlib import Path
from shutil import copyfile

import urllib3 as urllib
import yaml
from url_normalize import url_normalize

from .datasources.datasources import Datasources
from .namespace import Namespace
from .repo import Repo
from .teams_connector import TeamsConnector


class Config(Namespace):
    """Configuration object. Also contains datasources for use."""
    args: Namespace
    cwd: os.PathLike
    datasources: Datasources = Datasources()
    teams_connector: TeamsConnector
    GERRIT_HOST: str
    GERRIT_STATE_PATH: str
    GERRIT_USERNAME: str
    GERRIT_PASSWORD: str
    MS_TEAMS_NOTIFY_URL: str
    state_repo: Repo
    state_data: dict[str, Repo] = {}
    _state_ref: str = None
    qt5_default: dict[str, Repo] = {}
    suppress_warn: bool = False
    REPOS: list[str]
    NON_BLOCKING_REPOS: list[str] = []
    rewind_module: Repo = None
    drop_dependency: Repo = None
    drop_dependency_from: list[Repo] = None


def _load_config(file, args):
    """Load configuration from disk or environment"""
    cwd = Path(__file__).parent.parent
    file = cwd.joinpath(file)
    c = dict()
    if file.exists():
        with open(file) as config_file:
            c = yaml.load(config_file, Loader=yaml.SafeLoader)
    else:
        try:
            copyfile(file.parent / (file.name + ".template"), file)
            print("Config file not found, so we created 'config.yaml' from the template.")
            with open(file) as config_file:
                c = yaml.load(config_file)
        except FileNotFoundError:
            print("ERROR: Unable to load config because config.yaml, or config.yaml.template\n"
                  "was not found on disk. Please pull/checkout config.yaml.template from\n"
                  "the repo again.")

    for key in c.keys():
        if os.environ.get(key):
            print(f'Overriding config option {key} with environment variable.')
            c[key] = os.environ[key]
    config = Config(**c)
    config.cwd = cwd
    config.args = args
    config.GERRIT_HOST = url_normalize(config.GERRIT_HOST)
    config.teams_connector = TeamsConnector(config)
    ssh_file = Path(os.path.expanduser('~'), ".ssh", "config")
    if ssh_file.exists():
        with open(ssh_file) as ssh_config:
            contents = ssh_config.read()
            gerrit_base_url = urllib.util.parse_url(config.GERRIT_HOST).host
            loc = contents.find(gerrit_base_url)
            user_loc = contents.find("User", loc)
            user_name = contents[user_loc:contents.find("\n", user_loc)].split(" ")[1]
            if user_name:
                config._state_ref = f"refs/personal/{user_name or config.GERRIT_USERNAME}" \
                                    f"/submodule_updater"
    return config