import pyautogui
import time
import os
import subprocess
import datetime

BURP_JAR = "/headless/burpsuite_community.jar"
CONFIG_FILE = "/app/burp/project_options.json"

def start_burp():
    os.system("rm -rf /tmp/burp*")
    burp_process = subprocess.Popen([
        "java", "-jar", BURP_JAR,
        f"--config-file={CONFIG_FILE}"
    ])
    return burp_process

time.sleep(5)

print("Starting Burp Suite...")
burp_process = start_burp()
end_time = datetime.datetime.now() + datetime.timedelta(days=1)
button = None
while True:
    if datetime.datetime.now() > end_time:
        print("Burp Suite has been running for 24 hours, restarting...")
        burp_process.terminate()
        time.sleep(1)
        burp_process = start_burp()
        end_time = datetime.datetime.now() + datetime.timedelta(days=1)
    try:
        button = pyautogui.locateCenterOnScreen("/app/burp/next_button.png", confidence=0.8)
    except:
        pass
    if button:
        print("Clicking on the 'Next' button...")
        pyautogui.click(button)
        button = None
    
    try:
        button = pyautogui.locateCenterOnScreen("/app/burp/start_burp.png", confidence=0.8)
    except:
        pass
    if button:
        print("Clicking on the 'Start Burp' button...")
        pyautogui.click(button)
        button = None
     
    try:   
        button = pyautogui.locateCenterOnScreen("/app/burp/accept.png", confidence=0.8)
    except:
        pass
    if button:
        print("Clicking on the 'Accept' button...")
        pyautogui.click(button)
        button = None
    
    try:
        button = pyautogui.locateCenterOnScreen("/app/burp/proxy.png", confidence=0.8)
    except:
        pass
    if button:
        print("Clicking on the 'Proxy' button...")
        pyautogui.click(button)
        button = None
        
        try:
            button = pyautogui.locateCenterOnScreen("/app/burp/http_history.png", confidence=0.8)
        except:
            pass
        if button:
            print("Clicking on the 'HTTP History' button...")
            pyautogui.click(button)
            time.sleep(2)
            try:
                button = pyautogui.locateCenterOnScreen("/app/burp/sort.png", confidence=0.8)
            except:
                pass
            if button:
                print("Clicking on the 'Sorting' button...")
                pyautogui.click(button)

            # time.sleep(60*60*24)
            # burp_process.terminate()
            # print("Starting Burp Suite...")
            # burp_process = start_burp()