package cert

import (
	"crypto/tls"
	"sync"

	"github.com/Dreamacro/clash/component/trie"
)

// DomainTrieCertsStorage cache wildcard certificates
type DomainTrieCertsStorage struct {
	certsCache *trie.DomainTrie[*tls.Certificate]
	lock       sync.RWMutex
}

// Get gets the certificate from the storage
func (c *DomainTrieCertsStorage) Get(key string) (*tls.Certificate, bool) {
	c.lock.RLock()
	defer c.lock.RUnlock()

	ca := c.certsCache.Search(key)
	if ca == nil {
		return nil, false
	}
	return ca.Data, true
}

// Set saves the certificate to the storage
func (c *DomainTrieCertsStorage) Set(key string, cert *tls.Certificate) {
	c.lock.Lock()
	_ = c.certsCache.Insert(key, cert)
	c.lock.Unlock()
}

func NewDomainTrieCertsStorage() *DomainTrieCertsStorage {
	return &DomainTrieCertsStorage{
		certsCache: trie.New[*tls.Certificate](),
	}
}