githuh 0.2.0 → 0.2.1

data/lib/githuh.rb

@@ -84,6 +84,7 @@ module Githuh
 end
 
 require 'githuh/cli/commands/base'
-require 'githuh/cli/commands/user/info'
 require 'githuh/cli/commands/version'
+require 'githuh/cli/commands/user/info'
 require 'githuh/cli/commands/repo/list'
+require 'githuh/cli/commands/issue/export'

data/lib/githuh/cli/commands/base.rb

@@ -9,6 +9,7 @@ module Githuh
   module CLI
     module Commands
       DEFAULT_PAGE_SIZE = 20
+      DEFAULT_TITLE     = 'Operation Progress'
 
       class Base < Dry::CLI::Command
         extend Forwardable
@@ -38,7 +39,7 @@ module Githuh
           self.verbose  = verbose
           self.info     = info
           self.token    = api_token || token_from_gitconfig
-          self.per_page = per_page || DEFAULT_PAGE_SIZE
+          self.per_page = per_page.to_i || DEFAULT_PAGE_SIZE
           self.client   = Octokit::Client.new(access_token: token)
 
           print_userinfo if info
@@ -62,6 +63,26 @@ module Githuh
           80
         end
 
+        def bar(title = DEFAULT_TITLE)
+          @bar ||= create_progress_bar(title: title)
+        end
+
+        # Overwrite me
+        def bar_size
+          0
+        end
+
+        def create_progress_bar(size = bar_size, title: DEFAULT_TITLE)
+          return unless info || verbose
+
+          TTY::ProgressBar.new("[:bar]",
+                               title:    title,
+                               total:    size.to_i,
+                               width:    ui_width - 2,
+                               head:     '',
+                               complete: '▉'.magenta)
+        end
+
         private
 
         def print_userinfo
@@ -77,23 +98,23 @@ module Githuh
           lines << sprintf("         Followers: %s", h(user_info.followers.to_s))
           lines << sprintf("        Member For: %s", h(sprintf("%d years, %d months, %d days", years, months, days)))
 
-          self.box = TTY::Box.frame *lines,
-                                    padding: 1,
+          self.box = TTY::Box.frame(*lines,
+                                    padding: 0,
                                     width:   ui_width,
                                     align:   :left,
-                                    title:   { top_center: Githuh::BANNER },
+                                    title:   { top_center: "┤ #{Githuh::BANNER} ├" },
                                     style:   {
                                       fg:     :white,
                                       border: {
                                         fg: :bright_green
                                       }
-                                    }
+                                    })
 
           Githuh.stdout.print box
         end
 
         def h(arg)
-          arg.to_s.bold.blue
+          arg.to_s
         end
 
         def token_from_gitconfig

data/lib/githuh/cli/commands/issue/export.rb

@@ -0,0 +1,192 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+# vim: ft=ruby
+require 'bundler/setup'
+require 'dry/cli'
+require 'json'
+require 'tty/progressbar'
+require 'csv'
+require 'active_support/inflector'
+
+require_relative '../base'
+
+module Githuh
+  module CLI
+    module Commands
+      module Issue
+        class Export < Base
+          FORMATS = {
+            json: 'json',
+            csv:  'csv'
+          }.freeze
+
+          DEFAULT_FORMAT        = :csv
+          DEFAULT_OUTPUT_FORMAT = "<username>.<repo>.issues.<format>"
+
+          attr_accessor :filename, :file, :output, :repo, :issues, :format, :record_count
+
+          desc "Export Repo issues into a CSV or JSON format\n" \
+               "  Default output file is " + DEFAULT_OUTPUT_FORMAT.bold.yellow
+
+          argument :repo, type: :string, required: true, desc: 'Name of the repo, eg "rails/rails"'
+          option :file, required: false, desc: 'Output file, overrides ' + DEFAULT_OUTPUT_FORMAT
+          option :format, values: FORMATS.keys.map(&:to_s), default: DEFAULT_FORMAT.to_s, required: false, desc: 'Output format'
+
+          def call(repo: nil, file: nil, format: nil, **opts)
+            super(**opts)
+
+            self.record_count = 0
+            self.repo         = repo
+
+            raise ArgumentError, "argument <repo> is required" unless repo
+            raise ArgumentError, "argument <repo> is not a repository, expected eg 'rails/rails'" unless repo =~ %r{/}
+
+            self.issues = []
+            self.output = StringIO.new
+            self.format = (format || DEFAULT_FORMAT).to_sym
+
+            self.filename = file || file_name(repo)
+            self.file     = File.open(filename, 'w')
+
+            print_summary
+
+            # —————————— actually get all issues ———————————————
+            self.file.write send("render_as_#{format}", fetch_issues)
+            # ————————————————————————————————————————————————————————
+
+            print_conclusion
+          ensure
+            file.close if file.respond_to?(:close) && !file.closed?
+          end
+
+          def fetch_issues
+            client.auto_paginate = true
+            self.issues          = filter_issues(client.issues(repo, query: default_options)).tap do |issue_list|
+              self.record_count = issue_list.size
+              bar('Issues')&.advance
+            end
+          end
+
+          def filter_issues(issues_list)
+            issues_list.reject do |issue|
+              issue.html_url =~ /pull/
+            end
+          end
+
+          def bar_size
+            record_count + 1
+          end
+
+          def default_options
+            { state: 'open' }
+          end
+
+          def self.issue_labels(issue)
+            issue.labels.map(&:name)
+          end
+
+          def self.find_user(client, username)
+            @user_cache           ||= {}
+            @user_cache[username] ||= client.user(username).name
+          end
+
+          CSV_MAP = {
+            'Labels'        => ->(_client, issue) { issue_labels(issue).reject { |l| LABEL_ESTIMATES.key?(l) }.join(',').downcase },
+            'Type'          => ->(*) { 'feature' },
+            'Estimate'      => ->(_client, issue) do
+              el = issue_labels(issue).find { |l| LABEL_ESTIMATES.key?(l) }
+              el ? LABEL_ESTIMATES[el] : nil
+            end,
+            'Current State' => ->(*) { 'unstarted' },
+            'Requested By'  => ->(client, issue) do
+              find_user(client, issue.user.login)
+            end,
+            'Owned By'  => ->(client, issue) do
+              find_user(client, issue.user.login)
+            end,
+            'Description'   => ->(_client, issue) {
+              issue.body
+            },
+            'Created at'    => ->(_client, issue) { issue.created_at },
+          }.freeze
+
+          LABEL_ESTIMATES = {
+            'XXL(13 eng day)' => 15,
+            'XL(8 eng day)'   => 15,
+            'xs(<=1 eng day)' => 3,
+            'L(5 eng day)'    => 15,
+            'm(3 eng day)'    => 9,
+            's(2 eng day)'    => 6
+          }.freeze
+
+          CSV_HEADER = %w(Id Title Labels Type Estimate) +
+            ['Current State', 'Created at', 'Accepted at', 'Deadline', 'Requested By',
+             'Owned By', 'Description', 'Comment', 'Comment', 'Comment', 'Comment'].freeze
+
+          # Id,Title,Labels,Type,Estimate,Current State,Created at,Accepted at,Deadline,Requested By,Owned By,Description,Comment,Comment
+          # 100, existing started story,"label one,label two",feature,1,started,"Nov 22, 2007",,,user1,user2,this will update story 100,,
+          # ,new story,label one,feature,-1,unscheduled,,,,user1,,this will create a new story in the icebox,comment1,comment2
+          def render_as_csv(issue_list)
+            # puts "rendering issues as CVS:"
+            # pp issue_list
+            ::CSV.generate do |csv|
+              csv << CSV_HEADER
+              issue_list.each do |issue|
+                row = []
+                CSV_HEADER.each do |column|
+                  method = column.downcase.underscore.to_sym
+                  value  = if CSV_MAP[column]
+                    CSV_MAP[column][client, issue]
+                  else
+                    begin
+                      issue.to_h[method]
+                    rescue StandardError
+                      nil
+                    end
+                  end
+                  value  = value.strip if value.is_a?(String)
+                  row << value
+                end
+                csv << row
+                bar&.advance
+              end
+              bar.finish
+            end
+          end
+
+          def render_as_json(issue_list)
+            JSON.pretty_generate(issue_list.map(&:to_h))
+          end
+
+          private
+
+          def print_conclusion
+            puts
+            puts TTY::Box.info("Success: written a total of #{record_count} records to #{filename}",
+                               width: ui_width, padding: 1)
+            puts
+          end
+
+          def print_summary
+            puts
+            puts TTY::Box.info("Format : #{self.format}\n" \
+                               "File   : #{filename}\n" \
+                               "Repo   : #{repo}\n",
+                               width:   ui_width,
+                               padding: 1)
+            puts
+          end
+
+          def file_name(repo)
+            "#{repo.gsub(%r{/}, '.')}.issues.#{FORMATS[self.format.to_sym]}"
+          end
+        end
+      end
+
+      register 'issue', aliases: ['r'] do |prefix|
+        prefix.register 'export', Issue::Export
+      end
+    end
+  end
+end

data/lib/githuh/cli/commands/issue/export_paginated.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+require_relative 'export'
+
+module Githuh
+  module CLI
+    module Commands
+      module Issue
+        class ExportPaginated < Export
+          def fetch_issues
+            page = 0
+            bar  = nil
+
+            [].tap do |issue_list|
+              loop do
+                options = default_options.merge({
+                                                  page:     page,
+                                                  per_page: per_page,
+                                                })
+
+                puts "page: #{page}"
+                issues_page = client.issues(repo, **options)
+
+                break if issues_page.nil? || issues_page.empty?
+
+                issue_list.concat(issues_page)
+
+                bar("#{repo} Issues Export")&.advance
+                page              += 1
+                self.record_count += issues_page.size
+              end
+
+              bar&.finish; puts
+
+              issue_list << issues
+            end.flatten
+          end
+        end
+      end
+    end
+  end
+end

data/lib/githuh/cli/commands/repo/list.rb

@@ -67,7 +67,7 @@ module Githuh
 
           def repositories
             page = 0
-            bar = nil
+            bar  = nil
 
             [].tap do |repo_list|
               loop do
@@ -78,9 +78,8 @@ module Githuh
                 }
 
                 result = client.repos({}, query: options)
-                bar    = create_progress_bar if info && !verbose && page == 0
+                bar('Repositories')&.advance
 
-                bar&.advance
                 filter_result!(result)
 
                 break if result.empty?
@@ -98,14 +97,8 @@ module Githuh
             end.flatten.sort_by(&:stargazers_count).reverse.uniq(&:name)
           end
 
-          def create_progress_bar
-            number_of_pages = client.last_response.rels[:last].href.match(/page=(\d+).*$/)[1]
-            TTY::ProgressBar.new("[:bar]",
-                                 title:    'Fetching Repositories',
-                                 total:    number_of_pages.to_i,
-                                 width:    ui_width - 2,
-                                 head:     '',
-                                 complete: '▉'.magenta)
+          def bar_size
+            client.last_response.rels[:last].href.match(/page=(\d+).*$/)[1].to_i
           end
 
           def render_as_markdown(repositories)
@@ -138,22 +131,22 @@ module Githuh
           def filter_result!(result)
             result.reject! do |r|
               fork_reject = case forks
-                            when 'exclude'
-                              r.fork
-                            when 'only'
-                              !r.fork
-                            when 'include'
-                              false
-                            end
+              when 'exclude'
+                r.fork
+              when 'only'
+                !r.fork
+              when 'include'
+                false
+              end
 
               private_reject = case private
-                               when true
-                                 !r.private
-                               when false
-                                 r.private
-                               when nil
-                                 false
-                               end
+              when true
+                !r.private
+              when false
+                r.private
+              when nil
+                false
+              end
 
               fork_reject || private_reject
             end

data/lib/githuh/cli/launcher.rb

@@ -3,6 +3,7 @@
 require 'dry/cli'
 require 'forwardable'
 require 'tty/box'
+require 'tty/screen'
 
 require 'githuh'
 module Githuh
@@ -34,14 +35,27 @@ module Githuh
 
         self.command = ::Dry::CLI.new(::Githuh::CLI::Commands)
         command.call(arguments: argv, out: stdout, err: stderr)
-
       rescue StandardError => e
-        box = TTY::Box.frame('ERROR:', ' ', e.message, **BOX_OPTIONS)
-        stderr.print box
+        lines = [e.message.gsub(/\n/, ', ')]
+        if e.backtrace
+          lines << ''
+          lines.concat(e.backtrace)
+        end
 
+        box = TTY::Box.frame(*lines,
+                             **BOX_OPTIONS.merge(
+                               width: TTY::Screen.width,
+                               title: { top_center: "┤ #{e.class.name} ├" },
+                             ))
+        stderr.puts
+        stderr.print box
       ensure
         Githuh.restore_kernel_behavior!
-        exit(10) unless Githuh.in_test
+        exit(0) unless Githuh.in_test
+      end
+
+      def trace?
+        argv.include?('-t') || argv.include?('--trace')
       end
     end
 
@@ -55,16 +69,16 @@ module Githuh
     BOX_OPTIONS = {
       padding: 1,
       align:   :left,
-      title:   { top_center: Githuh::BANNER },
+      title:   { top_center: "┤ #{Githuh::BANNER} ├" },
       width:   80,
       style:   {
-        bg:     :red,
+        bg:     :yellow,
+        fg:     :black,
         border: {
-          fg: :bright_yellow,
-          bg: :red
+          fg: :red,
+          bg: :yellow
         }
       }
     }.freeze
-
   end
 end