spinner to let users know scan status

Author: lalithr95

app/assets/images/spinner.svg

@@ -1,8 +1,9 @@
 $(function () {
+  var loaded = false;
+  var sid = $('#vulnerability-container').data('scan');
   fetch_analytics();
 
   function fetch_analytics() {
-    var sid = $('#vulnerability-container').data('scan')
     if (sid == undefined) {
       return;
     }

app/assets/javascripts/scans.js

@@ -8,12 +8,21 @@ def index
 
   def show
     @vulnerabilities ||= @scan.vulnerabilities
+    respond_to do |format|
+      format.html
+      format.json {
+        render json: {
+          scan: @scan,
+          vulnerabilities: @vulnerabilities
+        }
+      }
+    end
   end
 
   def create
     url = params[:url]
     parameters = params[:parameters]
-    method = params[:method].nil? ? "" : JSON.generate(params[:method])
+    method = params[:method].nil? ? JSON.generate(['GET']) : JSON.generate(params[:method])
     headers = params[:headers]
     json = params[:json]
 

app/controllers/scans_controller.rb

@@ -5,6 +5,7 @@
         <th>#</th>
         <th>URL</th>
         <th>Vulnerability count</th>
+        <th>Status</th>
         <th></th>
       </tr>
     </thead>
@@ -14,6 +15,7 @@
       <td><%= link_to scan_path(scan.id), scan_path(scan.id) %></td>
       <td><%= link_to scan.url, scan.url %></td>
       <td><%= scan.vulnerabilities.count %></td>
+      <td><%= scan.status %></td>
       <td><%= link_to "Delete scan", scan_path(scan), method: :delete, class: 'btn btn-danger' %>
     </tr>
     <% end %>

app/views/scans/index.html.erb

@@ -1,8 +1,11 @@
-<div class=" col-md-12">
+<div class="col-md-12">
   <%= render partial: 'details', locals: { scan: @scan } %>
 
   <div class="card card-outline-secondary text-xs-center">
     <div class="card-block">
+      <div id="spinner" style="display: none;">
+        <%= image_tag 'spinner.svg' %>
+      </div>
       <table class="table">
         <thead>
           <tr>
@@ -34,3 +37,25 @@
     </div>
   </div>
 </div>
+
+<%= javascript_tag do %>
+  <% if @scan.status == 'processing' %>
+    function fetch_scan() {
+      $.ajax({
+        url: '/scans/' + sid + '.json',
+        method: 'GET',
+        beforeSend: function() {
+          $('#spinner').show();
+        },
+        success: function(data) {
+          var status = data['scan']['status'];
+          if (status == 'completed') {
+            location.reload();
+          }
+        }
+      });
+    }
+    var sid = $('#vulnerability-container').data('scan');
+    setInterval(fetch_scan, 3000);
+  <% end %>
+<% end %>

app/views/scans/show.html.erb

@@ -27,6 +27,7 @@ def perform(scan_id, options = {})
         end
       end
     end
+    scan.update_attributes(status: 'completed')
   rescue ActiveRecord::RecordNotFound => e
     Rails.logger.info e.message
   end

app/workers/scan_vulnerability_worker.rb

@@ -0,0 +1,9 @@
+class AddStatusToScans < ActiveRecord::Migration
+  def up
+    add_column :scans, :status, :string, default: 'processing'
+  end
+
+  def down
+    remove_column :scans, :status
+  end
+end

db/migrate/20161014024647_add_status_to_scans.rb

@@ -11,18 +11,19 @@
 #
 # It's strongly recommended that you check this file into your version control system.
 
-ActiveRecord::Schema.define(version: 20161011064845) do
+ActiveRecord::Schema.define(version: 20161014024647) do
 
   create_table "scans", force: :cascade do |t|
     t.string   "url"
     t.string   "sid"
     t.string   "parameters"
     t.string   "method"
     t.string   "cookies"
-    t.datetime "created_at",                 null: false
-    t.datetime "updated_at",                 null: false
+    t.datetime "created_at",                        null: false
+    t.datetime "updated_at",                        null: false
     t.boolean  "json",       default: false
     t.integer  "user_id"
+    t.string   "status",     default: "processing"
   end
 
   add_index "scans", ["user_id"], name: "index_scans_on_user_id"