vx-service_connector 0.2.12 → 0.2.13

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: a5b071bc9825ff7e1abdfc7a258f11cf19234bde
-  data.tar.gz: ad9c9f8ac7c25bfda55e2f691d6bc6f89d3ce2fc
+  metadata.gz: 54d71cae077981d1705c726931668bf9b2340cb9
+  data.tar.gz: c0dea00e6d39a55757f96b6bd6bc6cb9e4428182
 SHA512:
-  metadata.gz: 8da6953d253b5584a2533ca4f9c1a23b3400d77bb08c56ce0f1e239f44720198d0da9168a61130fb5420ec2e30bcae76cb7f2b98d19272d1145b47d06de1c3ce
-  data.tar.gz: 42bfeb21d21e951762be1b1e24febbe4670b690a58390f57680684f3e65736727c538727cf0486d18dff56257a83f5295e72b2d21f4354a26e06c503786af796
+  metadata.gz: bbb12a18dd04bb71320c037805d831f1e809fbe32b0d91055dfdcbd4287b6fd395dacb32fc67df1509f4654c0b19af8910f02b3bfc323f79910d4affc54d82da
+  data.tar.gz: 07315c7f4b03f472551857bbda073f7a8c934dce4fe324b15730194243fc5dd4d7de0fcf72089c08bf3ba34dca0e97d6ee9d7f9f2a33f2f8ee5339bb293a5fce

data/Gemfile

@@ -2,3 +2,5 @@ source 'https://rubygems.org'
 
 # Specify your gem's dependencies in vx-service_connector.gemspec
 gemspec
+
+gem 'pry'

data/lib/vx/service_connector.rb

@@ -9,6 +9,7 @@ module Vx
     autoload :Github,     File.expand_path("../service_connector/github",     __FILE__)
     autoload :GitlabV5,   File.expand_path("../service_connector/gitlab_v5",  __FILE__)
     autoload :GitlabV6,   File.expand_path("../service_connector/gitlab_v6",  __FILE__)
+    autoload :Bitbucket,  File.expand_path("../service_connector/bitbucket",  __FILE__)
     autoload :Model,      File.expand_path("../service_connector/model",      __FILE__)
 
     extend self
@@ -19,6 +20,8 @@ module Vx
         Github
       when :gitlab_v6
         GitlabV6
+      when :bitbucket
+        Bitbucket
       else
         raise ArgumentError, "Serivice for #{name.inspect} is not defined"
       end

data/lib/vx/service_connector/bitbucket.rb

@@ -0,0 +1,51 @@
+module Vx
+  module ServiceConnector
+    Bitbucket = Struct.new(:login, :options) do
+
+      include ServiceConnector::Base
+
+      def repos
+        Bitbucket::Repos.new(session).to_a
+      end
+
+      def organizations
+        []
+      end
+
+      def hooks(repo)
+        Bitbucket::Hooks.new(session, repo)
+      end
+
+      def deploy_keys(repo)
+        Bitbucket::DeployKeys.new(session, repo)
+      end
+
+      def notices(repo)
+        Bitbucket::Notices.new(session, repo)
+      end
+
+      def files(repo)
+        Bitbucket::Files.new(session, repo)
+      end
+
+      def payload(repo, params)
+        Bitbucket::Payload.new(session, params).build
+      end
+
+      def commits(repo, options = {})
+        Bitbucket::Commits.new(session, repo)
+      end
+
+      private
+
+        def create_session
+          self.class::Session.new(login, options)
+        end
+
+    end
+  end
+end
+
+Dir[File.expand_path("../bitbucket/*.rb", __FILE__)].each do |f|
+  require f
+end

data/lib/vx/service_connector/bitbucket/commits.rb

@@ -0,0 +1,30 @@
+module Vx
+  module ServiceConnector
+    class Bitbucket
+      Commits = Struct.new(:session, :repo) do
+        def last(options = {})
+          begin
+            commits = session.get "api/1.0/repositories/#{repo.full_name}/changesets/?limit=1"
+            commit  = commits["changesets"].first
+            author, email = commit["raw_author"].split(/[<>]/)
+            sha = commit["raw_node"]
+            branch = commit["branch"] || 'master'
+            Model::Payload.from_hash(
+              skip:          false,
+              pull_request?: false,
+              branch:        branch,
+              branch_label:  branch,
+              sha:           sha,
+              message:       commit["message"],
+              author:        author.strip,
+              author_email:  email,
+              web_url:       session.endpoint.to_s + "/#{repo.full_name}/commits/#{sha}"
+            )
+          rescue RequestError
+            nil
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vx/service_connector/bitbucket/deploy_keys.rb

@@ -0,0 +1,29 @@
+module Vx
+  module ServiceConnector
+    class Bitbucket
+      DeployKeys = Struct.new(:session, :repo) do
+
+        def all
+          begin
+            session.get "api/1.0/repositories/#{repo.full_name}/deploy-keys?pagelen=100"
+          rescue RequestError
+            []
+          end
+        end
+
+        def create(key_name, public_key)
+          session.post "api/1.0/repositories/#{repo.full_name}/deploy-keys", label: key_name, key: public_key
+        end
+
+        def destroy(key_name)
+          all.select do |key|
+            key['label'] == key_name
+          end.map do |key|
+            session.delete "api/1.0/repositories/#{repo.full_name}/deploy-keys/#{key['pk']}"
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/vx/service_connector/bitbucket/files.rb

@@ -0,0 +1,21 @@
+require 'base64'
+
+module Vx
+  module ServiceConnector
+    class Bitbucket
+      Files = Struct.new(:session, :repo) do
+
+        def get(sha, path)
+          begin
+            re = session.get("api/1.0/repositories/#{repo.full_name}/src/#{sha}/#{path}")
+            re['source']
+          rescue RequestError => e
+            $stderr.puts "ERROR: #{e.inspect}"
+            nil
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/vx/service_connector/bitbucket/hooks.rb

@@ -0,0 +1,58 @@
+module Vx
+  module ServiceConnector
+    class Bitbucket
+      Hooks = Struct.new(:session, :repo) do
+
+        def all
+          begin
+            session.get hooks_url
+          rescue RequestError
+            []
+          end
+        end
+
+        def create(url, token)
+          session.post(
+            hooks_url,
+            'type' => 'POST',
+            'URL'  => url
+          )
+          session.post(
+            hooks_url,
+            'type' => 'Pull Request POST',
+            'URL'  => url,
+            'create/edit/merge/decline' => 'on',
+            'comments'                  => 'off',
+            'approve/unapprove'         => 'off'
+          )
+        end
+
+        def destroy(url_mask)
+          all.select do |hook|
+            url = extract_url(hook)
+            url && url =~ /#{Regexp.escape url_mask}/
+          end.map do |hook|
+            session.delete hook_url(hook['id'])
+          end
+        end
+
+        private
+
+          def extract_url(hook)
+            fields = hook['service']['fields']
+            url = fields.select{|f| f['name'] == 'URL' }.map{|f| f['value'] }.first
+            url
+          end
+
+          def hooks_url
+            "api/1.0/repositories/#{repo.full_name}/services"
+          end
+
+          def hook_url(id)
+            "#{hooks_url}/#{id}"
+          end
+
+      end
+    end
+  end
+end

data/lib/vx/service_connector/bitbucket/notices.rb

@@ -0,0 +1,11 @@
+module Vx
+  module ServiceConnector
+    class Bitbucket
+      Notices = Struct.new(:session, :repo) do
+        def create(build_sha, build_status, build_url, description)
+          :not_available
+        end
+      end
+    end
+  end
+end

data/lib/vx/service_connector/bitbucket/payload.rb

@@ -0,0 +1,136 @@
+module Vx
+  module ServiceConnector
+    class Bitbucket
+      Payload = Struct.new(:session, :params) do
+
+        def build
+          ServiceConnector::Model::Payload.new(
+            !!ignore?,
+            pull_request?,
+            pull_request_number,
+            branch,
+            branch_label,
+            sha,
+            message,
+            author,
+            author_email,
+            web_url
+          )
+        end
+
+        private
+
+        def pull_request?
+          !key? 'repository'
+        end
+
+        def pull_request_number
+          if pull_request? && pull_request.key?('id')
+            pull_request['id']
+          end
+        end
+
+        def sha
+          if pull_request?
+            pull_request['source']['commit']['hash']
+          else
+            head_commit['raw_node']
+          end
+        end
+
+        def branch
+          @branch ||= begin
+            if pull_request?
+              pull_request['source']['branch']['name']
+            else
+              head_commit['branch']
+            end
+          end
+        end
+
+        def branch_label
+          branch
+        end
+
+        def web_url
+          if pull_request?
+            pull_request['links'] ? pull_request['links']['html']['href'] : nil
+          else
+            "https://bitbucket.org#{params['repository']['absolute_url']}commits/#{head_commit['raw_node']}"
+          end
+        end
+
+        def message
+          if pull_request?
+            commit_for_pull_request["message"]
+          else
+            head_commit['message']
+          end
+        end
+
+        def author
+          if pull_request?
+            commit_for_pull_request["author"]["user"]["display_name"]
+          else
+            head_commit['author']
+          end
+        end
+
+        def author_email
+          if pull_request?
+            commit_for_pull_request["author"]["raw"][/.*<([^>]*)/,1]
+          else
+            head_commit['raw_author'][/.*<([^>]*)/,1]
+          end
+        end
+
+        def close_pull_request?
+          pull_request? && (pull_request['state'] && %w(DECLINED MERGED).include?(pull_request['state']))
+        end
+
+        def pull_request_head_repo_full_name
+          pull_request['source']['repository']['full_name']
+        end
+
+        def pull_request_base_repo_full_name
+          pull_request['destination']['repository']['full_name']
+        end
+
+        def foreign_pull_request?
+          pull_request_head_repo_full_name != pull_request_base_repo_full_name
+        end
+
+        def ignore?
+          if pull_request?
+            close_pull_request? || !foreign_pull_request?
+          else
+            sha == '0000000000000000000000000000000000000000'
+          end
+        end
+
+        def commit_for_pull_request
+          @commit_for_pull_request ||= begin
+            session.get pull_request['source']['commit']['links']['self']['href']
+          end
+        end
+
+        def head_commit
+          params['commits'].last || {}
+        end
+
+        def pull_request
+          params.first.last
+        end
+
+        def key?(name)
+          params.key? name
+        end
+
+        def [](val)
+          params[val]
+        end
+
+      end
+    end
+  end
+end

data/lib/vx/service_connector/bitbucket/repos.rb

@@ -0,0 +1,56 @@
+module Vx
+  module ServiceConnector
+    class Bitbucket
+      Repos = Struct.new(:session) do
+
+        def to_a
+          @repos ||= user_repositories
+        end
+
+        private
+
+          def user_repositories
+            res = session.get("api/1.0/user/repositories")
+            res.select do |repo|
+              git?(repo) && repo_access?(repo['owner'])
+            end.map do |repo|
+              repo_to_model repo
+            end
+          end
+
+          def repo_to_model(repo)
+            name = repo['owner'] + "/" + repo['slug']
+            Model::Repo.new(
+              name,
+              name,
+              repo['is_private'],
+              "git@#{session.endpoint.host}/#{name}.git",
+              "#{session.endpoint}/#{name}",
+              repo['description']
+            )
+          end
+
+          def team_admin
+            @team_admin ||= begin
+              values = session.get("api/1.0/user/privileges")['teams']
+              values.select{|k,v| v == 'admin' }.keys
+            end
+          end
+
+          def login
+            session.login
+          end
+
+          def git?(repo)
+            repo['scm'] == 'git'
+          end
+
+          def repo_access?(repo_owner)
+            repo_owner == login ||
+              team_admin.include?(repo_owner)
+          end
+
+      end
+    end
+  end
+end

data/lib/vx/service_connector/bitbucket/session.rb

@@ -0,0 +1,113 @@
+require 'oauth'
+require 'json'
+require 'ostruct'
+require 'uri'
+
+module Vx
+  module ServiceConnector
+    class Bitbucket
+      Session = Struct.new(:login, :options) do
+
+        def get(url)
+          wrap do
+            res = agent.get request_url(url)
+            response! res
+          end
+        end
+
+        def post(url, options = {})
+          wrap do
+            res = agent.post request_url(url), options
+            response! res
+          end
+        end
+
+        def delete(url)
+          wrap do
+            res = agent.delete request_url(url)
+            response! res
+          end
+        end
+
+        def endpoint
+          @endpoint ||= URI("https://bitbucket.org")
+        end
+
+        def self.test
+          {
+            consumer_key:    "key",
+            consumer_secret: "secret",
+            token:           "token",
+            token_secret:    "token secret"
+          }
+        end
+
+        private
+
+
+          def request_url(url)
+            if url.include?(endpoint.to_s)
+              url
+            else
+              "#{endpoint}/#{url}"
+            end
+          end
+
+          def response!(res)
+            if (200..204).include?(res.code.to_i)
+              if res.header['Content-Type'].include?("application/json")
+                ::JSON.parse(res.body)
+              else
+                res.body
+              end
+            else
+              raise RequestError, res.body
+            end
+          end
+
+          def wrap
+            begin
+              yield
+            rescue Errno::ETIMEDOUT => e
+              raise RequestError, e
+            end
+          end
+
+          def validate_options!
+            unless options.is_a?(Hash) && options.keys.sort == [:consumer_key, :consumer_secret, :token, :token_secret]
+              raise InvalidArguments, options
+            end
+          end
+
+          def agent
+            @agent ||= begin
+              validate_options!
+              consumer = OAuth::Consumer.new(
+                options[:consumer_key], options[:consumer_secret],
+                site: endpoint.to_s
+              )
+              token = OAuth::AccessToken.new consumer, options[:token], options[:token_secret]
+=begin
+              Sawyer::Agent.new(endpoint) do |http|
+                http.headers['content-type']  = 'application/json'
+                http.headers['accept']        = 'application/json'
+                http.ssl.verify               = false
+                http.options[:timeout]        = 5
+                http.options[:open_timeout]   = 5
+                http.request :oauth,
+                  consumer_key:    "7j4RYZEd8YTQdfTWkJ",
+                  consumer_secret: "Y8fSdaDK4GxKzvJn3tKRzyyYXctYSbUV",
+                  token:           "WhFvryHuZ82XLKW6aP",
+                  token_secret:    "zLSQe2DwhA4mfzwnPA53pKqLSyZgX5XH"
+                  ignore_extra_keys: true
+                http.response :logger
+              end
+=end
+              token
+            end
+          end
+
+      end
+    end
+  end
+end