routes.py

from . import app
import sqlite3
from flask import request

class InterceptRequestMiddleware:
    def __init__(self, wsgi_app):
        self.wsgi_app = wsgi_app

    def __call__(self, environ, start_response):
        environ['Access-Control-Allow-Origin'] = '*'
        return self.wsgi_app(environ, start_response)





@app.route("/test", methods=["GET"])
def start():
    return "hello"





@app.route("/api/network", methods=["GET"])
def getNetworkTraffic():
    connection = sqlite3.connect("Simple-Network-Visualiser/analysis.db")
    cursor = connection.cursor()
    cursor.execute("SELECT * FROM data;")
    return cursor.fetchall()


@app.route("/api/nodes", methods=["GET"])
def getNodes():
        minTime = request.args.get("minTime") 
        maxTime = request.args.get("maxTime")
        
        query = "SELECT DISTINCT d.source, d.destination, dns.common_name FROM data d LEFT JOIN dns_records dns ON d.destination = dns.ip_address;"

        if request.args.get("minTime") != None and request.args.get("maxTime") != None:
            maxTime = round(float(maxTime) * 1000000)
            minTime = round(float(minTime) * 1000000)
            query = "SELECT DISTINCT d.source, d.destination, dns.common_name FROM data d LEFT JOIN dns_records dns ON d.destination = dns.ip_address WHERE time >=" + str(minTime) +" AND time <= " + str(maxTime) +" ;"

        connection = sqlite3.connect("Simple-Network-Visualiser/analysis.db")
        cursor = connection.cursor()
        cursor.execute(query)
        return cursor.fetchall()


@app.route("/api/time", methods=["GET"])
def get_time_range():
    connection = sqlite3.connect("Simple-Network-Visualiser/analysis.db")
    cursor = connection.cursor()
    cursor.execute("SELECT MIN(time) as smallest, MAX(time) as largest FROM data;")
    return cursor.fetchall()