forked from brianvoe/gofakeit
-
Notifications
You must be signed in to change notification settings - Fork 0
/
auth.go
163 lines (141 loc) · 4.65 KB
/
auth.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
package gofakeit
// Username will generate a random username based upon picking a random lastname and random numbers at the end
func Username() string {
return username(GlobalFaker)
}
// Username will generate a random username based upon picking a random lastname and random numbers at the end
func (f *Faker) Username() string {
return username(f)
}
func username(f *Faker) string {
return getRandValue(f, []string{"person", "last"}) + replaceWithNumbers(f, "####")
}
// Password will generate a random password.
// Minimum number length of 5 if less than.
func Password(lower bool, upper bool, numeric bool, special bool, space bool, num int) string {
return password(GlobalFaker, lower, upper, numeric, special, space, num)
}
// Password will generate a random password.
// Minimum number length of 5 if less than.
func (f *Faker) Password(lower bool, upper bool, numeric bool, special bool, space bool, num int) string {
return password(f, lower, upper, numeric, special, space, num)
}
func password(f *Faker, lower bool, upper bool, numeric bool, special bool, space bool, num int) string {
// Make sure the num minimum is at least 5
if num < 5 {
num = 5
}
// Setup weights
items := make([]any, 0)
weights := make([]float32, 0)
if lower {
items = append(items, "l")
weights = append(weights, 4)
}
if upper {
items = append(items, "u")
weights = append(weights, 4)
}
if numeric {
items = append(items, "n")
weights = append(weights, 3)
}
if special {
items = append(items, "e")
weights = append(weights, 2)
}
if space {
items = append(items, "a")
weights = append(weights, 1)
}
// If no items are selected then default to lower, upper, numeric
if len(items) == 0 {
items = append(items, "l", "u", "n")
weights = append(weights, 4, 4, 3)
}
// Create byte slice
b := make([]byte, num)
for i := 0; i <= num-1; i++ {
// Run weighted
weight, _ := weighted(f, items, weights)
switch weight.(string) {
case "l":
b[i] = lowerStr[f.Int64()%int64(len(lowerStr))]
case "u":
b[i] = upperStr[f.Int64()%int64(len(upperStr))]
case "n":
b[i] = numericStr[f.Int64()%int64(len(numericStr))]
case "e":
b[i] = specialSafeStr[f.Int64()%int64(len(specialSafeStr))]
case "a":
b[i] = spaceStr[f.Int64()%int64(len(spaceStr))]
}
}
// Shuffle bytes
for i := range b {
j := f.IntN(i + 1)
b[i], b[j] = b[j], b[i]
}
// Replace first or last character if it's a space, and other options are available
if b[0] == ' ' {
b[0] = password(f, lower, upper, numeric, special, false, 1)[0]
}
if b[len(b)-1] == ' ' {
b[len(b)-1] = password(f, lower, upper, numeric, special, false, 1)[0]
}
return string(b)
}
func addAuthLookup() {
AddFuncLookup("username", Info{
Display: "Username",
Category: "auth",
Description: "Unique identifier assigned to a user for accessing an account or system",
Example: "Daniel1364",
Output: "string",
Generate: func(f *Faker, m *MapParams, info *Info) (any, error) {
return username(f), nil
},
})
AddFuncLookup("password", Info{
Display: "Password",
Category: "auth",
Description: "Secret word or phrase used to authenticate access to a system or account",
Example: "EEP+wwpk 4lU-eHNXlJZ4n K9%v&TZ9e",
Output: "string",
Params: []Param{
{Field: "lower", Display: "Lower", Type: "bool", Default: "true", Description: "Whether or not to add lower case characters"},
{Field: "upper", Display: "Upper", Type: "bool", Default: "true", Description: "Whether or not to add upper case characters"},
{Field: "numeric", Display: "Numeric", Type: "bool", Default: "true", Description: "Whether or not to add numeric characters"},
{Field: "special", Display: "Special", Type: "bool", Default: "true", Description: "Whether or not to add special characters"},
{Field: "space", Display: "Space", Type: "bool", Default: "false", Description: "Whether or not to add spaces"},
{Field: "length", Display: "Length", Type: "int", Default: "12", Description: "Number of characters in password"},
},
Generate: func(f *Faker, m *MapParams, info *Info) (any, error) {
lower, err := info.GetBool(m, "lower")
if err != nil {
return nil, err
}
upper, err := info.GetBool(m, "upper")
if err != nil {
return nil, err
}
numeric, err := info.GetBool(m, "numeric")
if err != nil {
return nil, err
}
special, err := info.GetBool(m, "special")
if err != nil {
return nil, err
}
space, err := info.GetBool(m, "space")
if err != nil {
return nil, err
}
length, err := info.GetInt(m, "length")
if err != nil {
return nil, err
}
return password(f, lower, upper, numeric, special, space, length), nil
},
})
}