rhodecode/apps/ssh_support/lib/backends/git.py

# Copyright (C) 2016-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 sys
import logging
import subprocess

from vcsserver import hooks
from .base import SshVcsServer

log = logging.getLogger(__name__)


class GitTunnelWrapper(object):
    process = None

    def __init__(self, server):
        self.server = server
        self.stdin = sys.stdin
        self.stdout = sys.stdout

    def create_hooks_env(self):
        pass

    def command(self):
        root = self.server.get_root_store()
        command = "cd {root}; {git_path} {mode} '{root}{repo_name}'".format(
            root=root, git_path=self.server.git_path,
            mode=self.server.repo_mode, repo_name=self.server.repo_name)
        log.debug("Final CMD: %s", command)
        return command

    def run(self, extras):
        action = "push" if self.server.repo_mode == "receive-pack" else "pull"
        exit_code = self.server._check_permissions(action)
        if exit_code:
            return exit_code

        scm_extras = self.server.update_environment(action=action, extras=extras)

        if action == "pull":
            hook_response = hooks.git_pre_pull(scm_extras)
            pre_pull_messages = hook_response.output
            sys.stdout.write(pre_pull_messages)

        self.create_hooks_env()
        result = subprocess.run(self.command(), shell=True)
        result = result.returncode

        # Upload-pack == clone
        if action == "pull":
            hook_response = hooks.git_post_pull(scm_extras)
            post_pull_messages = hook_response.output
            sys.stderr.write(post_pull_messages)
        return result


class GitServer(SshVcsServer):
    backend = 'git'
    repo_user_agent = 'git'

    def __init__(self, store, ini_path, repo_name, repo_mode, user, user_permissions, settings, env):
        super().__init__(user, user_permissions, settings, env)

        self.store = store
        self.ini_path = ini_path
        self.repo_name = repo_name
        self._path = self.git_path = settings['ssh.executable.git']

        self.repo_mode = repo_mode
        self.tunnel = GitTunnelWrapper(server=self)