rhodecode/config/environment.py

# Copyright (C) 2010-2023 RhodeCode GmbH
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License, version 3
# (only), as published by the Free Software Foundation.
#
# This program 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 General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
# This program is dual-licensed. If you wish to learn more about the
# RhodeCode Enterprise Edition, including its added features, Support services,
# and proprietary license terms, please see https://rhodecode.com/licenses/

import logging

import rhodecode
import collections

from rhodecode.config import utils

from rhodecode.lib.utils import load_rcextensions
from rhodecode.lib.utils2 import str2bool
from rhodecode.lib.vcs import connect_vcs

log = logging.getLogger(__name__)


def propagate_rhodecode_config(global_config, settings, config):
    # Store the settings to make them available to other modules.
    settings_merged = global_config.copy()
    settings_merged.update(settings)
    if config:
        settings_merged.update(config)

    rhodecode.PYRAMID_SETTINGS = settings_merged
    rhodecode.CONFIG = settings_merged

    if 'default_user_id' not in rhodecode.CONFIG:
        rhodecode.CONFIG['default_user_id'] = utils.get_default_user_id()
    log.debug('set rhodecode.CONFIG data')


def load_pyramid_environment(global_config, settings):
    # Some parts of the code expect a merge of global and app settings.
    settings_merged = global_config.copy()
    settings_merged.update(settings)

    # TODO(marcink): probably not required anymore
    # configure channelstream,
    settings_merged['channelstream_config'] = {
        'enabled': str2bool(settings_merged.get('channelstream.enabled', False)),
        'server': settings_merged.get('channelstream.server'),
        'secret': settings_merged.get('channelstream.secret')
    }

    # If this is a test run we prepare the test environment like
    # creating a test database, test search index and test repositories.
    # This has to be done before the database connection is initialized.
    if rhodecode.is_test:
        rhodecode.disable_error_handler = True
        from rhodecode import authentication
        authentication.plugin_default_auth_ttl = 0

        utils.initialize_test_environment(settings_merged)

    # Initialize the database connection.
    utils.initialize_database(settings_merged)

    load_rcextensions(root_path=settings_merged['here'])

    # Limit backends to `vcs.backends` from configuration, and preserve the order
    for alias in list(rhodecode.BACKENDS.keys()):
        if alias not in settings['vcs.backends']:
            del rhodecode.BACKENDS[alias]

    _sorted_backend = sorted(rhodecode.BACKENDS.items(),
                             key=lambda item: settings['vcs.backends'].index(item[0]))
    rhodecode.BACKENDS = collections.OrderedDict(_sorted_backend)

    log.info('Enabled VCS backends: %s', list(rhodecode.BACKENDS.keys()))

    # initialize vcs client and optionally run the server if enabled
    vcs_server_uri = settings['vcs.server']
    vcs_server_enabled = settings['vcs.server.enable']

    utils.configure_vcs(settings)

    # first run, to store data...
    propagate_rhodecode_config(global_config, settings, {})

    if vcs_server_enabled:
        connect_vcs(vcs_server_uri, utils.get_vcs_server_protocol(settings))
    else:
        log.warning('vcs-server not enabled, vcs connection unavailable')