export-pull-requests 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 7bcd3a99bc36a8a9c42ae375bd29113f4b1de1f7
-  data.tar.gz: ea1fc9e512b536b7c5f6c7d553944d7b44783271
+  metadata.gz: 0a531cf40cb67b6e178bc75a254ee821758369d6
+  data.tar.gz: da773616d8a630e16bc43b68bf19a6dcddfca440
 SHA512:
-  metadata.gz: 56a71220ae9b7492d8ed4fa41f79a40e3d2eea3e2883b5e59d8efb9d3a8bce89820eaa5fa799e11f465a2417ddfdb265f4401335ed28203060f8f9dfe83a31c8
-  data.tar.gz: 793e9283143b22969f9663f355036c0d8c8da497181ddb520f4e445ec1a13571e16b9d0d675ab34404c986c91e407ef76c34ff15ab00cd7838e9b4dd74f92a16
+  metadata.gz: 859d3b348f3e95edefa52cf74de68a5a3ce253863ddd9c8695b663fa5b3488193d9672740822f51b6becf14558a9ac4d623d36f56df79fc2ddd23d365f2a3aea
+  data.tar.gz: 883f927bd24a27b417b085550fc4697e06ce4f631c3212ce54e53e8ce74f901df02c7b33ae9d7df99f9aa20b80cdc81288deadea57c1e0eb860f8a7f7af6ca20

data/bin/epr

@@ -3,10 +3,14 @@
 require "csv"
 require "optparse"
 require "time"
+require "logger"
+
 require "github_api"
 require "gitlab"
+require "bitbucket_rest_api"
 
-VERSION = "0.1.0"
+VERSION = "0.1.1"
+SERVICES = %w[github gitlab bitbucket]
 
 def localtime(t)
   Time.parse(t).localtime.strftime("%x %X")
@@ -36,6 +40,39 @@ def lookup_token
   end
 end
 
+def bitbucket(user, repo)
+  page = 0
+  rows = []
+
+  $bitbucket ||= BitBucket.new(:basic_auth => $token)
+  # TODO: make sure no need to translate any states
+  # https://developer.atlassian.com/bitbucket/api/2/reference/resource/repositories/%7Busername%7D/%7Brepo_slug%7D/pullrequests
+
+  loop do
+    page += 1
+
+    prs = $bitbucket.repos.pull_request.all(user, repo, :page => page, :state => $filter.upcase)
+    prs["values"].each do |pr|
+      next if pr.author && skip_user?(pr.author.username)
+
+      rows << [
+        "#{user}/#{repo}",
+        pr.id,
+        pr.author ? pr.author.username : "-",
+        pr.title,
+        pr.state,
+        localtime(pr.created_on),
+        localtime(pr.updated_on),
+        pr["links"].html.href
+      ]
+    end
+
+    break unless prs["next"]
+  end
+
+  rows
+end
+
 def github(user, repo)
   rows = []
 
@@ -99,12 +136,14 @@ def export_repos(repos)
       rows.concat(github(user, repo))
     when "gitlab"
       rows.concat(gitlab(user, repo))
+    when "bitbucket"
+      rows.concat(bitbucket(user, repo))
     else
-      abort "unknown service provider: #{service}"
+      abort "unknown service provider: #$provider"
     end
 
     rows.each do |r|
-      # Repository column is dropped if we only have one repo
+      # Drop Repository column if we only have one repo
       r.shift unless repos.size > 1
       puts r.to_csv
     end
@@ -112,10 +151,14 @@ def export_repos(repos)
   end
 end
 
+# Hashie is used by Bitbucket, and it spews warnings to stdout. We stop that,
+# specifically: https://github.com/bitbucket-rest-api/bitbucket/issues/87
+Hashie.logger = Logger.new(File::NULL) if defined?(Hashie)
+
 $exclude_users = []
 $include_users = []
 $filter = "open"
-$provider = "github"
+$provider = ENV["EPR_SERVICE"] || SERVICES[0]
 $token = lookup_token
 
 parser = OptionParser.new do |opts|
@@ -135,7 +178,7 @@ parser = OptionParser.new do |opts|
     $token = t
   end
 
-  opts.on "-p, --provider=NAME", %w[github gitlab], "Service provider: gitlab or github; defaults to github" do |name|
+  opts.on "-p, --provider=NAME", SERVICES, "Service provider, one of: #{SERVICES}" do |name|
     $provider = name
   end
 
@@ -144,7 +187,7 @@ parser = OptionParser.new do |opts|
   end
 
   opts.on "-v", "--version", "epr version" do
-    puts "v#{VERSION} (GitHub API v#{Github::VERSION}, GitLab API v#{Gitlab::VERSION})"
+    puts "v#{VERSION} (GitHub v#{Github::VERSION}, GitLab v#{Gitlab::VERSION}, Bitbucket v#{BitBucket::VERSION::STRING})"
     exit
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: export-pull-requests
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Skye Shaw
@@ -38,22 +38,36 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '4.0'
+- !ruby/object:Gem::Dependency
+  name: bitbucket_rest_api
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.1'
 - !ruby/object:Gem::Dependency
   name: rake
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.9.2
+        version: '0.9'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.9.2
-description: Program to export GitHub pull requests or GitLab merge requests to CSV
-  a file.
+        version: '0.9'
+description: Program to export GitHub, GitLab, or Bitbucket pull requests/merge requests
+  to CSV a file.
 email: skye.shaw@gmail.com
 executables:
 - epr