tasks/common/persistence/visibility/store/query/resolve.go

package query

import (
	"strings"

	enumspb "go.temporal.io/api/enums/v1"
	"github.com/hanzoai/tasks/chasm"
	"github.com/hanzoai/tasks/common/namespace"
	"github.com/hanzoai/tasks/common/searchattribute"
	"github.com/hanzoai/tasks/common/searchattribute/sadefs"
)

// ResolveSearchAttributeAlias resolves the search attribute alias for the given name. The process is:
//  1. If the name has the "Temporal", skip mapping to a custom search attribute.
//  2. If the search attribute exists in the visibility mapper, pass it through.
//  3. If the search attribute exists in the CHASM mapper, pass it through.
//  4. If it exists as a system / predefined attribute, map it.
//     4.1 Some pre-defined attributes are already defined with the Temporal prefix, so both options need to be checked.
func ResolveSearchAttributeAlias(
	name string,
	ns namespace.Name,
	mapper searchattribute.Mapper,
	saTypeMap searchattribute.NameTypeMap,
	chasmMapper *chasm.VisibilitySearchAttributesMapper,
) (string, enumspb.IndexedValueType, error) {
	if sadefs.IsMappable(name) {
		// First check if the visibility mapper can handle this field (e.g., custom search attributes)
		fieldName, fieldType := tryVisibilityMapper(name, ns, mapper, saTypeMap)
		if fieldName != "" {
			return fieldName, fieldType, nil
		}

		// Handle ScheduleID → WorkflowID transformation, but only if ScheduleID is not defined as a custom search attribute
		// This fallback only applies when the visibility mapper doesn't handle the field
		if name == sadefs.ScheduleID {
			// ScheduleID is not defined, transform to WorkflowID
			saType, _ := saTypeMap.GetType(sadefs.WorkflowID)
			return sadefs.WorkflowID, saType, nil
		}
	}

	fieldName, fieldType := tryChasmMapper(name, chasmMapper)
	if fieldName != "" {
		return fieldName, fieldType, nil
	}

	fieldName, fieldType, found := tryDirectAndPrefixedLookup(name, saTypeMap)
	if found {
		return fieldName, fieldType, nil
	}

	return "", enumspb.INDEXED_VALUE_TYPE_UNSPECIFIED, NewConverterError("invalid search attribute: %s", name)
}

// tryVisibilityMapper might find a successful match in which case we return it
// otherwise we continue with the fallback logic.
func tryVisibilityMapper(
	name string,
	ns namespace.Name,
	mapper searchattribute.Mapper,
	saTypeMap searchattribute.NameTypeMap,
) (string, enumspb.IndexedValueType) {
	if mapper == nil {
		return "", enumspb.INDEXED_VALUE_TYPE_UNSPECIFIED
	}

	fieldName, err := mapper.GetFieldName(name, ns.String())
	if err != nil {
		// If there is an error, we need to continue with the fallback logic
		// because this search attribute is not defined in the mapper, but might
		// exist with a different name in the namespace.
		return "", enumspb.INDEXED_VALUE_TYPE_UNSPECIFIED
	}

	// Mapper successfully resolved the field name, now check if it exists in the type map
	fieldType, err := saTypeMap.GetType(fieldName)
	if err != nil {
		// If the mapped field doesn't exist in type map, allow fallback to direct/prefixed lookup.
		return "", enumspb.INDEXED_VALUE_TYPE_UNSPECIFIED
	}

	return fieldName, fieldType
}

func tryChasmMapper(name string, chasmMapper *chasm.VisibilitySearchAttributesMapper) (string, enumspb.IndexedValueType) {
	if chasmMapper == nil {
		return "", enumspb.INDEXED_VALUE_TYPE_UNSPECIFIED
	}

	fieldName, err := chasmMapper.Field(name)
	if err != nil {
		return "", enumspb.INDEXED_VALUE_TYPE_UNSPECIFIED
	}

	fieldType, err := chasmMapper.ValueType(fieldName)
	if err != nil {
		return "", enumspb.INDEXED_VALUE_TYPE_UNSPECIFIED
	}
	return fieldName, fieldType
}

func tryDirectAndPrefixedLookup(name string, saTypeMap searchattribute.NameTypeMap) (string, enumspb.IndexedValueType, bool) {
	if saType, err := saTypeMap.GetType(name); err == nil {
		return name, saType, true
	}

	prefixedName := sadefs.ReservedPrefix + name
	if saType, err := saTypeMap.GetType(prefixedName); err == nil {
		return prefixedName, saType, true
	}

	strippedName := strings.TrimPrefix(name, sadefs.ReservedPrefix)
	if saType, err := saTypeMap.GetType(strippedName); err == nil {
		return strippedName, saType, true
	}

	return "", enumspb.INDEXED_VALUE_TYPE_UNSPECIFIED, false
}