import os
import subprocess
from datetime import datetime
from config import OS_COMMANDS, TEMP_DIR

def execute_command(command_type: str, command_text: str) -> str:
    try:
        if command_type == "open_app":
            for app, cmd in OS_COMMANDS.get("open", {}).items():
                if app in command_text.lower():
                    subprocess.Popen(cmd.split())
                    return f"✅ Opening {app.title()}..."

        elif command_type == "close_app":
            for app, cmd in OS_COMMANDS.get("close", {}).items():
                if app in command_text.lower():
                    os.system(cmd)
                    return f"🛑 Closing {app.title()}..."

        elif command_type == "screenshot":
            screenshot_path = os.path.join(TEMP_DIR, f"screenshot_{datetime.now().strftime('%Y%m%d_%H%M%S')}.png")
            cmd = OS_COMMANDS["screenshot"]["default"]
            cmd = cmd.replace("screenshot.png", screenshot_path)
            os.system(cmd)
            return f"📸 Screenshot saved to: {screenshot_path}"

        elif command_type == "create_file":
            filename = "new_file.txt"
            with open(filename, "w") as f:
                f.write("Hello from your AI assistant.")
            return f"✅ File '{filename}' created."

        elif command_type == "shutdown":
            os.system(OS_COMMANDS["shutdown"])
            return "🔌 Shutting down..."

        elif command_type == "restart":
            os.system(OS_COMMANDS["restart"])
            return "🔄 Restarting system..."

        elif command_type == "lock":
            os.system(OS_COMMANDS["lock"])
            return "🔒 System locked."

    except Exception as e:
        return f"❌ Error while executing command: {e}"

    return "⚠️ Command not recognized or not supported yet."