package machine

import "github.com/samber/lo"

// Machines defines which machines a configuration applies to.
type Machines struct {
	// Only specifies a list of machine IDs or aliases where the configuration should apply.
	Only *[]string `json:"only"   yaml:"only"`
	// Except specifies a list of machine IDs or aliases where the configuration should not apply.
	Except *[]string `json:"except" yaml:"except"`
}

// GetShouldRunOnMachine determines if a configuration should run on the current machine
// based on the Only and Except fields of the Machines struct.
// The aliases parameter is a map of friendly names to machine IDs. When checking,
// aliases are resolved first, falling back to treating the value as a literal machine ID.
func (m *Machines) GetShouldRunOnMachine(machineID string, aliases map[string]string) bool {
	if m == nil {
		return true
	}

	if m.Only != nil {
		return containsMachineID(*m.Only, machineID, aliases)
	}
	if m.Except != nil {
		return !containsMachineID(*m.Except, machineID, aliases)
	}
	return true
}

// containsMachineID checks if the machine ID is in the list, resolving aliases first.
func containsMachineID(list []string, machineID string, aliases map[string]string) bool {
	return lo.SomeBy(list, func(entry string) bool {
		if aliases != nil {
			if resolvedID, ok := aliases[entry]; ok {
				return resolvedID == machineID
			}
		}
		return entry == machineID
	})
}