hubstats 0.0.8 → 0.0.9

data/db/migrate/20140703200752_create_hubstats_labels_pull_requests.rb

@@ -0,0 +1,8 @@
+class CreateHubstatsLabelsPullRequests < ActiveRecord::Migration
+  def change
+    create_table :hubstats_labels_pull_requests do |t|
+      t.belongs_to :pull_request
+      t.belongs_to :label
+    end
+  end
+end

data/lib/hubstats/events_handler.rb

@@ -2,15 +2,14 @@ module Hubstats
   class EventsHandler
 
     def route(payload, type) 
-      puts 
       case type
-      when "issue_comment" || "IssueCommentEvent"
+      when "issue_comment", "IssueCommentEvent"
         comment_processor(payload,"Issue")
-      when "commit_comment" || "CommitCommentEvent"
+      when "commit_comment", "CommitCommentEvent"
         comment_processor(payload,"Commit")
-      when "pull_request" || "PullRequestEvent"
+      when "pull_request", "PullRequestEvent"
         pull_processor(payload)
-      when "pull_request_review_comment" || "PullRequestReviewCommentEvent"
+      when "pull_request_review_comment", "PullRequestReviewCommentEvent"
         comment_processor(payload,"PullRequest")
       end
     end
@@ -31,7 +30,6 @@ module Hubstats
       Hubstats::Comment.create_or_update(comment)
     end
 
-
     #grabs the number off of anyone of the various places it can be
     def get_pull_number(payload)
       if payload[:pull_request]

data/lib/hubstats/github_api.rb

@@ -82,6 +82,21 @@ module Hubstats
       end
     end
 
+    def self.update_labels
+      puts "Adding labels to pull requests"
+      Hubstats::Repo.all.each do |repo|
+        Hubstats::GithubAPI.client({:auto_paginate => true}).labels(repo.full_name).each do |label|
+          label_hash = label.to_h if label.respond_to? :to_h
+          label_data = label_hash.slice(*Hubstats::Label.column_names.map(&:to_sym))
+          label = Hubstats::Label.where(:name => label_data[:name]).first_or_create(label_data)
+        end
+
+        Hubstats::Label.all.each do |label|
+          inline(repo.full_name,'issues', labels: label.name, state: 'closed')
+          inline(repo.full_name,'issues', labels: label.name, state: 'open')
+        end
+      end
+    end
 
     def self.wait_limit(grab_size,rate_limit)
       if rate_limit.remaining < grab_size
@@ -102,6 +117,12 @@ module Hubstats
         Hubstats::Comment.create_or_update(HubHelper.comment_setup(object,repo.id,"Commit"))
       elsif kind == "pulls"
         Hubstats::PullRequest.create_or_update(HubHelper.pull_setup(object))
+      elsif kind == "issues"
+        if object[:pull_request]
+          repo = Hubstats::Repo.where(full_name: repo_name).first
+          pull_request = Hubstats::PullRequest.where(repo_id: repo.id).where(number: object[:number]).first
+          pull_request.add_labels(object[:labels])
+        end
       end
     end
   end

data/lib/hubstats/version.rb

@@ -1,3 +1,3 @@
 module Hubstats
-  VERSION = "0.0.8"
+  VERSION = "0.0.9"
 end

data/lib/tasks/populate_task.rake

@@ -29,25 +29,44 @@ namespace :hubstats do
       pull_requests = Hubstats::GithubAPI.inline(repo.full_name,'pulls', :state => "all")
     end
 
+    desc "Pull labels from Github saves in database"
+    task :labels, [:repo] => :environment do |t, args|
+      repo = repo_checker(args[:repo])
+      puts "Adding labels for to " + repo.full_name
+
+      Hubstats::GithubAPI.client({:auto_paginate => true}).labels(repo.full_name).each do |label|
+
+        label_hash = label.to_h if label.respond_to? :to_h
+        label_data = label_hash.slice(*Hubstats::Label.column_names.map(&:to_sym))
+        label = Hubstats::Label.where(:name => label_data[:name]).first_or_create(label_data)
+      end
+    end
+
     desc "Pull repos from Github save to database"
     task :all => :environment do
       get_repos.each do |repo|
         repo = Hubstats::Repo.create_or_update(repo)
         Hubstats::GithubAPI.create_hook(repo)
-
+        
         Rake::Task["hubstats:populate:users"].execute({repo: repo})
         Rake::Task["hubstats:populate:pulls"].execute({repo: repo})
         Rake::Task["hubstats:populate:comments"].execute({repo: repo})
       end
       puts "Finished with initial population, grabing extra info for pull requests"
-      Rake::Task["hubstats:populate:update"].execute
+      Rake::Task["hubstats:populate:update_labels"].execute
+      Rake::Task["hubstats:populate:update_pulls"].execute
     end
 
     desc "indivdually gets and updates pull requests"
-    task :update => :environment do
+    task :update_pulls => :environment do
       Hubstats::GithubAPI.update_pulls
     end
 
+    desc "sets labels for pull_requests"
+    task :update_labels => :environment do
+      Hubstats::GithubAPI.update_labels
+    end
+
     def repo_checker(args)
       raise ArgumentError, "Must be called with repo argument. [:org/:repo]" if args.nil?
       if args.is_a? String

data/test/dummy/db/migrate/{20140702234357_create_hubstats_pull_requests.hubstats.rb → 20140703201046_create_hubstats_pull_requests.hubstats.rb}

@@ -3,6 +3,18 @@ class CreateHubstatsPullRequests < ActiveRecord::Migration
   def change
     create_table :hubstats_pull_requests do |t|
       t.integer :id
+      t.integer :number
+      t.belongs_to :user
+      t.belongs_to :repo
+      t.datetime :created_at
+      t.datetime :updated_at
+      t.datetime :closed_at
+      t.integer :additions
+      t.integer :deletions
+      t.integer :comments
+      t.integer :commits
+      t.integer :changed_files
+
       t.string :url
       t.string :html_url
       t.string :diff_url
@@ -13,25 +25,13 @@ class CreateHubstatsPullRequests < ActiveRecord::Migration
       t.string :review_comment_url
       t.string :comments_url
       t.string :statuses_url
-      t.integer :number
       t.string :state
       t.string :title
       t.string :body
-      t.string :created_at
-      t.string :updated_at
-      t.string :closed_at
       t.string :merged_at
       t.string :merge_commit_sha
       t.string :merged
       t.string :mergeable
-      t.integer :comments
-      t.integer :commits
-      t.integer :additions
-      t.integer :deletions
-      t.integer :changed_files
-      t.belongs_to :user
-      t.belongs_to :repo
-      t.timestamps
     end
 
     add_index :hubstats_pull_requests, :user_id

data/test/dummy/db/migrate/{20140702234358_create_hubstats_users.hubstats.rb → 20140703201047_create_hubstats_users.hubstats.rb}

@@ -4,6 +4,10 @@ class CreateHubstatsUsers < ActiveRecord::Migration
     create_table :hubstats_users do |t|
       t.string :login
       t.integer :id
+      t.string :role
+      t.boolean :site_admin
+      t.timestamps
+      
       t.string :avatar_url
       t.string :gravatar_id
       t.string :url
@@ -17,9 +21,6 @@ class CreateHubstatsUsers < ActiveRecord::Migration
       t.string :repos_url
       t.string :events_url
       t.string :received_events_url
-      t.string :role
-      t.boolean :site_admin
-      t.timestamps
     end
   end
 end

data/test/dummy/db/migrate/{20140702234359_create_hubstats_comments.hubstats.rb → 20140703201048_create_hubstats_comments.hubstats.rb}

@@ -3,9 +3,14 @@ class CreateHubstatsComments < ActiveRecord::Migration
   def change
     create_table :hubstats_comments do |t|
       t.integer :id
-      t.string :html_url
-      t.string :url
-      t.string :pull_request_url
+      t.string :kind
+      t.belongs_to :user
+      t.belongs_to :pull_request
+      t.belongs_to :repo
+      t.datetime :created_at
+      t.datetime :updated_at
+      t.string :body
+
       t.string :diff_hunk
       t.integer :path
       t.integer :position
@@ -13,12 +18,10 @@ class CreateHubstatsComments < ActiveRecord::Migration
       t.string :line
       t.string :commit_id
       t.string :original_commit_id
-      t.string :body
-      t.string :kind
-      t.belongs_to :user
-      t.belongs_to :pull_request
-      t.belongs_to :repo
-      t.timestamps
+
+      t.string :html_url
+      t.string :url
+      t.string :pull_request_url
     end
 
     add_index :hubstats_comments, :user_id

data/test/dummy/db/migrate/{20140702234360_create_hubstats_repos.hubstats.rb → 20140703201049_create_hubstats_repos.hubstats.rb}

@@ -3,10 +3,25 @@ class CreateHubstatsRepos < ActiveRecord::Migration
   def change
     create_table :hubstats_repos do |t|
       t.integer :id
+      t.belongs_to :owner
       t.string :name
       t.string :full_name
+      t.datetime :pushed_at
+      t.datetime :created_at
+      t.datetime :updated_at
+
       t.string :homepage
       t.string :language
+      t.integer :forks_count
+      t.integer :stargazers_count
+      t.integer :watches_count
+      t.integer :size 
+      t.integer :open_issues_count
+      t.boolean :has_issues
+      t.boolean :has_wiki
+      t.boolean :has_downloads
+      t.boolean :private
+      t.boolean :fork
       t.string :description
       t.string :default_branch
       t.string :url
@@ -26,21 +41,6 @@ class CreateHubstatsRepos < ActiveRecord::Migration
       t.string :issues_url
       t.string :pulls_url
       t.string :labels_url
-      t.integer :forks_count
-      t.integer :stargazers_count
-      t.integer :watches_count
-      t.integer :size 
-      t.integer :open_issues_count
-      t.boolean :has_issues
-      t.boolean :has_wiki
-      t.boolean :has_downloads
-      t.boolean :private
-      t.boolean :fork
-      t.datetime :pushed_at
-      t.datetime :created_at
-      t.datetime :updated_at
-      t.belongs_to :owner
-      t.timestamps
     end
 
      add_index :hubstats_repos, :owner_id

data/test/dummy/db/migrate/20140703201050_create_hubstats_labels.hubstats.rb

@@ -0,0 +1,10 @@
+# This migration comes from hubstats (originally 20140703173123)
+class CreateHubstatsLabels < ActiveRecord::Migration
+  def change
+    create_table :hubstats_labels do |t|
+      t.string :name
+      t.string :color
+      t.string :url
+    end
+  end
+end

data/test/dummy/db/migrate/20140703201051_create_hubstats_labels_pull_requests.hubstats.rb

@@ -0,0 +1,9 @@
+# This migration comes from hubstats (originally 20140703200752)
+class CreateHubstatsLabelsPullRequests < ActiveRecord::Migration
+  def change
+    create_table :hubstats_labels_pull_requests do |t|
+      t.belongs_to :pull_request
+      t.belongs_to :label
+    end
+  end
+end

data/test/dummy/db/schema.rb

@@ -11,12 +11,16 @@
 #
 # It's strongly recommended to check this file into your version control system.
 
-ActiveRecord::Schema.define(:version => 20140702234360) do
+ActiveRecord::Schema.define(:version => 20140703201051) do
 
   create_table "hubstats_comments", :force => true do |t|
-    t.string   "html_url"
-    t.string   "url"
-    t.string   "pull_request_url"
+    t.string   "kind"
+    t.integer  "user_id"
+    t.integer  "pull_request_id"
+    t.integer  "repo_id"
+    t.datetime "created_at"
+    t.datetime "updated_at"
+    t.string   "body"
     t.string   "diff_hunk"
     t.integer  "path"
     t.integer  "position"
@@ -24,57 +28,78 @@ ActiveRecord::Schema.define(:version => 20140702234360) do
     t.string   "line"
     t.string   "commit_id"
     t.string   "original_commit_id"
-    t.string   "body"
-    t.string   "kind"
-    t.integer  "user_id"
-    t.integer  "pull_request_id"
-    t.integer  "repo_id"
-    t.datetime "created_at",         :null => false
-    t.datetime "updated_at",         :null => false
+    t.string   "html_url"
+    t.string   "url"
+    t.string   "pull_request_url"
   end
 
   add_index "hubstats_comments", ["pull_request_id"], :name => "index_hubstats_comments_on_pull_request_id"
   add_index "hubstats_comments", ["user_id"], :name => "index_hubstats_comments_on_user_id"
 
+  create_table "hubstats_labels", :force => true do |t|
+    t.string "name"
+    t.string "color"
+    t.string "url"
+  end
+
+  create_table "hubstats_labels_pull_requests", :force => true do |t|
+    t.integer "pull_request_id"
+    t.integer "label_id"
+  end
+
   create_table "hubstats_pull_requests", :force => true do |t|
-    t.string  "url"
-    t.string  "html_url"
-    t.string  "diff_url"
-    t.string  "patch_url"
-    t.string  "issue_url"
-    t.string  "commits_url"
-    t.string  "review_comments_url"
-    t.string  "review_comment_url"
-    t.string  "comments_url"
-    t.string  "statuses_url"
-    t.integer "number"
-    t.string  "state"
-    t.string  "title"
-    t.string  "body"
-    t.string  "created_at",          :null => false
-    t.string  "updated_at",          :null => false
-    t.string  "closed_at"
-    t.string  "merged_at"
-    t.string  "merge_commit_sha"
-    t.string  "merged"
-    t.string  "mergeable"
-    t.integer "comments"
-    t.integer "commits"
-    t.integer "additions"
-    t.integer "deletions"
-    t.integer "changed_files"
-    t.integer "user_id"
-    t.integer "repo_id"
+    t.integer  "number"
+    t.integer  "user_id"
+    t.integer  "repo_id"
+    t.datetime "created_at"
+    t.datetime "updated_at"
+    t.datetime "closed_at"
+    t.integer  "additions"
+    t.integer  "deletions"
+    t.integer  "comments"
+    t.integer  "commits"
+    t.integer  "changed_files"
+    t.string   "url"
+    t.string   "html_url"
+    t.string   "diff_url"
+    t.string   "patch_url"
+    t.string   "issue_url"
+    t.string   "commits_url"
+    t.string   "review_comments_url"
+    t.string   "review_comment_url"
+    t.string   "comments_url"
+    t.string   "statuses_url"
+    t.string   "state"
+    t.string   "title"
+    t.string   "body"
+    t.string   "merged_at"
+    t.string   "merge_commit_sha"
+    t.string   "merged"
+    t.string   "mergeable"
   end
 
   add_index "hubstats_pull_requests", ["repo_id"], :name => "index_hubstats_pull_requests_on_repo_id"
   add_index "hubstats_pull_requests", ["user_id"], :name => "index_hubstats_pull_requests_on_user_id"
 
   create_table "hubstats_repos", :force => true do |t|
+    t.integer  "owner_id"
     t.string   "name"
     t.string   "full_name"
+    t.datetime "pushed_at"
+    t.datetime "created_at"
+    t.datetime "updated_at"
     t.string   "homepage"
     t.string   "language"
+    t.integer  "forks_count"
+    t.integer  "stargazers_count"
+    t.integer  "watches_count"
+    t.integer  "size"
+    t.integer  "open_issues_count"
+    t.boolean  "has_issues"
+    t.boolean  "has_wiki"
+    t.boolean  "has_downloads"
+    t.boolean  "private"
+    t.boolean  "fork"
     t.string   "description"
     t.string   "default_branch"
     t.string   "url"
@@ -94,26 +119,16 @@ ActiveRecord::Schema.define(:version => 20140702234360) do
     t.string   "issues_url"
     t.string   "pulls_url"
     t.string   "labels_url"
-    t.integer  "forks_count"
-    t.integer  "stargazers_count"
-    t.integer  "watches_count"
-    t.integer  "size"
-    t.integer  "open_issues_count"
-    t.boolean  "has_issues"
-    t.boolean  "has_wiki"
-    t.boolean  "has_downloads"
-    t.boolean  "private"
-    t.boolean  "fork"
-    t.datetime "pushed_at"
-    t.datetime "created_at",        :null => false
-    t.datetime "updated_at",        :null => false
-    t.integer  "owner_id"
   end
 
   add_index "hubstats_repos", ["owner_id"], :name => "index_hubstats_repos_on_owner_id"
 
   create_table "hubstats_users", :force => true do |t|
     t.string   "login"
+    t.string   "role"
+    t.boolean  "site_admin"
+    t.datetime "created_at",          :null => false
+    t.datetime "updated_at",          :null => false
     t.string   "avatar_url"
     t.string   "gravatar_id"
     t.string   "url"
@@ -127,10 +142,6 @@ ActiveRecord::Schema.define(:version => 20140702234360) do
     t.string   "repos_url"
     t.string   "events_url"
     t.string   "received_events_url"
-    t.string   "role"
-    t.boolean  "site_admin"
-    t.datetime "created_at",          :null => false
-    t.datetime "updated_at",          :null => false
   end
 
 end