ghlabel 0.3.0 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 7d313e14a40128d38ca3c0f76cc0f118976726df
-  data.tar.gz: 401e4ec7d2a7e0e98677b16337786f0c05be1ee7
+  metadata.gz: a3dc39e3389a901e0a780af5e9e6d207b8602c02
+  data.tar.gz: 6f5e627c35607570b7fe6f6a294394ee09822e37
 SHA512:
-  metadata.gz: d58f4dc0abf2c4a77876c9049eeddb5d03736c06f5b5a9de389a983b74c1ece92891615f4baf8cf8ceb861a3384128f047904fa6f4b76b3cff26264bc4426b09
-  data.tar.gz: 058dbab02658d6a86dcdf337a62d1c337d8d1e8ac6bece44333606fbcf7b0e2db6ffe2eb796a5878354a8d97d932fa7536edca581f5c5b9315783120ec9e949f
+  metadata.gz: 6109bbd510aa68ecb5e74df688a1678dca1fb9a36b38a2277747cfd9f2f5077e5ea8663e7bb3f25d575b263eea44f9311e4eb7a1d8b573e54574b19f53f7a79c
+  data.tar.gz: d5db8a5c8d31fc7e249f4dbeaf3551e10195bfe1ddf6a78bd2cf8ab8f94c0d5103dd0bed9e7a1f508a0a196ad5b38e6a3c691bd845873395ace5e8a3a03b0575

data/exe/ghlabel

@@ -18,13 +18,14 @@ opts = Trollop::options do
   opt :organization, "Your organization name", default: nil, type: :string
   opt :add, "--add label1,label2,label3", type: :string, default: nil
   opt :remove, "--remove label1,label2,label3", type: :string, default: nil
+  opt :pr_number, "specify pr number, don't autodetect pr", type: :string, default: nil
   opt :with_references, "Also apply labels to title referenced #issues_numbers", default: true
 end
 
 abort "You need at least --add or --remove" if opts[:add] && opts[:remove] or !opts[:add] && !opts[:remove]
 
 gh = Ghlabel::Ghlabel.new(token: File.read(opts[:token_file]), repo: opts[:repository],
-                          organization: opts[:organization], with_references: opts[:with_references])
+                          organization: opts[:organization], with_references: opts[:with_references], pr_number: opts[:pr_number])
 
 if opts[:remove]
   labels = opts[:remove].split(',')

data/lib/ghlabel/version.rb

@@ -1,3 +1,3 @@
 module Ghlabel
-  VERSION = "0.3.0"
+  VERSION = "0.4.0"
 end

data/lib/ghlabel.rb

@@ -7,18 +7,15 @@ require 'awesome_print'
 
 module Ghlabel
   class Ghlabel
-    def initialize(token:, repo: nil, organization: nil, with_references: true)
+    def initialize(token:, repo: nil, organization: nil, with_references: true, pr_number: nil)
       @token = token
       @user = github.users.get.login
 
-      if organization.nil? && organizations.count > 1
-        warn "You have more than one organization (#{organizations}), please choose one as parameter or we'll use your personal one"
-      end
-      @organization = organization || @user 
-      @repo = repo || current_repo
+      @organization = organization || current_repo_info[:organization] 
+      @repo = repo || current_repo_info[:repo]
       @with_references = with_references
 
-      @pr = pr_from_ref(File.read('.git/HEAD').gsub('ref: ', '').strip)
+      @pr = pr_number ? pr_from_num(pr_number) : pr_from_ref(File.read('.git/HEAD').gsub('ref: ', '').strip)
     end
 
     def remove_labels(labels)
@@ -40,8 +37,15 @@ module Ghlabel
     end
 
     private
+
+    def pr_from_num(pr_number)
+      pr = github.pull_requests.get(user: @organization, repo: @repo, number: pr_number)
+      {id: pr.id, ref: pr.head.ref, title: pr.title, created_at: pr.created_at, merged_at: pr.merged_at,
+       href: pr['_links']['self']['href'], issue_number: /\/(\d+)$/.match(pr['_links']['issue']['href']).captures.first}
+    end
+
     def pr_from_ref(ref)
-      prs = github.pull_requests.list(user: @organization, repo: @repo, state: 'open', auto_pagination: true)
+      prs = github.pull_requests.list(user: @organization, repo: @repo, state: 'all', auto_pagination: true)
         .map{|x| {id: x.id, ref: x.head.ref, title: x.title, created_at: x.created_at, merged_at: x.merged_at,
                   href: x['_links']['self']['href'], issue_number: /\/(\d+)$/.match(x['_links']['issue']['href']).captures.first }}
       prs.find{|x| x[:ref] == ref.gsub('refs/heads/', '')}
@@ -52,8 +56,9 @@ module Ghlabel
       pr_issue.labels.map(&:name)
     end
 
-    def current_repo
-      @_repo_info ||= (/url = git@github.com:.*\/(.*).git/.match(File.read('.git/config')).captures.try(:first))
+    def current_repo_info
+      organization, repo = (/url = git@github.com:(.*)\/(.*).git/.match(File.read('.git/config')).captures)
+      {organization: organization, repo: repo}
     end
 
     def organizations

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ghlabel
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors:
 - pcboy
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2017-04-13 00:00:00.000000000 Z
+date: 2017-04-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler