from upgrades import coin_multiplier, lucky_coin, flip_chance
from game import wealth, coin_flip(propability, wealth, coins_per_head, heads_amount, multiplier)


base_lvl_coin_multiplier = coin_multiplier(0.5, 10)
base_lvl_flip_chance = flip_chance(10, 0)
base_lvl_lucky_coin = lucky_coin(0.5, 10)

def lvl_up_multiplier():
    lvl_up = True

    if lvl_up == True:
        base_lvl_coin_multiplier.cost *= 2.5
        base_lvl_coin_multiplier.multiplier *= 5
        print(f' CM: {base_lvl_coin_multiplier.cost}')
        print(f' MM: {base_lvl_coin_multiplier.multiplier}')

def lvl_up_flip_chance():
    lvl_up = True

    if lvl_up == True:
        base_lvl_flip_chance.cost += 1
        base_lvl_flip_chance.cost *= 6
        base_lvl_flip_chance.chance *= 1.5
        print(f" FC Cost: {base_lvl_flip_chance.cost}")
        print(f" FC Chance: {base_lvl_flip_chance.chance}")

def lvl_up_lucky_coin():
    lvl_up = True

    if lvl_up == True:
        base_lvl_lucky_coin.cost *= 3
        base_lvl_lucky_coin.chance *= 1.5
        print(f" LC Cost: {base_lvl_lucky_coin.cost}")
        print(f" LC Chance: {base_lvl_lucky_coin.chance}")

def homepage_upgrades(wealth):
    while True:
        answer = input(' Flip Chance: [1]\n Lucky Coin:  [2]\n Multiplier:  [3]\n Zum Game:  [4]')
        try:
            answer = int(answer)
            if answer == 1:
                if wealth >= base_lvl_flip_chance.cost:
                    lvl_up_flip_chance()
                    wealth -= base_lvl_flip_chance.cost
                    print('Erfolgreich geupgradet. Du hast',wealth, "Coins")
                    break
                else:
                    print('Du bist zu arm')
                    
            elif answer == 2:
                if wealth >= base_lvl_lucky_coin.cost:
                    lvl_up_lucky_coin()
                    wealth -= base_lvl_lucky_coin.cost
                    print('Erfolgreich geupgradet. Du hast',wealth, "Coins")
                    break
                else:
                    print('Du bist zu arm')
                
            elif answer == 3:
                if wealth >= base_lvl_coin_multiplier.cost:
                    lvl_up_multiplier()
                    wealth -= base_lvl_coin_multiplier.cost
                    print('Erfolgreich geupgradet. Du hast',wealth, "Coins")
                    break
                else:
                    print('Du bist zu arm')
            elif answer == 4:
                coin_flip(propability, wealth, coins_per_head, heads_amount, multiplier)



            elif answer not in (1, 2, 3, 4):
                print('Bitte 1, 2, 3 oder 4!')
            
        except ValueError:
            print('Nur Zahlen')

homepage_upgrades(wealth)