dotfiles/tmux.script.gitlab.rb at master · jorgeborges/dotfiles · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
#!/usr/bin/env ruby

# Gets merge requests assignned to me and pending reviews
# Outputs icons with total counts

require 'rest-client' # has to be gem installed
require 'json'
require 'date'
require 'yaml'

config = YAML.load_file(__dir__ + '/config/gitlab.yml')

merge_request_type = ARGV[0]

gitlab_api_base_url = 'https://gitlab.com/api/v4/merge_requests'
headers = {:Authorization => 'Bearer ' + config['personal_access_token']}
username = config['my_username']
ignore_review_for_merge_requests_iids = config['ignore_review_for_merge_requests_iids']

begin
  case merge_request_type
  when 'assigned'
    response = RestClient.get gitlab_api_base_url + '?scope=assigned_to_me&state=opened', headers
  when 'review'
    response = RestClient.get gitlab_api_base_url + '?scope=all&state=opened&reviewer_username=' + username, headers
  else
    raise 'Invalid argument'
  end
  rescue
    puts 'gitlab_display()'
    Kernel.abort
end

merge_requests = JSON.parse response
merge_requests = merge_requests.select { |merge_request| !ignore_review_for_merge_requests_iids.include? merge_request['iid'] }
merge_request_count = merge_requests.length()

merge_request_icons = {
  assigned: '華',
  review: ' ',
}

puts merge_request_icons[merge_request_type.to_sym] + merge_request_count.to_s