hubstats 0.0.19 → 0.0.20

checksums.yaml

@@ -1,15 +1,15 @@
 ---
 !binary "U0hBMQ==":
   metadata.gz: !binary |-
-    MDIwZjQ4NTg1MjA4NzA1OTQ3ZTk4ZTVhYWUwNzFmMjE2ZWQ3YzMyYw==
+    NGU1YzA2ZmVmOTA5NTMzNzdhMTk3NDllZjczNzI2NTc4YjY3ZDZlOQ==
   data.tar.gz: !binary |-
-    Nzg3Y2I2Y2UxYWY5MDdkZjBlYzc3NmMzZGNiNTk4MjkzNjk2NzEyZA==
+    MWJlNDA1MDAxZWZmNDFjNzE5ZjI0OGUxZjQxYmZhYzRmYmU4YzJmOA==
 SHA512:
   metadata.gz: !binary |-
-    ZTVlZWY4ZWQ0YmJjZDY5ZWNjNTYwMTg4YjFkNTliNTBiNzgyYTAyZDZlNzY1
-    Yjc5OTdhMzc0Yjg4NzdhYTNjYjg0YjM4ZmI0MGQ5MDQ4NzFmMDFjNGY0YzRm
-    OTJlOTVjY2UwMWY3OTU1M2I3NjBhZTQwZDEwYWFiOGU4M2I2Zjc=
+    YTQ1YmJlYmE3ZjkxNTcxMzliZTRhMWMwZGRkOTAyMTc1MmZmM2IyZmViNmNi
+    ZTJlMmNjOGQ4N2RkZGExNTUyMTA4YmQzYWE1ZmIzZGIwYjE2N2ExOTMzYjZk
+    N2I2MjRjMWM0NjkwZmQyNzI5MjA4YmQ3Y2IzZTlkMGE5NWU5Yjk=
   data.tar.gz: !binary |-
-    Y2I0ZDMzZDg0Mjc3YjhiYzRiYjhmMTlhOTMxMjYxZTFlYmMwNjhhMzI4Nzli
-    ODA4NGQ2YmRjMGVkMjQ1ZTUyNjE3ZDVhMjFjZGE2ZmU1ODhkZTdkNjJhNWEw
-    NGYxN2NlODVkOGM2ZDM3NjZkMmFhMzM1OWJhNmE2NjRlMGU2NTU=
+    MjJmYjg3MTE5N2Y5NTM5ZDRlMDY5ODgyMmQ0NTEyZDFhZGExYjNmNTUzZmUx
+    Y2RiN2Y5MzNlOWU2Mzg1ZTZjZWI5MjdkNmUyNzVkZDk0MWMyMTBjZGI2ODcy
+    NjAxNjcwMmJkZTU1ZjM3ODZkODI4NmJmYmU1YjE5ZTg2NTA3ZDQ=

data/app/models/hubstats/label.rb

@@ -17,5 +17,13 @@ module Hubstats
 
     has_and_belongs_to_many :pull_requests, :join_table => 'hubstats_labels_pull_requests'
 
+    def self.first_or_create(label)
+      if exists = Hubstats::Label.where(name: label[:name]).first
+        return exists
+      else
+        Hubstats::Label.new(name: label[:name], url: label[:url], color: label[:color])
+      end
+    end
+
   end
 end

data/app/models/hubstats/pull_request.rb

@@ -40,7 +40,7 @@ module Hubstats
     end
 
     def add_labels(labels)
-      labels.map!{|v| Hubstats::Label.where(name: v.name).first}
+      labels.map!{ |label| Hubstats::Label.first_or_create(label) }
       self.labels = labels
     end
 

data/lib/hubstats/events_handler.rb

@@ -17,8 +17,10 @@ module Hubstats
     def pull_processor(payload)
       pull_request = payload[:pull_request]
       pull_request[:repository] = payload[:repository]
-      
-      Hubstats::PullRequest.create_or_update(pull_request.with_indifferent_access)
+      new_pull = Hubstats::PullRequest.create_or_update(pull_request.with_indifferent_access)
+      repo_name = Hubstats::Repo.where(id: new_pull.repo_id).first.full_name
+      labels = Hubstats::GithubAPI.get_labels(repo_name, new_pull.number )
+      new_pull.add_labels(labels)
     end
 
     def comment_processor(payload,kind)

data/lib/hubstats/github_api.rb

@@ -111,6 +111,12 @@ module Hubstats
       end
     end
 
+    def self.get_labels(repo_name, pull_request_number)
+      issue = Hubstats::GithubAPI.client.issue(repo_name, pull_request_number)
+      issue[:labels]
+    end
+
+
     def self.wait_limit(grab_size,rate_limit)
       if rate_limit.remaining < grab_size
         puts "Hit Github rate limit, waiting #{Time.at(rate_limit.resets_in).utc.strftime("%H:%M:%S")} to get more"

data/lib/hubstats/version.rb

@@ -1,3 +1,3 @@
 module Hubstats
-  VERSION = "0.0.19"
+  VERSION = "0.0.20"
 end