# Based on the gitlab reporter from buildbot from __future__ import absolute_import from __future__ import print_function from twisted.internet import defer from twisted.python import log from buildbot.process.properties import Interpolate from buildbot.process.properties import Properties from buildbot.process.results import CANCELLED from buildbot.process.results import EXCEPTION from buildbot.process.results import FAILURE from buildbot.process.results import RETRY from buildbot.process.results import SKIPPED from buildbot.process.results import SUCCESS from buildbot.process.results import WARNINGS from buildbot.reporters import http from buildbot.util import httpclientservice from buildbot.util import unicode2NativeString import re class GiteaStatusPush(http.HttpStatusPushBase): name = "GiteaStatusPush" neededDetails = dict(wantProperties=True) ssh_url_match = re.compile(r"(ssh://)?[\w+\-\_]+@[\w+\.\-\_]:?\d*/(?P[\w_\-\.]+)/(?P[\w_\-\.]+)\.git") @defer.inlineCallbacks def reconfigService(self, baseURL, token, startDescription=None, endDescription=None, context=None, context_pr=None, verbose=False, warningAsSuccess=False, **kwargs): token = yield self.renderSecrets(token) yield http.HttpStatusPushBase.reconfigService(self, **kwargs) self.context = context or Interpolate('buildbot/%(prop:buildername)s') self.context_pr = context_pr or \ Interpolate('buildbot/pull_request/%(prop:buildername)s') self.startDescription = startDescription or 'Build started.' self.endDescription = endDescription or 'Build done.' if baseURL.endswith('/'): baseURL = baseURL[:-1] self.baseURL = baseURL self._http = yield httpclientservice.HTTPClientService.getService( self.master, baseURL, headers={'AuthorizationHeaderToken': 'token {}'.format(token)}, debug=self.debug, verify=self.verify) self.verbose = verbose self.project_ids = {} self.warningAsSuccess = warningAsSuccess def createStatus(self, project_owner, repo_name, sha, state, target_url=None, description=None, context=None): """ :param project_owner: username of the owning user or organization :param repo_name: name of the repository :param sha: Full sha to create the status for. :param state: one of the following 'pending', 'success', 'failed' or 'cancelled'. :param target_url: Target url to associate with this status. :param description: Short description of the status. :param context: Context of the result :return: A deferred with the result from GitLab. """ payload = {'state': state} if description is not None: payload['description'] = description if target_url is not None: payload['target_url'] = target_url if context is not None: payload['name'] = context return self._http.post( '/api/v1/repos/{owner}/{repository}/statuses/{sha}'.format( owner=project_owner, repository=repo_name, sha=sha ), json=payload) @defer.inlineCallbacks def send(self, build): props = Properties.fromDict(build['properties']) props.master = self.master if build['complete']: state = { SUCCESS: 'success', WARNINGS: 'success' if self.warningAsSuccess else 'warning', FAILURE: 'failure', SKIPPED: 'success', EXCEPTION: 'error', RETRY: 'pending', CANCELLED: 'error' }.get(build['results'], 'failure') description = yield props.render(self.endDescription) else: state = 'pending' description = yield props.render(self.startDescription) if 'pr_id' in props: context = yield props.render(self.context_pr) else: context = yield props.render(self.context) sourcestamps = build['buildset']['sourcestamps'] for sourcestamp in sourcestamps: sha = sourcestamp['revision'] if 'repository_name' in props: repository_name = props['repository_name'] else: match = re.match(self.ssh_url_match, sourcestamp['repository']) if match is not None: repository_name = match.group("repo_name") else: log.msg( "Could not send status, " "build has no repository_name property for Gitea.") continue if 'owner' in props: repository_owner = props['owner'] else: match = re.match(self.ssh_url_match, sourcestamp['repository']) if match is not None: repository_owner = match.group("owner") else: log.msg( "Could not send status, " "build has no owner property for Gitea.") continue try: sha = unicode2NativeString(sha) state = unicode2NativeString(state) target_url = unicode2NativeString(build['url']) context = unicode2NativeString(context) description = unicode2NativeString(description) res = yield self.createStatus( project_owner=repository_owner, repo_name=repository_name, sha=sha, state=state, target_url=target_url, context=context, description=description ) if res.code not in (200, 201, 204): message = yield res.json() message = message.get('message', 'unspecified error') log.msg( 'Could not send status "{state}" for ' '{repo} at {sha}: {message}'.format( state=state, repo=sourcestamp['repository'], sha=sha, message=message)) elif self.verbose: log.msg( 'Status "{state}" sent for ' '{repo} at {sha}.'.format( state=state, repo=sourcestamp['repository'], sha=sha)) except Exception as e: log.err( e, 'Failed to send status "{state}" for ' '{repo} at {sha}'.format( state=state, repo=sourcestamp['repository'], sha=sha ))