Source code for tool_shed.webapp.controllers.hg

import logging

from mercurial.hgweb.hgwebdir_mod import hgwebdir

from galaxy import web
from galaxy.exceptions import ObjectNotFound
from galaxy.model.base import transaction
from galaxy.webapps.base.controller import BaseUIController
from tool_shed.util.repository_util import get_repository_by_name_and_owner

log = logging.getLogger(__name__)


[docs]class PortAsStringMiddleware:
[docs]    def __init__(self, application):
        self.application = application

    def __call__(self, environ, start_response):
        environ["SERVER_PORT"] = str(environ["SERVER_PORT"])
        return self.application(environ, start_response)


[docs]class HgController(BaseUIController):
[docs]    @web.expose
    def handle_request(self, trans, **kwd):
        # The os command that results in this method being called will look something like:
        # hg clone http://test@127.0.0.1:9009/repos/test/convert_characters1
        hgweb_config = trans.app.hgweb_config_manager.hgweb_config

        def make_web_app():
            hgwebapp = hgwebdir(hgweb_config.encode("utf-8"))
            return hgwebapp

        wsgi_app = make_web_app()
        repository = None
        path_info = kwd.get("path_info", None)
        if path_info and len(path_info.split("/")) == 2:
            owner, name = path_info.split("/")
            repository = get_repository_by_name_and_owner(trans.app, name, owner)
        if repository:
            if repository.deprecated:
                raise ObjectNotFound("Requested repository not found or deprecated.")
            cmd = kwd.get("cmd", None)
            if cmd == "getbundle":
                times_downloaded = repository.times_downloaded
                times_downloaded += 1
                repository.times_downloaded = times_downloaded
                trans.sa_session.add(repository)
                with transaction(trans.sa_session):
                    trans.sa_session.commit()
        return PortAsStringMiddleware(wsgi_app)