Work in progress: bitbucket support

endpoints/trigger.py

@@ -7,10 +7,11 @@ import re
 import json
 
 from github import Github, UnknownObjectException, GithubException
+from bitbucket.bitbucket import Bitbucket
 from tempfile import SpooledTemporaryFile
 from jsonschema import validate
 
-from app import app, userfiles as user_files, github_trigger
+from app import app, userfiles as user_files, github_trigger, get_app_url
 from util.tarfileappender import TarfileAppender
 from util.ssh import generate_ssh_keypair
 
@@ -58,6 +59,9 @@ class EmptyRepositoryException(Exception):
 class RepositoryReadException(Exception):
   pass
 
+class TriggerProviderException(Exception):
+  pass
+
 
 class BuildTrigger(object):
   def __init__(self):
@@ -158,6 +162,87 @@ def get_trigger_config(trigger):
     return {}
 
 
+class BitbucketBuildTrigger(BuildTrigger):
+  """
+  BuildTrigger for Bitbucket.
+  """
+  @classmethod
+  def service_name(cls):
+    return 'bitbucket'
+
+  @staticmethod
+  def _get_authorized_client(trigger_uuid):
+    key = app.config.get('BITBUCKET_TRIGGER_CONFIG', {}).get('CONSUMER_KEY', '')
+    secret = app.config.get('BITBUCKET_TRIGGER_CONFIG', {}).get('CONSUMER_SECRET', '')
+
+    callback_url = '%s/oauth1/bitbucket/callback/trigger/%s' % (get_app_url(), trigger_uuid)
+    bitbucket_client = Bitbucket()
+    (result, err_message) = bitbucket_client.authorize(key, secret, callback_url)
+    if not result:
+      raise TriggerProviderException(err_message)
+
+    return bitbucket_client
+
+
+  @staticmethod
+  def get_oauth_url(trigger_uuid):
+    bitbucket_client = BitbucketBuildTrigger._get_authorized_client(trigger_uuid)
+    url = bitbucket_client.url('AUTHENTICATE', token=bitbucket_client.access_token)
+    return {
+      'access_token': bitbucket_client.access_token,
+      'access_token_secret': bitbucket_client.access_token_secret,
+      'url': url
+    }
+
+  @staticmethod
+  def exchange_verifier(trigger, verifier):
+    trigger_config = get_trigger_config(trigger.config)
+    bitbucket_client = BitbucketBuildTrigger._get_authorized_client(trigger.uuid)
+    print trigger.config
+    print trigger.auth_token
+    print bitbucket_client.verify(verifier, access_token=trigger_config.get('access_token', ''),
+                                  access_token_secret=trigger.auth_token)
+    return None
+    #(result, _) = bitbucket_client.verify(verifier)
+
+    #if not result:
+    #  return None
+
+    #return (bitbucket_client.access_token, bitbucket_client.access_token_secret)
+
+  def is_active(self, config):
+    return False
+
+  def activate(self, trigger_uuid, standard_webhook_url, auth_token, config):
+    return {}
+
+  def deactivate(self, auth_token, config):
+    return config
+
+  def list_build_sources(self, auth_token):
+    return []
+
+
+  def list_build_subdirs(self, auth_token, config):
+    raise RepositoryReadException('Not supported')
+
+  def dockerfile_url(self, auth_token, config):
+    return None
+
+  def load_dockerfile_contents(self, auth_token, config):
+    raise RepositoryReadException('Not supported')
+
+  @staticmethod
+  def _build_commit_info(repo, commit_sha):
+    return {}
+
+  def handle_trigger_request(self, request, trigger):
+    return
+
+  def manual_start(self, trigger, run_parameters=None):
+    return None
+
+
 class GithubBuildTrigger(BuildTrigger):
   """
   BuildTrigger for GitHub that uses the archive API and buildpacks.