Prototype for grade reports

mozilla · Jun 14, 2017 · 491d52e · 491d52e
1 parent 293dc72
commit 491d52e
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 1 deletion.
diff --git a/lib/ssh_scan_api/api.rb b/lib/ssh_scan_api/api.rb
@@ -195,7 +195,8 @@ class API < Sinatra::Base
           "RUNNING" => settings.db.run_count,
           "ERRORED" => settings.db.error_count,
           "COMPLETED" => settings.db.complete_count,
-          "TOTAL" => settings.db.total_count
+          "TOTAL" => settings.db.total_count,
+          "GRADE_REPORT" => settings.db.grade_report
         }.to_json
       end
 

diff --git a/lib/ssh_scan_api/database.rb b/lib/ssh_scan_api/database.rb
@@ -53,6 +53,10 @@ def get_scan(uuid)
       @database.get_scan(uuid)
     end
 
+    def grade_report
+      @database.grade_report
+    end
+
     def queue_scan(uuid, socket)
       @database.queue_scan(uuid, socket)
     end

diff --git a/lib/ssh_scan_api/database/mongo.rb b/lib/ssh_scan_api/database/mongo.rb
@@ -84,6 +84,17 @@ def error_scan(uuid, worker_id, result)
         )
       end
 
+      def grade_report
+        grades = ["A", "B", "C", "D", "F"]
+        histogram = {}
+
+        grades.each do |grade|
+          histogram[grade] = @collection.count("scan.compliance.grade": grade)
+        end
+
+        return histogram
+      end
+
       def next_scan_in_queue()
         @collection.find(status: "QUEUED").first
       end