energylink/pkg/login/ext_user.go

package login

import (
	"fmt"

	"github.com/grafana/grafana/pkg/bus"
	"github.com/grafana/grafana/pkg/log"
	m "github.com/grafana/grafana/pkg/models"
	"github.com/grafana/grafana/pkg/services/quota"
)

func UpsertUser(ctx *m.ReqContext, cmd *m.UpsertUserCommand) error {
	extUser := cmd.ExternalUser

	userQuery := m.GetUserByAuthInfoQuery{
		AuthModule: extUser.AuthModule,
		AuthId:     extUser.AuthId,
		UserId:     extUser.UserId,
		Email:      extUser.Email,
		Login:      extUser.Login,
	}
	err := bus.Dispatch(&userQuery)
	if err != nil {
		if err != m.ErrUserNotFound {
			return err
		}

		if !cmd.SignupAllowed {
			log.Warn(fmt.Sprintf("Not allowing %s login, user not found in internal user database and allow signup = false", extUser.AuthModule))
			return ErrInvalidCredentials
		}

		limitReached, err := quota.QuotaReached(ctx, "user")
		if err != nil {
			log.Warn("Error getting user quota", "err", err)
			return ErrGettingUserQuota
		}
		if limitReached {
			return ErrUsersQuotaReached
		}

		cmd.User, err = createUser(extUser)
		if err != nil {
			return err
		}
	} else {
		cmd.User = userQuery.User

		// sync user info
		err = updateUser(cmd.User, extUser)
		if err != nil {
			return err
		}
	}

	err = syncOrgRoles(cmd.User, extUser)
	if err != nil {
		return err
	}

	return nil
}

func createUser(extUser *m.ExternalUserInfo) (*m.User, error) {
	cmd := m.CreateUserCommand{
		Login: extUser.Login,
		Email: extUser.Email,
		Name:  extUser.Name,
	}
	if err := bus.Dispatch(&cmd); err != nil {
		return nil, err
	}

	cmd2 := m.SetAuthInfoCommand{
		UserId:     cmd.Result.Id,
		AuthModule: extUser.AuthModule,
		AuthId:     extUser.AuthId,
	}
	if err := bus.Dispatch(&cmd2); err != nil {
		return nil, err
	}

	return &cmd.Result, nil
}

func updateUser(user *m.User, extUser *m.ExternalUserInfo) error {
	// sync user info
	if user.Login != extUser.Login || user.Email != extUser.Email || user.Name != extUser.Name {
		log.Debug("Syncing user info", "id", user.Id, "login", extUser.Login, "email", extUser.Email)
		updateCmd := m.UpdateUserCommand{
			UserId: user.Id,
			Login:  extUser.Login,
			Email:  extUser.Email,
			Name:   extUser.Name,
		}
		err := bus.Dispatch(&updateCmd)
		if err != nil {
			return err
		}
	}

	return nil
}

func syncOrgRoles(user *m.User, extUser *m.ExternalUserInfo) error {
	if len(extUser.OrgRoles) == 0 {
		// log.Warn("No group mappings defined")
		return nil
	}

	orgsQuery := m.GetUserOrgListQuery{UserId: user.Id}
	if err := bus.Dispatch(&orgsQuery); err != nil {
		return err
	}

	handledOrgIds := map[int64]bool{}
	deleteOrgIds := []int64{}

	// update existing org roles
	for _, org := range orgsQuery.Result {
		handledOrgIds[org.OrgId] = true

		if extUser.OrgRoles[org.OrgId] == "" {
			deleteOrgIds = append(deleteOrgIds, org.OrgId)
		} else if extUser.OrgRoles[org.OrgId] != org.Role {
			// update role
			cmd := m.UpdateOrgUserCommand{OrgId: org.OrgId, UserId: user.Id, Role: extUser.OrgRoles[org.OrgId]}
			if err := bus.Dispatch(&cmd); err != nil {
				return err
			}
		}
	}

	// add any new org roles
	for orgId, orgRole := range extUser.OrgRoles {
		if _, exists := handledOrgIds[orgId]; exists {
			continue
		}

		// add role
		cmd := m.AddOrgUserCommand{UserId: user.Id, Role: orgRole, OrgId: orgId}
		err := bus.Dispatch(&cmd)
		if err != nil && err != m.ErrOrgNotFound {
			return err
		}
	}

	// delete any removed org roles
	for _, orgId := range deleteOrgIds {
		cmd := m.RemoveOrgUserCommand{OrgId: orgId, UserId: user.Id}
		if err := bus.Dispatch(&cmd); err != nil {
			return err
		}
	}

	return nil
}