80 lines
No EOL
2.8 KiB
Python
80 lines
No EOL
2.8 KiB
Python
import logging
|
|
import stripe
|
|
import urlparse
|
|
|
|
from flask import request, make_response, Blueprint
|
|
|
|
from data import model
|
|
from auth.auth import process_auth
|
|
from auth.permissions import ModifyRepositoryPermission
|
|
from util.invoice import renderInvoiceToHtml
|
|
from util.email import send_invoice_email
|
|
from util.names import parse_repository_name
|
|
from util.http import abort
|
|
from endpoints.trigger import BuildTrigger
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
webhooks = Blueprint('webhooks', __name__)
|
|
|
|
|
|
@webhooks.route('/stripe', methods=['POST'])
|
|
def stripe_webhook():
|
|
request_data = request.get_json()
|
|
logger.debug('Stripe webhook call: %s' % request_data)
|
|
|
|
event_type = request_data['type'] if 'type' in request_data else None
|
|
if event_type == 'charge.succeeded':
|
|
data = request_data['data'] if 'data' in request_data else {}
|
|
obj = data['object'] if 'object' in data else {}
|
|
invoice_id = obj['invoice'] if 'invoice' in obj else None
|
|
customer_id = obj['customer'] if 'customer' in obj else None
|
|
|
|
if invoice_id and customer_id:
|
|
# Find the user associated with the customer ID.
|
|
user = model.get_user_or_org_by_customer_id(customer_id)
|
|
if user and user.invoice_email:
|
|
# Lookup the invoice.
|
|
invoice = stripe.Invoice.retrieve(invoice_id)
|
|
if invoice:
|
|
invoice_html = renderInvoiceToHtml(invoice, user)
|
|
send_invoice_email(user.email, invoice_html)
|
|
|
|
return make_response('Okay')
|
|
|
|
|
|
@webhooks.route('/push/<path:repository>/trigger/<trigger_uuid>',
|
|
methods=['POST'])
|
|
@process_auth
|
|
@parse_repository_name
|
|
def github_push_webhook(namespace, repository, trigger_uuid):
|
|
logger.debug('Webhook received for %s/%s with uuid %s', namespace,
|
|
repository, trigger_uuid)
|
|
permission = ModifyRepositoryPermission(namespace, repository)
|
|
if permission.can():
|
|
try:
|
|
trigger = model.get_build_trigger(namespace, repository, trigger_uuid)
|
|
except model.InvalidBuildTriggerException:
|
|
abort(404)
|
|
|
|
handler = BuildTrigger.get_trigger_for_service(trigger.service.name)
|
|
|
|
logger.debug('Passing webhook request to handler %s', handler)
|
|
df_id, tag, name = handler.handle_trigger_request(request,
|
|
trigger.auth_token,
|
|
trigger.config)
|
|
|
|
host = urlparse.urlparse(request.url).netloc
|
|
full_tag = '%s/%s/%s:%s' % (host, trigger.repository.namespace,
|
|
trigger.repository.name, tag)
|
|
|
|
token = model.create_access_token(trigger.repository, 'write')
|
|
logger.debug('Creating build %s with full_tag %s and dockerfile_id %s',
|
|
name, full_tag, df_id)
|
|
model.create_repository_build(trigger.repository, token, df_id, full_tag,
|
|
name)
|
|
|
|
return make_response('Okay')
|
|
|
|
abort(403) |