rhodecode/apps/repository/views/repo_caches.py

# Copyright (C) 2011-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 os
import logging

from pyramid.httpexceptions import HTTPFound


from rhodecode.apps._base import RepoAppView
from rhodecode.lib.auth import (
    LoginRequired, HasRepoPermissionAnyDecorator, CSRFRequired)
from rhodecode.lib import helpers as h, rc_cache
from rhodecode.lib import system_info
from rhodecode.model.meta import Session
from rhodecode.model.scm import ScmModel

log = logging.getLogger(__name__)


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

    @LoginRequired()
    @HasRepoPermissionAnyDecorator('repository.admin')
    def repo_caches(self):
        c = self.load_default_context()
        c.active = 'caches'
        cached_diffs_dir = c.rhodecode_db_repo.cached_diffs_dir
        c.cached_diff_count = len(c.rhodecode_db_repo.cached_diffs())
        c.cached_diff_size = 0
        if os.path.isdir(cached_diffs_dir):
            c.cached_diff_size = system_info.get_storage_size(cached_diffs_dir)
        c.shadow_repos = c.rhodecode_db_repo.shadow_repos()

        cache_namespace_uid = f'repo.{self.db_repo.repo_id}'
        c.region = rc_cache.get_or_create_region('cache_repo', cache_namespace_uid)
        c.backend = c.region.backend
        c.repo_keys = sorted(c.region.backend.list_keys(prefix=cache_namespace_uid))

        return self._get_template_context(c)

    @LoginRequired()
    @HasRepoPermissionAnyDecorator('repository.admin')
    @CSRFRequired()
    def repo_caches_purge(self):
        _ = self.request.translate
        c = self.load_default_context()
        c.active = 'caches'
        invalidated = 0

        try:
            ScmModel().mark_for_invalidation(self.db_repo_name, delete=True)
            Session().commit()
            invalidated +=1
        except Exception:
            log.exception("Exception during cache invalidation")
            h.flash(_('An error occurred during cache invalidation'),
                    category='error')

        try:
            invalidated += 1
            self.rhodecode_vcs_repo.vcsserver_invalidate_cache(delete=True)
        except Exception:
            log.exception("Exception during vcsserver cache invalidation")
            h.flash(_('An error occurred during vcsserver cache invalidation'),
                    category='error')

        if invalidated:
            h.flash(_('Cache invalidation successful. Stages {}/2').format(invalidated),
                    category='success')

        raise HTTPFound(h.route_path(
            'edit_repo_caches', repo_name=self.db_repo_name))