dri 0.1.2 → 0.3.0

data/lib/dri/commands/fetch/failures.rb

@@ -1,19 +1,28 @@
 # frozen_string_literal: true
 
 require_relative '../../command'
-
-require 'tty-table'
+require_relative '../../utils/table'
 
 module Dri
   module Commands
     class Fetch
       class Failures < Dri::Command
+        include Dri::Utils::Table
+        using Refinements
+
+        SORT_BY_OPTIONS = {
+          title: 0,
+          triaged: 1,
+          author: 2,
+          url: 3
+        }.freeze
+
         def initialize(options)
           @options = options
           @today_iso_format = Time.now.strftime('%Y-%m-%dT00:00:00Z')
         end
 
-        def execute(input: $stdin, output: $stdout)
+        def execute(input: $stdin, output: $stdout) # rubocop:disable Metrics/AbcSize, Metrics/CyclomaticComplexity, Metrics/PerceivedComplexity, Metrics/MethodLength
           verify_config_exists
 
           title = add_color('Title', :bright_yellow)
@@ -23,63 +32,63 @@ module Dri
 
           failures = []
           urgent = []
-          labels = [ title, triaged, author, url ]
+          labels = [title, triaged, author, url]
           triaged_counter = 0
 
-          logger.info "Fetching today\'s failures..."
-          
-          spinner.run do 
+          logger.info "Fetching today's failures..."
 
+          spinner.run do # rubocop:disable Metrics/BlockLength
             response = api_client.fetch_failures(date: @today_iso_format, state: 'opened')
 
             if response.nil?
-              logger.info "Life is great, there are no new failures today!"
+              logger.info 'Life is great, there are no new failures today!'
               exit 0
             end
 
             response.each do |failure|
-              title = truncate(failure["title"], 60)
-              author = failure["author"]["username"]
-              url = failure["web_url"]
-              award_emoji_url = failure["_links"]["award_emoji"]
+              title = failure['title'].truncate(60)
+              author = failure['author']['username']
+              url = failure['web_url']
+              award_emoji_url = failure['_links']['award_emoji']
               triaged = add_color('x', :red)
 
-              emoji_awards = api_client.fetch_awarded_emojis(award_emoji_url)
+              emoji_awards = api_client.fetch_awarded_emojis(award_emoji_url).find do |e|
+                (e['name'] == emoji) && (e['user']['username'] == username)
+              end
 
-              triaged = add_color('✓', :green) && triaged_counter += 1 if emoji_awards.find do |e| 
-                e['name'] == emoji && e['user']['username'] == @username 
+              if emoji_awards
+                triaged = add_color('✓', :green)
+                triaged_counter += 1
               end
 
               if @options[:urgent]
-                labels = failure["labels"]
+                labels = failure['labels']
 
                 labels.each do |label|
-                  if label.include? ("found:canary.gitlab.com" && "found:canary.staging.gitlab.com")
+                  if label.include?('found:canary.gitlab.com' && 'found:canary.staging.gitlab.com')
                     urgent << [title, triaged, author, url]
                   end
                 end
               end
 
               failures << [title, triaged, author, url]
+
+              failures.sort_by! { |e| e[SORT_BY_OPTIONS[@options[:sort_by].to_sym]] } if @options[:sort_by]
             end
           end
 
           if @options[:urgent]
-            table = TTY::Table.new(labels, urgent)
-            puts table.render(:ascii, resize: true, alignments: [:center, :center, :center, :center])
-            output.puts "\nFound: #{urgent.size} urgent failures, ocurring in both canary.gitlab.com and canary.staging.gitlab.com."
+            print_table(labels, urgent, alignments: [:left, :center, :center, :left])
+            output.puts(<<~MSG)
+              Found: #{urgent.size} urgent failures, occurring in both canary.gitlab.com and canary.staging.gitlab.com.
+            MSG
           else
-            table = TTY::Table.new(labels, failures)
-            puts table.render(:ascii, resize: true, alignments: [:center, :center, :center, :center])
-            output.puts "\nFound: #{failures.size} failures, of these #{triaged_counter} have been triaged with a #{emoji}."
+            print_table(labels, failures, alignments: [:left, :center, :center, :left])
+            output.puts(<<~MSG)
+              Found: #{failures.size} failures, of these #{triaged_counter} have been triaged with a #{emoji}.
+            MSG
           end
         end
-         
-        private
-
-        def truncate(string, max)
-          string.length > max ? "#{string[0...max]}..." : string
-        end
       end
     end
   end

data/lib/dri/commands/fetch/featureflags.rb

@@ -0,0 +1,95 @@
+# frozen_string_literal: true
+
+require_relative '../../command'
+require_relative '../../utils/table'
+
+module Dri
+  module Commands
+    class Fetch
+      class FeatureFlags < Dri::Command
+        include Dri::Utils::Table
+
+        PRODUCTION  = 'host::gitlab.com'
+        STAGING     = 'host::staging.gitlab.com'
+        STAGING_REF = 'host::staging-ref.gitlab.com'
+        PREPROD     = 'host::pre.gitlab.com'
+
+        def initialize(options)
+          @options = options
+          @today_iso_format = Time.now.strftime('%Y-%m-%dT00:00:00Z')
+        end
+
+        def execute(input: $stdin, output: $stdout) # rubocop:disable Metrics/AbcSize, Metrics/CyclomaticComplexity, Metrics/PerceivedComplexity, Metrics/MethodLength
+          verify_config_exists
+
+          summary = add_color('Summary', :bright_yellow)
+          changed_on = add_color('Changed(UTC)', :bright_yellow)
+          url = add_color('URL', :bright_yellow)
+
+          prod_feature_flags = []
+          staging_feature_flags = []
+          staging_ref_feature_flags = []
+          preprod_feature_flags = []
+
+          headers = [summary, changed_on, url]
+
+          logger.info "Fetching today's feature flag changes..."
+
+          spinner.run do # rubocop:disable Metrics/BlockLength
+            page = 1
+            loop do # rubocop:disable Metrics/BlockLength
+              response = api_client.fetch_feature_flag_logs(date: @today_iso_format, page: page)
+
+              if response.nil? && page == 1
+                logger.info 'It\'s been quiet...no feature flag changes for today 👀'
+                exit 0
+              end
+
+              response.each do |feature_flag|
+                summary = feature_flag['title']
+
+                substrings = ["set to \"true\"", "set to \"false\""]
+                next unless substrings.any? { |substr| summary.include?(substr) }
+
+                changed_on = feature_flag['description'][/(?<=Changed on \(in UTC\): ).+?(?=\n)/].delete('`')
+                url = feature_flag['web_url']
+
+                feature_flag_data = [summary, changed_on, url]
+
+                labels = feature_flag['labels']
+                host_label = labels.select { |label| /^host::/.match(label) }.join('')
+
+                case host_label
+                when PRODUCTION
+                  prod_feature_flags << feature_flag_data
+                when STAGING
+                  staging_feature_flags << feature_flag_data
+                when STAGING_REF
+                  staging_ref_feature_flags << feature_flag_data
+                when PREPROD
+                  preprod_feature_flags << feature_flag_data
+                end
+              end
+
+              page += 1
+              break if response.count < 100 || response.nil?
+            end
+          end
+
+          print_results('Production', headers, prod_feature_flags, output) unless prod_feature_flags.empty?
+          print_results('Staging', headers, staging_feature_flags, output) unless staging_feature_flags.empty?
+          print_results('Staging Ref', headers, staging_ref_feature_flags, output) unless staging_ref_feature_flags.empty? # rubocop:disable Layout/LineLength
+          print_results('Preprod', headers, preprod_feature_flags, output) unless preprod_feature_flags.empty?
+        end
+
+        private
+
+        def print_results(env, headers, rows, output_src)
+          puts "\n#{add_color(env, :bright_blue)}"
+          print_table(headers, rows)
+          output_src.puts "\nFound: #{rows.count} feature flag change(s) set to true or false on #{env}."
+        end
+      end
+    end
+  end
+end

data/lib/dri/commands/fetch/quarantines.rb

@@ -0,0 +1,53 @@
+# frozen_string_literal: true
+
+require_relative '../../command'
+require_relative '../../utils/table'
+
+module Dri
+  module Commands
+    class Fetch
+      class Quarantines < Dri::Command
+        include Dri::Utils::Table
+        using Refinements
+
+        def initialize(options, search: '[QUARANTINE]')
+          @options = options
+          @search = search
+
+          @today_iso_format = Time.now.strftime('%Y-%m-%dT00:00:00Z')
+        end
+
+        def execute(input: $stdin, output: $stdout)
+          verify_config_exists
+          title = add_color('Example name', :bright_yellow)
+          url = add_color('URL', :bright_yellow)
+
+          headers = [title, url]
+          mrs = []
+
+          logger.info "Fetching #{@search} MRs..."
+
+          spinner.run do
+            response = api_client.fetch_mrs(
+              project_id: 'gitlab-org/gitlab',
+              labels: 'QA,Quality',
+              search: @search,
+              in: :title,
+              state: :opened
+            )
+
+            mrs = response.each_with_object([]) do |mr, found_mrs|
+              title = mr['title'][13..].truncate(60) # remove the "[QUARANTINE] " prefix
+              url = mr['web_url']
+
+              found_mrs << [title, url]
+            end
+          end
+
+          print_table(headers, mrs, alignments: [:left, :left])
+          output.puts "Found #{mrs.size} open #{@search} MRs"
+        end
+      end
+    end
+  end
+end

data/lib/dri/commands/fetch/testcases.rb

@@ -8,13 +8,21 @@ module Dri
   module Commands
     class Fetch
       class Testcases < Dri::Command
-
         def initialize(options)
           @options = options
-          @available_pipelines = %w(main canary master nightly production staging-canary staging-orchestrated staging-ref staging)
+          @available_pipelines = %w[
+            main
+            canary
+            master
+            nightly
+            production
+            staging-canary
+            staging-ref
+            staging
+          ]
         end
 
-        def execute(input: $stdin, output: $stdout)
+        def execute(input: $stdin, output: $stdout) # rubocop:disable Metrics/AbcSize
           verify_config_exists
 
           if @options[:filter_pipelines]
@@ -31,22 +39,20 @@ module Dri
 
           pipelines = @options[:filter_pipelines] ? filtered_pipelines : @available_pipelines
 
-          if pipelines.empty?
-            logger.error "No pipelines selected to continue to fetch testcases."
-          end
+          logger.error "No pipelines selected to continue to fetch testcases." if pipelines.empty?
 
           pipelines.each do |pipeline|
             logger.info "Fetching failing testcases in #{pipeline}\n"
             response = api_client.fetch_failing_testcases(pipeline, state: 'opened')
-        
+
             output.puts "♦♦♦♦♦ #{add_color(pipeline, :black, :on_white)}♦♦♦♦♦\n\n"
 
             response.each do |pipeline|
-              output.puts "#{title_label} #{pipeline["title"]}\n#{url_label} #{pipeline["web_url"]}"
+              output.puts "#{title_label} #{pipeline['title']}\n#{url_label} #{pipeline['web_url']}"
               output.puts "#{divider}\n"
-            end 
+            end
           end
-          
+
           spinner.stop
         end
       end

data/lib/dri/commands/fetch/triaged.rb

@@ -1,26 +1,31 @@
+# frozen_string_literal: true
+
 require_relative '../../command'
-require 'tty-table'
+require_relative '../../utils/table'
 
 module Dri
   module Commands
     class Fetch
       class Triaged < Dri::Command
+        include Dri::Utils::Table
+        using Refinements
+
         def initialize(options)
           @options = options
         end
 
-        def execute(input: $stdin, output: $stdout)
+        def execute(input: $stdin, output: $stdout) # rubocop:disable Metrics/AbcSize
           verify_config_exists
 
           title = add_color('Title', :magenta)
           url = add_color('URL', :magenta)
           type_label = add_color('Type', :magenta)
 
-          table_labels = [ title, url, type_label ]
+          table_labels = [title, url, type_label]
           failures_triaged = []
 
           logger.info "Fetching your triaged failures..."
-          spinner.start 
+          spinner.start
 
           response = api_client.fetch_triaged_failures(emoji: emoji, state: 'opened')
 
@@ -30,7 +35,7 @@ module Dri
           end
 
           response.each do |triaged|
-            title = truncate(triaged["title"], 70)
+            title = triaged["title"].truncate(70)
             url = triaged["web_url"]
             labels = triaged["labels"]
             type = ""
@@ -39,21 +44,12 @@ module Dri
               type = label.gsub!('failure::', ' ').to_s if label.include? "failure::"
             end
 
-            labels = triaged["labels"]
-  
             failures_triaged << [title, url, type]
           end
 
           spinner.stop
 
-          table = TTY::Table.new(table_labels,failures_triaged)
-          puts table.render(:ascii, resize: true, alignments: [:center, :center, :center])
-        end
-
-        private
-
-        def truncate(string, max)
-          string.length > max ? "#{string[0...max]}..." : string
+          print_table(table_labels, failures_triaged)
         end
       end
     end

data/lib/dri/commands/fetch.rb

@@ -7,6 +7,18 @@ module Dri
     class Fetch < Thor
       namespace :fetch
 
+      desc 'featureflags', 'Display feature flag changes for today'
+      method_option :help, aliases: '-h', type: :boolean,
+                           desc: 'Display usage information'
+      def featureflags(*)
+        if options[:help]
+          invoke :help, ['featureflags']
+        else
+          require_relative 'fetch/featureflags'
+          Dri::Commands::Fetch::FeatureFlags.new(options).execute
+        end
+      end
+
       desc 'triaged', 'Command description...'
       method_option :help, aliases: '-h', type: :boolean,
                            desc: 'Display usage information'
@@ -23,7 +35,7 @@ module Dri
       method_option :help, aliases: '-h', type: :boolean,
                            desc: 'Display usage information'
       method_option :filter_pipelines, type: :boolean,
-                           desc: 'Filter by pipeline'
+                                       desc: 'Filter by pipeline'
       def testcases(*)
         if options[:help]
           invoke :help, ['testcases']
@@ -37,7 +49,9 @@ module Dri
       method_option :help, aliases: '-h', type: :boolean,
                            desc: 'Display usage information'
       method_option :urgent, type: :boolean,
-                           desc: 'Shows failures that quickly escalated'
+                             desc: 'Shows failures that quickly escalated'
+      method_option :sort_by, type: :string,
+                              desc: 'Shows failures in specified order'
       def failures(*)
         if options[:help]
           invoke :help, ['failures']
@@ -46,6 +60,28 @@ module Dri
           Dri::Commands::Fetch::Failures.new(options).execute
         end
       end
+
+      desc 'quarantines', 'Display open quarantine MRs'
+      method_option :help, aliases: '-h', type: :boolean,
+                           desc: 'Display usage information'
+      def quarantines(*)
+        if options[:help]
+          invoke :help, ['quarantines']
+        else
+          require_relative 'fetch/quarantines'
+          Dri::Commands::Fetch::Quarantines.new(options, search: '[QUARANTINE]').execute
+        end
+      end
+
+      desc 'dequarantines', 'Display open dequarantine MRs'
+      method_option :help, aliases: '-h', type: :boolean,
+                           desc: 'Display usage information'
+      def dequarantines(*)
+        return invoke :help, %w[quarantines] if options[:help]
+
+        require_relative 'fetch/quarantines'
+        Dri::Commands::Fetch::Quarantines.new(options, search: '[DEQUARANTINE]').execute
+      end
     end
   end
 end

data/lib/dri/commands/incidents.rb

@@ -0,0 +1,59 @@
+# frozen_string_literal: true
+
+require_relative '../command'
+require_relative '../utils/table'
+
+module Dri
+  module Commands
+    class Incidents < Dri::Command
+      include Dri::Utils::Table
+      using Refinements
+
+      def initialize(options)
+        @options = options
+      end
+
+      def execute(input: $stdin, output: $stdout) # rubocop:disable Metrics/AbcSize
+        verify_config_exists
+
+        incident = add_color('Incident', :bright_yellow)
+        service = add_color('Service', :bright_yellow)
+        status = add_color('Status', :bright_yellow)
+        url = add_color('URL', :bright_yellow)
+
+        header = [incident, service, status, url]
+
+        logger.info "Looking for open incidents..."
+        incidents = []
+
+        spinner.run do
+          response = api_client.incidents
+
+          if response.nil?
+            logger.info 'Hooray, no active incidents 🎉.'
+            exit 0
+          end
+
+          response.each do |incident|
+            title = incident['title'].truncate(70)
+            url = incident['web_url']
+            labels = incident['labels']
+            status = "N/A"
+            service = "N/A"
+
+            labels.each do |label|
+              status = label.gsub!('Incident::', ' ').to_s if label.include? "Incident::"
+              service = label.gsub!('Service::', ' ').to_s if label.include? "Service::"
+            end
+
+            incidents << [title, service, status, url]
+          end
+        end
+        print_table(header, incidents, alignments: [:left, :center, :center, :center])
+        output.puts(<<~MSG)
+          Found: #{incidents.size} incident(s).
+        MSG
+      end
+    end
+  end
+end

data/lib/dri/commands/init.rb

@@ -10,11 +10,11 @@ module Dri
       def initialize(options)
         @options = options
 
-        font = TTY::Font.new(:doom)       
+        font = TTY::Font.new(:doom)
         puts pastel.yellow(font.write("DRI"))
       end
 
-      def execute(input: $stdin, output: $stdout)
+      def execute(input: $stdin, output: $stdout) # rubocop:disable Metrics/AbcSize
         output.puts "🤖 Welcome to DRI 🤖\n"
 
         logger.info "🔎 Scanning for existing configurations...\n"
@@ -22,14 +22,16 @@ module Dri
         if config.exist?
           overwrite = prompt.yes?("There is already a configuration initialized. Would you like to overwrite it?")
           unless overwrite
-            output.puts "Using existing configuration. To view configuration in use try #{add_color('dri profile', :yellow)}."
+            output.puts(
+              "Using existing configuration. To view configuration in use try #{add_color('dri profile', :yellow)}."
+            )
             exit 0
           end
         end
 
         @username = prompt.ask("What is your GitLab username?")
         @token = prompt.mask("Please provide your GitLab personal access token:")
-        @timezone = prompt.select("Choose your current timezone?", %w(EMEA AMER APAC))
+        @timezone = prompt.select("Choose your current timezone?", %w[EMEA AMER APAC])
         @emoji = prompt.ask("Have a triage emoji?")
 
         if (@emoji || @token || @username).nil?
@@ -43,7 +45,7 @@ module Dri
         config.set(:settings, :emoji, value: @emoji)
         config.write(force: true)
 
-        logger.success "✅ We're ready to go 🚀"      
+        logger.success "✅ We're ready to go 🚀"
       end
     end
   end

data/lib/dri/commands/profile.rb

@@ -20,10 +20,15 @@ module Dri
         logger.info "🔎 Looking for profiles...\n"
 
         if config.exist?
-          box = TTY::Box.frame(width: 30, height: 10, align: :center, padding: 1, title: {top_left: add_color('PROFILE:', :bright_cyan)}, border: :thick) do
-            pretty_print_profile
-          end
-          print box
+          frame_args = {
+            width: 30,
+            height: 10,
+            align: :center,
+            padding: 1,
+            border: :thick,
+            title: { top_left: add_color('PROFILE:', :bright_cyan) }
+          }
+          output.print TTY::Box.frame(**frame_args) { pretty_print_profile.strip }
           output.puts "☝️ To modify this profile try passing #{add_color('dri profile --edit', :yellow)}.\n"
         else
           logger.error "Oops.. Profile not found. Try creating one using #{add_color('dri init', :yellow)}."
@@ -31,7 +36,11 @@ module Dri
       end
 
       def pretty_print_profile
-        "#{add_color('User:', :bright_cyan)} #{username}\n #{add_color('Token:', :bright_cyan)} #{token}\n #{add_color('Timezone:', :bright_cyan)} #{timezone}\n #{add_color('Emoji:', :bright_cyan)} #{emoji}"
+        <<~PROFILE
+          #{add_color('User:', :bright_cyan)} #{username}\n #{add_color('Token:', :bright_cyan)} #{token}
+          #{add_color('Timezone:', :bright_cyan)} #{timezone}
+          #{add_color('Emoji:', :bright_cyan)} #{emoji}
+        PROFILE
       end
     end
   end