package lite

import (
	"fmt"
	"sync"

	"github.com/tendermint/tendermint/libs/log"
	"github.com/tendermint/tendermint/types"
)

// Provider provides information for the lite client to sync validators.
// Examples: MemProvider, files.Provider, client.Provider, CacheProvider.
type Provider interface {

	// LatestFullCommit returns the latest commit with minHeight <= height <=
	// maxHeight.
	// If maxHeight is zero, returns the latest where minHeight <= height.
	LatestFullCommit(chainID string, minHeight, maxHeight int64) (FullCommit, error)

	// Get the valset that corresponds to chainID and height and return.
	// Height must be >= 1.
	ValidatorSet(chainID string, height int64) (*types.ValidatorSet, error)

	// Set a logger.
	SetLogger(logger log.Logger)
}

// A provider that can also persist new information.
// Examples: MemProvider, files.Provider, CacheProvider.
type PersistentProvider interface {
	Provider

	// SaveFullCommit saves a FullCommit (without verification).
	SaveFullCommit(fc FullCommit) error
}

// A provider that can update itself w/ more recent commit data.
type UpdatingProvider interface {
	Provider

	// Update internal information by fetching information somehow.
	// UpdateToHeight will block until the request is complete,
	// or returns an error if the request cannot complete.
	// Generally, one must call UpdateToHeight(h) before LatestFullCommit(_,h,h)
	// will return this height.
	//
	// NOTE: Behavior with concurrent requests is undefined.  To make
	// concurrent calls safe, look at the struct `ConcurrentUpdatingProvider`.
	UpdateToHeight(chainID string, height int64) error
}

//----------------------------------------

type ConcurrentProvider struct {
	UpdatingProvider

	// pending map to synchronize concurrent verification requests
	mtx                  sync.Mutex
	pendingVerifications map[pendingKey]*pendingResult
}

// convenience to create the key for the lookup map
type pendingKey struct {
	chainID string
	height  int64
}

// used to cache the result from underlying UpdatingProvider.
type pendingResult struct {
	wait chan struct{}
	err  error // cached result.
}

func NewConcurrentUpdatingProvider(up UpdatingProvider) ConcurrentProvider {
	return ConcurrentProvider{
		UpdatingProvider:     up,
		pendingVerifications: make(map[pendingKey]*pendingResult),
	}
}

// Returns the unique pending request for all identical calls to
// joinConcurrency(chainID,height), and returns true for isFirstCall only for the
// first call, which should call the returned callback w/ results if any.
// The callback must be called, otherwise there will be memory leaks.
// Other subsequent calls should just return uniq.err.
// NOTE: This is a separate function, primarily to make mtx unlocking more obviously safe via defer.
func (cp ConcurrentProvider) joinConcurrency(chainID string, height int64) (uniq *pendingResult, isFirstCall bool, callback func(error)) {
	cp.mtx.Lock()
	defer cp.mtx.Unlock()

	var pkey = pendingKey{chainID, height}
	cp.mtx.Lock()
	if uniq = cp.pendingVerifications[pkey]; uniq != nil {
		cp.mtx.Unlock()
		<-uniq.wait // uniq.wait is of type `chan struct{}`
		return uniq, false, nil
	} else {
		uniq = &pendingResult{wait: make(chan struct{}), err: nil}
		cp.pendingVerifications[pkey] = uniq
		cp.mtx.Unlock()
		// The caller must call this, otherwise there will be memory leaks.
		return uniq, true, func(err error) {
			// NOTE: other result parameters can be added here.
			uniq.err = err
			// *After* setting the results, *then* call close(uniq.wait).
			close(uniq.wait)
			cp.mtx.Lock() // temporarily acquire lock to remove this item
			delete(cp.pendingVerifications, pkey)
			cp.mtx.Unlock() // and release lock
		}
	}
}

func (cp *ConcurrentProvider) UpdateToHeight(chainID string, height int64) error {

	// Performs synchronization for multi-threads verification at the same height.
	var presult *pendingResult
	var isFirstCall bool
	var callback func(error)
	presult, isFirstCall, callback = cp.joinConcurrency(chainID, height)

	if isFirstCall {
		var err error
		// Use a defer in case UpdateToHeight itself fails.
		defer func() {
			if r := recover(); r != nil {
				err = fmt.Errorf("Recovered from panic: %v", r)
			}
			callback(err)
		}()
		err = cp.UpdatingProvider.UpdateToHeight(chainID, height)
		return err
	} else {
		// Is not the first call, so return the error from previous concurrent calls.
		return presult.err
	}
}