Address PR review: graceful shutdown, Role/NewUser, drop SQLite.

internal/web/memstore_test.go

@@ -18,11 +18,11 @@ import (
 // memDB is an in-memory store.DB for tests.
 type memDB struct {
 	mu        sync.Mutex
-	users     map[string]*store.User            // id -> user
-	byName    map[string]string                 // username -> id
-	questions map[string]*store.RankedQuestion  // id -> question
-	votes     map[string]int                    // userID|questionID -> value
-	answers   map[string]*store.Answer          // questionID -> answer
+	users     map[string]*store.User           // id -> user
+	byName    map[string]string                // username -> id
+	questions map[string]*store.RankedQuestion // id -> question
+	votes     map[string]int                   // userID|questionID -> value
+	answers   map[string]*store.Answer         // questionID -> answer
 }
 
 func newMemDB() *memDB {
@@ -39,23 +39,22 @@ func voteKey(userID, questionID string) string {
 	return userID + "|" + questionID
 }
 
-func (m *memDB) CreateUser(_ context.Context, username, passwordHash string, asAdmin bool) (*store.User, error) {
+func (m *memDB) CreateUser(_ context.Context, nu store.NewUser) (*store.User, error) {
 	m.mu.Lock()
 	defer m.mu.Unlock()
-	username = store.NormalizeUsername(username)
+	username := store.NormalizeUsername(nu.Username)
 	if _, ok := m.byName[username]; ok {
 		return nil, fmt.Errorf("username taken")
 	}
-	role := "user"
-	if asAdmin {
-		role = "admin"
+	if nu.Role != store.RoleUser && nu.Role != store.RoleAdmin {
+		return nil, fmt.Errorf("invalid role")
 	}
 	u := &store.User{
 		ID:           uuid.NewString(),
 		Username:     username,
 		Name:         username,
-		Role:         role,
-		PasswordHash: passwordHash,
+		Role:         nu.Role,
+		PasswordHash: nu.PasswordHash,
 		CreatedAt:    time.Now().UTC().Format(time.RFC3339),
 	}
 	m.users[u.ID] = u
@@ -92,7 +91,7 @@ func (m *memDB) CountAdmins(_ context.Context) (int, error) {
 	defer m.mu.Unlock()
 	n := 0
 	for _, u := range m.users {
-		if u.Role == "admin" {
+		if u.Role == store.RoleAdmin {
 			n++
 		}
 	}
@@ -114,8 +113,8 @@ func (m *memDB) ListUsers(_ context.Context) ([]store.User, error) {
 	return out, nil
 }
 
-func (m *memDB) SetRole(_ context.Context, userID, role string) error {
-	if role != "user" && role != "admin" {
+func (m *memDB) SetRole(_ context.Context, userID string, role store.Role) error {
+	if role != store.RoleUser && role != store.RoleAdmin {
 		return fmt.Errorf("invalid role")
 	}
 	m.mu.Lock()
@@ -124,10 +123,10 @@ func (m *memDB) SetRole(_ context.Context, userID, role string) error {
 	if !ok {
 		return sql.ErrNoRows
 	}
-	if u.Role == "admin" && role == "user" {
+	if u.Role == store.RoleAdmin && role == store.RoleUser {
 		n := 0
 		for _, x := range m.users {
-			if x.Role == "admin" {
+			if x.Role == store.RoleAdmin {
 				n++
 			}
 		}