37 lines
932 B
Go
37 lines
932 B
Go
package scripts
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/andrew/go-computer/internal/db"
|
|
"golang.org/x/crypto/bcrypt"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
// Seed only creates the requested account; it never clears existing records.
|
|
func Seed(ctx context.Context, d *db.DB) error {
|
|
email, password := strings.ToLower(strings.TrimSpace(os.Getenv("SEED_EMAIL"))), os.Getenv("SEED_PASSWORD")
|
|
if email == "" || password == "" {
|
|
return fmt.Errorf("SEED_EMAIL and SEED_PASSWORD are required")
|
|
}
|
|
repo := db.NewUserRepo(d)
|
|
if _, e := repo.FindByEmail(ctx, email); e == nil {
|
|
return nil
|
|
}
|
|
hash, e := bcrypt.GenerateFromPassword([]byte(password), 12)
|
|
if e != nil {
|
|
return e
|
|
}
|
|
name := os.Getenv("SEED_NAME")
|
|
if name == "" {
|
|
name = "Admin"
|
|
}
|
|
role := "USER"
|
|
if db.IsAdminEmail(email) {
|
|
role = "ADMIN"
|
|
}
|
|
_, e = repo.Create(ctx, &db.NewUserInput{ID: db.NewID(), Name: name, Email: email, Password: string(hash), Role: role})
|
|
return e
|
|
}
|