from ispoof.objects.player import Player
from ispoof.spoofer.scraper import Scraper
from ispoof.spoofer.device import Device
from ispoof.lists.pokemonlist import PokemonList
from ispoof.data import Database

if __name__ == "__main__":
    device = Device()
    device.mount_image()
    database = Database()
    database.initialize_database()
    engine = database.get_engine()
    player = Player(engine=engine)

    query = input("Enter your location: ")
    player.set_location_with_query(query)

    scraper = Scraper()
    pokemon_lst = PokemonList(engine=engine)
    while True:
        pokemons = scraper.get_hundos()
        pokemon_lst.insert_to_database(pokemons)
        print("1. Sort by name")
        print("2. Sort by level")
        print("3. Sort by CP")
        print("4. Sort by distance")
        print("5. Search name")
        choice = int(input("Your choice: "))
        if choice == 1:
            pokemons = pokemon_lst.sort_by_name()
        elif choice == 2:
            pokemons = pokemon_lst.sort_by_level()
        elif choice == 3:
            pokemons = pokemon_lst.sort_by_cp()
        elif choice == 4:
            pokemons = pokemon_lst.sort_by_distance(player.location)
        else:
            pokemons = pokemon_lst.search_by_name(input("Enter query: "))
        print("Pokemon List:")
        for i, pokemon in enumerate(pokemons):
            print(i, pokemon)
        i = int(input("Choose pokemon: "))
        pokemon = pokemons[i]
        location = pokemon.location
        device.spoof_gps(location)
        did_activity = None
        print("Spoof to")
        print(pokemon)
        while True:
            activity = input("Did you do any cooldown activities? [Y/N] ").lower()
            if activity in ("y", "n"):
                did_activity = activity == "y"
                break
        player.change_gps_by_location(location, did_activity)
        print(f"Current cooldown: {player.get_current_cooldown()} min")

        while True:
            continue_prompt = input("Continue? [Y/N] ").lower()
            if continue_prompt in ("y", "n"):
                if continue_prompt == "n":
                    device.stop_spoofing()
                    exit(0)
                else:
                    break