energylink/vendor/github.com/brianvoe/gofakeit/payment.go

package gofakeit

import "strconv"
import "time"

var currentYear = time.Now().Year() - 2000

// CreditCardInfo is a struct containing credit variables
type CreditCardInfo struct {
	Type   string
	Number int
	Exp    string
	Cvv    string
}

// CreditCard will generate a struct full of credit card information
func CreditCard() *CreditCardInfo {
	return &CreditCardInfo{
		Type:   CreditCardType(),
		Number: CreditCardNumber(),
		Exp:    CreditCardExp(),
		Cvv:    CreditCardCvv(),
	}
}

// CreditCardType will generate a random credit card type string
func CreditCardType() string {
	return getRandValue([]string{"payment", "card_type"})
}

// CreditCardNumber will generate a random credit card number int
func CreditCardNumber() int {
	integer, _ := strconv.Atoi(replaceWithNumbers(getRandValue([]string{"payment", "number"})))
	return integer
}

// CreditCardNumberLuhn will generate a random credit card number int that passes luhn test
func CreditCardNumberLuhn() int {
	cc := ""
	for i := 0; i < 100000; i++ {
		cc = replaceWithNumbers(getRandValue([]string{"payment", "number"}))
		if luhn(cc) {
			break
		}
	}
	integer, _ := strconv.Atoi(cc)
	return integer
}

// CreditCardExp will generate a random credit card expiration date string
// Exp date will always be a future date
func CreditCardExp() string {
	month := strconv.Itoa(randIntRange(1, 12))
	if len(month) == 1 {
		month = "0" + month
	}
	return month + "/" + strconv.Itoa(randIntRange(currentYear+1, currentYear+10))
}

// CreditCardCvv will generate a random CVV number - Its a string because you could have 017 as an exp date
func CreditCardCvv() string {
	return Numerify("###")
}

// luhn check is used for checking if credit card is valid
func luhn(s string) bool {
	var t = [...]int{0, 2, 4, 6, 8, 1, 3, 5, 7, 9}
	odd := len(s) & 1
	var sum int
	for i, c := range s {
		if c < '0' || c > '9' {
			return false
		}
		if i&1 == odd {
			sum += t[c-'0']
		} else {
			sum += int(c - '0')
		}
	}
	return sum%10 == 0
}