IOT/SERVER/IotAdmin/app/system/service/sys_user.go

package service

import (
	"IotAdmin/app/system/models"
	"IotAdmin/app/system/service/dto"
	"errors"

	log "IotAdmin/core/logger"
	"IotAdmin/core/sdk/pkg"
	"IotAdmin/core/sdk/service"

	"gorm.io/gorm"

	cDto "IotAdmin/common/dto"
	"IotAdmin/common/permission"
)

type SysUserService struct {
	service.Service
}

// GetPage 获取SysUser列表
func (e *SysUserService) GetPage(c *dto.SysUserGetPageReq, p *permission.DataPermission, list *[]models.SysUser, count *int64) error {
	var err error
	var data models.SysUser

	err = e.Orm.Debug().
		Preload("Org").
		Scopes(
			cDto.MakeCondition(c.GetNeedSearch()),
			cDto.Paginate(c.GetPageSize(), c.GetPageIndex()),
			permission.Permission(data.TableName(), p),
		).
		Find(list).Limit(-1).Offset(-1).
		Count(count).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return err
	}
	return nil
}

// Get 获取SysUser对象
func (e *SysUserService) Get(d *dto.SysUserById, p *permission.DataPermission, model *models.SysUser) error {
	var data models.SysUser

	err := e.Orm.Model(&data).Debug().Preload("Org").Scopes(
		permission.Permission(data.TableName(), p),
	).First(model, d.GetId()).Error
	if err != nil && errors.Is(err, gorm.ErrRecordNotFound) {
		err = errors.New("查看对象不存在或无权查看")
		e.Log.Errorf("db error: %s", err)
		return err
	}
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return err
	}
	return nil
}

// GetInfo 获取SysUser对象
func (e *SysUserService) GetInfo(d *dto.SysUserById, model *models.SysUser) error {
	var data models.SysUser

	err := e.Orm.Model(&data).Debug().Preload("Org").First(model, d.GetId()).Error
	if err != nil && errors.Is(err, gorm.ErrRecordNotFound) {
		err = errors.New("查看对象不存在或无权查看")
		e.Log.Errorf("db error: %s", err)
		return err
	}
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return err
	}
	return nil
}

func (e *SysUserService) GetUser(d *dto.SysUserById, p *permission.DataPermission, model *models.SysUser) error {
	if d.GetId() != 0 {
		var data models.SysUser
		err := e.Orm.Model(&data).Debug().Scopes(
			permission.Permission(data.TableName(), p),
		).First(model, d.GetId()).Error
		if err != nil && errors.Is(err, gorm.ErrRecordNotFound) {
			err = errors.New("查看对象不存在或无权查看")
			e.Log.Errorf("db error: %s", err)
			return err
		}
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			return err
		}
	}

	var role models.SysRole
	err := e.Orm.Model(&role).Debug().Scopes(
		permission.Permission(role.TableName(), p),
	).Select("role_id,role_name,status").Find(&(model.Roles)).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return err
	}
	var post models.SysPost
	err = e.Orm.Model(&post).Debug().Scopes(
		permission.Permission(post.TableName(), p),
	).Select("post_id,post_name,status").Find(&(model.Posts)).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return err
	}
	return nil
}

// Insert 创建SysUser对象
func (e *SysUserService) Insert(c *dto.SysUserInsertReq) error {
	var err error
	var data models.SysUser
	var i int64
	err = e.Orm.Model(&data).Where("username = ?", c.Username).Count(&i).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return err
	}
	if i > 0 {
		err := errors.New("用户名已存在！")
		e.Log.Errorf("db error: %s", err)
		return err
	}
	c.Generate(&data)
	err = e.Orm.Create(&data).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return err
	}
	return nil
}

// Update 修改SysUser对象
func (e *SysUserService) Update(c *dto.SysUserUpdateReq, p *permission.DataPermission) error {
	var err error
	var model models.SysUser
	db := e.Orm.Scopes(
		permission.Permission(model.TableName(), p),
	).First(&model, c.GetId())
	if err = db.Error; err != nil {
		e.Log.Errorf("Service UpdateSysUser error: %s", err)
		return err
	}
	if db.RowsAffected == 0 {
		return errors.New("无权更新该数据")

	}
	c.Generate(&model)
	update := e.Orm.Model(&model).Where("user_id = ?", &model.UserId).Omit("password", "salt").Updates(&model)
	if err = update.Error; err != nil {
		e.Log.Errorf("db error: %s", err)
		return err
	}
	if update.RowsAffected == 0 {
		err = errors.New("update userinfo error")
		log.Warnf("db update error")
		return err
	}
	return nil
}

// UpdateAvatar 更新用户头像
func (e *SysUserService) UpdateAvatar(c *dto.UpdateSysUserAvatarReq, p *permission.DataPermission) error {
	var err error
	var model models.SysUser
	db := e.Orm.Scopes(
		permission.Permission(model.TableName(), p),
	).First(&model, c.GetId())
	if err = db.Error; err != nil {
		e.Log.Errorf("Service UpdateSysUser error: %s", err)
		return err
	}
	if db.RowsAffected == 0 {
		return errors.New("无权更新该数据")

	}
	err = e.Orm.Table(model.TableName()).Where("user_id =? ", c.UserId).Updates(c).Error
	if err != nil {
		e.Log.Errorf("Service UpdateSysUser error: %s", err)
		return err
	}
	return nil
}

// UpdateStatus 更新用户状态
func (e *SysUserService) UpdateStatus(c *dto.UpdateSysUserStatusReq, p *permission.DataPermission) error {
	var err error
	var model models.SysUser
	db := e.Orm.Scopes(
		permission.Permission(model.TableName(), p),
	).First(&model, c.GetId())
	if err = db.Error; err != nil {
		e.Log.Errorf("Service UpdateSysUser error: %s", err)
		return err
	}
	if db.RowsAffected == 0 {
		return errors.New("无权更新该数据")

	}
	err = e.Orm.Table(model.TableName()).Where("user_id =? ", c.UserId).Updates(c).Error
	if err != nil {
		e.Log.Errorf("Service UpdateSysUser error: %s", err)
		return err
	}
	return nil
}

// ResetPwd 重置用户密码
func (e *SysUserService) ResetPwd(c *dto.ResetSysUserPwdReq, p *permission.DataPermission) error {
	var err error
	var model models.SysUser
	db := e.Orm.Scopes(
		permission.Permission(model.TableName(), p),
	).First(&model, c.GetId())
	if err = db.Error; err != nil {
		e.Log.Errorf("At Service ResetSysUserPwd error: %s", err)
		return err
	}
	if db.RowsAffected == 0 {
		return errors.New("无权更新该数据")
	}
	c.Generate(&model)
	err = e.Orm.Omit("username", "nick_name", "phone", "role_id", "avatar", "sex").Save(&model).Error
	if err != nil {
		e.Log.Errorf("At Service ResetSysUserPwd error: %s", err)
		return err
	}
	return nil
}

// Remove 删除SysUser
func (e *SysUserService) Remove(c *dto.SysUserById, p *permission.DataPermission) error {
	var err error
	var data models.SysUser

	db := e.Orm.Model(&data).
		Scopes(
			permission.Permission(data.TableName(), p),
		).Delete(&data, c.GetId())
	if err = db.Error; err != nil {
		e.Log.Errorf("Error found in  RemoveSysUser : %s", err)
		return err
	}
	if db.RowsAffected == 0 {
		return errors.New("无权删除该数据")
	}
	return nil
}

// UpdatePwd 修改SysUser对象密码
func (e *SysUserService) UpdatePwd(id int, oldPassword, newPassword string, p *permission.DataPermission) error {
	var err error

	if newPassword == "" {
		return nil
	}
	c := &models.SysUser{}

	err = e.Orm.Model(c).
		Scopes(
			permission.Permission(c.TableName(), p),
		).Select("UserId", "Password", "Salt").
		First(c, id).Error
	if err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return errors.New("无权更新该数据")
		}
		e.Log.Errorf("db error: %s", err)
		return err
	}
	var ok bool
	ok, err = pkg.CompareHashAndPassword(c.Password, oldPassword)
	if err != nil {
		e.Log.Errorf("CompareHashAndPassword error, %s", err.Error())
		return err
	}
	if !ok {
		err = errors.New("incorrect Password")
		e.Log.Warnf("user[%d] %s", id, err.Error())
		return err
	}
	c.Password = newPassword
	db := e.Orm.Model(c).Where("user_id = ?", id).
		Select("Password", "Salt").
		Updates(c)
	if err = db.Error; err != nil {
		e.Log.Errorf("db error: %s", err)
		return err
	}
	if db.RowsAffected == 0 {
		err = errors.New("set password error")
		log.Warnf("db update error")
		return err
	}
	return nil
}

func (e *SysUserService) GetProfile(c *dto.SysUserById, user *models.SysUser, roles *[]models.SysRole, posts *[]models.SysPost) error {
	err := e.Orm.Preload("Org").First(user, c.GetId()).Error
	if err != nil {
		return err
	}
	err = e.Orm.Find(roles, user.RoleId).Error
	if err != nil {
		return err
	}
	err = e.Orm.Find(posts, user.PostIds).Error
	if err != nil {
		return err
	}

	return nil
}

// GetUserRole 获取用户角色
func (e *SysUserService) GetUserRole(c *dto.SysUserById, p *permission.DataPermission, data *models.SysUser) error {
	var err error
	var role models.SysRole
	db := e.Orm.Model(data).
		Scopes(
			permission.Permission(data.TableName(), p),
		).First(data, c.GetId())

	if err = db.Error; err != nil {
		e.Log.Errorf("Error found in  RemoveSysUser : %s", err)
		return err
	}
	err = e.Orm.Model(&role).Scopes(
		permission.Permission(data.TableName(), p),
	).Find(&(data.Roles)).Error

	return nil
}