rhodecode/apps/repository/views/repo_settings_issue_trackers.py

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

# Copyright (C) 2017-2017 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

from pyramid.httpexceptions import HTTPFound
from pyramid.view import view_config

from rhodecode.apps._base import RepoAppView
from rhodecode.lib import audit_logger
from rhodecode.lib import helpers as h
from rhodecode.lib.auth import (
    LoginRequired, HasRepoPermissionAnyDecorator, CSRFRequired)
from rhodecode.model.forms import IssueTrackerPatternsForm
from rhodecode.model.meta import Session
from rhodecode.model.settings import IssueTrackerSettingsModel

log = logging.getLogger(__name__)


class RepoSettingsIssueTrackersView(RepoAppView):
    def load_default_context(self):
        c = self._get_local_tmpl_context()

        # TODO(marcink): remove repo_info and use c.rhodecode_db_repo instead
        c.repo_info = self.db_repo

        self._register_global_c(c)
        return c

    @LoginRequired()
    @HasRepoPermissionAnyDecorator('repository.admin')
    @view_config(
        route_name='edit_repo_issuetracker', request_method='GET',
        renderer='rhodecode:templates/admin/repos/repo_edit.mako')
    def repo_issuetracker(self):
        c = self.load_default_context()
        c.active = 'issuetracker'
        c.data = 'data'

        c.settings_model = IssueTrackerSettingsModel(repo=self.db_repo)
        c.global_patterns = c.settings_model.get_global_settings()
        c.repo_patterns = c.settings_model.get_repo_settings()

        return self._get_template_context(c)

    @LoginRequired()
    @HasRepoPermissionAnyDecorator('repository.admin')
    @CSRFRequired()
    @view_config(
        route_name='edit_repo_issuetracker_test', request_method='POST',
        xhr=True, renderer='string')
    def repo_issuetracker_test(self):
        return h.urlify_commit_message(
            self.request.POST.get('test_text', ''),
            self.db_repo_name)

    @LoginRequired()
    @HasRepoPermissionAnyDecorator('repository.admin')
    @CSRFRequired()
    @view_config(
        route_name='edit_repo_issuetracker_delete', request_method='POST',
        renderer='rhodecode:templates/admin/repos/repo_edit.mako')
    def repo_issuetracker_delete(self):
        _ = self.request.translate
        uid = self.request.POST.get('uid')
        repo_settings = IssueTrackerSettingsModel(repo=self.db_repo_name)
        try:
            repo_settings.delete_entries(uid)
        except Exception:
            h.flash(_('Error occurred during deleting issue tracker entry'),
                    category='error')
        else:
            h.flash(_('Removed issue tracker entry'), category='success')
        raise HTTPFound(
            h.route_path('edit_repo_issuetracker', repo_name=self.db_repo_name))

    def _update_patterns(self, form, repo_settings):
        for uid in form['delete_patterns']:
            repo_settings.delete_entries(uid)

        for pattern_data in form['patterns']:
            for setting_key, pattern, type_ in pattern_data:
                sett = repo_settings.create_or_update_setting(
                    setting_key, pattern.strip(), type_)
                Session().add(sett)

            Session().commit()

    @LoginRequired()
    @HasRepoPermissionAnyDecorator('repository.admin')
    @CSRFRequired()
    @view_config(
        route_name='edit_repo_issuetracker_update', request_method='POST',
        renderer='rhodecode:templates/admin/repos/repo_edit.mako')
    def repo_issuetracker_update(self):
        _ = self.request.translate
        # Save inheritance
        repo_settings = IssueTrackerSettingsModel(repo=self.db_repo_name)
        inherited = (
            self.request.POST.get('inherit_global_issuetracker') == "inherited")
        repo_settings.inherit_global_settings = inherited
        Session().commit()

        form = IssueTrackerPatternsForm()().to_python(self.request.POST)
        if form:
            self._update_patterns(form, repo_settings)

        h.flash(_('Updated issue tracker entries'), category='success')
        raise HTTPFound(
            h.route_path('edit_repo_issuetracker', repo_name=self.db_repo_name))