main.py

import sys
import os
from browser_artifact_scanning import main as artifact_main
from file_metadata_scanning import main as meta_main

# clear screen helper function
def clear_screen():
    os.system("cls" if os.name=="nt" else "clear")

# wait for input helper function
def wait():
    input("Press any key")

def exit_script():
    print("Exiting...")
    sys.exit(0)

def main():
    menu_actions = {
        "1": artifact_main.main, 
        "2": meta_main.main,
        "3": print("3"), #lambda: analyze_form_history(browser_files),
        "4": print("4"), #lambda: analyze_history(browser_files),
        "5": exit_script
    }

    while True:
        print("\nBen's Forensic CLI Toolkit")
        print("1. Browser Artifact Analysis")
        print("2. File Metadata Analysis")
        print("3. System Artifact Analysis")
        print("4. Memory Analysis")
        print("5. Exit")
        choice = input("Enter choice: ").strip()
        clear_screen()

        action = menu_actions.get(choice)
        if action:
            action()
        else:
            print("Invalid Choice, try again.")
            wait()

if __name__ == "__main__":
    main()