gokemon_pokemon.go

package gokemon

import (
	"encoding/json"
	"errors"
	"fmt"
	"net/http"
	"strings"
)

// A Pokemon resource represent a single Pokémon
type Pokemon struct {
	Name         string         `json:"name"`
	ID           int            `json:"national_id"`
	URL          string         `json:"resource_uri"`
	Descriptions []*Description `json:"descriptions"`
	Abilities    []*Ability     `json:"abilities"`
	Types        []*Type        `json:"types"`
	Moves        []*Move        `json:"moves"`
	Eggs         []*Egg         `json:"egg_groups"`
	Sprites      []*Sprite      `json:"sprites"`
}

// Get detailed pokemon data
func (pokemon *Pokemon) Get() *Pokemon {
	temp, err := getPokemon(API_URL + pokemon.URL)
	if err != nil {
		return nil
	}
	return temp
}

// Print pokemon data in string format
func (poke Pokemon) String() string {
	str := ""
	str += fmt.Sprintf("Pokemon: %s\r\n", strings.Title(poke.Name))
	str += fmt.Sprintf("  URL: %s\r\n", poke.URL)
	if poke.ID != 0 {
		str += fmt.Sprintf("  ID: %d\r\n", poke.ID)
	}
	str += type2string(poke.Types, "Types")
	return str
}

// Return a Pokemon with Basic Information using pokemon name
func GetPokemon(name string) (*Pokemon, error) {
	name = strings.ToLower(name)
	url := fmt.Sprintf(API_URL+"/api/v1/pokemon/%s/", name)
	return getPokemon(url)
}

// Return a Pokemon with Basic Information using pokemon ID
func GetPokemonByID(id int) (*Pokemon, error) {
	url := fmt.Sprintf(API_URL+"/api/v1/pokemon/%d/", id)
	return getPokemon(url)
}

// Pokemon API request
func getPokemon(url string) (*Pokemon, error) {
	resp, err := http.Get(url)
	if err != nil {
		return nil, err
	}
	defer resp.Body.Close()
	if resp.StatusCode != http.StatusOK {
		return nil, errors.New(resp.Status)
	}
	pokemon := new(Pokemon)
	err = json.NewDecoder(resp.Body).Decode(pokemon)
	if err != nil {
		return nil, err
	}
	return pokemon, nil
}