Feat: word loader

wordloader/wordloader.go

@@ -0,0 +1,80 @@
+package wordloader
+
+import (
+	"golias/models"
+	"math/rand/v2"
+	"os"
+	"strings"
+)
+
+type Loader struct {
+	FilePath   string
+	BlackCount uint8
+	WhiteCount uint8
+	RedCount   uint8
+	BlueCount  uint8
+}
+
+func InitDefaultLoader(fpath string) *Loader {
+	return &Loader{
+		FilePath:   fpath,
+		BlackCount: 1,
+		WhiteCount: 7,
+		RedCount:   8,
+		BlueCount:  9,
+	}
+}
+
+func (l *Loader) WholeCount() uint8 {
+	return l.BlackCount + l.WhiteCount + l.RedCount + l.BlueCount
+}
+
+func (l *Loader) Load() ([]models.WordCard, error) {
+	data, err := os.ReadFile(l.FilePath)
+	if err != nil {
+		return nil, err
+	}
+	lines := strings.Split(string(data), "\n")
+	rand.Shuffle(len(lines), func(i, j int) {
+		lines[i], lines[j] = lines[j], lines[i]
+	})
+	words := lines[:int(l.WholeCount())]
+	cards := make([]models.WordCard, l.WholeCount())
+	for i, word := range words {
+		if l.BlackCount > 0 {
+			cards[i] = models.WordCard{
+				Word:  word,
+				Color: "black",
+			}
+			l.BlackCount--
+			continue
+		}
+		if l.WhiteCount > 0 {
+			cards[i] = models.WordCard{
+				Word:  word,
+				Color: "white",
+			}
+			l.WhiteCount--
+			continue
+		}
+		if l.RedCount > 0 {
+			cards[i] = models.WordCard{
+				Word:  word,
+				Color: "red",
+			}
+			l.RedCount--
+			continue
+		}
+		if l.BlueCount > 0 {
+			cards[i] = models.WordCard{
+				Word:  word,
+				Color: "blue",
+			}
+			l.BlueCount--
+		}
+	}
+	rand.Shuffle(len(cards), func(i, j int) {
+		cards[i], cards[j] = cards[j], cards[i]
+	})
+	return cards, nil
+}