Source code for toxicbuild.slave.protocols

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

# Copyright 2015, 2016 Juca Crispim <juca@poraodojuca.net>

# This file is part of toxicbuild.

# toxicbuild is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.

# toxicbuild 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 General Public License
# along with toxicbuild. If not, see <http://www.gnu.org/licenses/>.

import asyncio
import traceback
from toxicbuild.core.protocol import BaseToxicProtocol
from toxicbuild.core.utils import log, datetime2string, now
from toxicbuild.slave import BuildManager, settings
from toxicbuild.slave.exceptions import (BadData, BadBuilderConfig)


[docs]class BuildServerProtocol(BaseToxicProtocol): """ A simple server for build requests. """ salt = settings.BCRYPT_SALT encrypted_token = settings.ACCESS_TOKEN @asyncio.coroutine
[docs] def client_connected(self): try: self.log('executing {} for {}'.format(self.action, self.peername)) status = 0 if self.action == 'healthcheck': yield from self.healthcheck() elif self.action == 'list_builders': yield from self.list_builders() elif self.action == 'build': # build has a strange behavior. It sends messages to the client # directly. I think it should be an iterable here and the # protocol should send the messages. But how do to that? # I tought, instead of the iterable, use messages sent from # BuildManager and captured by the protocol, but didn't try # that yet. Any ideas? build_info = yield from self.build() yield from self.send_response(code=0, body=build_info) else: msg = 'Action {} does not exist'.format(self.action) self.log(msg, level='error') yield from self.send_response(code=1, body={'error': msg}) except BadData: msg = 'Something wrong with your data {}'.format(self.raw_data) self.log('bad data', level='error') yield from self.send_response(code=1, body={'error': msg}) status = 1 except Exception as e: self.log(e.args[0], level='error') msg = traceback.format_exc() status = 1 yield from self.send_response(code=1, body={'error': msg}) finally: self.close_connection() return status
@asyncio.coroutine
[docs] def healthcheck(self): """ Informs that the server is up and running """ yield from self.send_response(code=0, body='I\'m alive!')
@asyncio.coroutine
[docs] def list_builders(self): """ Informs all builders' names for this repo/branch/named_tree """ with (yield from self.get_buildmanager()) as manager: # We do not work after wait because if we wait for it # the other instance working is in the same named_tree yield from manager.update_and_checkout(work_after_wait=False) builder_names = manager.list_builders() yield from self.send_response(code=0, body={'builders': builder_names})
@asyncio.coroutine
[docs] def build(self): """ Performs a build requested by the client using the params sent in the request data """ with (yield from self.get_buildmanager()) as manager: yield from manager.update_and_checkout(work_after_wait=False) try: builder_name = self.data['body']['builder_name'] except KeyError: raise BadData("No builder name for build.") try: builder = manager.load_builder(builder_name) except BadBuilderConfig: build_info = {'steps': [], 'status': 'exception', 'started': datetime2string(now()), 'finished': datetime2string(now()), 'branch': manager.branch, 'named_tree': manager.named_tree} else: build_info = yield from builder.build() return build_info
@asyncio.coroutine
[docs] def get_buildmanager(self): """ Returns the builder manager for this request """ try: repo_url = self.data['body']['repo_url'] branch = self.data['body']['branch'] vcs_type = self.data['body']['vcs_type'] named_tree = self.data['body']['named_tree'] except KeyError: raise BadData('Bad data!') manager = BuildManager(self, repo_url, vcs_type, branch, named_tree) return manager
[docs] def log(self, msg, level='info'): log('[{}] {} '.format(type(self).__name__, msg), level)