# Binarian
# Copyright (C) 2025 Sarahtonin & Lyxminx
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.

import readchar
from gamemodes import norToBin, binToNor
from functions import clear
from translate import translateEnToBin, translateBinToEn  # Ensure these are correctly imported

def entoBin():
    options = ["Level 1", "Level 2", "Level 3", "Level 4", "Level 5", "Back to main menu"]
    selected = 0

    while True:
        clear()
        print("Select a level to start translating!")
        print("Use arrow keys to navigate and Enter to select:\n")

        for i, option in enumerate(options):
            prefix = "> " if i == selected else "  "
            print(f"{prefix}{option}")

        key = readchar.readkey()

        if key == readchar.key.UP and selected > 0:
            selected -= 1
        elif key == readchar.key.DOWN and selected < len(options) - 1:
            selected += 1
        elif key == readchar.key.ENTER:
            if selected < 5:  # If it's a level
                clear()
                norToBin(selected + 1)
            else:  # Back to main menu
                print("Exiting...")
                break

def binToEn():
    options = ["Level 1", "Level 2", "Level 3", "Level 4", "Level 5", "Back to main menu"]
    selected = 0

    while True:
        clear()
        print("Select a level to start translating!")
        print("Use arrow keys to navigate and Enter to select:\n")

        for i, option in enumerate(options):
            prefix = "> " if i == selected else "  "
            print(f"{prefix}{option}")

        key = readchar.readkey()

        if key == readchar.key.UP and selected > 0:
            selected -= 1
        elif key == readchar.key.DOWN and selected < len(options) - 1:
            selected += 1
        elif key == readchar.key.ENTER:
            if selected < 5:
                clear()
                binToNor(selected + 1)
            else:
                print("Exiting...")
                break

def translateMenu():
    options = ["Binarian to English", "English to Binarian", "Back to main menu"]
    selected = 0

    while True:
        clear()
        print("Select if you want to translate from or to Binarian!")
        print("Use arrow keys to navigate and Enter to select:\n")

        for i, option in enumerate(options):
            prefix = "> " if i == selected else "  "
            print(f"{prefix}{option}")

        key = readchar.readkey()

        if key == readchar.key.UP and selected > 0:
            selected -= 1
        elif key == readchar.key.DOWN and selected < len(options) - 1:
            selected += 1
        elif key == readchar.key.ENTER:
            if selected == 0:
                clear()
                translateBinToEn()
            elif selected == 1:
                clear()
                translateEnToBin()
            else:
                print("Exiting...")
                break