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

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

# Copyright (C) 2016-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 os
import sys
import shutil
import logging
import tempfile
import textwrap

from .base import VcsServer

log = logging.getLogger(__name__)


class MercurialTunnelWrapper(object):
    process = None

    def __init__(self, server):
        self.server = server
        self.stdin = sys.stdin
        self.stdout = sys.stdout
        self.svn_conf_fd, self.svn_conf_path = tempfile.mkstemp()
        self.hooks_env_fd, self.hooks_env_path = tempfile.mkstemp()

    def create_hooks_env(self):

        content = textwrap.dedent(
            '''
            # SSH hooks version=1.0.0
            [hooks]
            pretxnchangegroup.ssh_auth=python:vcsserver.hooks.pre_push_ssh_auth
            pretxnchangegroup.ssh=python:vcsserver.hooks.pre_push_ssh
            changegroup.ssh=python:vcsserver.hooks.post_push_ssh
            
            preoutgoing.ssh=python:vcsserver.hooks.pre_pull_ssh
            outgoing.ssh=python:vcsserver.hooks.post_pull_ssh

            '''
        )

        with os.fdopen(self.hooks_env_fd, 'w') as hooks_env_file:
            hooks_env_file.write(content)
        root = self.server.get_root_store()

        hgrc_custom = os.path.join(
            root, self.server.repo_name, '.hg', 'hgrc_rhodecode')
        log.debug('Wrote custom hgrc file under %s', hgrc_custom)
        shutil.move(
            self.hooks_env_path, hgrc_custom)

        hgrc_main = os.path.join(
            root, self.server.repo_name, '.hg', 'hgrc')
        include_marker = '%include hgrc_rhodecode'

        if not os.path.isfile(hgrc_main):
            os.mknod(hgrc_main)

        with open(hgrc_main, 'rb') as f:
            data = f.read()
            has_marker = include_marker in data

        if not has_marker:
            log.debug('Adding include marker for hooks')
            with open(hgrc_main, 'wa') as f:
                f.write(textwrap.dedent('''
                # added by RhodeCode
                {}
                '''.format(include_marker)))

    def command(self):
        root = self.server.get_root_store()

        command = (
            "cd {root}; {hg_path} -R {root}{repo_name} "
            "serve --stdio".format(
                root=root, hg_path=self.server.hg_path,
                repo_name=self.server.repo_name))
        log.debug("Final CMD: %s", command)
        return command

    def run(self, extras):
        # at this point we cannot tell, we do further ACL checks
        # inside the hooks
        action = '?'
        # permissions are check via `pre_push_ssh_auth` hook
        self.server.update_environment(action=action, extras=extras)
        self.create_hooks_env()
        return os.system(self.command())


class MercurialServer(VcsServer):
    backend = 'hg'

    def __init__(self, store, ini_path, repo_name,
                 user, user_permissions, config, env):
        super(MercurialServer, self).\
            __init__(user, user_permissions, config, env)

        self.store = store
        self.ini_path = ini_path
        self.repo_name = repo_name
        self._path = self.hg_path = config.get(
            'app:main', 'ssh.executable.hg')

        self.tunnel = MercurialTunnelWrapper(server=self)