groups.go

// Copyright (c) HashiCorp, Inc.
// SPDX-License-Identifier: MPL-2.0

package groups

import (
	"context"
	"fmt"
	"math/rand"

	"github.com/hashicorp/go-azure-helpers/lang/pointer"
	"github.com/hashicorp/go-azure-helpers/lang/response"
	"github.com/hashicorp/go-azure-sdk/microsoft-graph/common-types/beta"
	groupBeta "github.com/hashicorp/go-azure-sdk/microsoft-graph/groups/beta/group"
	memberBeta "github.com/hashicorp/go-azure-sdk/microsoft-graph/groups/beta/member"
)

func groupDefaultMailNickname() string {
	charSet := "0123456789abcdef"
	result := make([]byte, 9)
	for i := 0; i < 9; i++ {
		result[i] = charSet[rand.Intn(len(charSet))]
	}
	resultString := string(result)
	return resultString[:8] + "-" + resultString[8:]
}

func groupFindByName(ctx context.Context, client *groupBeta.GroupClient, displayName string) (*[]beta.Group, error) {
	options := groupBeta.ListGroupsOperationOptions{
		Filter: pointer.To(fmt.Sprintf("displayName eq '%s'", displayName)),
	}

	resp, err := client.ListGroups(ctx, options)
	if err != nil {
		return nil, fmt.Errorf("unable to list Groups with filter %q: %v", *options.Filter, err)
	}

	result := make([]beta.Group, 0)
	if resp.Model != nil {
		for _, group := range *resp.Model {
			if group.DisplayName != nil && group.DisplayName.GetOrZero() == displayName {
				result = append(result, group)
			}
		}
	}

	return &result, nil
}

func groupGetAdditional(ctx context.Context, client *groupBeta.GroupClient, id beta.GroupId) (*beta.Group, error) {
	options := groupBeta.GetGroupOperationOptions{
		Select: &[]string{
			"allowExternalSenders",
			"autoSubscribeNewMembers",
			"hideFromAddressLists",
			"hideFromOutlookClients",
		},
	}

	resp, err := client.GetGroup(ctx, id, options)
	if err != nil {
		if response.WasNotFound(resp.HttpResponse) {
			// API returns 404 when these M365-only fields are requested for a group in a non-M365 tenant, so we
			// don't raise an error in this case and proceed as if they are not set.
			// See https://github.com/microsoftgraph/msgraph-metadata/issues/333
			return nil, nil
		}

		return nil, fmt.Errorf("retrieving additional fields: %+v", err)
	}

	return resp.Model, nil
}

func groupGetMember(ctx context.Context, client *memberBeta.MemberClient, id beta.GroupIdMemberId) (*beta.DirectoryObject, error) {
	options := memberBeta.ListMembersOperationOptions{
		Filter: pointer.To(fmt.Sprintf("id eq '%s'", id.DirectoryObjectId)),
	}

	resp, err := client.ListMembers(ctx, beta.NewGroupID(id.GroupId), options)
	if err != nil {
		if response.WasNotFound(resp.HttpResponse) {
			return nil, nil
		} else {
			return nil, err
		}
	}

	if resp.Model != nil {
		for _, member := range *resp.Model {
			if member.DirectoryObject().Id != nil && *member.DirectoryObject().Id == id.DirectoryObjectId {
				return &member, nil
			}
		}
	}

	return nil, nil
}