card.go

//go:generate stringer -type=Suit,Rank

package deck

import (
	"fmt"
	"math/rand"
	"sort"
	"time"
)

type Suit uint8

const (
	Spade Suit = iota
	Diamond 
	Club 
	Heart 
	Joker
)

var suits = [...]Suit{
	Spade,
	Diamond,
	Club,
	Heart,
}


type Rank uint8

const (
	_ Rank = iota
	Ace 
	Two
	Three
	Four
	Five
	Six
	Seven
	Eight
	Nine
	Ten
	Jack
	Queen
	King
)

const (
	minRank = Ace
	maxRank = King
)


type Card struct {
	Suit 
	Rank
	//suit
	//rank

}




func (c Card) String() string {
	if c.Suit == Joker {
		return c.Suit.String()
	}

	

	return fmt.Sprintf("%s of %ss", c.Rank.String(), c.Suit.String())
}




// type Deck []Card




func New(opts ...func([]Card) []Card) []Card {
	var cards []Card
 for _, suit := range suits {
	for rank := minRank; rank <= maxRank; rank++ {
		cards = append(cards, Card{Suit: suit, Rank: rank})
	}
 }

 for _,opt := range opts {
	cards = opt(cards)
 }
	return cards
}

func DefaultSort(cards []Card) []Card {
	sort.Slice(cards, Less(cards))
	return cards
}


func Sort(less func(cards []Card) func(i,j int) bool) func([]Card) []Card {
	return func(cards []Card) []Card {
		sort.Slice(cards, less(cards))
		return cards
	}
}



func Less(cards []Card) func(i,j int) bool {
	return func(i,j int) bool {
		return absRank(cards[i]) < absRank(cards[j])
	}
}

func absRank(c Card) int {
	return int(c.Suit)*int(maxRank) + int(c.Rank)
}

func Shuffle(cards []Card) []Card {
	ret := make([]Card, len(cards))
	r := rand.New(rand.NewSource(time.Now().Unix()))
	perm := r.Perm(len(cards))
	for i,j := range perm {
		ret[i] = cards[j]
	}
	return ret
}



func Jokers(n int) func([]Card) []Card {
	return func(cards []Card) []Card {
		for i:= 0; i< n; i++ {
			cards = append(cards, Card{
				Suit: Joker,
				Rank: Rank(i),
			})
		}
		return cards
	}
}


func Filter(f func(card Card) bool) func([]Card) []Card {
	return func(cards []Card) []Card {
		var ret []Card
		for _,c := range cards {
			if !f(c) {
				ret = append(ret, c)
			}
		}
		return ret
	}
}


func Deck(n int) func([]Card) []Card{
	return func(card []Card) []Card{
		var ret []Card
		for i :=0; i<n;i++ {
			ret = append(ret,card...)
		}

		return  ret
	}

}