fakermaker: add support for invite code and domain suffix

This gets the tool working with, eg, staging

cmd/fakermaker/README.md

···
     export GOLOG_LOG_LEVEL=info
 
     # setup and create initial accounts; 100 by default
+    # supply --use-invite-code and/or --domain-suffix SUFFIX as needed
 	go run ./cmd/fakermaker/ gen-accounts > data/fakermaker/accounts.json
 
     # create or update profiles for all the accounts

cmd/fakermaker/main.go

···
 package main
 
 import (
+	"context"
 	"encoding/json"
 	"fmt"
 	"os"
 	"runtime"
 
+	comatproto "github.com/bluesky-social/indigo/api/atproto"
 	"github.com/bluesky-social/indigo/fakedata"
 	"github.com/bluesky-social/indigo/util/cliutil"
 	"github.com/bluesky-social/indigo/util/version"
···
 					Name:  "count-celebrities",
 					Usage: "number of accounts as 'celebrities' (many followers)",
 					Value: 10,
+				},
+				&cli.StringFlag{
+					Name:  "domain-suffix",
+					Usage: "domain to register handle under",
+					Value: "test",
+				},
+				&cli.BoolFlag{
+					Name:  "use-invite-code",
+					Usage: "create and use an invite code",
+					Value: false,
 				},
 			},
 		},
···
 
 	countTotal := cctx.Int("count")
 	countCelebrities := cctx.Int("count-celebrities")
+	domainSuffix := cctx.String("domain-suffix")
 	if countCelebrities > countTotal {
 		return fmt.Errorf("more celebrities than total accounts!")
 	}
 	countRegulars := countTotal - countCelebrities
 
+	var inviteCode *string = nil
+	if cctx.Bool("use-invite-code") {
+		resp, err := comatproto.ServerCreateInviteCodes(context.TODO(), xrpcc, &comatproto.ServerCreateInviteCodes_Input{
+			UseCount:    int64(countTotal),
+			ForAccounts: nil,
+			CodeCount:   1,
+		})
+		if err != nil {
+			return err
+		}
+		if len(resp.Codes) != 1 || len(resp.Codes[0].Codes) != 1 {
+			return fmt.Errorf("expected a single invite code")
+		}
+		inviteCode = &resp.Codes[0].Codes[0]
+	}
+
 	// call helper to do actual creation
 	var usr *fakedata.AccountContext
 	var line []byte
 	t1 := fakedata.MeasureIterations("register celebrity accounts")
 	for i := 0; i < countCelebrities; i++ {
-		if usr, err = fakedata.GenAccount(xrpcc, i, "celebrity"); err != nil {
+		if usr, err = fakedata.GenAccount(xrpcc, i, "celebrity", domainSuffix, inviteCode); err != nil {
 			return err
 		}
 		// compact single-line JSON by default
···
 
 	t2 := fakedata.MeasureIterations("register regular accounts")
 	for i := 0; i < countRegulars; i++ {
-		if usr, err = fakedata.GenAccount(xrpcc, i, "regular"); err != nil {
+		if usr, err = fakedata.GenAccount(xrpcc, i, "regular", domainSuffix, inviteCode); err != nil {
 			return err
 		}
 		// compact single-line JSON by default

fakedata/generators.go

···
 	}
 }
 
-func GenAccount(xrpcc *xrpc.Client, index int, accountType string) (*AccountContext, error) {
-	var suffix string
+func GenAccount(xrpcc *xrpc.Client, index int, accountType, domainSuffix string, inviteCode *string) (*AccountContext, error) {
+	if domainSuffix == "" {
+		domainSuffix = "test"
+	}
+	var handleSuffix string
 	if accountType == "celebrity" {
-		suffix = "C"
+		handleSuffix = "C"
 	} else {
-		suffix = ""
+		handleSuffix = ""
 	}
 	prefix := gofakeit.Username()
 	if len(prefix) > 10 {
 		prefix = prefix[0:10]
 	}
-	handle := fmt.Sprintf("%s-%s%d.test", prefix, suffix, index)
+	handle := fmt.Sprintf("%s-%s%d.%s", prefix, handleSuffix, index, domainSuffix)
 	email := gofakeit.Email()
 	password := gofakeit.Password(true, true, true, true, true, 24)
 	ctx := context.TODO()
 	resp, err := comatproto.ServerCreateAccount(ctx, xrpcc, &comatproto.ServerCreateAccount_Input{
-		Email:    email,
-		Handle:   handle,
-		Password: password,
+		Email:      email,
+		Handle:     handle,
+		InviteCode: inviteCode,
+		Password:   password,
 	})
 	if err != nil {
 		return nil, err

testing/pds_fakedata_test.go

···
 
 	var catalog fakedata.AccountCatalog
 	for i := 0; i < celebCount; i++ {
-		usr, err := fakedata.GenAccount(&xrpccAdmin, i, "celebrity")
+		usr, err := fakedata.GenAccount(&xrpccAdmin, i, "celebrity", "", nil)
 		if err != nil {
 			t.Fatal(err)
 		}
 		catalog.Celebs = append(catalog.Celebs, *usr)
 	}
 	for i := 0; i < regularCount; i++ {
-		usr, err := fakedata.GenAccount(&xrpccAdmin, i, "regular")
+		usr, err := fakedata.GenAccount(&xrpccAdmin, i, "regular", "", nil)
 		if err != nil {
 			t.Fatal(err)
 		}