dri 0.7.0 → 0.9.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e6ee21fc2a0243e0850f7bab281b0611021807a1a4559e89433741b1faed40f2
-  data.tar.gz: 5b97afcdbc1e1e9307b084eab6e2afddbff94556cc102f2b174a0dedc3f51c33
+  metadata.gz: 2b398cef45fa101592e0abecf4d014969e4e26dac5b5e3fd9f7f774b6108618f
+  data.tar.gz: 96a6b402d7dfcccf9fe2e69e01a4ece8af13d56be0213dc8ffa3824ea9a5ed2a
 SHA512:
-  metadata.gz: f5b54fee2e4f0d99d0d013d0c9a211306eda149dabb6c0cedf05ec0c5a92274dd4038377fc270b6185782ba2dfdbe5c2be6028136c77a4e77dceb21989a82894
-  data.tar.gz: c2d5056697c04ea976e00db46f183cd716091e3681687ad369c6dc3c11dab7ecd5592022460948a19ce779da7b114c2c90c4ba3c712a0bfe6256524cf6859934
+  metadata.gz: 9d94573de9142a648b643819381244382a65fedd7c180ae5b78013ea2738204e54e543e43ed5ad6209c2a3ae5a3af4cd1ecb2a7937afa5607b3c04435ce5a2a2
+  data.tar.gz: 48e010a476d8b1a2b48b3b74ba485e7a9c533aa17f96e4639eead23e486f2a309a8f60fed3c282a9b85ec38154cc2bb0b26fdbf28d3389981d1bf26578336b87

data/.gitlab/merge_request_templates/Release.md

@@ -0,0 +1,28 @@
+## Diff
+<!-- Replace `v0.9.0` with the previous DRI release, and `6a9c4fcc150741d578ec75141d3706891b6694bc` with the latest commit from https://gitlab.com/gitlab-org/quality/dri/-/commits/master that will be included in the release -->
+https://gitlab.com/gitlab-org/quality/dri/-/compare/v0.9.0...6a9c4fcc150741d578ec75141d3706891b6694bc
+
+## Release Notes
+
+```markdown
+### Features
+- !aaa <Title of the aaa MR>
+
+### Bug Fixes
+- !bbb <Title of the bbb MR>
+
+### Maintenance
+- !ccc <Title of the ccc MR>
+```
+
+## Checklists 
+
+### Before merging:
+  - [ ] MR title is set to `Bump version to <version>`, where `<version>` is the new version, according to [SemVer](https://semver.org/).
+  - [ ] Diff link is up-to-date.
+  - [ ] Release notes are accurate.
+  - [ ] `lib/dri/version.rb` is updated with the new version number.
+  - [ ] `bundle update` was run to update `Gemfile.lock` with the new version number.
+
+### After merging:
+  - [ ] Update the release notes for the [newly created tag](https://gitlab.com/gitlab-org/quality/dri/-/tags) with the release notes from this MR.

data/.gitlab-ci.yml

@@ -15,18 +15,18 @@
       - vendor/bundle
     policy: pull
   rules:
-    - if: '$CI_COMMIT_BRANCH == $CI_DEFAULT_BRANCH'
+    - if: '$CI_PIPELINE_SOURCE == "push" && $CI_COMMIT_BRANCH == $CI_DEFAULT_BRANCH'
     - if: '$CI_PIPELINE_SOURCE == "merge_request_event"'
 
 include:
   - project: gitlab-org/quality/pipeline-common
-    ref: 0.15.1
+    ref: 1.2.3
     file:
       - /ci/gem-release.yml
       - /ci/ref-update.gitlab-ci.yml
 
 variables:
-  RUBY_VERSION: "2.7"
+  RUBY_VERSION: "3.0"
 
 stages:
   - build

data/.ruby-version

@@ -1 +1 @@
-3.0.4
+3.0.5

data/.tool-versions

@@ -1 +1 @@
-ruby 3.0.4
+ruby 3.0.5

data/Gemfile.lock

@@ -1,19 +1,20 @@
 PATH
   remote: .
   specs:
-    dri (0.7.0)
+    dri (0.9.0)
       amatch (~> 0.4.1)
-      gitlab (~> 4.18)
-      httparty (~> 0.20.0)
+      gitlab (~> 4.19)
+      httparty (~> 0.21.0)
       json (~> 2.6.1)
       markdown-tables (~> 1.1.1)
       pastel (~> 0.8.0)
-      thor (~> 1.0.1)
+      thor (~> 1.2.0)
       tty-box (~> 0.7.0)
-      tty-config (~> 0.4.0)
-      tty-editor (~> 0.6)
+      tty-config (~> 0.6.0)
+      tty-editor (~> 0.7)
       tty-font (~> 0.5)
       tty-logger (~> 0.6.0)
+      tty-markdown (~> 0.7.0)
       tty-prompt (~> 0.23.1)
       tty-spinner (~> 0.9)
       tty-table (~> 0.12.0)
@@ -21,19 +22,19 @@ PATH
 GEM
   remote: https://rubygems.org/
   specs:
-    activesupport (7.0.2.3)
+    activesupport (7.0.4.2)
       concurrent-ruby (~> 1.0, >= 1.0.2)
       i18n (>= 1.6, < 2)
       minitest (>= 5.1)
       tzinfo (~> 2.0)
-    addressable (2.8.0)
-      public_suffix (>= 2.0.2, < 5.0)
+    addressable (2.8.1)
+      public_suffix (>= 2.0.2, < 6.0)
     amatch (0.4.1)
       mize
       tins (~> 1.0)
     ast (2.4.2)
     coderay (1.1.3)
-    concurrent-ruby (1.1.10)
+    concurrent-ruby (1.2.2)
     crack (0.4.5)
       rexml
     diff-lcs (1.5.0)
@@ -48,37 +49,38 @@ GEM
       rubocop-rails (~> 2.9)
       rubocop-rspec (~> 1.44)
     hashdiff (1.0.1)
-    httparty (0.20.0)
-      mime-types (~> 3.0)
+    httparty (0.21.0)
+      mini_mime (>= 1.0.0)
       multi_xml (>= 0.5.2)
-    i18n (1.10.0)
+    i18n (1.12.0)
       concurrent-ruby (~> 1.0)
-    json (2.6.2)
+    json (2.6.3)
+    kramdown (2.4.0)
+      rexml
     markdown-tables (1.1.1)
     method_source (1.0.0)
-    mime-types (3.4.1)
-      mime-types-data (~> 3.2015)
-    mime-types-data (3.2022.0105)
-    minitest (5.15.0)
-    mize (0.4.0)
+    mini_mime (1.1.2)
+    minitest (5.18.0)
+    mize (0.4.1)
       protocol (~> 2.0)
     multi_xml (0.6.0)
     parallel (1.22.1)
-    parser (3.1.1.0)
+    parser (3.2.1.0)
       ast (~> 2.4.1)
     pastel (0.8.0)
       tty-color (~> 0.5)
     protocol (2.0.0)
       ruby_parser (~> 3.0)
-    pry (0.14.1)
+    pry (0.14.2)
       coderay (~> 1.1)
       method_source (~> 1.0)
-    public_suffix (4.0.6)
-    rack (2.2.3)
+    public_suffix (5.0.1)
+    rack (3.0.4.2)
     rainbow (3.1.1)
     rake (13.0.6)
-    regexp_parser (2.2.1)
+    regexp_parser (2.7.0)
     rexml (3.2.5)
+    rouge (3.30.0)
     rspec (3.10.0)
       rspec-core (~> 3.10.0)
       rspec-expectations (~> 3.10.0)
@@ -101,11 +103,11 @@ GEM
       rubocop-ast (>= 0.6.0)
       ruby-progressbar (~> 1.7)
       unicode-display_width (>= 1.4.0, < 2.0)
-    rubocop-ast (1.16.0)
-      parser (>= 3.1.1.0)
+    rubocop-ast (1.27.0)
+      parser (>= 3.2.1.0)
     rubocop-gitlab-security (0.1.1)
       rubocop (>= 0.51)
-    rubocop-graphql (0.14.2)
+    rubocop-graphql (0.19.0)
       rubocop (>= 0.87, < 2)
     rubocop-performance (1.9.2)
       rubocop (>= 0.90.0, < 2.0)
@@ -117,8 +119,8 @@ GEM
     rubocop-rspec (1.44.1)
       rubocop (~> 0.87)
       rubocop-ast (>= 0.7.1)
-    ruby-progressbar (1.11.0)
-    ruby_parser (3.19.1)
+    ruby-progressbar (1.13.0)
+    ruby_parser (3.20.0)
       sexp_processor (~> 4.16)
     sexp_processor (4.16.1)
     strings (0.2.1)
@@ -129,22 +131,29 @@ GEM
     sync (0.5.0)
     terminal-table (3.0.2)
       unicode-display_width (>= 1.1.1, < 3)
-    thor (1.0.1)
-    timecop (0.9.5)
-    tins (1.31.1)
+    thor (1.2.1)
+    timecop (0.9.6)
+    tins (1.32.1)
       sync
     tty-box (0.7.0)
       pastel (~> 0.8)
       strings (~> 0.2.0)
       tty-cursor (~> 0.7)
     tty-color (0.6.0)
-    tty-config (0.4.0)
+    tty-config (0.6.0)
     tty-cursor (0.7.1)
     tty-editor (0.7.0)
       tty-prompt (~> 0.22)
     tty-font (0.5.0)
     tty-logger (0.6.0)
       pastel (~> 0.8)
+    tty-markdown (0.7.1)
+      kramdown (>= 1.16.2, < 3.0)
+      pastel (~> 0.8)
+      rouge (~> 3.14)
+      strings (~> 0.2.0)
+      tty-color (~> 0.5)
+      tty-screen (~> 0.8)
     tty-prompt (0.23.1)
       pastel (~> 0.8)
       tty-reader (~> 0.8)
@@ -159,11 +168,11 @@ GEM
       pastel (~> 0.8)
       strings (~> 0.2.0)
       tty-screen (~> 0.8)
-    tzinfo (2.0.4)
+    tzinfo (2.0.6)
       concurrent-ruby (~> 1.0)
     unicode-display_width (1.8.0)
     unicode_utils (1.4.0)
-    webmock (3.14.0)
+    webmock (3.18.1)
       addressable (>= 2.8.0)
       crack (>= 0.3.2)
       hashdiff (>= 0.4.0, < 2.0.0)
@@ -182,4 +191,4 @@ DEPENDENCIES
   webmock (~> 3.5)
 
 BUNDLED WITH
-   2.3.16
+   2.4.7

data/README.md

@@ -59,6 +59,7 @@ $ dri profile
   - quarantines
   - dequarantines
   - featureflags
+  - runbooks
 - [4. publish](#4-publish)
   - report
 - [5. rm](#5-rm)
@@ -139,6 +140,13 @@ $ dri fetch pipelines
 Fetches a table containing last executed pipeline and its test report link for all monitored pipelines.
 The timestamps are in UTC
 
+```shell
+$ dri fetch runbooks [runbook]
+```
+
+Fetches a runbook from the [runbooks](https://gitlab.com/gitlab-org/quality/runbooks) project.
+If `[runbook]` is omitted, the command will prompt the user to select from a list of available runbooks.
+
 #### 4. publish
 
 ```shell
@@ -156,12 +164,13 @@ $ dri publish report --format=table # formats the report in a table (default)
 $ dri publish report --dry-run # the report is only generated locally
 $ dri publish report --actions # activate the actions prompt for each failure
 $ dri publish report --feature-flags # includes a summary of the feature flag changes on each environment
+$ dri publish report --update # update the report note if has already been posted
 ```
 
 **Note:** These options above can be combined like:
 
 ```shell
-$ dri publish report --format=list --dry-run --actions
+$ dri publish report --format=list --dry-run --actions --update
 ```
 
 **Actions**

data/dri.gemspec

@@ -23,17 +23,18 @@ Gem::Specification.new do |spec|
   spec.require_paths = ['lib']
 
   spec.add_dependency 'amatch', '~> 0.4.1'
-  spec.add_dependency "gitlab", "~> 4.18"
-  spec.add_dependency 'httparty', '~> 0.20.0'
+  spec.add_dependency "gitlab", "~> 4.19"
+  spec.add_dependency 'httparty', '~> 0.21.0'
   spec.add_dependency 'json', '~> 2.6.1'
   spec.add_dependency 'markdown-tables', '~> 1.1.1'
   spec.add_dependency 'pastel', '~> 0.8.0'
-  spec.add_dependency 'thor', '~> 1.0.1'
+  spec.add_dependency 'thor', '~> 1.2.0'
   spec.add_dependency 'tty-box', '~> 0.7.0'
-  spec.add_dependency 'tty-config', '~> 0.4.0'
-  spec.add_dependency 'tty-editor', '~> 0.6'
+  spec.add_dependency 'tty-config', '~> 0.6.0'
+  spec.add_dependency 'tty-editor', '~> 0.7'
   spec.add_dependency 'tty-font', '~> 0.5'
   spec.add_dependency 'tty-logger', '~> 0.6.0'
+  spec.add_dependency 'tty-markdown', '~> 0.7.0'
   spec.add_dependency 'tty-prompt', '~> 0.23.1'
   spec.add_dependency 'tty-spinner', '~> 0.9'
   spec.add_dependency 'tty-table', '~> 0.12.0'

data/lib/dri/api_client.rb

@@ -6,16 +6,16 @@ require "tty-config"
 require "cgi"
 require "gitlab"
 
+require_relative './utils/constants'
+
 module Dri
   TokenNotProvidedError = Class.new(StandardError)
   class ApiClient # rubocop:disable Metrics/ClassLength
+    include Dri::Utils::Constants::ProjectIDs
+    include Dri::Utils::Constants::Triage::Labels
+
     API_URL                     = "https://gitlab.com/api/v4"
     OPS_API_URL                 = "https://ops.gitlab.net/api/v4"
-    TESTCASES_PROJECT_ID        = 11229385
-    TRIAGE_PROJECT_ID           = 15291320
-    GITLAB_PROJECT_ID           = 278964
-    FEATURE_FLAG_LOG_PROJECT_ID = 15208716
-    INFRA_TEAM_PROD_PROJECT_ID  = 7444821
 
     def initialize(config, ops = false)
       @token = config.read.dig("settings", "token")
@@ -28,14 +28,31 @@ module Dri
       @ops_instance = ops
     end
 
-    # Fetch triaged failures
+    # Fetch all triaged failures
     #
     # @param [String] emoji
     # @param [String] state
-    # @return [Gitlab::ObjectifiedHash]
-    def fetch_triaged_failures(emoji:, state:)
+    # @return [Array<Gitlab::ObjectifiedHash>]
+    def fetch_all_triaged_failures(emoji:, state:)
+      project_ids = [GITLAB_PROJECT_ID, CUSTOMERSDOT_PROJECT_ID]
+      failures = []
+
+      project_ids.each do |project_id|
+        failures += fetch_triaged_failures(project_id: project_id, emoji: emoji, state: state)
+      end
+
+      failures
+    end
+
+    # Fetch triaged failures for a given project
+    #
+    # @param [Integer] project_id
+    # @param [String] emoji
+    # @param [String] state
+    # @return [Array<Gitlab::ObjectifiedHash>]
+    def fetch_triaged_failures(project_id:, emoji:, state:)
       gitlab.issues(
-        GITLAB_PROJECT_ID,
+        project_id,
         order_by: "updated_at",
         my_reaction_emoji: emoji,
         scope: "all",
@@ -61,7 +78,7 @@ module Dri
     # @param [Integer] issue_iid
     # @param [Integer] project_id
     # @return [Array<Gitlab::ObjectifiedHash>]
-    def fetch_awarded_emojis(issue_iid, project_id: GITLAB_PROJECT_ID)
+    def fetch_awarded_emojis(issue_iid, project_id:)
       gitlab.award_emojis(project_id, issue_iid, "issue")
     end
 
@@ -76,14 +93,14 @@ module Dri
         labels: "#{pipeline}::failed",
         state: state,
         scope: "all",
-        "not[labels]": "quarantine"
+        "not[labels]": QUARANTINE
       ).auto_paginate
     end
 
     # Fetch issues related to failing test cases
     #
     # @return [Array<Gitlab::ObjectifiedHash>]
-    def fetch_test_failure_issues(labels: 'failure::new')
+    def fetch_test_failure_issues(labels: FAILURE_NEW)
       gitlab.issues(
         GITLAB_PROJECT_ID,
         labels: labels,
@@ -94,10 +111,11 @@ module Dri
 
     # Fetch related issue mrs
     #
+    # @param [Integer] project_id
     # @param [Integer] issue_iid
     # @return [Array<Gitlab::ObjectifiedHash>]
-    def fetch_related_mrs(issue_iid)
-      gitlab.related_issue_merge_requests(GITLAB_PROJECT_ID, issue_iid)
+    def fetch_related_mrs(project_id, issue_iid)
+      gitlab.related_issue_merge_requests(project_id, issue_iid)
     end
 
     # Fetch MRs
@@ -130,15 +148,42 @@ module Dri
       gitlab.create_issue_note(TRIAGE_PROJECT_ID, iid, body)
     end
 
-    # Fetch new failures
+    # Update triage report note
+    #
+    # @param [Integer] iid
+    # @param [Integer] note_id
+    # @param [String] body
+    # @return [Gitlab::ObjectifiedHash]
+    def update_triage_report_note(iid:, note_id:, body:)
+      gitlab.edit_issue_note(TRIAGE_PROJECT_ID, iid, note_id, body)
+    end
+
+    # Fetch all new failures
     #
     # @param [String] date
     # @param [String] state
     # @return [Array<Gitlab::ObjectifiedHash>]
-    def fetch_failures(date:, state:)
+    def fetch_all_new_failures(date:, state:)
+      project_ids = [GITLAB_PROJECT_ID, CUSTOMERSDOT_PROJECT_ID]
+      failures = []
+
+      project_ids.each do |project_id|
+        failures += fetch_new_failures(project_id: project_id, date: date, state: state)
+      end
+
+      failures
+    end
+
+    # Fetch new failures for a given project
+    #
+    # @param [Integer] project_id
+    # @param [String] date
+    # @param [String] state
+    # @return [Array<Gitlab::ObjectifiedHash>]
+    def fetch_new_failures(project_id:, date:, state:)
       gitlab.issues(
-        GITLAB_PROJECT_ID,
-        labels: "failure::new",
+        project_id,
+        labels: FAILURE_NEW,
         order_by: "updated_at",
         state: state,
         scope: "all",
@@ -149,10 +194,11 @@ module Dri
 
     # Fetch failure notes
     #
+    # @param [Integer] project_id
     # @param [Integer] issue_iid
     # @return [Array<Gitlab::ObjectifiedHash>]
-    def fetch_failure_notes(issue_iid)
-      gitlab.issue_notes(GITLAB_PROJECT_ID, issue_iid, per_page: 100).auto_paginate
+    def fetch_failure_notes(project_id, issue_iid)
+      gitlab.issue_notes(project_id, issue_iid, per_page: 100).auto_paginate
     end
 
     # Delete award emoji
@@ -233,6 +279,28 @@ module Dri
       gitlab.pipeline_jobs(project_id, pipeline_id, options).auto_paginate
     end
 
+    # Fetch runbooks from the runbooks project
+    #
+    # @param [Integer] project_id
+    # @return [Array<Gitlab::ObjectifedHash>]
+    def list_runbooks(project_id = RUNBOOKS_PROJECT_ID)
+      tree = gitlab.tree(project_id, { recursive: true, ref: 'main' }).auto_paginate
+
+      tree.select do |node|
+        node.type == 'tree' && !node.name.start_with?('_')
+      end
+    end
+
+    # Fetches file contents at <path>
+    #
+    # @param [String] path
+    # @param [String] ref
+    # @param [Integer] project_id
+    # @return [Gitlab::ObjectifiedHash]
+    def get_file(path, ref:, project_id:)
+      gitlab.get_file(project_id, path, ref)
+    end
+
     private
 
     attr_reader :token, :ops_token

data/lib/dri/commands/analyze/stack_traces.rb

@@ -2,6 +2,7 @@
 
 require_relative '../../command'
 require_relative '../../utils/table'
+require_relative '../../utils/constants'
 require 'amatch'
 require 'fileutils'
 
@@ -11,10 +12,11 @@ module Dri
       class StackTraces < Dri::Command
         include Amatch
         include Dri::Utils::Table
+        include Dri::Utils::Constants::Triage::Labels
 
         def initialize(options)
           @options = options
-          @labels = options[:labels] || 'failure::new'
+          @labels = options[:labels] || FAILURE_NEW
           @similarity_score_threshold = options[:similarity_score_threshold] || 0.9
         end
 
@@ -29,7 +31,7 @@ module Dri
             logger.info "#{Time.now.utc} API response completed"
 
             if response.empty?
-              logger.info 'There are no failure::new issues identified!'
+              logger.info "There are no #{FAILURE_NEW} issues identified!"
               exit 0
             end
 

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

@@ -2,12 +2,14 @@
 
 require_relative '../../command'
 require_relative '../../utils/table'
+require_relative '../../utils/constants'
 
 module Dri
   module Commands
     class Fetch
       class Failures < Dri::Command
         include Dri::Utils::Table
+        include Dri::Utils::Constants::Triage::Labels
         using Refinements
 
         SORT_BY_OPTIONS = {
@@ -34,33 +36,34 @@ module Dri
           author = add_color('Author', :bright_yellow)
           url = add_color('URL', :bright_yellow)
 
-          failures = []
+          sorted_failures = []
           labels = [title, triaged, environment, author, url]
           triaged_counter = 0
 
           logger.info "Fetching today's failures..."
 
           spinner.run do # rubocop:disable Metrics/BlockLength
-            response = api_client.fetch_failures(date: @today_iso_format, state: 'opened')
+            failures = api_client.fetch_all_new_failures(date: @today_iso_format, state: 'opened')
 
-            if response.empty?
+            if failures.empty?
               logger.info 'Life is great, there are no new failures today!'
               exit 0
             end
 
-            response.each do |failure|
+            failures.each do |failure|
+              project_id = failure.project_id
               title = failure.title.truncate(60)
               author = failure.to_h.dig('author', 'username')
               url = failure.web_url
               triaged = add_color('x', :red)
-              envs = failure.labels.select { |l| l.include?('found:') }.map do |l|
+              envs = failure.labels.select { |l| l.include?(FOUND) }.map do |l|
                 env = l.split(':').last.gsub('.gitlab.com', '')
 
                 env == 'gitlab.com' ? 'production' : env
               end
               urgent = urgent_environments.all? { |env| envs.include?(env) }
 
-              emoji_awards = api_client.fetch_awarded_emojis(failure.iid).find do |e|
+              emoji_awards = api_client.fetch_awarded_emojis(failure.iid, project_id: project_id).find do |e|
                 e.name == emoji && e.to_h.dig('user', 'username') == username
               end
 
@@ -70,26 +73,26 @@ module Dri
               end
 
               if @options[:urgent]
-                failures << [title, triaged, envs.first, author, url] if urgent
+                sorted_failures << [title, triaged, envs.first, author, url] if urgent
               else
-                failures << [title, triaged, envs.first, author, url]
+                sorted_failures << [title, triaged, envs.first, author, url]
               end
             end
 
-            failures.sort_by! { |failure| failure[SORT_BY_OPTIONS[@options[:sort_by]&.to_sym || :environment]] }
+            sorted_failures.sort_by! { |failure| failure[SORT_BY_OPTIONS[@options[:sort_by]&.to_sym || :environment]] }
           end
 
           msg = if @options[:urgent]
                   <<~MSG
-                    Found: #{failures.size} urgent failures, occurring in both canary.gitlab.com and canary.staging.gitlab.com.
+                    Found: #{sorted_failures.size} urgent failures, occurring in both canary.gitlab.com and canary.staging.gitlab.com.
                   MSG
                 else
                   <<~MSG
-                    Found: #{failures.size} failures, of these #{triaged_counter} have been triaged with a #{emoji}.
+                    Found: #{sorted_failures.size} failures, of these #{triaged_counter} have been triaged with a #{emoji}.
                   MSG
                 end
 
-          print_table(labels, failures, alignments: [:left, :center, :center, :left])
+          print_table(labels, sorted_failures, alignments: [:left, :center, :center, :left])
           output.puts(msg)
         end
       end

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

@@ -2,7 +2,7 @@
 
 require_relative '../../command'
 require_relative '../../utils/table'
-require_relative '../../utils/feature_flag_consts'
+require_relative '../../utils/constants'
 require_relative '../../feature_flag_report'
 
 module Dri
@@ -10,7 +10,7 @@ module Dri
     class Fetch
       class FeatureFlags < Dri::Command
         include Dri::Utils::Table
-        include Dri::Utils::FeatureFlagConsts
+        include Dri::Utils::Constants::FeatureFlag
 
         def initialize(options)
           @options = options

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

@@ -10,6 +10,7 @@ module Dri
     class Fetch
       class Pipelines < Dri::Command # rubocop:disable Metrics/ClassLength
         include Dri::Utils::Table
+        include Dri::Utils::Constants
         using Refinements
 
         NUM_OF_TESTS_LIVE_ENV = 1000
@@ -27,7 +28,7 @@ module Dri
           table_labels = define_table_labels
 
           spinner.run do
-            Dri::Utils::Constants::PIPELINE_ENVIRONMENTS.each do |environment, details|
+            PIPELINE_ENVIRONMENTS.each do |environment, details|
               logger.info "Fetching last executed #{environment} pipeline"
               pipelines << fetch_pipeline(pipeline_name: environment.to_s, details: details)
               logger.info "Fetching complete for #{environment} ✓"

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

@@ -2,12 +2,14 @@
 
 require_relative '../../command'
 require_relative '../../utils/table'
+require_relative '../../utils/constants'
 
 module Dri
   module Commands
     class Fetch
       class Quarantines < Dri::Command
         include Dri::Utils::Table
+        include Dri::Utils::Constants::Triage::Labels
         using Refinements
 
         def initialize(options, search: '[QUARANTINE]')
@@ -30,7 +32,7 @@ module Dri
           spinner.run do
             response = api_client.fetch_mrs(
               project_id: 'gitlab-org/gitlab',
-              labels: 'QA,Quality',
+              labels: "#{QA},#{QUALITY}",
               search: @search,
               in: :title,
               state: :opened

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

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+
+require_relative '../../command'
+require_relative '../../utils/constants'
+require 'tty-markdown'
+require 'tty-prompt'
+require 'base64'
+
+module Dri
+  module Commands
+    class Fetch
+      class Runbooks < Dri::Command
+        def initialize(options)
+          @options = options
+        end
+
+        def execute(folder: nil, input: $stdin, output: $stdout)
+          return fetch_summary(output: output) unless folder
+
+          fetch_runbook("#{folder}/index.md", output: output)
+        end
+
+        def fetch_summary(output:)
+          logger.info 'Fetching runbooks'
+          prompt = TTY::Prompt.new
+          runbooks = api_client.list_runbooks
+
+          return output.puts 'No runbooks could be found' if runbooks.empty?
+
+          choices = runbooks.map(&:path)
+          path = prompt.select('Which runbook would you like to look at?', choices)
+
+          fetch_runbook("#{path}/index.md", output: output)
+        end
+
+        def fetch_runbook(path, output:)
+          runbook = nil
+          spinner.run do
+            runbook = api_client.get_file(
+              path,
+              ref: 'main',
+              project_id: Utils::Constants::ProjectIDs::RUNBOOKS_PROJECT_ID
+            )
+          rescue Gitlab::Error::NotFound
+            logger.error "The runbook #{path} could not be found."
+          end
+
+          return unless runbook
+
+          content = Base64.decode64(runbook.content)
+          markdown(content, output)
+        end
+
+        private
+
+        def markdown(str, io)
+          io.puts TTY::Markdown.parse(str, symbols: :ascii)
+        end
+      end
+    end
+  end
+end

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

@@ -2,12 +2,14 @@
 
 require_relative '../../command'
 require_relative '../../utils/table'
+require_relative '../../utils/constants'
 
 module Dri
   module Commands
     class Fetch
       class Triaged < Dri::Command
         include Dri::Utils::Table
+        include Dri::Utils::Constants::Triage::Labels
         using Refinements
 
         def initialize(options)
@@ -27,21 +29,21 @@ module Dri
           logger.info "Fetching your triaged failures..."
           spinner.start
 
-          response = api_client.fetch_triaged_failures(emoji: emoji, state: 'opened')
+          failures = api_client.fetch_all_triaged_failures(emoji: emoji, state: 'opened')
 
-          if response.empty?
+          if failures.empty?
             logger.info "There are no failures triaged yet with #{add_color(emoji, :black, :on_white)}."
             exit 0
           end
 
-          response.each do |triaged|
+          failures.each do |triaged|
             title = triaged["title"].truncate(70)
             url = triaged["web_url"]
             labels = triaged["labels"]
             type = ""
 
             labels.each do |label|
-              type = label.gsub!('failure::', ' ').to_s if label.include? "failure::"
+              type = label.gsub!(FAILURE, ' ').to_s if label.include? FAILURE
             end
 
             failures_triaged << [title, url, type]

data/lib/dri/commands/fetch.rb

@@ -94,6 +94,17 @@ module Dri
         require_relative 'fetch/pipelines'
         Dri::Commands::Fetch::Pipelines.new(options).execute
       end
+
+      desc 'runbooks', 'Fetch runbooks'
+      method_option :help, aliases: '-h', type: :boolean,
+                           desc: 'Display usage information'
+
+      def runbooks(*args)
+        return invoke :help, %w[runbooks] if options[:help]
+
+        require_relative 'fetch/runbooks'
+        Dri::Commands::Fetch::Runbooks.new(options).execute(folder: args.first)
+      end
     end
   end
 end

data/lib/dri/commands/incidents.rb

@@ -2,11 +2,13 @@
 
 require_relative '../command'
 require_relative '../utils/table'
+require_relative '../utils/constants'
 
 module Dri
   module Commands
     class Incidents < Dri::Command
       include Dri::Utils::Table
+      include Dri::Utils::Constants::Triage::Labels
       using Refinements
 
       def initialize(options)
@@ -42,8 +44,8 @@ module Dri
             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::"
+              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]

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

@@ -2,7 +2,7 @@
 
 require_relative '../../command'
 require_relative '../../utils/markdown_lists'
-require_relative '../../utils/feature_flag_consts'
+require_relative '../../utils/constants'
 require_relative '../../report'
 require_relative '../../feature_flag_report'
 
@@ -14,7 +14,7 @@ module Dri
   module Commands
     class Publish
       class Report < Dri::Command # rubocop:disable Metrics/ClassLength
-        include Dri::Utils::FeatureFlagConsts
+        include Dri::Utils::Constants::FeatureFlag
 
         def initialize(options)
           @options = options
@@ -32,13 +32,13 @@ module Dri
 
           spinner.start
 
-          issues = api_client.fetch_triaged_failures(emoji: emoji, state: 'opened')
+          failures = api_client.fetch_all_triaged_failures(emoji: emoji, state: 'opened')
           incidents = api_client.fetch_triaged_incidents(emoji: emoji)
 
           spinner.stop
 
-          if issues.empty? && incidents.empty?
-            logger.warn "Found no issues nor incidents associated with \"#{emoji}\" emoji. Will exit. Bye 👋"
+          if failures.empty? && incidents.empty?
+            logger.warn "Found no failures nor incidents associated with \"#{emoji}\" emoji. Will exit. Bye 👋"
             exit 1
           end
 
@@ -58,18 +58,18 @@ module Dri
 
           spinner.start
 
-          issues.each do |issue|
+          failures.each do |failure|
             actions = []
 
             if @options[:actions]
               actions = prompt.multi_select(
-                "Please mark the actions on #{add_color(issue.title, :yellow)}: ",
+                "Please mark the actions on #{add_color(failure.title, :yellow)}: ",
                 action_options,
                 per_page: 9
               )
             end
 
-            report.add_failure(issue, actions)
+            report.add_failure(failure, actions)
           end
 
           unless incidents.empty?
@@ -180,11 +180,29 @@ module Dri
           issues = api_client.fetch_current_triage_issue
           current_issue_iid = issues.first.iid
 
-          api_client.post_triage_report_note(iid: current_issue_iid, body: note)
+          note_action = 'posted'
+          posted_note = nil
+          if @options[:update]
+            report_file = File.read("handover_reports/.tmp/report-#{@date}.json")
+            report_json = JSON.parse(report_file) if report_file
+            posted_note = api_client.update_triage_report_note(
+              iid: current_issue_iid, note_id: report_json['id'], body: note
+            )
+            note_action = 'updated'
+          else
+            posted_note = api_client.post_triage_report_note(iid: current_issue_iid, body: note)
+
+            FileUtils.mkdir_p("#{Dir.pwd}/handover_reports/.tmp")
+            report_path = "handover_reports/.tmp/report-#{@date}.json"
+
+            File.open(report_path, 'w') do |out_file|
+              out_file.write(posted_note.to_h.to_json)
+            end
+          end
 
           output.puts "Done! ✅\n"
           logger.success(<<~MSG)
-            Thanks @#{username}, your report was posted at https://gitlab.com/gitlab-org/quality/pipeline-triage/-/issues/#{current_issue_iid} 🎉
+            Thanks @#{username}, your report was #{note_action} at https://gitlab.com/gitlab-org/quality/pipeline-triage/-/issues/#{current_issue_iid}#note_#{posted_note.id} 🎉
           MSG
         end
 

data/lib/dri/commands/publish.rb

@@ -16,6 +16,9 @@ module Dri
                               desc: 'Updates actions on failures'
       method_option :feature_flags, type: :boolean,
                                     desc: 'Adds summary of feature flag changes'
+      method_option :update, aliases: '-u', type: :boolean,
+                             desc: 'Updates an existing report'
+
       def report(*)
         if options[:help]
           invoke :help, ['report']

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

@@ -24,7 +24,7 @@ module Dri
 
           spinner.start
 
-          failures_with_award_emoji = api_client.fetch_triaged_failures(emoji: emoji, state: 'opened')
+          failures_with_award_emoji = api_client.fetch_all_triaged_failures(emoji: emoji, state: 'opened')
           incidents_with_award_emoji = api_client.fetch_triaged_incidents(emoji: emoji)
 
           issues_with_award_emoji = failures_with_award_emoji + incidents_with_award_emoji

data/lib/dri/feature_flag_report.rb

@@ -1,10 +1,10 @@
 # frozen_string_literal: true
 
-require_relative './utils/feature_flag_consts'
+require_relative './utils/constants'
 
 module Dri
   class FeatureFlagReport
-    include Dri::Utils::FeatureFlagConsts
+    include Dri::Utils::Constants::FeatureFlag::Labels
 
     attr_reader :header, :labels, :prod, :staging, :staging_ref, :preprod
 

data/lib/dri/report.rb

@@ -1,7 +1,10 @@
 # frozen_string_literal: true
 
+require_relative './utils/constants'
+
 module Dri
   class Report # rubocop:disable Metrics/ClassLength
+    include Dri::Utils::Constants::Triage::Labels
     using Refinements
 
     attr_reader :header, :failures, :labels, :labels_incidents, :incidents
@@ -31,24 +34,23 @@ module Dri
       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::"
+        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
 
     def add_failure(failure, actions_opts = [])
+      project_id = failure["project_id"]
       iid = failure["iid"]
       title = format_title(failure["title"])
       link = failure["web_url"]
       labels = failure["labels"]
-      created_at = failure["created_at"]
-      assignees = failure["assignees"]
       description = failure["description"]
 
-      related_mrs = @api_client.fetch_related_mrs(iid)
-      emoji = classify_failure_emoji(created_at)
+      related_mrs = @api_client.fetch_related_mrs(project_id, iid)
+      emoji = classify_failure_emoji(failure["created_at"])
       emojified_link = "#{emoji} #{link}"
 
       stack_blob = if description.empty?
@@ -60,10 +62,10 @@ module Dri
       stack_trace = ":link:[`#{stack_blob}...`](#{link}#stack-trace)"
 
       failure_type = filter_failure_type_labels(labels)
-      assigned_status = assigned?(assignees)
+      assigned_status = assigned?(failure["assignees"])
       pipelines = filter_pipeline_labels(labels)
 
-      linked_pipelines = link_pipelines(iid, pipelines, description)
+      linked_pipelines = link_pipelines(project_id, iid, pipelines, description)
 
       actions_status = actions_status_template(failure_type, assigned_status, actions_opts)
       actions_fixes = actions_fixes_template(related_mrs)
@@ -73,7 +75,7 @@ module Dri
 
     private
 
-    def link_pipelines(iid, pipelines, description) # rubocop:disable Metrics/CyclomaticComplexity, Metrics/PerceivedComplexity, Metrics/MethodLength
+    def link_pipelines(project_id, iid, pipelines, description) # rubocop:disable Metrics/CyclomaticComplexity, Metrics/PerceivedComplexity, Metrics/MethodLength
       linked = []
       label_pipeline_map = {
         'gitlab.com' => '/quality/production',
@@ -88,7 +90,7 @@ module Dri
         'release' => '/quality/release'
       }
 
-      failure_notes = @api_client.fetch_failure_notes(iid)
+      failure_notes = @api_client.fetch_failure_notes(project_id, iid)
 
       return if pipelines.empty?
 
@@ -173,10 +175,10 @@ module Dri
       pipelines = []
 
       labels.each do |label|
-        matchers = { 'found:' => ' ' }
+        matchers = { FOUND => ' ' }
 
-        if label.include? "found:"
-          pipeline = label.gsub(/found:/) { |match| matchers[match] }
+        if label.include? FOUND
+          pipeline = label.gsub(/#{FOUND}/o) { |match| matchers[match] }
           pipelines << pipeline.strip
         end
       end
@@ -185,7 +187,7 @@ module Dri
 
     def filter_failure_type_labels(labels)
       labels.each do |label|
-        @type = label.gsub!('failure::', ' ').to_s if label.include? "failure::"
+        @type = label.gsub!(FAILURE, ' ').to_s if label.include? FAILURE
       end
       @type
     end

data/lib/dri/utils/constants.rb

@@ -3,6 +3,40 @@
 module Dri
   module Utils
     module Constants
+      module ProjectIDs
+        TESTCASES_PROJECT_ID        = 11229385
+        TRIAGE_PROJECT_ID           = 15291320
+        GITLAB_PROJECT_ID           = 278964
+        CUSTOMERSDOT_PROJECT_ID     = 2670515
+        FEATURE_FLAG_LOG_PROJECT_ID = 15208716
+        INFRA_TEAM_PROD_PROJECT_ID  = 7444821
+        RUNBOOKS_PROJECT_ID         = 41045213
+      end
+
+      module Triage
+        module Labels
+          FOUND = 'found:'
+          FAILURE = 'failure::'
+          FAILURE_NEW = "#{FAILURE}new"
+          INCIDENT = 'Incident::'
+          SERVICE = 'Service::'
+          QA = 'QA'
+          QUALITY = 'Quality'
+          QUARANTINE = 'quarantine'
+        end
+      end
+
+      module FeatureFlag
+        module Labels
+          PRODUCTION  = 'host::gitlab.com'
+          STAGING     = 'host::staging.gitlab.com'
+          STAGING_REF = 'host::staging-ref.gitlab.com'
+          PREPROD     = 'host::pre.gitlab.com'
+        end
+
+        TITLE_SUBSTRINGS = ["set to \"true\"", "set to \"false\""].freeze
+      end
+
       PIPELINE_ENVIRONMENTS =
         {
           production: {

data/lib/dri/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Dri
-  VERSION = "0.7.0"
+  VERSION = "0.9.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: dri
 version: !ruby/object:Gem::Version
-  version: 0.7.0
+  version: 0.9.0
 platform: ruby
 authors:
 - GitLab Quality
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2022-09-12 00:00:00.000000000 Z
+date: 2023-03-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: amatch
@@ -30,28 +30,28 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '4.18'
+        version: '4.19'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '4.18'
+        version: '4.19'
 - !ruby/object:Gem::Dependency
   name: httparty
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.20.0
+        version: 0.21.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.20.0
+        version: 0.21.0
 - !ruby/object:Gem::Dependency
   name: json
   requirement: !ruby/object:Gem::Requirement
@@ -100,14 +100,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.0.1
+        version: 1.2.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.0.1
+        version: 1.2.0
 - !ruby/object:Gem::Dependency
   name: tty-box
   requirement: !ruby/object:Gem::Requirement
@@ -128,28 +128,28 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.4.0
+        version: 0.6.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.4.0
+        version: 0.6.0
 - !ruby/object:Gem::Dependency
   name: tty-editor
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.6'
+        version: '0.7'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.6'
+        version: '0.7'
 - !ruby/object:Gem::Dependency
   name: tty-font
   requirement: !ruby/object:Gem::Requirement
@@ -178,6 +178,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 0.6.0
+- !ruby/object:Gem::Dependency
+  name: tty-markdown
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.7.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.7.0
 - !ruby/object:Gem::Dependency
   name: tty-prompt
   requirement: !ruby/object:Gem::Requirement
@@ -315,6 +329,7 @@ files:
 - ".editorconfig"
 - ".gitignore"
 - ".gitlab-ci.yml"
+- ".gitlab/merge_request_templates/Release.md"
 - ".rspec"
 - ".rubocop.yml"
 - ".ruby-version"
@@ -339,6 +354,7 @@ files:
 - lib/dri/commands/fetch/featureflags.rb
 - lib/dri/commands/fetch/pipelines.rb
 - lib/dri/commands/fetch/quarantines.rb
+- lib/dri/commands/fetch/runbooks.rb
 - lib/dri/commands/fetch/testcases.rb
 - lib/dri/commands/fetch/triaged.rb
 - lib/dri/commands/incidents.rb
@@ -355,7 +371,6 @@ files:
 - lib/dri/refinements/truncate.rb
 - lib/dri/report.rb
 - lib/dri/utils/constants.rb
-- lib/dri/utils/feature_flag_consts.rb
 - lib/dri/utils/markdown_lists.rb
 - lib/dri/utils/table.rb
 - lib/dri/version.rb

data/lib/dri/utils/feature_flag_consts.rb

@@ -1,13 +0,0 @@
-# frozen_string_literal: true
-
-module Dri
-  module Utils
-    module FeatureFlagConsts
-      PRODUCTION  = 'host::gitlab.com'
-      STAGING     = 'host::staging.gitlab.com'
-      STAGING_REF = 'host::staging-ref.gitlab.com'
-      PREPROD     = 'host::pre.gitlab.com'
-      TITLE_SUBSTRINGS = ["set to \"true\"", "set to \"false\""].freeze
-    end
-  end
-end