sso/service/login_service.go

package service

import (
	"SSO/models"

	"github.com/jinzhu/gorm"
)

func IsPasswordRight(mobile string, password string) bool {
	var count int
	readUserDb.Model(&models.AdminUser{}).Where("mobile = ? AND password = ? AND status = 1", mobile, password).Count(&count)
	return count == 1
}

func IsSuperAdmin(mobile string) bool {
	var count int
	readUserDb.Model(&models.AdminUser{}).Where("mobile = ? AND is_super_admin = 1 AND status = 1", mobile).Count(&count)
	return count == 1
}

func GetAdminUserWithMobile(mobile string) *models.AdminUser {
	var user models.AdminUser
	readUserDb.Where("mobile = ? AND status = 1", mobile).Find(&user)
	if user.Id == 0 {
		return nil
	} else {
		return &user
	}
}

func GetValidAdminUserByMobileReturnErr(mobile string) (*models.AdminUser, error) {
	var user models.AdminUser
	err := readUserDb.Where("mobile = ? AND status = 1", mobile).First(&user).Error
	if err != nil {
		if err == gorm.ErrRecordNotFound {
			return nil, nil
		} else {
			return nil, err
		}
	}
	return &user, nil
}

func DidAdminUserCreateOrg(adminUserID int) (bool, error) {
	var count int
	err := readUserDb.Model(&models.Org{}).Where("creator = ? AND status <> 0", adminUserID).Count(&count).Error
	return count > 0, err
}

func DidAdminUserOrgCreateApp(adminUserID int, appType int) (bool, error) {
	var count int
	db := readUserDb.Model(&models.OrgApp{}).Where("creator = ? AND status = 1", adminUserID)
	if appType != 0 {
		db = db.Where("app_type = ?", appType)
	}
	err := db.Count(&count).Error
	return count > 0, err
}

func GetAdminUserLastLoginLog(adminUserID int, appType int) (*models.AdminUserLoginLog, error) {
	var record models.AdminUserLoginLog
	err := readUserDb.Last(&record, "admin_user_id = ? AND operate_type <> 2 AND app_type = ?", adminUserID, appType).Error
	if err == gorm.ErrRecordNotFound {
		return nil, nil
	}
	return &record, err
}

func GetAdminUserPrioritizedAppType(adminUserID int) (int8, error) {
	rows, err := readUserDb.Raw("SELECT app.app_type FROM sgj_user_org_app AS app, sgj_user_admin_role AS r WHERE r.admin_user_id = ? AND r.app_id = app.id AND app.status = 1 AND r.status = 1 ORDER BY app.org_id ASC, app.app_type ASC LIMIT 1;", adminUserID).Rows()
	defer rows.Close()
	if err != nil {
		return 0, err
	} else {
		if rows.Next() {
			var appType int8
			rows.Scan(&appType)
			return appType, nil
		} else {
			return 0, nil
		}
	}
}

func GetAdminUserAllOrgWithUID(adminUserID int, isSuperAdmin bool) ([]*models.Org, error) {
	if isSuperAdmin {
		var org models.Org
		err := readUserDb.Preload("OrgGallery", "status = 1").Where("creator = ? AND status <> 0", adminUserID).First(&org).Error
		if err != nil {
			return nil, err
		}
		return []*models.Org{&org}, nil

	} else {
		rows, err := readUserDb.Raw("SELECT org.*, COUNT(DISTINCT org.id) FROM sgj_user_admin_role AS u_r, sgj_user_org AS org WHERE u_r.admin_user_id = ? AND u_r.org_id = org.id AND u_r.status = 1 AND org.status <> 0 GROUP BY org.id ORDER BY org.id;", adminUserID).Rows()
		defer rows.Close()
		if err != nil {
			return nil, err
		} else {
			orgs := make([]*models.Org, 0)
			for rows.Next() {
				var org models.Org
				if scanErr := readUserDb.Preload("OrgGallery", "status = 1").ScanRows(rows, &org); scanErr != nil {
					return nil, scanErr

				} else {
					orgs = append(orgs, &org)
				}
			}
			return orgs, nil
		}
	}
}

func GetAdminUserAllOrgApp(adminUserID int, orgID int) ([]*models.OrgApp, error) {
	rows, err := readUserDb.Raw("SELECT app.* from sgj_user_admin_role AS u_r, sgj_user_org_app AS app WHERE u_r.admin_user_id = ? AND u_r.org_id = ? AND u_r.app_id = app.id AND u_r.status = 1 AND app.status = 1 ORDER BY app_type ASC;", adminUserID, orgID).Rows()
	defer rows.Close()
	if err != nil {
		return nil, err
	} else {
		apps := make([]*models.OrgApp, 0)
		for rows.Next() {
			var app models.OrgApp
			if scanErr := readUserDb.ScanRows(rows, &app); scanErr != nil {
				return nil, scanErr
			} else {
				apps = append(apps, &app)
			}
		}
		return apps, nil
	}
}

func InsertLoginLog(loginLog *models.AdminUserLoginLog) error {
	tx := writeUserDb.Begin()
	if err := tx.Create(loginLog).Error; err != nil {
		tx.Rollback()
		return err
	}
	return tx.Commit().Error
}

func GetAppRole(adminUserID int, orgID int, appID int) (*models.App_Role, error) {
	var appRole models.App_Role
	err := readUserDb.Where("admin_user_id = ? AND org_id = ? AND app_id = ? AND status = 1", adminUserID, orgID, appID).First(&appRole).Error
	if err != nil {
		return nil, err
	}
	return &appRole, nil
}

// 获取最近被创建的血透系统管理员角色（AppType为3）
func GetLastXTAdminRole(adminUserID int, appType int) (*models.App_Role, error) {
	var appRole models.App_Role
	// 	select a_r.* from sgj_user_admin_role as a_r
	// join sgj_user_admin as a on a.id = a_r.admin_user_id
	// join sgj_user_org_app as app on app.id = a_r.app_id and app.status = 1 and app.app_type = 3
	// where a_r.status = 1 and a_r.admin_user_id = 380
	err := readUserDb.Table("sgj_user_admin_role").
		Select("sgj_user_admin_role.*").
		Joins("join sgj_user_admin as a on a.id = sgj_user_admin_role.admin_user_id").
		Joins("join sgj_user_org_app as app on app.id = sgj_user_admin_role.app_id and app.status = 1 and app.app_type = ?", appType).
		Where("sgj_user_admin_role.status = 1 and sgj_user_admin_role.admin_user_id = ?", adminUserID).
		Order("sgj_user_admin_role.id desc").
		First(&appRole).
		Error
	if err != nil {
		if err == gorm.ErrRecordNotFound {
			return nil, nil
		} else {
			return nil, err
		}
	}
	return &appRole, nil
}

func GetOrgById(orgID int) (*models.Org, error) {
	var org models.Org
	err := readUserDb.Model(&models.Org{}).Where("id = ?", orgID).First(&org).Error
	if err != nil {
		if err == gorm.ErrRecordNotFound {
			return nil, nil
		} else {
			return nil, err
		}
	}
	return &org, nil
}

func GetAppById(appID int) (*models.OrgApp, error) {
	var app models.OrgApp
	err := readUserDb.Model(&models.OrgApp{}).Where("id = ?", appID).First(&app).Error
	if err != nil {
		if err == gorm.ErrRecordNotFound {
			return nil, nil
		} else {
			return nil, err
		}
	}
	return &app, nil
}