i2_bot/lib/getters/pokememes.go

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

package getters

import (
	// stdlib
	"log"
	"strconv"
	// local
	"../dbmapping"
)

// Internal functions

func (g *Getters) formFullPokememes(pokememes []dbmapping.Pokememe) ([]dbmapping.PokememeFull, bool) {
	pokememes_full := []dbmapping.PokememeFull{}
	elements := []dbmapping.Element{}
	err := c.Db.Select(&elements, "SELECT * FROM elements")
	if err != nil {
		log.Println(err)
		return pokememes_full, false
	}
	locations := []dbmapping.Location{}
	err = c.Db.Select(&locations, "SELECT * FROM locations")
	if err != nil {
		log.Println(err)
		return pokememes_full, false
	}
	pokememes_elements := []dbmapping.PokememeElement{}
	err = c.Db.Select(&pokememes_elements, "SELECT * FROM pokememes_elements")
	if err != nil {
		log.Println(err)
		return pokememes_full, false
	}
	pokememes_locations := []dbmapping.PokememeLocation{}
	err = c.Db.Select(&pokememes_locations, "SELECT * FROM pokememes_locations")
	if err != nil {
		log.Println(err)
		return pokememes_full, false
	}

	for i := range pokememes {
		full_pokememe := dbmapping.PokememeFull{}
		elements_listed := []dbmapping.Element{}
		locations_listed := []dbmapping.Location{}

		for j := range pokememes_locations {
			if pokememes_locations[j].Pokememe_id == pokememes[i].Id {
				for l := range locations {
					if pokememes_locations[j].Location_id == locations[l].Id {
						locations_listed = append(locations_listed, locations[l])
					}
				}
			}
		}

		for k := range pokememes_elements {
			if pokememes_elements[k].Pokememe_id == pokememes[i].Id {
				for e := range elements {
					if pokememes_elements[k].Element_id == elements[e].Id {
						elements_listed = append(elements_listed, elements[e])
					}
				}
			}
		}

		full_pokememe.Pokememe = pokememes[i]
		full_pokememe.Elements = elements_listed
		full_pokememe.Locations = locations_listed

		pokememes_full = append(pokememes_full, full_pokememe)
	}

	return pokememes_full, true
}

// External functions

func (g *Getters) GetPokememes() ([]dbmapping.PokememeFull, bool) {
	pokememes_full := []dbmapping.PokememeFull{}
	pokememes := []dbmapping.Pokememe{}
	err := c.Db.Select(&pokememes, "SELECT * FROM pokememes ORDER BY grade asc, name asc")
	if err != nil {
		log.Println(err)
		return pokememes_full, false
	}

	pokememes_full, ok := g.formFullPokememes(pokememes)
	return pokememes_full, ok
}

func (g *Getters) GetBestPokememes(player_id int) ([]dbmapping.PokememeFull, bool) {
	pokememes_full := []dbmapping.PokememeFull{}
	player_raw, ok := g.GetPlayerByID(player_id)
	if !ok {
		return pokememes_full, ok
	}
	profile_raw, ok := g.GetProfile(player_id)
	if !ok {
		return pokememes_full, ok
	}

	if player_raw.League_id == 0 {
		return pokememes_full, false
	}

	// TODO: make it more complicated
	pokememes := []dbmapping.Pokememe{}
	err := c.Db.Select(&pokememes, c.Db.Rebind("SELECT p.* FROM pokememes p, pokememes_elements pe, elements e WHERE e.league_id = ? AND p.grade = ? AND pe.element_id = e.id AND pe.pokememe_id = p.id ORDER BY p.attack DESC"), player_raw.League_id, profile_raw.Level_id+1)
	if err != nil {
		log.Println(err)
		return pokememes_full, false
	}

	pokememes_full, ok = g.formFullPokememes(pokememes)
	return pokememes_full, ok
}

func (g *Getters) GetPokememeByID(pokememe_id string) (dbmapping.PokememeFull, bool) {
	pokememe_full := dbmapping.PokememeFull{}
	pokememe := dbmapping.Pokememe{}
	err := c.Db.Get(&pokememe, c.Db.Rebind("SELECT * FROM pokememes WHERE id=?"), pokememe_id)
	if err != nil {
		log.Println(err)
		return pokememe_full, false
	}
	elements := []dbmapping.Element{}
	err = c.Db.Select(&elements, "SELECT * FROM elements")
	if err != nil {
		log.Println(err)
		return pokememe_full, false
	}
	locations := []dbmapping.Location{}
	err = c.Db.Select(&locations, "SELECT * FROM locations")
	if err != nil {
		log.Println(err)
		return pokememe_full, false
	}
	pokememes_elements := []dbmapping.PokememeElement{}
	err = c.Db.Select(&pokememes_elements, "SELECT * FROM pokememes_elements WHERE pokememe_id='"+strconv.Itoa(pokememe.Id)+"'")
	if err != nil {
		log.Println(err)
		return pokememe_full, false
	}
	pokememes_locations := []dbmapping.PokememeLocation{}
	err = c.Db.Select(&pokememes_locations, "SELECT * FROM pokememes_locations WHERE pokememe_id='"+strconv.Itoa(pokememe.Id)+"'")
	if err != nil {
		log.Println(err)
		return pokememe_full, false
	}

	elements_listed := []dbmapping.Element{}
	locations_listed := []dbmapping.Location{}

	for j := range pokememes_locations {
		if pokememes_locations[j].Pokememe_id == pokememe.Id {
			for l := range locations {
				if pokememes_locations[j].Location_id == locations[l].Id {
					locations_listed = append(locations_listed, locations[l])
				}
			}
		}
	}

	for k := range pokememes_elements {
		if pokememes_elements[k].Pokememe_id == pokememe.Id {
			for e := range elements {
				if pokememes_elements[k].Element_id == elements[e].Id {
					elements_listed = append(elements_listed, elements[e])
				}
			}
		}
	}

	pokememe_full.Pokememe = pokememe
	pokememe_full.Elements = elements_listed
	pokememe_full.Locations = locations_listed

	return pokememe_full, true
}