archetype.go

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.

package alzlib

import (
	"strings"

	mapset "github.com/deckarep/golang-set/v2"
)

// Archetype represents an archetype definition that hasn't been assigned to a management group
// The contents of the sets represent the map keys of the corresponding AlzLib maps.
// Do not creaste this struct directly, use NewArchetype instead.
type Archetype struct {
	PolicyDefinitions    mapset.Set[string]
	PolicyAssignments    mapset.Set[string]
	PolicySetDefinitions mapset.Set[string]
	RoleDefinitions      mapset.Set[string]
	name                 string
}

// NewArchetype creates a new Archetype with the given name.
func NewArchetype(name string) *Archetype {
	return &Archetype{
		PolicyDefinitions:    mapset.NewThreadUnsafeSet[string](),
		PolicyAssignments:    mapset.NewThreadUnsafeSet[string](),
		PolicySetDefinitions: mapset.NewThreadUnsafeSet[string](),
		RoleDefinitions:      mapset.NewThreadUnsafeSet[string](),
		name:                 name,
	}
}

// Name returns the name of the archetype.
func (a *Archetype) Name() string {
	return a.name
}

// copy creates a deep copy of the archetype.
func (a *Archetype) copy() *Archetype {
	return &Archetype{
		PolicyDefinitions:    a.PolicyDefinitions.Clone(),
		PolicyAssignments:    a.PolicyAssignments.Clone(),
		PolicySetDefinitions: a.PolicySetDefinitions.Clone(),
		RoleDefinitions:      a.RoleDefinitions.Clone(),
		name:                 a.name,
	}
}

// SplitNameAndVersion splits a resource reference into its name and version components.
// If no version is present, the second return value will be nil.
func SplitNameAndVersion(ref string) (string, *string) {
	split := strings.Split(ref, "@")
	if len(split) == 1 {
		return split[0], nil
	}

	return split[0], &split[1]
}

// JoinNameAndVersion joins a resource name and version into a single string.
// If the version is nil, only the name is returned.
func JoinNameAndVersion(name string, version *string) string {
	if version == nil {
		return name
	}

	return name + "@" + *version
}