Update Webhook response variabel

This commit is contained in:
shekhar-cis 2023-01-06 20:38:43 +05:30
parent 1b29a5639c
commit 89305ec602
Signed by untrusted user: shekhar-cis
GPG Key ID: F4F00AB04E83F9A7
1 changed files with 72 additions and 114 deletions

View File

@ -1,18 +1,14 @@
"""Transifex webhook handler """
import sys
import os import os
import base64 import base64
import time import time
import json import json
import re
import hmac import hmac
import pprint
import hashlib import hashlib
import requests
from subprocess import call from subprocess import call
from base64 import b64encode from base64 import b64encode
import requests
from buildbot.process.properties import Properties from buildbot.process.properties import Properties
from buildbot.util import bytes2unicode, unicode2bytes from buildbot.util import bytes2unicode, unicode2bytes
@ -25,152 +21,94 @@ from dateutil.parser import parse as dateparse
_HEADER_USER_AGENT = 'User-Agent' _HEADER_USER_AGENT = 'User-Agent'
_HEADER_SIGNATURE = 'X-TX-Signature' _HEADER_SIGNATURE = 'X-TX-Signature'
_HEADER_URL_PATH = 'X-TX-Url'
HTTP_DATE = 'date'
_EVENT_KEY = 'event' _EVENT_KEY = 'event'
transifexSecret = "" transifex_request_data = {}
transifexUsername = ""
transifexPassword = ""
transifex_dict = {}
secret = ""
master = ""
gitHubToken = os.environ('gitHubToken')
class TransifexHandler(BaseHookHandler): class TransifexHandler(BaseHookHandler):
def __init__(self, master, secret, transifex_dict): def __init__(self, master, secret, options, transifex_dict):
if not options:
options = {}
self.secret = secret self.secret = secret
self.master = master self.master = master
self.options = options
self.transifex_dict = transifex_dict self.transifex_dict = transifex_dict
def returnMessage(self, status = False, message = "Unimplemented"): def returnMessage(self, status = False, message = "Unimplemented"):
output = json.dumps({"status": "OK" if status else "FAIL", "message": message}) output = json.dumps({"status": "OK" if status else "FAIL", "message": message})
return [output, [('Content-type', 'text/plain'), return [output, [('Content-type', 'application/json')]]
('Content-Length', str(len(output)))
]]
def verifyTransifexSignature( def verifyTransifexSignature(
self, request, content, rendered_secret, signature, header_signature self, request, content, signature, header_signature
): ):
http_verb = 'POST' http_verb = 'POST'
http_url_path = request.headers('X-TX-Url') http_url_path = request.getHeader(_HEADER_URL_PATH)
http_gmt_date = request.headers('Date') http_gmt_date = request.getHeader(HTTP_DATE)
content_md5 = hashlib.md5(content).hexdigest() content_md5 = hashlib.md5(content).hexdigest()
msg = b'\n'.join([ msg = b'\n'.join([
http_verb, http_url_path, http_gmt_date, content_md5 http_verb, http_url_path, http_gmt_date, content_md5
]) ])
tx_signature = base64.b64encode( tx_signature = base64.b64encode(
hmac.new( hmac.new(
key=rendered_secret, key=self.rendered_secret,
msg=msg, msg=msg,
digestmod=hashlib.sha256 digestmod=hashlib.sha256
).digest() ).digest()
) )
if tx_signature() != header_signature: if tx_signature != header_signature:
raise ValueError('Invalid secret')
try:
if signature != os.environ.get('HTTP_X_TX_SIGNATURE'):
return False
return True
except:
return False return False
def downloadTranslatedLanguage(self, ts, lang): if signature != request.getHeader(_HEADER_SIGNATURE):
headers = {"Authorization": "Basic " + b64encode(transifexUsername + ":" + transifexPassword)} return False
fname = "bitmessage_" + lang.lower() + ".po"
with open("src/translations/" + fname, "wt") as handle:
response = requests.get("https://www.transifex.com/api/2/project/pybitmessage/resource/pybitmessage/translation/" + lang + "/",
headers=headers)
if response.ok:
content = json.loads(response.content)["content"]
handle.write(content.encode("utf-8"))
return response
def commitTranslatedLanguage(self, ts, lang):
call(["kivy", "src/translations/messages.pro"])
call(["git", "add", "src/translations/bitmessage_" + lang + ".ts", "src/translations/bitmessage_" + lang + ".qm"])
call(["git", "commit", "-q", "-S", "-m", "Auto-updated language %s from transifex" % (lang)])
newbranch = "translate_" + lang + "_" + str(ts)
call(["git", "push", "-q", "translations", newbranch + ":" + newbranch])
branch = transifex_dict['branch']
request = {
"title": "Translation update " + lang,
"body": "Auto-updated from transifex",
"head": "PyBitmessageTranslations:" + newbranch,
"base": branch
}
headers = {"Authorization": "token " + gitHubToken}
response = requests.post("https://api.github.com/repos/Bitmessage/PyBitmessage/pulls",
headers=headers, data=json.dumps(request))
return response
def process_translation_completed(self, payload, transifex_dict, event_type, codebase): def process_translation_completed(self, payload, transifex_dict, event_type, codebase):
changes = [] changes = []
transifex_response = self._transform_variables(payload, transifex_dict) transifex_response = self._transform_variables(payload, transifex_dict)
if 'pybitmessage-test' in transifex_response['project'] and 'messagespot' in transifex_response['resource']: if 'pybitmessage-test' in transifex_response['project'] and 'messagespot' in transifex_response['resource']:
if 'translation_completed' in transifex_response['event'] and 100 in transifex_response['translated']: if 'translation_completed' in transifex_response['event']:
ts = int(time.time()) ts = int(time.time())
lang = transifex_response['language'] lang = transifex_response['language']
branch = transifex_dict['branch'] return
self.downloadTranslatedLanguage(ts, lang.lower())
response = self.commitTranslatedLanguage(ts, lang.lower())
if response.ok:
output, responseHeaders = self.returnMessage(True, "Processed.")
else:
output, responseHeaders = self.returnMessage(False, "Error: %i." % (response.status_code))
else:
output, responseHeaders = self.returnMessage(False, "Nothing to do")
else:
output, responseHeaders = self.returnMessage(False, "Nothing to do")
# if isinstance(self.options, dict) and self.options.get('onlyIncludePushCommit', False): # if isinstance(self.options, dict):
# commits = commits[:1] # commits = commits[:1]
# for commit in commits: # # for commit in commits:
# files = [] # # files = []
# for kind in ('added', 'modified', 'removed'): # # for kind in ('added', 'modified', 'removed'):
# files.extend(commit.get(kind, []) or []) # # files.extend(commit.get(kind, []) or [])
# timestamp = dateparse(commit['timestamp']) change = {
# change = { 'author': 'buildbot-transifex,
# 'author': '{} <{}>'.format(commit['author']['name'], 'resource': transifex_response['resource'],
# commit['author']['email']), 'branch': transifex_dict['branch'],
# 'files': files, 'project': transifex_response['project'],
# 'comments': commit['message'], 'event': event_type,
# 'revision': commit['id'], 'properties': {
# 'when_timestamp': timestamp, 'branch': branch,
# 'branch': branch, 'revision': revision,
# 'revlink': commit['url'], 'language': lang,
# 'repository': repo_url, 'resource': resource,
# 'project': project, 'project': project
# 'category': event_type, }
# 'properties': { }
# 'event': event_type, if codebase is not None:
# 'repository_name': repository['name'], change['codebase'] = codebase
# 'owner': repository["owner"]["username"] changes.insert(0, change)
# },
# }
# if codebase is not None:
# change['codebase'] = codebase
# changes.insert(0, change)
return changes return changes
def process_review_completed(self, payload, transifex_data):
pass
def _transform_variables(self, payload, transifex_dict): def _transform_variables(self, payload, transifex_dict):
transifex_variables = { project = payload.get('project', 'None')
'project': payload['project'], transform_values = {
"translated": payload['translated'], project: {
"resource": payload['resource'], "repository": "https://github.com/Bitmessage/PyBitmessage",
"event": payload['event'], "branch": "v0.6"
"language": payload['language'] }
} }
return transform_values
return transifex_variables
@defer.inlineCallbacks @defer.inlineCallbacks
def getChanges(self, request): def getChanges(self, request):
@ -187,19 +125,39 @@ class TransifexHandler(BaseHookHandler):
if self.secret is not None: if self.secret is not None:
p = Properties() p = Properties()
p.master = self.master p.master = self.master
rendered_secret = yield p.render(self.secret) option = self.options
self.rendered_secret = yield p.render(self.secret)
signature = hmac.new( signature = hmac.new(
unicode2bytes(rendered_secret), unicode2bytes(self.rendered_secret),
unicode2bytes(content_text.strip()), unicode2bytes(content_text.strip()),
digestmod=hashlib.sha256) digestmod=hashlib.sha256)
header_signature = bytes2unicode( header_signature = bytes2unicode(
request.getHeader(_HEADER_SIGNATURE)) request.getHeader(_HEADER_SIGNATURE))
self.verifyTransifexSignature( self.verifyTransifexSignature(
request, content, rendered_secret, request, content, self.rendered_secret,
signature, header_signature signature, header_signature
) )
event_type = bytes2unicode(payload.get(_EVENT_KEY), "None") event_type = payload.get("event", "None")
language = payload.get("language", 'None')
project = payload.get("project", 'None')
resource = payload.get("resource", 'None')
transifex_request_data['branch'] = "v0.6"
transifex_request_data['revision'] = ""
transifex_request_data["properties"] = "langugage"
transifex_request_data["properties"] = "resource"
transifex_request_data["properties"] = "project"
transifex_request_data["properties"] = {
"branch": branch,
"revision": revision
}
transifex_request_data["changes"] = {
"author": "buildbot-transifex",
"repository": project,
}
log.msg("Received event '{}' from transifex".format(event_type)) log.msg("Received event '{}' from transifex".format(event_type))
codebase = "" codebase = ""