autobrr/pkg/regexcache/regex.go

// Copyright (c) 2021 - 2025, Ludvig Lundgren and the autobrr contributors.
// SPDX-License-Identifier: GPL-2.0-or-later

package regexcache

import (
	"regexp"
	"time"

	"github.com/autobrr/autobrr/pkg/ttlcache"
)

var cache = ttlcache.New[string, *regexp.Regexp](
	ttlcache.Options[string, *regexp.Regexp]{}.
		SetTimerResolution(5 * time.Minute).
		SetDefaultTTL(15 * time.Minute),
)

func MustCompilePOSIX(pattern string) *regexp.Regexp {
	item, ok := cache.Get(pattern)
	if ok {
		return item
	}

	reg := regexp.MustCompilePOSIX(pattern)
	cache.Set(pattern, reg, ttlcache.NoTTL)
	return reg
}

func MustCompile(pattern string) *regexp.Regexp {
	item, ok := cache.Get(pattern)
	if ok {
		return item
	}

	reg := regexp.MustCompile(pattern)
	cache.Set(pattern, reg, ttlcache.NoTTL)
	return reg
}

func CompilePOSIX(pattern string) (*regexp.Regexp, error) {
	item, ok := cache.Get(pattern)
	if ok {
		return item, nil
	}

	reg, err := regexp.CompilePOSIX(pattern)
	if err != nil {
		return nil, err
	}

	cache.Set(pattern, reg, ttlcache.DefaultTTL)
	return reg, nil
}

func Compile(pattern string) (*regexp.Regexp, error) {
	item, ok := cache.Get(pattern)
	if ok {
		return item, nil
	}

	reg, err := regexp.Compile(pattern)
	if err != nil {
		return nil, err
	}

	cache.Set(pattern, reg, ttlcache.DefaultTTL)
	return reg, nil
}

func SubmitOriginal(plain string, reg *regexp.Regexp) {
	cache.Set(plain, reg, ttlcache.DefaultTTL)
}

func FindOriginal(plain string) (*regexp.Regexp, bool) {
	item, ok := cache.Get(plain)
	if ok {
		return item, true
	}

	return nil, false
}