// i2_bot – Instinct PokememBro Bot
// Copyright (c) 2017 Vladimir "fat0troll" Hodakov

package getters

import (
    // stdlib
    "log"
)

func (g *Getters) PossibilityRequiredPokeballs(location int, grade int, lvl int) int {
    var base_possibility float64 = 0.00
    var required_pokeballs int = 0

    if lvl > 3 {
        switch {
        case grade == (lvl + 1):
            base_possibility = 0.05
        case grade == lvl:
            base_possibility = 0.5
        case grade == (lvl - 1):
            base_possibility = 0.3
        case grade == (lvl - 2):
            base_possibility = 0.1
        case grade == (lvl - 3):
            base_possibility = 0.05
        default:
            base_possibility = 0.00
        }
    } else if lvl == 3 {
        switch grade {
        case 4:
            base_possibility = 0.05
        case 3:
            base_possibility = 0.5
        case 2:
            base_possibility = 0.3
        case 1:
            base_possibility = 0.15
        default:
            base_possibility = 0.00
        }
    } else if lvl == 2 {
        switch grade {
        case 3:
            base_possibility = 0.05
        case 2:
            base_possibility = 0.70
        case 1:
            base_possibility = 0.25
        default:
            base_possibility = 0.00
        }
    } else if lvl == 1 {
        switch grade {
        case 2:
            base_possibility = 0.80
        case 1:
            base_possibility = 0.20
        default:
            base_possibility = 0.00
        }
    }

    var number_of_pokememes int = 0

    err := c.Db.Get(&number_of_pokememes, c.Db.Rebind("SELECT count(*) FROM pokememes p, pokememes_locations pl WHERE p.grade = ? AND pl.location_id = ? AND pl.pokememe_id = p.id;"), grade, location)
    if err != nil {
        log.Println(err)
    }

    if base_possibility != 0 && number_of_pokememes != 0 {
        required_pokeballs = int(1.0 / (base_possibility / float64(number_of_pokememes)))
    }

    return required_pokeballs
}