main.py

#!/usr/bin/env python3

import subprocess
import os
import sqlite3
import sys
from rich.panel import Panel
from rich.prompt import Prompt
from rich.layout import Layout
from theme import console

# Paths
BASE_DIR = os.path.dirname(os.path.realpath(__file__))
DB_FILE = os.path.join(BASE_DIR, "commands.db")
HISTORY_SCRIPT = os.path.join(BASE_DIR, "history.py")
VIEW_SCRIPT = os.path.join(BASE_DIR, "view_history.py")
CLEAR_DB_SCRIPT = os.path.join(BASE_DIR, "clear_db.py")

def clear_screen():
    os.system("clear" if os.name == "posix" else "cls")


def database_exists():
    if not os.path.exists(DB_FILE):
        return False

    try:
        conn = sqlite3.connect(DB_FILE)
        cursor = conn.cursor()
        cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='commands'")
        table_exists = cursor.fetchone()
        conn.close()
        return bool(table_exists)
    except sqlite3.Error:
        return False


def get_command_stats():
    try:
        conn = sqlite3.connect(DB_FILE)
        cursor = conn.cursor()
        cursor.execute("SELECT COUNT(*) FROM commands")
        total_unique = cursor.fetchone()[0]
        cursor.execute("SELECT SUM(count) FROM commands")
        total = cursor.fetchone()[0] or 0
        conn.close()
        return total, total_unique
    except sqlite3.Error as e:
        console.print(f"[error]Database error: {e}[/]")
        return (0, 0)

def show_initial_menu():
    menu_text = "[header bold]First Time Setup[/]\n\n"
    menu_text += "[header]1.[/] Import Shell History (initialize database)\n"
    menu_text += "[header]q.[/] Quit"
    return Panel(menu_text, title="[header]No Database Found[/]", border_style="border")

def show_full_menu():
    menu_text = "[header bold]Command Tracker Menu[/]\n\n"
    menu_text += "[header]1.[/] Update commands\n"
    menu_text += "[header]2.[/] View Commands\n"
    menu_text += "[header]3.[/] Clear Command History\n"
    menu_text += "[header]q.[/] Quit"
    return Panel(menu_text, title="[header]Main Menu[/]", border_style="border")

def show_stats_panel():
    if database_exists():
        total, unique = get_command_stats()
        content = f"[cell]Tracked Commands:[/] [header]{total}[/] total ([header]{unique}[/] unique)"
    else:
        content = "[error]⚠️  No database found. Please import history first."
    return Panel(content, title="[header]Stats[/]", border_style="border")

def first_time_menu():
    while True:
        clear_screen()
        layout = Layout()
        layout.split_column(
            Layout(show_stats_panel(), size=3),
            Layout(show_initial_menu(), size=8)
        )
        console.print(layout)

        choice = Prompt.ask("[prompt]Choose an option[/]", choices=["1", "q"], default="q")
        if choice == "1":
            subprocess.run(["python3", HISTORY_SCRIPT])
            input("\nPress Enter to continue...")
            break
        elif choice == "q":
            console.print("\n[header]Goodbye![/]")
            exit()

def main_menu():
    while True:
        clear_screen()
        layout = Layout()
        layout.split_column(
            Layout(show_stats_panel(), size=3),
            Layout(show_full_menu(), size=10)
        )
        console.print(layout)

        choice = Prompt.ask("[prompt]Choose an option[/]", choices=["1", "2", "3", "q"], default="q")

        if choice == "1":
            clear_choice = Prompt.ask(
                "[prompt]Clear saved command history before importing? (not doing so will lead to existing and new commands being appended)[/]",
                choices=["y", "n"], default="y"
            )
            if clear_choice == "y":
                subprocess.run(["python3", CLEAR_DB_SCRIPT])
            subprocess.run(["python3", HISTORY_SCRIPT])
            input("\nPress Enter to return to menu...")

        elif choice == "2":
            subprocess.run(["python3", VIEW_SCRIPT])
            input("\nPress Enter to return to menu...")

        elif choice == "3":
            confirm = Prompt.ask("[prompt]Delete all command history (only the `showcmm` history will be removed)?[/]", choices=["y", "n"], default="n")
            if confirm == "y":
                subprocess.run(["python3", CLEAR_DB_SCRIPT])
                console.print("[header]✅ Database cleared successfully.[/]")
            else:
                console.print("[cell]Cancelled.[/]")
            input("\nPress Enter to return to menu...")

        elif choice == "q":
            console.print("\n[header]Goodbye![/]")
            break

# --- CLI Subcommand Support ---
def run_subcommand():
    def run_view():
        if not database_exists():
            console.print("[error]⚠️ No database found. Please run 'showcmm update' first.[/]")
            return
        args = sys.argv[2:]  # skip "showcmm view"
        subprocess.run(["python3", VIEW_SCRIPT] + args)

    def run_update():
        if database_exists():
            confirm = input("Clear existing database first? [y/N]: ").strip().lower()
            if confirm == "y":
                subprocess.run(["python3", CLEAR_DB_SCRIPT])
        subprocess.run(["python3", HISTORY_SCRIPT])

    def run_safe_clear():
        confirm = input("⚠️  Are you sure you want to clear all saved command history? [y/N]: ").strip().lower()
        if confirm == "y":
            subprocess.run(["python3", CLEAR_DB_SCRIPT])
            console.print("[header]✅ Database cleared successfully.[/]")
        else:
            console.print("[cell]Cancelled.[/]")


    commands = {
        "view": run_view,
        "update": run_update,
        "clear": run_safe_clear,
    }

    if len(sys.argv) > 1:
        cmd = sys.argv[1]
        if cmd in commands:
            commands[cmd]()
        else:
            console.print(f"[error]Unknown command: {cmd}[/]")
    else:
        if not database_exists():
            first_time_menu()
        main_menu()

if __name__ == "__main__":
    run_subcommand()