Show More
hg.py
123 lines
| 4.0 KiB
| text/x-python
|
PythonLexer
r2187 | # -*- 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) | ||||