zrok/controller/changePassword.go

package controller

import (
	"github.com/go-openapi/runtime/middleware"
	"github.com/openziti/zrok/controller/config"
	"github.com/openziti/zrok/rest_model_zrok"
	"github.com/openziti/zrok/rest_server_zrok/operations/account"
	"github.com/sirupsen/logrus"
)

type changePasswordHandler struct {
	cfg *config.Config
}

func newChangePasswordHandler(cfg *config.Config) *changePasswordHandler {
	return &changePasswordHandler{
		cfg: cfg,
	}
}

func (handler *changePasswordHandler) Handle(params account.ChangePasswordParams, principal *rest_model_zrok.Principal) middleware.Responder {
	if params.Body == nil || params.Body.Email == "" || params.Body.OldPassword == "" || params.Body.NewPassword == "" {
		logrus.Error("missing email, old, or new password")
		return account.NewChangePasswordUnauthorized()
	}
	logrus.Infof("received change password request for email '%v'", params.Body.Email)

	tx, err := str.Begin()
	if err != nil {
		logrus.Errorf("error starting transaction: %v", err)
		return account.NewChangePasswordUnauthorized()
	}
	defer func() { _ = tx.Rollback() }()

	a, err := str.FindAccountWithEmail(params.Body.Email, tx)
	if err != nil {
		logrus.Errorf("error finding account '%v': %v", params.Body.Email, err)
		return account.NewChangePasswordUnauthorized()
	}
	ohpwd, err := rehashPassword(params.Body.OldPassword, a.Salt)
	if err != nil {
		logrus.Errorf("error hashing password for '%v': %v", params.Body.Email, err)
		return account.NewChangePasswordUnauthorized()
	}
	if a.Password != ohpwd.Password {
		logrus.Errorf("password mismatch for account '%v'", params.Body.Email)
		return account.NewChangePasswordUnauthorized()
	}

	if err := validatePassword(handler.cfg, params.Body.NewPassword); err != nil {
		logrus.Errorf("password not valid for request '%v': %v", a.Email, err)
		return account.NewChangePasswordUnprocessableEntity().WithPayload(rest_model_zrok.ErrorMessage(err.Error()))
	}

	nhpwd, err := hashPassword(params.Body.NewPassword)
	if err != nil {
		logrus.Errorf("error hashing password for '%v': %v", a.Email, err)
		return account.NewChangePasswordInternalServerError()
	}
	a.Salt = nhpwd.Salt
	a.Password = nhpwd.Password

	if _, err := str.UpdateAccount(a, tx); err != nil {
		logrus.Errorf("error updating for '%v': %v", a.Email, err)
		return account.NewChangePasswordInternalServerError()
	}

	if err := tx.Commit(); err != nil {
		logrus.Errorf("error committing '%v': %v", a.Email, err)
		return account.NewChangePasswordInternalServerError()
	}

	logrus.Infof("change password for '%v'", a.Email)
	return account.NewChangePasswordOK()
}