package common

import (
	"time"

	"github.com/tidbcloud/account/pkg/grpc/middleware/ratelimit"
)

// GetContextFromState generate a new limiter.Context from given state.
func GetContextFromState(now time.Time, rate ratelimit.Rate, expiration time.Time, count int64) ratelimit.Context {
	limit := rate.Limit
	remaining := int64(0)
	reached := true

	if count <= limit {
		remaining = limit - count
		reached = false
	}

	reset := expiration.Unix()

	return ratelimit.Context{
		Limit:     limit,
		Remaining: remaining,
		Reset:     reset,
		Reached:   reached,
	}
}