package streammanager

import (
	"container/list"
	"sync"
	"time"

	"github.com/libp2p/go-libp2p/core/peer"

	"github.com/whyrusleeping/timecache"
)

const (
	coolDownPeriod = 1 * time.Minute
)

type coolDownCache struct {
	mu        sync.Mutex
	timeCache *timecache.TimeCache
}

func newCoolDownCache() *coolDownCache {
	tl := timecache.NewTimeCache(coolDownPeriod)
	return &coolDownCache{
		timeCache: tl,
	}
}

// Has check and add the peer ID to the cache
func (cache *coolDownCache) Has(id peer.ID) bool {
	cache.mu.Lock()
	defer cache.mu.Unlock()
	has := cache.timeCache.Has(string(id))
	return has
}

// Add adds the peer ID to the cache
func (cache *coolDownCache) Add(id peer.ID) {
	cache.mu.Lock()
	defer cache.mu.Unlock()
	has := cache.timeCache.Has(string(id))
	if !has {
		cache.timeCache.Add(string(id))
	}
}

// Reset the cool down cache
func (cache *coolDownCache) Reset() {
	cache.mu.Lock()
	defer cache.mu.Unlock()
	cache.timeCache.Q = list.New()
	cache.timeCache.M = make(map[string]time.Time)
}