diff --git a/app/models/cisa_known_exploit.rb b/app/models/cisa_known_exploit.rb index 27e0b2c..4ae7e21 100644 --- a/app/models/cisa_known_exploit.rb +++ b/app/models/cisa_known_exploit.rb @@ -12,4 +12,12 @@ class CisaKnownExploit < ActiveRecord::Base def self.cves_from_year(year) last.vulnerabilities.select { |vuln| vuln if vuln.dig('cve_id') =~ /CVE-#{year}-\d{4,7}/ } end + + def self.by_product(product_name) + last.vulnerabilities.select {|vuln| vuln if vuln.dig('product') == product_name } + end + + def self.by_due_date(due_date) + last.vulnerabilities.select {|vuln| vuln if vuln.dig('due_date') == due_date } + end end