from config import ADMIN_PASSWORD


class CLI:
    def __init__(self, tracker) -> None:
        self.tracker = tracker
        self.current_user = None

    def run(self):
        while True:
            self._clear_screen()
            self._show_main_menu()

    def _clear_screen(self):
        import subprocess
        import os

        subprocess.run("cls" if os.name == "nt" else "clear")

    def _show_main_menu(self):
        print("=" * 50)
        print("DEBT TRACKER")
        print("=" * 50)

        if self.current_user:
            user = self.tracker.get_user_by_id(self.current_user)
            debt = self.tracker.get_user_debt(self.current_user)
            print(f"\nLogged in as: {user.name}")
            print(f"Current Debt: ${debt:.2f}\n")

        print("1. Select user")
        print("2. View all debts")
        print("3. Admin panel")
        print("4. Exit")
        print()

        choice = input("Choose Option: ").strip()

        if choice == "1":
            self._select_user()
        elif choice == "2":
            self._view_all_debts()
        elif choice == "3":
            self._admin_panel()
        elif choice == "4":
            print("\nGoodbye!")
            exit()
        else:
            self._invalid_choice()

    def _select_user(self):
        self._clear_screen()
        print("=" * 50)
        print("SELECT USER")
        print("=" * 50)
        print()

        users = self.tracker.get_all_users()
        user_list = list(users.items())

        for i, (user_id, user) in enumerate(user_list, 1):
            debt = self.tracker.get_user_debt(user_id)
            print(f"{i}. {user.name} (Debt: ${debt:.2f})")

        print(f"{len(user_list) + 1}. Back to main menu")
        print()

        try:
            choice = int(input("Choose user: ").strip())

            if choice == len(user_list) + 1:
                return

            if 1 <= choice <= len(user_list):
                self.current_user = user_list[choice - 1][0]
                self._user_menu()
            else:
                self._invalid_choice()
        except ValueError:
            self._invalid_choice()

    def _user_menu(self):
        while True:
            self._clear_screen()
            user = self.tracker.get_user_by_id(self.current_user)
            debt = self.tracker.get_user_debt(self.current_user)

            print("=" * 50)
            print(f"USER MENU - {user.name}")
            print("=" * 50)
            print(f"Current debt: ${debt:.2f}\n")

            print("1. Add purchase")
            print("2. View my debt")
            print("3. Logout")
            print()

            choice = input("Choose option: ").strip()

            if choice == "1":
                self._add_purchase()
            elif choice == "2":
                self._view_user_debt()
            elif choice == "3":
                self.current_user = None
                break
            else:
                self._invalid_choice()

    def _add_purchase(self):
        self._clear_screen()
        print("=" * 50)
        print("ADD PURCHASE")
        print("=" * 50)
        print()

        menu_items = self.tracker.get_items()
        item_list = list(menu_items.items())

        for i, (_item_id, item) in enumerate(item_list, 1):
            print(f"{i}. {item.name} - ${item.price:.2f}")

        print(f"{len(item_list) + 1}. Cancel")
        print()

        try:
            choice = int(input("Choose item: ").strip())

            if choice == len(item_list) + 1:
                return

            if 1 <= choice <= len(item_list):
                selected_item_id = item_list[choice - 1][0]
                self._confirm_purchase(selected_item_id)
            else:
                self._invalid_choice()
        except ValueError:
            self._invalid_choice()

    def _confirm_purchase(self, item_id):
        self._clear_screen()

        user = self.tracker.get_user_by_id(self.current_user)
        item = self.tracker.get_items()[item_id]

        print("=" * 50)
        print("CONFIRM PURCHASE")
        print("=" * 50)
        print()
        print(f"User: {user.name}")
        print(f"Item: {item.name}")
        print(f"Price: ${item.price:.2f}")
        print()

        confirm = input("Add to debt? (y/n): ").strip().lower()

        if confirm == "y":
            result = self.tracker.add_purchase(self.current_user, item_id)

            self._clear_screen()
            print("=" * 50)
            print("PURCHASE ADDED")
            print("=" * 50)
            print()
            print(f"✓ {result['message']}")
            print(f"New debt: ${result['new_debt']:.2f}")
            print()

            input("Press Enter to continue...")
        else:
            print("Purchase cancelled.")
            input("Press Enter to continue...")

    def _view_user_debt(self):
        self._clear_screen()

        user = self.tracker.get_user_by_id(self.current_user)
        debt = self.tracker.get_user_debt(self.current_user)

        print("=" * 50)
        print(f"DEBT - {user.name}")
        print("=" * 50)
        print()
        print(f"Amount owed: ${debt:.2f}")
        print()

        input("Press Enter to continue...")

    def _view_all_debts(self):
        self._clear_screen()
        print("=" * 50)
        print("ALL DEBTS")
        print("=" * 50)
        print()

        all_debts = self.tracker.get_all_debts()

        if not all_debts:
            print("No users with debt.")
        else:
            total = 0
            for _user_id, debt_info in all_debts.items():
                amount = debt_info["debt"]
                print(f"{debt_info['name']:.<30} ${amount:>7.2f}")
                total += amount

            print("-" * 50)
            print(f"{'TOTAL':.<30} ${total:>7.2f}")

        print()
        input("Press Enter to continue...")

    def _admin_panel(self):
        self._clear_screen()
        print("=" * 50)
        print("ADMIN PANEL")
        print("=" * 50)
        print()

        password = input("Enter admin password: ").strip()

        if password != ADMIN_PASSWORD:
            print("Invalid password!")
            input("Press Enter to continue...")
            return

        while True:
            self._clear_screen()
            print("=" * 50)
            print("ADMIN PANEL")
            print("=" * 50)
            print()

            print("1. View all debts")
            print("2. Adjust user debt")
            print("3. Clear user debt")
            print("4. View transaction history")
            print("5. Logout")
            print()

            choice = input("Choose option: ").strip()

            if choice == "1":
                self._view_all_debts()
            elif choice == "2":
                self._adjust_debt()
            elif choice == "3":
                self._clear_user_debt()
            elif choice == "4":
                self._view_transaction_history()
            elif choice == "5":
                break
            else:
                self._invalid_choice()

    def _adjust_debt(self):
        self._clear_screen()
        print("=" * 50)
        print("ADJUST DEBT")
        print("=" * 50)
        print()

        users = self.tracker.get_all_users()
        user_list = list(users.items())

        for i, (user_id, user) in enumerate(user_list, 1):
            debt = self.tracker.get_user_debt(user_id)
            print(f"{i}. {user.name} - ${debt:.2f}")

        print()

        try:
            choice = int(input("Choose user: ").strip())

            if 1 <= choice <= len(user_list):
                user_id = user_list[choice - 1][0]

                amount_str = input("Amount paid: $").strip()
                amount = float(amount_str)

                result = self.tracker.adjust_debt(user_id, amount, ADMIN_PASSWORD)

                self._clear_screen()
                print("=" * 50)
                print("DEBT ADJUSTED")
                print("=" * 50)
                print()
                print(f"{result['message']}")
                print(f"New debt: ${result['new_debt']:.2f}")
                print()

                input("Press Enter to continue...")
            else:
                self._invalid_choice()
        except ValueError:
            print("Invalid input!")
            input("Press Enter to continue...")

    def _clear_user_debt(self):
        self._clear_screen()
        print("=" * 50)
        print("CLEAR USER DEBT")
        print("=" * 50)
        print()

        users = self.tracker.get_all_users()
        user_list = list(users.items())

        for i, (user_id, user) in enumerate(user_list, 1):
            debt = self.tracker.get_user_debt(user_id)
            print(f"{i}. {user.name} - ${debt:.2f}")

        print()

        try:
            choice = int(input("Choose user: ").strip())

            if 1 <= choice <= len(user_list):
                user_id = user_list[choice - 1][0]
                debt_amount = self.tracker.get_user_debt(user_id)

                confirm = input(f"Clear ${debt_amount:.2f}? (y/n): ").strip().lower()

                if confirm == "y":
                    _result = self.tracker.adjust_debt(
                        user_id, debt_amount, ADMIN_PASSWORD
                    )

                    self._clear_screen()
                    print("=" * 50)
                    print("DEBT CLEARED")
                    print("=" * 50)
                    print()
                    print(
                        f"Debt cleared for {self.tracker.get_user_by_id(user_id).name}"
                    )
                    print()

                    input("Press Enter to continue...")
            else:
                self._invalid_choice()
        except ValueError:
            print("Invalid input!")
            input("Press Enter to continue...")

    def _view_transaction_history(self):
        self._clear_screen()
        print("=" * 50)
        print("TRANSACTION HISTORY")
        print("=" * 50)
        print()

        transactions = self.tracker.transactions

        if not transactions:
            print("No transactions yet.")
        else:
            for t in sorted(transactions, key=lambda x: x.timestamp, reverse=True):
                user = self.tracker.get_user_by_id(t.user_id)
                item = self.tracker.get_items()[t.item_id]
                print(
                    f"{t.timestamp[:10]} | {user.name:.<20} {item.name:.<15} ${t.amount:.2f}"
                )

        print()
        input("Press Enter to continue...")

    def _invalid_choice(self):
        print("Invalid choice. Please try again.")
        input("Press Enter to continue...")