zitadel/internal/api/grpc/user/user_grant.go

package user

import (
	"github.com/caos/zitadel/internal/api/grpc/object"
	"github.com/caos/zitadel/internal/domain"
	usr_grant_model "github.com/caos/zitadel/internal/usergrant/model"
	user_pb "github.com/caos/zitadel/pkg/grpc/user"
)

func UserGrantsToPb(grants []*usr_grant_model.UserGrantView) []*user_pb.UserGrant {
	u := make([]*user_pb.UserGrant, len(grants))
	for i, grant := range grants {
		u[i] = UserGrantToPb(grant)
	}
	return u
}

func UserGrantToPb(grant *usr_grant_model.UserGrantView) *user_pb.UserGrant {
	return &user_pb.UserGrant{
		Id:             grant.ID,
		UserId:         grant.UserID,
		State:          ModelUserGrantStateToPb(grant.State),
		RoleKeys:       grant.RoleKeys,
		UserName:       grant.UserName,
		FirstName:      grant.FirstName,
		LastName:       grant.LastName,
		Email:          grant.Email,
		DisplayName:    grant.DisplayName,
		OrgId:          grant.ResourceOwner,
		OrgDomain:      grant.OrgPrimaryDomain,
		OrgName:        grant.OrgName,
		ProjectId:      grant.ProjectID,
		ProjectName:    grant.ProjectName,
		ProjectGrantId: grant.GrantID,
		AvatarUrl:      grant.AvatarURL,
		Details: object.ToViewDetailsPb(
			grant.Sequence,
			grant.CreationDate,
			grant.ChangeDate,
			grant.ResourceOwner,
		),
	}
}

func UserGrantQueriesToModel(queries []*user_pb.UserGrantQuery) []*usr_grant_model.UserGrantSearchQuery {
	q := make([]*usr_grant_model.UserGrantSearchQuery, len(queries))
	for i, query := range queries {
		q[i] = UserGrantQueryToModel(query)
	}
	return q
}

func UserGrantQueryToModel(query *user_pb.UserGrantQuery) *usr_grant_model.UserGrantSearchQuery {
	switch q := query.Query.(type) {
	case *user_pb.UserGrantQuery_DisplayNameQuery:
		return UserGrantDisplayNameQueryToModel(q.DisplayNameQuery)
	case *user_pb.UserGrantQuery_EmailQuery:
		return UserGrantEmailQueryToModel(q.EmailQuery)
	case *user_pb.UserGrantQuery_FirstNameQuery:
		return UserGrantFirstNameQueryToModel(q.FirstNameQuery)
	case *user_pb.UserGrantQuery_LastNameQuery:
		return UserGrantLastNameQueryToModel(q.LastNameQuery)
	case *user_pb.UserGrantQuery_OrgDomainQuery:
		return UserGrantOrgDomainQueryToModel(q.OrgDomainQuery)
	case *user_pb.UserGrantQuery_OrgNameQuery:
		return UserGrantOrgNameQueryToModel(q.OrgNameQuery)
	case *user_pb.UserGrantQuery_ProjectGrantIdQuery:
		return UserGrantProjectGrantIDQueryToModel(q.ProjectGrantIdQuery)
	case *user_pb.UserGrantQuery_ProjectIdQuery:
		return UserGrantProjectIDQueryToModel(q.ProjectIdQuery)
	case *user_pb.UserGrantQuery_ProjectNameQuery:
		return UserGrantProjectNameQueryToModel(q.ProjectNameQuery)
	case *user_pb.UserGrantQuery_RoleKeyQuery:
		return UserGrantRoleKeyQueryToModel(q.RoleKeyQuery)
	case *user_pb.UserGrantQuery_UserIdQuery:
		return UserGrantUserIDQueryToModel(q.UserIdQuery)
	case *user_pb.UserGrantQuery_UserNameQuery:
		return UserGrantUserNameQueryToModel(q.UserNameQuery)
	case *user_pb.UserGrantQuery_WithGrantedQuery:
		return UserGrantWithGrantedQueryToModel(q.WithGrantedQuery)
	default:
		return nil
	}
}

func UserGrantDisplayNameQueryToModel(q *user_pb.UserGrantDisplayNameQuery) *usr_grant_model.UserGrantSearchQuery {
	return &usr_grant_model.UserGrantSearchQuery{
		Key:    usr_grant_model.UserGrantSearchKeyDisplayName,
		Method: object.TextMethodToModel(q.Method),
		Value:  q.DisplayName,
	}
}

func UserGrantEmailQueryToModel(q *user_pb.UserGrantEmailQuery) *usr_grant_model.UserGrantSearchQuery {
	return &usr_grant_model.UserGrantSearchQuery{
		Key:    usr_grant_model.UserGrantSearchKeyEmail,
		Method: object.TextMethodToModel(q.Method),
		Value:  q.Email,
	}
}

func UserGrantFirstNameQueryToModel(q *user_pb.UserGrantFirstNameQuery) *usr_grant_model.UserGrantSearchQuery {
	return &usr_grant_model.UserGrantSearchQuery{
		Key:    usr_grant_model.UserGrantSearchKeyFirstName,
		Method: object.TextMethodToModel(q.Method),
		Value:  q.FirstName,
	}
}

func UserGrantLastNameQueryToModel(q *user_pb.UserGrantLastNameQuery) *usr_grant_model.UserGrantSearchQuery {
	return &usr_grant_model.UserGrantSearchQuery{
		Key:    usr_grant_model.UserGrantSearchKeyLastName,
		Method: object.TextMethodToModel(q.Method),
		Value:  q.LastName,
	}
}

func UserGrantOrgDomainQueryToModel(q *user_pb.UserGrantOrgDomainQuery) *usr_grant_model.UserGrantSearchQuery {
	return &usr_grant_model.UserGrantSearchQuery{
		Key:    usr_grant_model.UserGrantSearchKeyOrgDomain,
		Method: object.TextMethodToModel(q.Method),
		Value:  q.OrgDomain,
	}
}

func UserGrantOrgNameQueryToModel(q *user_pb.UserGrantOrgNameQuery) *usr_grant_model.UserGrantSearchQuery {
	return &usr_grant_model.UserGrantSearchQuery{
		Key:    usr_grant_model.UserGrantSearchKeyOrgName,
		Method: object.TextMethodToModel(q.Method),
		Value:  q.OrgName,
	}
}

func UserGrantProjectIDQueryToModel(q *user_pb.UserGrantProjectIDQuery) *usr_grant_model.UserGrantSearchQuery {
	return &usr_grant_model.UserGrantSearchQuery{
		Key:    usr_grant_model.UserGrantSearchKeyProjectID,
		Method: domain.SearchMethodEquals,
		Value:  q.ProjectId,
	}
}

func UserGrantProjectGrantIDQueryToModel(q *user_pb.UserGrantProjectGrantIDQuery) *usr_grant_model.UserGrantSearchQuery {
	return &usr_grant_model.UserGrantSearchQuery{
		Key:    usr_grant_model.UserGrantSearchKeyGrantID,
		Method: domain.SearchMethodEquals,
		Value:  q.ProjectGrantId,
	}
}

func UserGrantProjectNameQueryToModel(q *user_pb.UserGrantProjectNameQuery) *usr_grant_model.UserGrantSearchQuery {
	return &usr_grant_model.UserGrantSearchQuery{
		Key:    usr_grant_model.UserGrantSearchKeyProjectName,
		Method: object.TextMethodToModel(q.Method),
		Value:  q.ProjectName,
	}
}

func UserGrantRoleKeyQueryToModel(q *user_pb.UserGrantRoleKeyQuery) *usr_grant_model.UserGrantSearchQuery {
	return &usr_grant_model.UserGrantSearchQuery{
		Key:    usr_grant_model.UserGrantSearchKeyRoleKey,
		Method: domain.SearchMethodListContains,
		Value:  q.RoleKey,
	}
}

func UserGrantUserIDQueryToModel(q *user_pb.UserGrantUserIDQuery) *usr_grant_model.UserGrantSearchQuery {
	return &usr_grant_model.UserGrantSearchQuery{
		Key:    usr_grant_model.UserGrantSearchKeyUserID,
		Method: domain.SearchMethodEquals,
		Value:  q.UserId,
	}
}

func UserGrantUserNameQueryToModel(q *user_pb.UserGrantUserNameQuery) *usr_grant_model.UserGrantSearchQuery {
	return &usr_grant_model.UserGrantSearchQuery{
		Key:    usr_grant_model.UserGrantSearchKeyUserName,
		Method: object.TextMethodToModel(q.Method),
		Value:  q.UserName,
	}
}

func UserGrantWithGrantedQueryToModel(q *user_pb.UserGrantWithGrantedQuery) *usr_grant_model.UserGrantSearchQuery {
	return &usr_grant_model.UserGrantSearchQuery{
		Key:    usr_grant_model.UserGrantSearchKeyWithGranted,
		Method: domain.SearchMethodEquals,
		Value:  q.WithGranted,
	}
}