diff --git a/grading_data_generator.go b/grading_data_generator.go index 490c865..3ba3d98 100644 --- a/grading_data_generator.go +++ b/grading_data_generator.go @@ -535,6 +535,7 @@ func GenerateUserData(config Config) []User { LastName: gofakeit.LastName(), Status: weightedStatuses[rand.Intn(len(weightedStatuses))], ExamID: examIDs[i], + Username: gofakeit.Username(), APIKey: strings.ReplaceAll(gofakeit.UUID(), "-", ""), PasswordHash: gofakeit.Password(true, true, true, true, false, 16), } @@ -553,11 +554,11 @@ func SaveGeneratedUsers(users []User) { defer f.Close() w := csv.NewWriter(f) - if err := w.Write([]string{"uid", "first_name", "last_name", "status", "exam_id", "api_key", "password_hash"}); err != nil { + if err := w.Write([]string{"uid", "first_name", "last_name", "status", "exam_id", "username", "api_key", "password_hash"}); err != nil { log.Fatalf("failed to write csv header: %v", err) } for _, u := range users { - if err := w.Write([]string{u.UID, u.FirstName, u.LastName, u.Status, u.ExamID, u.APIKey, u.PasswordHash}); err != nil { + if err := w.Write([]string{u.UID, u.FirstName, u.LastName, u.Status, u.ExamID, u.Username, u.APIKey, u.PasswordHash}); err != nil { log.Fatalf("failed to write csv row: %v", err) } } diff --git a/prompts/add_username.md b/prompts/add_username.md new file mode 100644 index 0000000..48d543d --- /dev/null +++ b/prompts/add_username.md @@ -0,0 +1 @@ +Add Username to the GenerateUserData function using gofakeit. Also add this to SaveGeneratedUsers.