Add Bitbucket server pull requests formatter

lib/pronto.rb

@@ -14,6 +14,7 @@
 require 'pronto/config'
 
 require 'pronto/clients/bitbucket_client'
+require 'pronto/clients/bitbucket_server_client'
 
 require 'pronto/git/repository'
 require 'pronto/git/patches'
@@ -30,6 +31,7 @@
 require 'pronto/github'
 require 'pronto/gitlab'
 require 'pronto/bitbucket'
+require 'pronto/bitbucket_server'
 
 require 'pronto/formatter/colorizable'
 require 'pronto/formatter/base'
@@ -44,6 +46,7 @@
 require 'pronto/formatter/gitlab_formatter'
 require 'pronto/formatter/bitbucket_formatter'
 require 'pronto/formatter/bitbucket_pull_request_formatter'
+require 'pronto/formatter/bitbucket_server_pull_request_formatter'
 require 'pronto/formatter/checkstyle_formatter'
 require 'pronto/formatter/null_formatter'
 require 'pronto/formatter/formatter'

lib/pronto/bitbucket_server.rb

@@ -0,0 +1,34 @@
+module Pronto
+  class BitbucketServer < Bitbucket
+    def pull_comments(sha)
+      @comment_cache["#{pull_id}/#{sha}"] ||= begin
+        client.pull_comments(slug, pull_id).map do |comment|
+          if comment['commentAnchor']
+            Comment.new(sha, comment['comment']['text'], comment['commentAnchor']['path'], comment['commentAnchor']['line'])
+          else
+            Comment.new(sha, comment['comment']['text'])
+          end
+        end
+      end
+    end
+
+    private
+
+    def client
+      @client ||= BitbucketServerClient.new(@config.bitbucket_username,
+                                            @config.bitbucket_password,
+                                            @config.bitbucket_api_endpoint)
+    end
+
+    def pull
+      @pull ||=
+        if env_pull_id
+          pull_requests.find { |pr| pr.id.to_i == env_pull_id.to_i }
+        elsif @repo.branch
+          pull_requests.find do |pr|
+            pr['fromRef']['displayId'] == @repo.branch
+          end
+        end
+    end
+  end
+end

lib/pronto/clients/bitbucket_server_client.rb

@@ -0,0 +1,64 @@
+class BitbucketServerClient
+  include HTTParty
+
+  def initialize(username, password, endpoint)
+    self.class.base_uri(endpoint)
+    self.class.basic_auth(username, password)
+    @headers = { 'Content-Type' => 'application/json' }
+  end
+
+  def pull_comments(slug, pr_id)
+    project_key, repository_key = slug.split('/')
+    url = "/projects/#{project_key}/repos/#{repository_key}/pull-requests/#{pr_id}/activities"
+    response = paged_request(url)
+    response.select { |activity| activity.action == 'COMMENTED' }
+  end
+
+  def pull_requests(slug)
+    project_key, repository_key = slug.split('/')
+    url = "/projects/#{project_key}/repos/#{repository_key}/pull-requests"
+    paged_request(url, state: 'OPEN')
+  end
+
+  def create_pull_comment(slug, pull_id, body, path, position)
+    project_key, repository_key = slug.split('/')
+    url = "/projects/#{project_key}/repos/#{repository_key}/pull-requests/#{pull_id}/comments"
+    post(url, body, path, position)
+  end
+
+  private
+
+  def openstruct(response)
+    response.map { |r| OpenStruct.new(r) }
+  end
+
+  def post(url, body, path, position)
+    body = {
+      text: body,
+      anchor: {
+        line: position,
+        lineType: 'ADDED',
+        path: path,
+        srcPath: path
+      }
+    }
+    self.class.post(url, body: body.to_json, headers: @headers)
+  end
+
+  def paged_request(url, query = {})
+    Enumerator.new do |y|
+      next_page_start = 0
+      loop do
+        response = self.class.get(url, query.merge(start: next_page_start)).parsed_response
+        break if response['values'].nil?
+
+        response['values'].each do |item|
+          y << OpenStruct.new(item)
+        end
+
+        next_page_start = response['nextPageStart']
+        break unless next_page_start
+      end
+    end
+  end
+end

lib/pronto/config_file.rb

@@ -22,6 +22,7 @@ class ConfigFile
         'slug' => nil,
         'username' => nil,
         'password' => nil,
+        'api_endpoint' => nil,
         'web_endpoint' => 'https://bitbucket.org/'
       },
       'text' => {

lib/pronto/formatter/bitbucket_server_pull_request_formatter.rb

@@ -0,0 +1,17 @@
+module Pronto
+  module Formatter
+    class BitbucketServerPullRequestFormatter < PullRequestFormatter
+      def client_module
+        BitbucketServer
+      end
+
+      def pretty_name
+        'BitBucket Server'
+      end
+
+      def line_number(message, _)
+        message.line.line.new_lineno if message.line
+      end
+    end
+  end
+end

lib/pronto/formatter/formatter.rb

@@ -17,6 +17,7 @@ def self.names
       'gitlab' => GitlabFormatter,
       'bitbucket' => BitbucketFormatter,
       'bitbucket_pr' => BitbucketPullRequestFormatter,
+      'bitbucket_server_pr' => BitbucketServerPullRequestFormatter,
       'json' => JsonFormatter,
       'checkstyle' => CheckstyleFormatter,
       'text' => TextFormatter,