cmd/sojudb: use User.SetPassword

This commit is contained in:
Simon Ser 2023-10-31 23:59:29 +01:00
parent 4f7b577734
commit edaae4d06b
1 changed files with 11 additions and 17 deletions

View File

@ -9,7 +9,6 @@ import (
"log" "log"
"os" "os"
"golang.org/x/crypto/bcrypt"
"golang.org/x/crypto/ssh/terminal" "golang.org/x/crypto/ssh/terminal"
"git.sr.ht/~emersion/soju/config" "git.sr.ht/~emersion/soju/config"
@ -69,14 +68,11 @@ func main() {
log.Fatalf("failed to read password: %v", err) log.Fatalf("failed to read password: %v", err)
} }
hashed, err := bcrypt.GenerateFromPassword(password, bcrypt.DefaultCost)
if err != nil {
log.Fatalf("failed to hash password: %v", err)
}
user := database.NewUser(username) user := database.NewUser(username)
user.Password = string(hashed)
user.Admin = *admin user.Admin = *admin
if err := user.SetPassword(password); err != nil {
log.Fatalf("failed to set user password: %v", err)
}
if err := db.StoreUser(ctx, user); err != nil { if err := db.StoreUser(ctx, user); err != nil {
log.Fatalf("failed to create user: %v", err) log.Fatalf("failed to create user: %v", err)
} }
@ -97,12 +93,10 @@ func main() {
log.Fatalf("failed to read password: %v", err) log.Fatalf("failed to read password: %v", err)
} }
hashed, err := bcrypt.GenerateFromPassword(password, bcrypt.DefaultCost) if err := user.SetPassword(password); err != nil {
if err != nil { log.Fatalf("failed to set user password: %v", err)
log.Fatalf("failed to hash password: %v", err)
} }
user.Password = string(hashed)
if err := db.StoreUser(ctx, user); err != nil { if err := db.StoreUser(ctx, user); err != nil {
log.Fatalf("failed to update password: %v", err) log.Fatalf("failed to update password: %v", err)
} }
@ -114,7 +108,7 @@ func main() {
} }
} }
func readPassword() ([]byte, error) { func readPassword() (string, error) {
var password []byte var password []byte
var err error var err error
fd := int(os.Stdin.Fd()) fd := int(os.Stdin.Fd())
@ -123,7 +117,7 @@ func readPassword() ([]byte, error) {
fmt.Printf("Password: ") fmt.Printf("Password: ")
password, err = terminal.ReadPassword(int(os.Stdin.Fd())) password, err = terminal.ReadPassword(int(os.Stdin.Fd()))
if err != nil { if err != nil {
return nil, err return "", err
} }
fmt.Printf("\n") fmt.Printf("\n")
} else { } else {
@ -132,16 +126,16 @@ func readPassword() ([]byte, error) {
scanner := bufio.NewScanner(os.Stdin) scanner := bufio.NewScanner(os.Stdin)
if !scanner.Scan() { if !scanner.Scan() {
if err := scanner.Err(); err != nil { if err := scanner.Err(); err != nil {
return nil, err return "", err
} }
return nil, io.ErrUnexpectedEOF return "", io.ErrUnexpectedEOF
} }
password = scanner.Bytes() password = scanner.Bytes()
if len(password) == 0 { if len(password) == 0 {
return nil, fmt.Errorf("zero length password") return "", fmt.Errorf("zero length password")
} }
} }
return password, nil return string(password), nil
} }