fix: update prompt string plus some refactoring

- Updated the prompt string.
- Move the main package to cmd/pokecli.
- Added a loop function in the main repl function.
This commit is contained in:
Dan Anglin 2024-09-23 22:05:30 +01:00
parent 0f5826a7de
commit 23d6ddbdc8
Signed by: dananglin
GPG key ID: 0C1D44CFBEE68638
5 changed files with 43 additions and 32 deletions

2
.gitignore vendored
View file

@ -1 +1 @@
pokecli /pokecli

5
cmd/pokecli/main.go Normal file
View file

@ -0,0 +1,5 @@
package main
func main() {
repl()
}

View file

@ -3,7 +3,6 @@ package main
import ( import (
"bufio" "bufio"
"fmt" "fmt"
"maps"
"os" "os"
"strings" "strings"
"time" "time"
@ -18,11 +17,7 @@ type command struct {
callback commands.CommandFunc callback commands.CommandFunc
} }
func main() { func repl() {
run()
}
func run() {
var ( var (
cacheCleanupInterval = 30 * time.Minute cacheCleanupInterval = 30 * time.Minute
httpTimeout = 10 * time.Second httpTimeout = 10 * time.Second
@ -80,52 +75,46 @@ func run() {
callback: commands.HelpFunc(summaries), callback: commands.HelpFunc(summaries),
} }
fmt.Printf("\nWelcome to the Pokedex!\n")
fmt.Print("\npokedex > ")
scanner := bufio.NewScanner(os.Stdin) scanner := bufio.NewScanner(os.Stdin)
for scanner.Scan() { loopFunc := func() {
defer printPrompt()
input := scanner.Text() input := scanner.Text()
command, args := parseArgs(input) command, args := parseInput(input)
cmd, ok := commandMap[command] cmd, ok := commandMap[command]
if !ok { if !ok {
fmt.Println("ERROR: Unrecognised command.") fmt.Println("ERROR: Unrecognised command.")
fmt.Print("\npokedex > ") return
continue
} }
if cmd.callback == nil { if cmd.callback == nil {
fmt.Println("ERROR: This command is defined but does not have a callback function.") fmt.Println("ERROR: This command is defined but does not have a callback function.")
fmt.Print("\npokedex > ") return
continue
} }
if err := commandMap[command].callback(args); err != nil { if err := commandMap[command].callback(args); err != nil {
fmt.Printf("ERROR: %v.\n", err) fmt.Printf("ERROR: %v.\n", err)
return
} }
}
fmt.Print("pokedex > ") fmt.Printf("\nWelcome to the Pokemon world!\n")
printPrompt()
for scanner.Scan() {
loopFunc()
} }
} }
func summaryMap(commandMap map[string]command) map[string]string { func parseInput(input string) (string, []string) {
summaries := make(map[string]string) input = strings.TrimSpace(input)
input = strings.ToLower(input)
for key, value := range maps.All(commandMap) {
summaries[key] = value.description
}
return summaries
}
func parseArgs(input string) (string, []string) {
split := strings.Split(input, " ") split := strings.Split(input, " ")
if len(split) == 0 { if len(split) == 0 {
@ -138,3 +127,7 @@ func parseArgs(input string) (string, []string) {
return split[0], split[1:] return split[0], split[1:]
} }
func printPrompt() {
fmt.Print("pokecli > ")
}

13
cmd/pokecli/summaries.go Normal file
View file

@ -0,0 +1,13 @@
package main
import "maps"
func summaryMap(commandMap map[string]command) map[string]string {
summaries := make(map[string]string)
for key, value := range maps.All(commandMap) {
summaries[key] = value.description
}
return summaries
}

View file

@ -56,7 +56,7 @@ func Lint() error {
// To rebuild packages that are already up-to-date set POKECLI_BUILD_REBUILD_ALL=1 // To rebuild packages that are already up-to-date set POKECLI_BUILD_REBUILD_ALL=1
// To enable verbose mode set POKECLI_BUILD_VERBOSE=1 // To enable verbose mode set POKECLI_BUILD_VERBOSE=1
func Build() error { func Build() error {
main := "main.go" path := "./cmd/" + app
flags := ldflags() flags := ldflags()
build := sh.RunCmd("go", "build") build := sh.RunCmd("go", "build")
args := []string{"-ldflags=" + flags, "-o", binary} args := []string{"-ldflags=" + flags, "-o", binary}
@ -69,7 +69,7 @@ func Build() error {
args = append(args, "-v") args = append(args, "-v")
} }
args = append(args, main) args = append(args, path)
return build(args...) return build(args...)
} }