probo/pkg/validator/validator_string.go

// Copyright (c) 2025 Probo Inc <hello@getprobo.com>.
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.
//
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
// PERFORMANCE OF THIS SOFTWARE.

package validator

import (
	"fmt"
	"reflect"
	"regexp"
	"strings"
)

var (
	alphaNumericRegex = regexp.MustCompile(`^[a-zA-Z0-9]+$`)
	slugRegex         = regexp.MustCompile(`^[a-z0-9]+(?:-[a-z0-9]+)*$`)
)

// MinLen validates that a string has at least the specified minimum length.
func MinLen(minLength int) ValidatorFunc {
	return func(value any) *ValidationError {
		actualValue, isNil := dereferenceValue(value)
		if isNil {
			return nil
		}

		str, ok := actualValue.(string)
		if !ok {
			return newValidationError(ErrorCodeInvalidFormat, "value must be a string")
		}

		if len(str) < minLength {
			return newValidationError(
				ErrorCodeTooShort,
				fmt.Sprintf("must be at least %d characters", minLength),
			)
		}

		return nil
	}
}

// MaxLen validates that a string does not exceed the specified maximum length.
func MaxLen(maxLength int) ValidatorFunc {
	return func(value any) *ValidationError {
		actualValue, isNil := dereferenceValue(value)
		if isNil {
			return nil
		}

		str, ok := actualValue.(string)
		if !ok {
			return newValidationError(ErrorCodeInvalidFormat, "value must be a string")
		}

		if len(str) > maxLength {
			return newValidationError(
				ErrorCodeTooLong,
				fmt.Sprintf("must be at most %d characters", maxLength),
			)
		}

		return nil
	}
}

// Pattern validates that a string matches the specified regular expression pattern.
func Pattern(pattern string, message string) ValidatorFunc {
	regex := regexp.MustCompile(pattern)

	return func(value any) *ValidationError {
		actualValue, isNil := dereferenceValue(value)
		if isNil {
			return nil
		}

		str, ok := actualValue.(string)
		if !ok {
			return newValidationError(ErrorCodeInvalidFormat, "value must be a string")
		}

		if !regex.MatchString(str) {
			if message == "" {
				message = fmt.Sprintf("must match pattern: %s", pattern)
			}
			return newValidationError(ErrorCodeInvalidFormat, message)
		}

		return nil
	}
}

// AlphaNumeric validates that a string contains only letters and numbers.
func AlphaNumeric() ValidatorFunc {
	return func(value any) *ValidationError {
		actualValue, isNil := dereferenceValue(value)
		if isNil {
			return nil
		}

		str, ok := actualValue.(string)
		if !ok {
			return newValidationError(ErrorCodeInvalidFormat, "value must be a string")
		}

		if str == "" {
			return nil
		}

		if !alphaNumericRegex.MatchString(str) {
			return newValidationError(ErrorCodeInvalidFormat, "must contain only letters and numbers")
		}

		return nil
	}
}

// NoSpaces validates that a string does not contain any spaces.
func NoSpaces() ValidatorFunc {
	return func(value any) *ValidationError {
		actualValue, isNil := dereferenceValue(value)
		if isNil {
			return nil
		}

		str, ok := actualValue.(string)
		if !ok {
			return newValidationError(ErrorCodeInvalidFormat, "value must be a string")
		}

		if str == "" {
			return nil
		}

		if strings.Contains(str, " ") {
			return newValidationError(ErrorCodeInvalidFormat, "must not contain spaces")
		}

		return nil
	}
}

// Slug validates that a string is a valid URL slug (lowercase letters, numbers, and hyphens).
func Slug() ValidatorFunc {
	return func(value any) *ValidationError {
		actualValue, isNil := dereferenceValue(value)
		if isNil {
			return nil
		}

		str, ok := actualValue.(string)
		if !ok {
			return newValidationError(ErrorCodeInvalidFormat, "value must be a string")
		}

		if str == "" {
			return nil
		}

		if !slugRegex.MatchString(str) {
			return newValidationError(ErrorCodeInvalidFormat, "must be a valid slug (lowercase letters, numbers, and hyphens)")
		}

		return nil
	}
}

// OneOfSlice validates that a value is one of the allowed values in the slice.
// Accepts a slice of any type. Compares by value first, then by string representation.
func OneOfSlice[T any](allowed []T) ValidatorFunc {
	// Build allowed map with string keys for flexible comparison
	allowedMap := make(map[string]bool)
	allowedStrings := make([]string, 0, len(allowed))

	for _, v := range allowed {
		str := fmt.Sprint(v)
		allowedMap[str] = true
		allowedStrings = append(allowedStrings, str)
	}

	return func(value any) *ValidationError {
		// Handle nil values first
		if value == nil {
			return nil
		}

		// Dereference all pointer levels
		actualValue := value
		val := reflect.ValueOf(value)
		for val.Kind() == reflect.Ptr {
			if val.IsNil() {
				return nil
			}
			val = val.Elem()
			actualValue = val.Interface()
		}

		// First try exact match with DeepEqual
		for _, allowedVal := range allowed {
			if reflect.DeepEqual(actualValue, allowedVal) {
				return nil
			}
		}

		// Then try string comparison (for custom string types)
		valueStr := fmt.Sprint(actualValue)
		if allowedMap[valueStr] {
			return nil
		}

		return newValidationError(
			ErrorCodeInvalidEnum,
			fmt.Sprintf("must be one of: %s", strings.Join(allowedStrings, ", ")),
		)
	}
}

// NotOneOfSlice validates that a value is not one of the values in the slice.
// Accepts a slice of any type. Compares by value first, then by string representation.
func NotOneOfSlice[T any](disallowed []T) ValidatorFunc {
	// Build disallowed map with string keys for flexible comparison
	disallowedMap := make(map[string]bool)
	disallowedStrings := make([]string, 0, len(disallowed))

	for _, v := range disallowed {
		str := fmt.Sprint(v)
		disallowedMap[str] = true
		disallowedStrings = append(disallowedStrings, str)
	}

	return func(value any) *ValidationError {
		// Handle nil values first
		if value == nil {
			return nil
		}

		// Dereference all pointer levels
		actualValue := value
		val := reflect.ValueOf(value)
		for val.Kind() == reflect.Ptr {
			if val.IsNil() {
				return nil
			}
			val = val.Elem()
			actualValue = val.Interface()
		}

		// First try exact match with DeepEqual
		for _, disallowedVal := range disallowed {
			if reflect.DeepEqual(actualValue, disallowedVal) {
				return newValidationError(
					ErrorCodeInvalidEnum,
					fmt.Sprintf("must not be one of: %s", strings.Join(disallowedStrings, ", ")),
				)
			}
		}

		// Then try string comparison (for custom string types)
		valueStr := fmt.Sprint(actualValue)
		if disallowedMap[valueStr] {
			return newValidationError(
				ErrorCodeInvalidEnum,
				fmt.Sprintf("must not be one of: %s", strings.Join(disallowedStrings, ", ")),
			)
		}

		return nil
	}
}

// OneOf validates that a value is one of the allowed values.
// Accepts strings or types that implement fmt.Stringer as variadic arguments.
func OneOf(allowed ...any) ValidatorFunc {
	allowedMap := make(map[string]bool)
	allowedStrings := make([]string, 0, len(allowed))

	for _, v := range allowed {
		var str string
		switch val := v.(type) {
		case string:
			str = val
		case fmt.Stringer:
			str = val.String()
		default:
			str = fmt.Sprint(val)
		}
		allowedMap[str] = true
		allowedStrings = append(allowedStrings, str)
	}

	return func(value any) *ValidationError {
		actualValue, isNil := dereferenceValue(value)
		if isNil {
			return nil
		}

		var str string
		switch v := actualValue.(type) {
		case string:
			str = v
		default:
			if stringer, ok := actualValue.(fmt.Stringer); ok {
				str = stringer.String()
			} else {
				return newValidationError(ErrorCodeInvalidEnum, "value must be a string or implement fmt.Stringer")
			}
		}

		if !allowedMap[str] {
			return newValidationError(
				ErrorCodeInvalidEnum,
				fmt.Sprintf("must be one of: %s", strings.Join(allowedStrings, ", ")),
			)
		}

		return nil
	}
}