|
|
# -*- coding: utf-8 -*-
|
|
|
|
|
|
# Copyright (C) 2010-2020 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
|
|
|
|
|
|
from pyramid.events import ApplicationCreated
|
|
|
from pyramid.settings import asbool
|
|
|
|
|
|
from rhodecode.apps._base import ADMIN_PREFIX
|
|
|
from rhodecode.lib.ext_json import json
|
|
|
from rhodecode.lib.str_utils import safe_str
|
|
|
|
|
|
|
|
|
def url_gen(request):
|
|
|
registry = request.registry
|
|
|
longpoll_url = registry.settings.get('channelstream.longpoll_url', '')
|
|
|
ws_url = registry.settings.get('channelstream.ws_url', '')
|
|
|
proxy_url = request.route_url('channelstream_proxy')
|
|
|
urls = {
|
|
|
'connect': request.route_path('channelstream_connect'),
|
|
|
'subscribe': request.route_path('channelstream_subscribe'),
|
|
|
'longpoll': longpoll_url or proxy_url,
|
|
|
'ws': ws_url or proxy_url.replace('http', 'ws')
|
|
|
}
|
|
|
return safe_str(json.dumps(urls))
|
|
|
|
|
|
|
|
|
PLUGIN_DEFINITION = {
|
|
|
'name': 'channelstream',
|
|
|
'config': {
|
|
|
'javascript': [],
|
|
|
'css': [],
|
|
|
'template_hooks': {
|
|
|
'plugin_init_template': 'rhodecode:templates/channelstream/plugin_init.mako'
|
|
|
},
|
|
|
'url_gen': url_gen,
|
|
|
'static': None,
|
|
|
'enabled': False,
|
|
|
'server': '',
|
|
|
'secret': ''
|
|
|
}
|
|
|
}
|
|
|
|
|
|
|
|
|
def maybe_create_history_store(event):
|
|
|
# create plugin history location
|
|
|
settings = event.app.registry.settings
|
|
|
history_dir = settings.get('channelstream.history.location', '')
|
|
|
if history_dir and not os.path.exists(history_dir):
|
|
|
os.makedirs(history_dir, 0o750)
|
|
|
|
|
|
|
|
|
def includeme(config):
|
|
|
from rhodecode.apps.channelstream.views import ChannelstreamView
|
|
|
|
|
|
settings = config.registry.settings
|
|
|
PLUGIN_DEFINITION['config']['enabled'] = asbool(
|
|
|
settings.get('channelstream.enabled'))
|
|
|
PLUGIN_DEFINITION['config']['server'] = settings.get(
|
|
|
'channelstream.server', '')
|
|
|
PLUGIN_DEFINITION['config']['secret'] = settings.get(
|
|
|
'channelstream.secret', '')
|
|
|
PLUGIN_DEFINITION['config']['history.location'] = settings.get(
|
|
|
'channelstream.history.location', '')
|
|
|
config.register_rhodecode_plugin(
|
|
|
PLUGIN_DEFINITION['name'],
|
|
|
PLUGIN_DEFINITION['config']
|
|
|
)
|
|
|
config.add_subscriber(maybe_create_history_store, ApplicationCreated)
|
|
|
|
|
|
config.add_route(
|
|
|
name='channelstream_connect',
|
|
|
pattern=ADMIN_PREFIX + '/channelstream/connect')
|
|
|
config.add_view(
|
|
|
ChannelstreamView,
|
|
|
attr='channelstream_connect',
|
|
|
route_name='channelstream_connect', renderer='json_ext')
|
|
|
|
|
|
config.add_route(
|
|
|
name='channelstream_subscribe',
|
|
|
pattern=ADMIN_PREFIX + '/channelstream/subscribe')
|
|
|
config.add_view(
|
|
|
ChannelstreamView,
|
|
|
attr='channelstream_subscribe',
|
|
|
route_name='channelstream_subscribe', renderer='json_ext')
|
|
|
|
|
|
config.add_route(
|
|
|
name='channelstream_proxy',
|
|
|
pattern=settings.get('channelstream.proxy_path') or '/_channelstream')
|
|
|
|
|
|
|
