PWGen/pwgenerator/funcs_charset.go

89 lines
1.5 KiB
Go
Raw Normal View History

2022-03-02 06:24:55 -05:00
package pwgenerator
2022-03-03 05:51:22 -05:00
/*
Has returns true if a CharSet contains char.
char can be one of: Char, byte, rune, or a string (of length 1).
*/
func (c *CharSet) Has(char interface{}) (contains bool, err error) {
var chkChar Char
switch t := char.(type) {
case Char:
chkChar = t
case byte:
chkChar = Char(t)
case rune:
chkChar = Char(t)
case string:
if len(t) != 1 {
err = ErrBadType
return
}
chkChar = Char(t[0])
default:
err = ErrBadType
return
}
for _, i := range *c {
if chkChar == i {
contains = true
break
}
}
return
}
2022-03-02 06:24:55 -05:00
// Len returns the length of a CharSet (needed for sort.Interface).
func (c *CharSet) Len() (l int) {
l = len(*c)
return
}
// Less returns true if item at index i is "less than" (sorts before) item at index j (needed for sort.Interface).
func (c *CharSet) Less(i, j int) (isBefore bool) {
if (*c)[i] < (*c)[j] {
isBefore = true
}
return
}
// RandChar returns a random character from a CharSet.
func (c *CharSet) RandChar() (char Char, err error) {
var selectIdx int
2022-03-03 05:51:22 -05:00
if selectIdx, err = saferRandInt(len(*c)); err != nil {
return
}
char = (*c)[selectIdx]
return
}
2022-03-02 06:24:55 -05:00
// String returns a string from a CharSet.
func (c *CharSet) String() (s string) {
for _, i := range *c {
s += string(i)
}
return
}
// Swap will swap the position of the item at index i and the item at index j in a CharSet (needed for sort.Interface).
func (c *CharSet) Swap(i, j int) {
(*c)[j], (*c)[i] = (*c)[i], (*c)[j]
2022-03-02 06:24:55 -05:00
return
}