VulnAPI Implementation

2019-07-18 20:32:15 +01:00
parent 84422b10c8
commit 463e77f0a5
60 changed files with 1126 additions and 223 deletions
--- a/app/models/theme.rb
+++ b/app/models/theme.rb
@@ -21,9 +21,16 @@ module WPScan
        parse_style
      end

+      # Retrieve the metadata from the vuln API if available (and a valid token is given),
+      # or the local metadata db otherwise
      # @return [ JSON ]
      def metadata
-        @metadata ||= DB::Theme.metadata_at(slug)
+        @metadata ||= db_data.empty? ? DB::Theme.metadata_at(slug) : db_data
+      end
+
+      # @return [ Hash ]
+      def db_data
+        @db_data ||= DB::VulnApi.theme_data(slug)
      end

      # @param [ Hash ] opts