U Webhook触发的自动发布读取commit信息作为申请标题

pull/462/head
vapao 2022-03-10 14:48:13 +08:00
parent 957ad48e8c
commit 5963c1c09a
1 changed files with 67 additions and 55 deletions

View File

@ -13,18 +13,17 @@ import json
def auto_deploy(request, deploy_id, kind):
if not _is_valid_token(request):
repo, body = _parse_request(request)
if not repo:
return HttpResponseForbidden()
try:
body = json.loads(request.body)
if not body['ref'].startswith('refs/'): # Compatible with gogs
body['ref'] = 'refs/tags/' + body['ref']
try:
_, _kind, ref = body['ref'].split('/', 2)
if kind == 'branch' and _kind == 'heads':
commit_id = body['after']
if commit_id != '0000000000000000000000000000000000000000' and ref == request.GET.get('name'):
Thread(target=_dispatch, args=(deploy_id, ref, commit_id)).start()
message = _parse_message(body, repo)[:20]
Thread(target=_dispatch, args=(deploy_id, ref, commit_id, message)).start()
return HttpResponse(status=202)
elif kind == 'tag' and _kind == 'tags':
Thread(target=_dispatch, args=(deploy_id, ref)).start()
@ -34,70 +33,83 @@ def auto_deploy(request, deploy_id, kind):
return HttpResponseBadRequest(e)
def _is_valid_token(request):
def _parse_request(request):
api_key = AppSetting.get_default('api_key')
# Gitlab Gitee Aliyun(Codeup)
token, repo, body = None, None, None
token = request.headers.get('X-Gitlab-Token')
token = token or request.headers.get('X-Gitee-Token')
token = token or request.headers.get('X-Codeup-Token')
# Compatible the old version of gitlab
token = token or request.GET.get('token')
if token:
return token == api_key
if 'X-Gitlab-Token' in request.headers:
token = request.headers['X-Gitlab-Token']
repo = 'Gitlab'
elif 'X-Gitee-Token' in request.headers:
token = request.headers['X-Gitee-Token']
repo = 'Gitee'
elif 'X-Codeup-Token' in request.headers:
token = request.headers['X-Codeup-Token']
repo = 'Codeup'
elif 'X-Gogs-Signature' in request.headers:
token = request.headers['X-Gogs-Signature']
repo = 'Gogs'
elif 'X-Hub-Signature-256' in request.headers:
token = request.headers['X-Hub-Signature-256'].replace('sha256=', '')
repo = 'Github'
elif 'token' in request.GET: # Compatible the old version of gitlab
token = request.GET.get('token')
repo = 'Gitlab'
# Gogs Github
token = request.headers.get('X-Gogs-Signature')
token = token or request.headers.get('X-Hub-Signature-256', '').replace('sha256=', '')
if token:
return token == hmac.new(api_key.encode(), request.body, hashlib.sha256).hexdigest()
return False
if repo in ['Gitlab', 'Gitee', 'Codeup']:
if token != api_key:
return None, None
elif repo in ['Github', 'Gogs']:
en_api_key = hmac.new(api_key.encode(), request.body, hashlib.sha256).hexdigest()
if token != en_api_key:
return None, None
else:
return None, None
body = json.loads(request.body)
if repo == 'Gogs' and not body['ref'].startswith('refs/'):
body['ref'] = 'refs/tags/' + body['ref']
return repo, body
def _dispatch(deploy_id, ref, commit_id=None):
def _parse_message(body, repo):
message = None
if repo in ['Gitee', 'Github']:
message = body.get('head_commit', {}).get('message', '')
elif repo in ['Gitlab', 'Codeup', 'Gogs']:
if body.get('commits'):
message = body['commits'][0].get('message', '')
else:
raise ValueError(f'repo {repo} is not supported')
return message
def _dispatch(deploy_id, ref, commit_id=None, message=None):
deploy = Deploy.objects.filter(pk=deploy_id).first()
if not deploy:
raise Exception(f'no such deploy id for {deploy_id}')
if deploy.extend == '1':
_deploy_extend_1(deploy, ref, commit_id)
else:
_deploy_extend_2(deploy, ref, commit_id)
def _deploy_extend_1(deploy, ref, commit_id=None):
if commit_id:
extra = ['branch', ref, commit_id]
version = f'{ref}#{commit_id[:6]}'
else:
extra = ['tag', ref, None]
version = ref
req = DeployRequest.objects.create(
req = DeployRequest(
type='3',
status='0' if deploy.is_audit else '2',
deploy=deploy,
name=version,
extra=json.dumps(extra),
version=version,
spug_version=Repository.make_spug_version(deploy.id),
host_ids=deploy.host_ids,
created_by=deploy.created_by
)
if req.status == '2':
deploy_dispatch(req)
def _deploy_extend_2(deploy, ref, commit_id=None):
version = f'{ref}#{commit_id[:6]}' if commit_id else ref
req = DeployRequest.objects.create(
type='3',
status='0' if deploy.is_audit else '2',
deploy=deploy,
name=version,
version=version,
spug_version=Repository.make_spug_version(deploy.id),
host_ids=deploy.host_ids,
created_by=deploy.created_by
)
if commit_id: # branch
req.version = f'{ref}#{commit_id[:6]}'
req.name = message or req.version
if deploy.extend == '1':
req.extra = json.dumps(['branch', ref, commit_id])
else: # tag
req.version = ref
req.name = ref
if deploy.extend == '1':
req.extra = json.dumps(['tag', ref, None])
req.save()
if req.status == '2':
deploy_dispatch(req)