internal/provider/services.go

package googleworkspace

import (
	"log"

	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"

	directory "google.golang.org/api/admin/directory/v1"
	"google.golang.org/api/chromepolicy/v1"
	"google.golang.org/api/gmail/v1"
	"google.golang.org/api/groupssettings/v1"
)

func GetChromePoliciesService(chromePolicyService *chromepolicy.Service) (*chromepolicy.CustomersPoliciesService, diag.Diagnostics) {
	var diags diag.Diagnostics

	log.Printf("[INFO] Instantiating Google Admin Chrome Policies service")
	customersService := chromePolicyService.Customers
	if customersService == nil || customersService.Policies == nil {
		diags = append(diags, diag.Diagnostic{
			Severity: diag.Error,
			Summary:  "Chrome Policies Service could not be created.",
		})

		return nil, diags
	}

	return customersService.Policies, diags
}

func GetChromePolicySchemasService(chromePolicyService *chromepolicy.Service) (*chromepolicy.CustomersPolicySchemasService, diag.Diagnostics) {
	var diags diag.Diagnostics

	log.Printf("[INFO] Instantiating Google Admin Chrome Policy Schemas service")
	customersService := chromePolicyService.Customers
	if customersService == nil || customersService.PolicySchemas == nil {
		diags = append(diags, diag.Diagnostic{
			Severity: diag.Error,
			Summary:  "Chrome Policy Schemas Service could not be created.",
		})

		return nil, diags
	}

	return customersService.PolicySchemas, diags
}

func GetDomainAliasesService(directoryService *directory.Service) (*directory.DomainAliasesService, diag.Diagnostics) {
	var diags diag.Diagnostics

	log.Printf("[INFO] Instantiating Google Admin Domain Aliases service")
	domainAliasesService := directoryService.DomainAliases
	if domainAliasesService == nil {
		diags = append(diags, diag.Diagnostic{
			Severity: diag.Error,
			Summary:  "Domain Aliases Service could not be created.",
		})

		return nil, diags
	}

	return domainAliasesService, diags
}

func GetDomainsService(directoryService *directory.Service) (*directory.DomainsService, diag.Diagnostics) {
	var diags diag.Diagnostics

	log.Printf("[INFO] Instantiating Google Admin Domains service")
	domainsService := directoryService.Domains
	if domainsService == nil {
		diags = append(diags, diag.Diagnostic{
			Severity: diag.Error,
			Summary:  "Domains Service could not be created.",
		})

		return nil, diags
	}

	return domainsService, diags
}

func GetGroupsService(directoryService *directory.Service) (*directory.GroupsService, diag.Diagnostics) {
	var diags diag.Diagnostics

	log.Printf("[INFO] Instantiating Google Admin Groups service")
	groupsService := directoryService.Groups
	if groupsService == nil {
		diags = append(diags, diag.Diagnostic{
			Severity: diag.Error,
			Summary:  "Groups Service could not be created.",
		})

		return nil, diags
	}

	return groupsService, diags
}

func GetGroupsSettingsService(groupsSettingsService *groupssettings.Service) (*groupssettings.GroupsService, diag.Diagnostics) {
	var diags diag.Diagnostics

	log.Printf("[INFO] Instantiating Google Admin Groups Settings Groups service")
	groupsService := groupsSettingsService.Groups
	if groupsService == nil {
		diags = append(diags, diag.Diagnostic{
			Severity: diag.Error,
			Summary:  "Groups Service could not be created.",
		})

		return nil, diags
	}

	return groupsService, diags
}

func GetGmailSendAsAliasService(gmailService *gmail.Service) (*gmail.UsersSettingsSendAsService, diag.Diagnostics) {
	var diags diag.Diagnostics

	log.Printf("[INFO] Instantiating Google Admin Gmail Send As Alias service")
	usersService := gmailService.Users
	if usersService == nil || usersService.Settings == nil || usersService.Settings.SendAs == nil {
		diags = append(diags, diag.Diagnostic{
			Severity: diag.Error,
			Summary:  "Send As Alias Service could not be created.",
		})

		return nil, diags
	}

	return usersService.Settings.SendAs, diags
}

func GetGroupAliasService(groupsService *directory.GroupsService) (*directory.GroupsAliasesService, diag.Diagnostics) {
	var diags diag.Diagnostics

	log.Printf("[INFO] Instantiating Google Admin Group Alias service")
	aliasesService := groupsService.Aliases
	if aliasesService == nil {
		diags = append(diags, diag.Diagnostic{
			Severity: diag.Error,
			Summary:  "Groups Aliases Service could not be created.",
		})

		return nil, diags
	}

	return aliasesService, diags
}

func GetMembersService(directoryService *directory.Service) (*directory.MembersService, diag.Diagnostics) {
	var diags diag.Diagnostics

	log.Printf("[INFO] Instantiating Google Admin Members service")
	membersService := directoryService.Members
	if membersService == nil {
		diags = append(diags, diag.Diagnostic{
			Severity: diag.Error,
			Summary:  "Members Service could not be created.",
		})

		return nil, diags
	}

	return membersService, diags
}

func GetOrgUnitsService(directoryService *directory.Service) (*directory.OrgunitsService, diag.Diagnostics) {
	var diags diag.Diagnostics

	log.Printf("[INFO] Instantiating Google Admin OrgUnits service")
	ousService := directoryService.Orgunits
	if ousService == nil {
		diags = append(diags, diag.Diagnostic{
			Severity: diag.Error,
			Summary:  "OrgUnits Service could not be created.",
		})

		return nil, diags
	}

	return ousService, diags
}

func GetPrivilegesService(directoryService *directory.Service) (*directory.PrivilegesService, diag.Diagnostics) {
	var diags diag.Diagnostics

	log.Printf("[INFO] Instantiating Google Admin Privileges service")
	privilegesService := directoryService.Privileges
	if privilegesService == nil {
		diags = append(diags, diag.Diagnostic{
			Severity: diag.Error,
			Summary:  "Privileges Service could not be created.",
		})

		return nil, diags
	}

	return privilegesService, diags
}

func GetRoleAssignmentsService(directoryService *directory.Service) (*directory.RoleAssignmentsService, diag.Diagnostics) {
	var diags diag.Diagnostics

	log.Printf("[INFO] Instantiating Google Admin RoleAssignments service")
	roleAssignmentsService := directoryService.RoleAssignments
	if roleAssignmentsService == nil {
		diags = append(diags, diag.Diagnostic{
			Severity: diag.Error,
			Summary:  "RoleAssignments Service could not be created.",
		})

		return nil, diags
	}

	return roleAssignmentsService, diags
}

func GetRolesService(directoryService *directory.Service) (*directory.RolesService, diag.Diagnostics) {
	var diags diag.Diagnostics

	log.Printf("[INFO] Instantiating Google Admin Roles service")
	rolesService := directoryService.Roles
	if rolesService == nil {
		diags = append(diags, diag.Diagnostic{
			Severity: diag.Error,
			Summary:  "Roles Service could not be created.",
		})

		return nil, diags
	}

	return rolesService, diags
}

func GetSchemasService(directoryService *directory.Service) (*directory.SchemasService, diag.Diagnostics) {
	var diags diag.Diagnostics

	log.Printf("[INFO] Instantiating Google Admin Schemas service")
	schemasService := directoryService.Schemas
	if schemasService == nil {
		diags = append(diags, diag.Diagnostic{
			Severity: diag.Error,
			Summary:  "Schemas Service could not be created.",
		})

		return nil, diags
	}

	return schemasService, diags
}

func GetUsersService(directoryService *directory.Service) (*directory.UsersService, diag.Diagnostics) {
	var diags diag.Diagnostics

	log.Printf("[INFO] Instantiating Google Admin Users service")
	usersService := directoryService.Users
	if usersService == nil {
		diags = append(diags, diag.Diagnostic{
			Severity: diag.Error,
			Summary:  "Users Service could not be created.",
		})

		return nil, diags
	}

	return usersService, diags
}

func GetUserAliasService(usersService *directory.UsersService) (*directory.UsersAliasesService, diag.Diagnostics) {
	var diags diag.Diagnostics

	log.Printf("[INFO] Instantiating Google Admin User Alias service")
	aliasesService := usersService.Aliases
	if aliasesService == nil {
		diags = append(diags, diag.Diagnostic{
			Severity: diag.Error,
			Summary:  "Users Aliases Service could not be created.",
		})

		return nil, diags
	}

	return aliasesService, diags
}