rhodecode/svn_support/utils.py

# -*- coding: utf-8 -*-

# Copyright (C) 2016-2016  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 codecs
import logging
import os
from pyramid.renderers import render

from rhodecode.events import trigger
from rhodecode.lib.utils import get_rhodecode_realm, get_rhodecode_base_path
from rhodecode.model.db import RepoGroup

from . import config_keys
from .events import ModDavSvnConfigChange


log = logging.getLogger(__name__)


def generate_mod_dav_svn_config(registry):
    """
    Generate the configuration file for use with subversion's mod_dav_svn
    module. The configuration has to contain a <Location> block for each
    available repository group because the mod_dav_svn module does not support
    repositories organized in sub folders.
    """
    settings = registry.settings
    config = _render_mod_dav_svn_config(
        parent_path_root=get_rhodecode_base_path(),
        list_parent_path=settings[config_keys.list_parent_path],
        location_root=settings[config_keys.location_root],
        repo_groups=RepoGroup.get_all_repo_groups(),
        realm=get_rhodecode_realm())
    _write_mod_dav_svn_config(config, settings[config_keys.config_file_path])

    # Trigger an event on mod dav svn configuration change.
    trigger(ModDavSvnConfigChange(), registry)


def _render_mod_dav_svn_config(
        parent_path_root, list_parent_path, location_root, repo_groups, realm):
    """
    Render mod_dav_svn configuration to string.
    """
    repo_group_paths = []
    for repo_group in repo_groups:
        group_path = repo_group.full_path_splitted
        location = os.path.join(location_root, *group_path)
        parent_path = os.path.join(parent_path_root, *group_path)
        repo_group_paths.append((location, parent_path))

    context = {
        'location_root': location_root,
        'parent_path_root': parent_path_root,
        'repo_group_paths': repo_group_paths,
        'svn_list_parent_path': list_parent_path,
        'rhodecode_realm': realm,
    }

    # Render the configuration template to string.
    template = 'rhodecode:svn_support/templates/mod-dav-svn.conf.mako'
    return render(template, context)


def _write_mod_dav_svn_config(config, filepath):
    """
    Write mod_dav_svn config to file.
    """
    with codecs.open(filepath, 'w', encoding='utf-8') as f:
        f.write(config)