package mongodb

import (
	"go.mongodb.org/mongo-driver/bson/primitive"
	"regexp"
	"strings"
	"unicode"

	"go.mongodb.org/mongo-driver/bson"
	"golang.org/x/text/runes"
	"golang.org/x/text/transform"
	"golang.org/x/text/unicode/norm"
)

// GenerateQuerySearchString ...
func GenerateQuerySearchString(s string) bson.M {
	return bson.M{
		"$regex": primitive.Regex{Pattern: NonAccentVietnamese(s), Options: "i"},
	}
}

// NonAccentVietnamese ...
func NonAccentVietnamese(str string) string {
	str = strings.ToLower(str)
	str = replaceStringWithRegex(str, `đ`, "d")
	t := transform.Chain(norm.NFD, runes.Remove(runes.In(unicode.Mn)), norm.NFC)
	result, _, _ := transform.String(t, str)
	result = replaceStringWithRegex(result, `[^a-zA-Z0-9\s]`, "")

	return result
}

// replaceStringWithRegex ...
func replaceStringWithRegex(src string, regex string, replaceText string) string {
	reg := regexp.MustCompile(regex)
	return reg.ReplaceAllString(src, replaceText)
}