summaryrefslogtreecommitdiff
path: root/lib/gitlab/ci/reports/security/reports.rb
blob: a7a6e5b259328a6bac21eeaeae9a99ba66ae15b7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
# frozen_string_literal: true

module Gitlab
  module Ci
    module Reports
      module Security
        class Reports
          attr_reader :reports, :pipeline

          delegate :each, :empty?, to: :reports

          def initialize(pipeline)
            @reports = {}
            @pipeline = pipeline
          end

          def get_report(report_type, report_artifact)
            reports[report_type] ||= Report.new(report_type, pipeline, report_artifact.created_at)
          end

          def findings
            reports.values.flat_map(&:findings)
          end

          def violates_default_policy_against?(target_reports, vulnerabilities_allowed)
            unsafe_findings_count(target_reports) > vulnerabilities_allowed
          end

          private

          def findings_diff(target_reports)
            findings - target_reports&.findings.to_a
          end

          def unsafe_findings_count(target_reports)
            findings_diff(target_reports).count(&:unsafe?)
          end
        end
      end
    end
  end
end