Created choose package

Author: RedSkiesReaperr

choose/game.go

@@ -0,0 +1,45 @@
+package choose
+
+import (
+	"notion-igdb-autocomplete/igdb"
+	"sort"
+
+	"github.com/agnivade/levenshtein"
+)
+
+type gameComparisons []gameComparison
+type gameComparison struct {
+	index int
+	game  *igdb.Game
+}
+
+// Implements interface sort.Interface
+func (gc gameComparisons) Len() int {
+	return len(gc)
+}
+
+// Implements interface sort.Interface
+func (gc gameComparisons) Swap(i, j int) {
+	gc[i], gc[j] = gc[j], gc[i]
+}
+
+// Implements interface sort.Interface
+func (gc gameComparisons) Less(i, j int) bool {
+	return gc[i].index < gc[j].index
+}
+
+// Game returns the best choice in the list for the given search
+func Game(search string, list igdb.Games) *igdb.Game {
+	var comparisons gameComparisons
+
+	for _, game := range list {
+		comparisons = append(comparisons, gameComparison{
+			game:  game,
+			index: levenshtein.ComputeDistance(search, game.Name),
+		})
+	}
+
+	sort.Sort(comparisons)
+
+	return comparisons[0].game
+}

choose/game_test.go

@@ -0,0 +1,53 @@
+package choose
+
+import (
+	"notion-igdb-autocomplete/igdb"
+	"sort"
+	"testing"
+)
+
+func TestGameComparisonsSorting(t *testing.T) {
+	comparisons := gameComparisons{
+		gameComparison{index: 89},
+		gameComparison{index: 4},
+		gameComparison{index: 64},
+		gameComparison{index: 5},
+	}
+
+	expected := []gameComparison{
+		{index: 4},
+		{index: 5},
+		{index: 64},
+		{index: 89},
+	}
+
+	sort.Sort(comparisons)
+
+	for i, v := range comparisons {
+		if v != expected[i] {
+			t.Errorf("Unexpected comparison[%d]: expected=%v, got=%v", i, expected[i], v)
+		}
+	}
+}
+
+func TestGameChoice(t *testing.T) {
+	search := "world oF warcraft"
+	choices := igdb.Games{
+		&igdb.Game{Name: "World of Warcraft: Battle for Azeroth"},
+		&igdb.Game{Name: "World of Warcraft: Wrath of the Lich King"},
+		&igdb.Game{Name: "World of Warcraft: Warlords of Draenor"},
+		&igdb.Game{Name: "World of Warcraft: Legion"},
+		&igdb.Game{Name: "World of Warcraft"},
+		&igdb.Game{Name: "World of Warcraft: Dragonflight"},
+		&igdb.Game{Name: "World of Warcraft: Cataclysm"},
+		&igdb.Game{Name: "World of Warcraft: The Burning Crusade"},
+		&igdb.Game{Name: "World of Warcraft: Mists of Pandaria"},
+		&igdb.Game{Name: "World of Warcraft: Shadowlands"},
+	}
+	expectedName := "World of Warcraft"
+	result := Game(search, choices)
+
+	if result.Name != expectedName {
+		t.Errorf("Unexpected choice: expected:%s, got:%s", expectedName, result.Name)
+	}
+}

cmd/updater/main.go

@@ -4,12 +4,11 @@ import (
 	"fmt"
 	"log"
 	"net/http"
+	"notion-igdb-autocomplete/choose"
 	"notion-igdb-autocomplete/config"
 	"notion-igdb-autocomplete/igdb"
 	"notion-igdb-autocomplete/notion"
-	"sort"
 
-	"github.com/agnivade/levenshtein"
 	"github.com/gin-gonic/gin"
 )
 
@@ -76,41 +75,5 @@ func searchIgdbGame(gameName string, client *igdb.Client) (*igdb.Game, error) {
 		return nil, fmt.Errorf("cannot find game '%s'", gameName)
 	}
 
-	return findBestGame(gameName, results), nil
-}
-
-type ComparedGames []ComparedGame
-type ComparedGame struct {
-	Game  igdb.Game
-	Index int
-}
-
-// Implements interface sort.Interface
-func (cg ComparedGames) Len() int {
-	return len(cg)
-}
-
-// Implements interface sort.Interface
-func (cg ComparedGames) Swap(i, j int) {
-	cg[i], cg[j] = cg[j], cg[i]
-}
-
-// Implements interface sort.Interface
-func (cg ComparedGames) Less(i, j int) bool {
-	return cg[i].Index < cg[j].Index
-}
-
-func findBestGame(search string, games igdb.Games) *igdb.Game {
-	var comparisons ComparedGames
-
-	for _, game := range games {
-		comparisons = append(comparisons, ComparedGame{
-			Game:  game,
-			Index: levenshtein.ComputeDistance(search, game.Name),
-		})
-	}
-
-	sort.Sort(comparisons)
-
-	return &comparisons[0].Game
+	return choose.Game(gameName, results), nil
 }

igdb/game.go

@@ -36,7 +36,7 @@ type Cover struct {
 	ImageID string `json:"image_id,required"`
 }
 
-type Games []Game
+type Games []*Game
 type Platforms []Platform
 type Franchises []Franchise
 type Genres []Genre