Feat: add extra/cluedo [WIP]

This commit is contained in:
Grail Finder
2025-02-26 16:02:48 +03:00
parent 97a1fc507e
commit 49409f5d94
2 changed files with 123 additions and 0 deletions

73
extra/cluedo.go Normal file
View File

@@ -0,0 +1,73 @@
package extra
import (
"math/rand"
"strings"
)
var (
rooms = []string{"HALL", "LOUNGE", "DINING ROOM", "KITCHEN", "BALLROOM", "CONSERVATORY", "BILLIARD ROOM", "LIBRARY", "STUDY"}
weapons = []string{"CANDLESTICK", "DAGGER", "LEAD PIPE", "REVOLVER", "ROPE", "SPANNER"}
people = []string{"Miss Scarlett", "Colonel Mustard", "Mrs. White", "Reverend Green", "Mrs. Peacock", "Professor Plum"}
)
type MurderTrifecta struct {
Murderer string
Weapon string
Room string
}
type CluedoRoundInfo struct {
Answer MurderTrifecta
PlayersCards map[string][]string
}
func (c *CluedoRoundInfo) GetPlayerCards(player string) string {
// maybe format it a little
return "cards of " + player + "are " + strings.Join(c.PlayersCards[player], ",")
}
func CluedoPrepCards(playerOrder []string) *CluedoRoundInfo {
res := &CluedoRoundInfo{}
// Select murder components
trifecta := MurderTrifecta{
Murderer: people[rand.Intn(len(people))],
Weapon: weapons[rand.Intn(len(weapons))],
Room: rooms[rand.Intn(len(rooms))],
}
// Collect non-murder cards
var notInvolved []string
for _, room := range rooms {
if room != trifecta.Room {
notInvolved = append(notInvolved, room)
}
}
for _, weapon := range weapons {
if weapon != trifecta.Weapon {
notInvolved = append(notInvolved, weapon)
}
}
for _, person := range people {
if person != trifecta.Murderer {
notInvolved = append(notInvolved, person)
}
}
// Shuffle and distribute cards
rand.Shuffle(len(notInvolved), func(i, j int) {
notInvolved[i], notInvolved[j] = notInvolved[j], notInvolved[i]
})
players := map[string][]string{}
cardsPerPlayer := len(notInvolved) / len(playerOrder)
// playerOrder := []string{"{{user}}", "{{char}}", "{{char2}}"}
for i, player := range playerOrder {
start := i * cardsPerPlayer
end := (i + 1) * cardsPerPlayer
if end > len(notInvolved) {
end = len(notInvolved)
}
players[player] = notInvolved[start:end]
}
res.Answer = trifecta
res.PlayersCards = players
return res
}

50
extra/cluedo_test.go Normal file
View File

@@ -0,0 +1,50 @@
package extra
import (
"testing"
)
func TestPrepCards(t *testing.T) {
// Run the function to get the murder combination and player cards
roundInfo := CluedoPrepCards([]string{"{{user}}", "{{char}}", "{{char2}}"})
// Create a map to track all distributed cards
distributedCards := make(map[string]bool)
// Check that the murder combination cards are not distributed to players
murderCards := []string{roundInfo.Answer.Murderer, roundInfo.Answer.Weapon, roundInfo.Answer.Room}
for _, card := range murderCards {
if distributedCards[card] {
t.Errorf("Murder card %s was distributed to a player", card)
}
}
// Check each player's cards
for player, cards := range roundInfo.PlayersCards {
for _, card := range cards {
// Ensure the card is not part of the murder combination
for _, murderCard := range murderCards {
if card == murderCard {
t.Errorf("Player %s has a murder card: %s", player, card)
}
}
// Ensure the card is unique and not already distributed
if distributedCards[card] {
t.Errorf("Card %s is duplicated in player %s's hand", card, player)
}
distributedCards[card] = true
}
}
// Verify that all non-murder cards are distributed
allCards := append(append([]string{}, rooms...), weapons...)
allCards = append(allCards, people...)
for _, card := range allCards {
isMurderCard := false
for _, murderCard := range murderCards {
if card == murderCard {
isMurderCard = true
break
}
}
if !isMurderCard && !distributedCards[card] {
t.Errorf("Card %s was not distributed to any player", card)
}
}
}