initial commit

Author: tobiwild

.env

@@ -0,0 +1,7 @@
+REVIEWBOARD_URI=http://localhost:8000
+REVIEWBOARD_USER=admin
+REVIEWBOARD_PASSWORD=admin
+GITLAB_ENDPOINT=http://localhost:10080/api/v3
+GITLAB_PRIVATE_TOKEN=tdUyLoh4vMhDpnMGusx1
+GITLAB_TRIGGER_LABEL=reviewboard
+GIT_REPO_PATH=repositories

.gitignore

@@ -0,0 +1 @@
+/repositories/

Gemfile

@@ -0,0 +1,15 @@
+source 'https://rubygems.org'
+gem 'sinatra'
+gem 'thin'
+gem 'httmultiparty'
+gem 'gitlab'
+gem 'dotenv'
+gem 'pry'
+gem 'awesome_print'
+group :development do
+  gem 'rubocop'
+  gem 'minitest'
+  gem 'minitest-reporters'
+  gem 'minitest-filesystem'
+  gem 'rake'
+end

Gemfile.lock

@@ -0,0 +1,82 @@
+GEM
+  remote: https://rubygems.org/
+  specs:
+    ansi (1.5.0)
+    ast (2.0.0)
+    astrolabe (1.3.0)
+      parser (>= 2.2.0.pre.3, < 3.0)
+    awesome_print (1.6.1)
+    builder (3.2.2)
+    coderay (1.1.0)
+    daemons (1.2.2)
+    dotenv (2.0.2)
+    eventmachine (1.0.7)
+    gitlab (3.4.0)
+      httparty
+      terminal-table
+    httmultiparty (0.3.16)
+      httparty (>= 0.7.3)
+      mimemagic
+      multipart-post
+    httparty (0.13.5)
+      json (~> 1.8)
+      multi_xml (>= 0.5.2)
+    json (1.8.3)
+    method_source (0.8.2)
+    mimemagic (0.3.0)
+    minitest (5.4.3)
+    minitest-filesystem (1.2.0)
+    minitest-reporters (1.0.17)
+      ansi
+      builder
+      minitest (>= 5.0)
+      ruby-progressbar
+    multi_xml (0.5.5)
+    multipart-post (2.0.0)
+    parser (2.3.0.pre.2)
+      ast (>= 1.1, < 3.0)
+    powerpack (0.1.1)
+    pry (0.10.1)
+      coderay (~> 1.1.0)
+      method_source (~> 0.8.1)
+      slop (~> 3.4)
+    rack (1.6.1)
+    rack-protection (1.5.3)
+      rack
+    rainbow (2.0.0)
+    rake (10.4.2)
+    rubocop (0.32.1)
+      astrolabe (~> 1.3)
+      parser (>= 2.2.2.5, < 3.0)
+      powerpack (~> 0.1)
+      rainbow (>= 1.99.1, < 3.0)
+      ruby-progressbar (~> 1.4)
+    ruby-progressbar (1.7.5)
+    sinatra (1.4.6)
+      rack (~> 1.4)
+      rack-protection (~> 1.4)
+      tilt (>= 1.3, < 3)
+    slop (3.6.0)
+    terminal-table (1.4.5)
+    thin (1.6.3)
+      daemons (~> 1.0, >= 1.0.9)
+      eventmachine (~> 1.0)
+      rack (~> 1.0)
+    tilt (2.0.1)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  awesome_print
+  dotenv
+  gitlab
+  httmultiparty
+  minitest
+  minitest-filesystem
+  minitest-reporters
+  pry
+  rake
+  rubocop
+  sinatra
+  thin

Rakefile

@@ -0,0 +1,11 @@
+require 'bundler'
+require 'rake/testtask'
+
+desc 'Default Task'
+task default: [:test]
+
+Rake::TestTask.new(:test) do |t|
+  t.libs << 'lib'
+  t.libs << 'test'
+  t.pattern = 'test/**/test_*.rb'
+end

app.rb

@@ -0,0 +1,33 @@
+$LOAD_PATH << File.expand_path(File.dirname(__FILE__)) + '/lib'
+
+require 'hooks/reviewboard'
+require 'git_service'
+require 'reviewboard/client'
+
+Dotenv.load
+
+post '/' do
+  request.body.rewind
+  @request_payload = JSON.parse request.body.read
+  # ap @request_payload
+
+  reviewboard_hook.process(@request_payload)
+end
+
+def reviewboard_hook
+  @reviewboard_hook ||= Hooks::Reviewboard.new(
+    Reviewboard::Client.new(
+      base_uri: ENV['REVIEWBOARD_URI'],
+      basic_auth: {
+        username: ENV['REVIEWBOARD_USER'],
+        password: ENV['REVIEWBOARD_PASSWORD']
+      }
+    ),
+    Gitlab.client(
+      endpoint: ENV['GITLAB_ENDPOINT'],
+      private_token: ENV['GITLAB_PRIVATE_TOKEN']
+    ),
+    GitService.new(ENV['GIT_REPO_PATH']),
+    label: ENV['GITLAB_TRIGGER_LABEL']
+  )
+end

config.ru

@@ -0,0 +1,7 @@
+require 'rubygems'
+require 'bundler'
+
+Bundler.require
+
+require './app'
+run Sinatra::Application

lib/git_service.rb

@@ -0,0 +1,44 @@
+require 'digest/sha1'
+require 'open3'
+require 'tempfile'
+
+# class
+class GitService
+  class Exception < StandardError
+  end
+
+  def initialize(repo_path)
+    @repo_path = repo_path
+  end
+
+  def create_diff(repo, from, to)
+    Dir.mkdir(@repo_path) unless Dir.exist?(@repo_path)
+    hash = Digest::SHA1.hexdigest(repo)
+    repo_path = File.join(@repo_path, hash)
+    if Dir.exist?(repo_path)
+      command 'git remote update --prune', chdir: repo_path
+    else
+      command 'git', 'clone', '--mirror', repo, hash, chdir: @repo_path
+    end
+    file = Tempfile.new('git_diff')
+
+    begin
+      command "git diff --full-index #{from}...#{to} > #{file.path}", chdir: repo_path
+
+      yield file.path
+    ensure
+      file.unlink
+    end
+  end
+
+  private
+
+  def command(*command)
+    ret = ''
+    Open3.popen3(*command) do |_stdin, stdout, stderr, wait_thr|
+      ret = stdout.gets
+      raise Exception.new(stderr.gets.chomp) unless wait_thr.value == 0
+    end
+    ret
+  end
+end

lib/hooks/reviewboard.rb

@@ -0,0 +1,102 @@
+module Hooks
+  # Transfers merge request changes to reviewboard
+  class Reviewboard
+    def initialize(reviewboard, gitlab, git, options = {})
+      @reviewboard = reviewboard
+      @gitlab = gitlab
+      @git = git
+      @options = options
+    end
+
+    def process(merge_request)
+      attr = merge_request.fetch('object_attributes')
+
+      return unless label_matches?(attr)
+
+      review_id = review_id_from_comments(
+        attr.fetch('source_project_id'),
+        attr.fetch('id')
+      )
+
+      unless review_id.nil?
+        review = @reviewboard.review_request(review_id)
+        return if review.fetch('commit_id') == attr.fetch('last_commit').fetch('id')
+      end
+
+      @git.create_diff(
+        attr.fetch('source').fetch('ssh_url'),
+        attr.fetch('target_branch'),
+        attr.fetch('source_branch')
+      ) do |file|
+        if review_id.nil?
+          review_id = create_review_request(merge_request, file)
+          @gitlab.create_merge_request_comment(
+            attr.fetch('source_project_id'),
+            attr.fetch('id'),
+            "REVIEW_ID: #{review_id}"
+          )
+        else
+          update_review_request(review_id, merge_request, file)
+        end
+      end
+
+      @gitlab.create_merge_request_comment(
+        attr.fetch('source_project_id'),
+        attr.fetch('id'),
+        @reviewboard.last_diff_uri(review_id)
+      )
+    end
+
+    def label_matches?(attr)
+      return true unless @options.key?(:label)
+
+      @gitlab.merge_request(
+        attr.fetch('source_project_id'),
+        attr.fetch('id')
+      ).labels.include?(@options[:label])
+    end
+
+    def update_review_request(review_id, merge_request, file)
+      attr = merge_request.fetch('object_attributes')
+
+      @reviewboard.sync_review_request(
+        review_id: review_id,
+        draft_params:
+        {
+          public: true,
+          commit_id: attr.fetch('last_commit').fetch('id')
+        },
+        diff: file
+      )
+    end
+
+    def create_review_request(merge_request, file)
+      attr = merge_request.fetch('object_attributes')
+      repo = attr.fetch('source')
+
+      @reviewboard.sync_review_request(
+        create_params:
+        {
+          repository: repo.fetch('name'),
+          submit_as: merge_request.fetch('user').fetch('username')
+        },
+        draft_params:
+        {
+          branch: attr.fetch('source_branch'),
+          summary: attr.fetch('title'),
+          description: attr.fetch('url'),
+          public: true,
+          commit_id: attr.fetch('last_commit').fetch('id')
+        },
+        diff: file
+      )
+    end
+
+    def review_id_from_comments(pid, mid)
+      @gitlab.merge_request_comments(pid, mid).each do |comment|
+        return $1.to_i if /REVIEW_ID: (\d+)/.match(comment.note)
+      end
+      nil
+    end
+  end
+end