woj-server/internal/service/user/profile.go

package user

import (
	"errors"
	"git.0x7f.app/WOJ/woj-server/internal/e"
	"git.0x7f.app/WOJ/woj-server/internal/model"
	"go.uber.org/zap"
	"gorm.io/gorm"
)

func (s *service) Profile(uid uint) (*model.User, e.Status) {
	user := new(model.User)

	err := s.db.Get().First(&user, uid).Error
	if errors.Is(err, gorm.ErrRecordNotFound) {
		return nil, e.UserNotFound
	}
	if err != nil {
		s.log.Warn("DatabaseError", zap.Error(err), zap.Any("uid", uid))
		return nil, e.DatabaseError
	}

	return user, e.Success
}

func (s *service) ProfileOrCreate(data *CreateData) (*model.User, e.Status) {
	user := &model.User{
		UserName:  data.UserName,
		NickName:  data.NickName,
		Role:      model.RoleGeneral,
		IsEnabled: true,
	}

	// Notice: FirstOrCreate will not update the record if it exists, and also we should not update the record
	// Notice: OAuth2 created user will not have password
	err := s.db.Get().Where(model.User{UserName: data.UserName}).FirstOrCreate(&user, data).Error
	if err != nil {
		s.log.Warn("DatabaseError", zap.Error(err), zap.Any("user", user))
		return nil, e.DatabaseError
	}

	return user, e.Success
}