dri 0.1.1 → 0.2.0

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

@@ -1,19 +1,21 @@
 # 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
+
         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,38 +25,40 @@ 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
@@ -65,21 +69,17 @@ module Dri
           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, found 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,22 @@ 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-orchestrated
+            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,18 +40,20 @@ module Dri
 
           pipelines = @options[:filter_pipelines] ? filtered_pipelines : @available_pipelines
 
+          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', :bright_yellow)
-          url = add_color('URL', :bright_yellow)
-          type_label = add_color('Type', :bright_yellow)
+          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"], 90)
+            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,7 @@ 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'
       def failures(*)
         if options[:help]
           invoke :help, ['failures']
@@ -46,6 +58,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/init.rb

@@ -3,17 +3,18 @@
 require_relative '../command'
 
 require "tty-font"
-require "pastel"
 
 module Dri
   module Commands
     class Init < Dri::Command
       def initialize(options)
-        font = TTY::Font.new(:doom)       
+        @options = options
+
+        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"
@@ -21,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?
@@ -42,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

data/lib/dri/commands/publish/report.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require_relative '../../command'
 require_relative '../../utils/markdown_lists'
 require_relative "../../report"
@@ -17,14 +19,14 @@ module Dri
           @time = Time.now.to_i
         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
           report = Dri::Report.new(config)
 
           logger.info "Fetching triaged failures with award emoji #{emoji}..."
 
           spinner.start
-          issues = api_client.fetch_triaged_failures(emoji: emoji, state: 'opened') 
+          issues = api_client.fetch_triaged_failures(emoji: emoji, state: 'opened')
           spinner.stop
 
           if issues.empty?
@@ -33,7 +35,7 @@ module Dri
           end
 
           logger.info "Assembling the report... "
-          # sets each failure on the table 
+          # sets each failure on the table
           action_options = ["pinged SET", "reproduced", "transient", "quarantined"]
 
           spinner.start
@@ -41,17 +43,27 @@ module Dri
             actions = []
 
             if @options[:actions]
-              actions = prompt.multi_select("Please mark the actions on #{add_color(issue['title'], :yellow)}: ", action_options)
+              actions = prompt.multi_select(
+                "Please mark the actions on #{add_color(issue['title'], :yellow)}: ",
+                action_options
+              )
             end
+
             report.add_failure(issue, actions)
           end
 
           if @options[:format] == 'list'
             # generates markdown list with failures
             format_style = Utils::MarkdownLists.make_list(report.labels, report.failures) unless report.failures.empty?
-          else 
+          else
             # generates markdown table with rows as failures
-            format_style = MarkdownTables.make_table(report.labels, report.failures, is_rows: true, align: %w[c l c l l]) unless report.failures.empty?
+            unless report.failures.empty?
+              format_style = MarkdownTables.make_table(
+                report.labels,
+                report.failures,
+                is_rows: true, align: %w[l l l l l]
+              )
+            end
           end
 
           report.set_header(timezone, username)
@@ -71,7 +83,7 @@ module Dri
             File.open(report_path, 'a') do |out_file|
               out_file.puts note
             end
-          
+
             spinner.stop
 
             output.puts "Done! ✅\n"
@@ -82,10 +94,13 @@ module Dri
           # sends note to the weekly triage report
           issues = api_client.fetch_current_triage_issue
           current_issue_iid = issues[0]["iid"]
-          response = api_client.post_triage_report_note(iid: current_issue_iid, body: note)
+
+          api_client.post_triage_report_note(iid: current_issue_iid, body: note)
 
           output.puts "Done! ✅\n"
-          logger.success "Thanks @#{username}, your report was posted at https://gitlab.com/gitlab-org/quality/pipeline-triage/-/issues/#{current_issue_iid} 🎉"
+          logger.success(<<~MSG)
+            Thanks @#{username}, your report was posted at https://gitlab.com/gitlab-org/quality/pipeline-triage/-/issues/#{current_issue_iid} 🎉
+          MSG
         end
       end
     end

data/lib/dri/commands/publish.rb

@@ -5,16 +5,15 @@ require 'thor'
 module Dri
   module Commands
     class Publish < Thor
-
       namespace :publish
 
       desc 'report', 'Generate a report'
       method_option :dry_run, type: :boolean,
-                           desc: 'Generates a report locally'
-      method_option :format, aliases: '-f', type: :string, :default => "table",
-                           desc: 'Formats the report'
+                              desc: 'Generates a report locally'
+      method_option :format, aliases: '-f', type: :string, default: "table",
+                             desc: 'Formats the report'
       method_option :actions, type: :boolean,
-                           desc: 'Updates actions on failures'
+                              desc: 'Updates actions on failures'
       def report(*)
         if options[:help]
           invoke :help, ['report']

data/lib/dri/commands/rm/emoji.rb

@@ -10,9 +10,9 @@ module Dri
           @options = options
         end
 
-        def execute(input: $stdin, output: $stdout)
+        def execute(input: $stdin, output: $stdout) # rubocop:disable Metrics/AbcSize
           verify_config_exists
-          
+
           remove = prompt.yes? "Are you sure you want to remove all #{emoji} award emojis from issues?"
 
           unless remove
@@ -29,7 +29,7 @@ module Dri
           spinner.stop
 
           issues_with_award_emoji.each do |issue|
-            logger.info "Removing #{emoji} from #{issue["web_url"]}..."
+            logger.info "Removing #{emoji} from #{issue['web_url']}..."
 
             award_emoji_url = issue["_links"]["award_emoji"]
 
@@ -37,8 +37,8 @@ module Dri
 
             emoji_found = response.find { |e| e['name'] == emoji && e['user']['username'] == username }
 
-            if !emoji_found.nil?
-              url = "#{award_emoji_url}/#{emoji_found["id"]}"
+            unless emoji_found.nil?
+              url = "#{award_emoji_url}/#{emoji_found['id']}"
               api_client.delete_award_emoji(url)
             end
           end