Added Stop Scan functionality

Dashboard/app.py

@@ -31,7 +31,7 @@ def index():
     queued = queuedScans.find()
     running = runningScans.find()
     completed = completedScans.find()
-    return render_template('index.html', prequeued=prequeued, queued=queued, running=running, completed=completed, newScanUrl=url_for('newScan'))
+    return render_template('index.html', prequeued=prequeued, queued=queued, running=running, completed=completed, newScanUrl=url_for('newScan'),stopscan=url_for('stopscan',id='<id>'))
 
 
 def new_scan():
@@ -64,6 +64,15 @@ def upload_files():
 
     return redirect(url_for('dashboard'))
 
+def stopscan(id):
+    if request.method == 'GET': 
+        print(id)
+        try:
+            runningScans.delete_one({'_id':id})
+        except:
+            print("Error")
+
+    return redirect(url_for('dashboard'))
 
 def progress():
     def generate():
@@ -75,7 +84,6 @@ def generate():
                     (item["files"]["completed"] / item["files"]["total"]) * 100)
 
             yield "data:" + json.dumps(x) + "\n\n"
-
     return Response(generate(), mimetype='text/event-stream')
 
 
@@ -86,5 +94,7 @@ def generate():
                  view_func=progress, methods=['GET'])
 app.add_url_rule("/upload", endpoint="upload",
                  view_func=upload_files, methods=['GET', 'POST'])
+app.add_url_rule("/stopscan/<id>", endpoint="stopscan",
+                 view_func=stopscan, methods=['GET'])
 if __name__ == "__main__":
     app.run(host="0.0.0.0",debug=True)

Dashboard/templates/index.html

@@ -78,6 +78,7 @@ <h6>Number of files: {{ item['files']['total'] }}</h6>
                     <div class="progress">
                         <div id={{ item["_id"] }} class="progress-bar" style="width: 0%;"></div>
                     </div>
+                    <button type="button" name="button" class="btn btn-danger mt-3" onclick="stopscan('{{ item['_id'] }}')">Stop</button>
                 </div>
             </div>
             {% endfor %}
@@ -109,6 +110,13 @@ <h6>Number of files: {{ item['files']['total'] }}</h6>
         function openUploader() {
             window.location = "{{ newScanUrl }}";
         }
+
+        async function stopscan(id){
+            let to_do = confirm("Do you want to stop scan")
+            if(to_do){
+                await fetch(`http://localhost:5000/stopscan/${id}`)
+            }
+        }
     </script>
 
     <script>