feat: added catch, inspect, and pokedex commands

commands.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"fmt"
+	"math/rand"
 	"os"
 	"sort"
 	"time"
@@ -11,11 +12,12 @@ import (
 )
 
 type context struct {
-	commands   map[string]cliCommand
-	client     *pokeapi.Client
-	prevOffset int
-	nextOffset int
-	args       []string
+	commands      map[string]cliCommand
+	client        *pokeapi.Client
+	prevOffset    int
+	nextOffset    int
+	args          []string
+	caughtPokemon map[string]pokeapi.Pokemon
 }
 
 func (r *context) register(c cliCommand) {
@@ -55,11 +57,31 @@ func NewContext() *context {
 		callback:    commandExplore,
 	}
 
+	catchCommand := cliCommand{
+		name:        "catch",
+		description: "Attempt to catch pokemon with given name",
+		callback:    commandCatch,
+	}
+
+	inspectCommand := cliCommand{
+		name:        "inspect",
+		description: "Inspect information about pokemon with given name if caught",
+		callback:    commandInspect,
+	}
+
+	pokedexCommand := cliCommand{
+		name:        "pokedex",
+		description: "Display the names of all pokemon you've caught",
+		callback:    commandPokedex,
+	}
+
 	c := &context{
-		commands:   map[string]cliCommand{},
-		client:     pokeapi.NewClient(pokecache.NewCache(5 * time.Second)),
-		prevOffset: 0,
-		nextOffset: 0,
+		commands:      map[string]cliCommand{},
+		client:        pokeapi.NewClient(pokecache.NewCache(5 * time.Second)),
+		prevOffset:    0,
+		nextOffset:    0,
+		args:          []string{},
+		caughtPokemon: map[string]pokeapi.Pokemon{},
 	}
 
 	c.register(exitCommand)
@@ -67,6 +89,9 @@ func NewContext() *context {
 	c.register(mapCommand)
 	c.register(mapBackCommand)
 	c.register(exploreCommand)
+	c.register(catchCommand)
+	c.register(inspectCommand)
+	c.register(pokedexCommand)
 
 	return c
 }
@@ -165,3 +190,84 @@ func commandExplore(c *context) error {
 
 	return nil
 }
+
+func commandCatch(c *context) error {
+	// TODO: Probably want to allow user to decide
+	// if they want to catch a pokemon they've
+	// already caught. Maybe present xp of current
+	// vs encountered then let them choose.
+
+	if len(c.args) < 1 {
+		fmt.Println("Please provide pokemon name. i.e. catch <pokemon_name>")
+		return nil
+	}
+
+	pokemon, err := c.client.Pokemon.Get(c.args[0])
+
+	if err != nil {
+		return err
+	}
+
+	fmt.Printf("Throwing a Pokeball at %s...\n", pokemon.Name)
+
+	roll := rand.Intn(100)
+	baseChance := 75.0
+	pokemonXpMinValue := 0
+	pokemonXpMaxValue := 1000
+	pokemonXp := min(max(pokemon.BaseExperience, pokemonXpMinValue), pokemonXpMaxValue)
+
+	modifier := int(float64(pokemonXp) / float64(pokemonXpMaxValue) * float64(baseChance))
+	chance := int(baseChance) - modifier
+
+	if roll > chance {
+		c.caughtPokemon[pokemon.Name] = pokemon
+		fmt.Printf("You caught %s!\n", pokemon.Name)
+	} else {
+		fmt.Printf("%s was able to break free!\n", pokemon.Name)
+	}
+
+	return nil
+}
+
+func commandInspect(c *context) error {
+	if len(c.args) < 1 {
+		fmt.Println("Please provide pokemon name. i.e. inspect <pokemon_name>")
+		return nil
+	}
+
+	pokemonToInspect := c.args[0]
+
+	pokemon, caught := c.caughtPokemon[pokemonToInspect]
+
+	if !caught {
+		fmt.Printf("You have not previously caught a %s", pokemonToInspect)
+		return nil
+	}
+
+	fmt.Printf("Name:   %s\n", pokemon.Name)
+	fmt.Printf("Height: %d\n", pokemon.Height)
+	fmt.Printf("Weight: %d\n", pokemon.Weight)
+	fmt.Println("Stats:")
+
+	for _, stat := range pokemon.Stats {
+		fmt.Printf("  - %s: %d\n", stat.Stat.Name, stat.BaseStat)
+	}
+
+	fmt.Println("Types:")
+
+	for _, t := range pokemon.Types {
+		fmt.Printf("  - %s\n", t.Type.Name)
+	}
+
+	return nil
+}
+
+func commandPokedex(c *context) error {
+	fmt.Println("Your Pokedex:")
+
+	for _, pokemon := range c.caughtPokemon {
+		fmt.Printf("  - %s\n", pokemon.Name)
+	}
+
+	return nil
+}