89 lines
1.5 KiB
Go
89 lines
1.5 KiB
Go
package pwgenerator
|
|
|
|
/*
|
|
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
|
|
}
|
|
|
|
// 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
|
|
|
|
if selectIdx, err = saferRandInt(len(*c)); err != nil {
|
|
return
|
|
}
|
|
|
|
char = (*c)[selectIdx]
|
|
|
|
return
|
|
}
|
|
|
|
// 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]
|
|
|
|
return
|
|
}
|