123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990 |
- package setup
- import (
- "context"
- "errors"
- "fmt"
- "golang.org/x/crypto/bcrypt"
- "github.com/usememos/memos/api"
- "github.com/usememos/memos/common"
- )
- func Execute(
- ctx context.Context,
- store store,
- hostUsername, hostPassword string,
- ) error {
- s := setupService{store: store}
- return s.Setup(ctx, hostUsername, hostPassword)
- }
- type store interface {
- FindUserList(ctx context.Context, find *api.UserFind) ([]*api.User, error)
- CreateUser(ctx context.Context, create *api.UserCreate) (*api.User, error)
- }
- type setupService struct {
- store store
- }
- func (s setupService) Setup(
- ctx context.Context,
- hostUsername, hostPassword string,
- ) error {
- if err := s.makeSureHostUserNotExists(ctx); err != nil {
- return err
- }
- if err := s.createUser(ctx, hostUsername, hostPassword); err != nil {
- return fmt.Errorf("create user: %w", err)
- }
- return nil
- }
- func (s setupService) makeSureHostUserNotExists(ctx context.Context) error {
- hostUserType := api.Host
- existedHostUsers, err := s.store.FindUserList(ctx, &api.UserFind{
- Role: &hostUserType,
- })
- if err != nil {
- return fmt.Errorf("find user list: %w", err)
- }
- if len(existedHostUsers) != 0 {
- return errors.New("host user already exists")
- }
- return nil
- }
- func (s setupService) createUser(
- ctx context.Context,
- hostUsername, hostPassword string,
- ) error {
- userCreate := &api.UserCreate{
- Username: hostUsername,
- // The new signup user should be normal user by default.
- Role: api.Host,
- Nickname: hostUsername,
- Password: hostPassword,
- OpenID: common.GenUUID(),
- }
- if err := userCreate.Validate(); err != nil {
- return fmt.Errorf("validate: %w", err)
- }
- passwordHash, err := bcrypt.GenerateFromPassword([]byte(hostPassword), bcrypt.DefaultCost)
- if err != nil {
- return fmt.Errorf("hash password: %w", err)
- }
- userCreate.PasswordHash = string(passwordHash)
- if _, err := s.store.CreateUser(ctx, userCreate); err != nil {
- return fmt.Errorf("create user: %w", err)
- }
- return nil
- }
|