// Tested on etcd 3.1+
package etcd3locker

import (
	"context"
	"time"

	"github.com/tus/tusd"
	"go.etcd.io/etcd/clientv3/concurrency"
)

type etcd3Lock struct {
	Id      string
	Mutex   *concurrency.Mutex
	Session *concurrency.Session
}

func newEtcd3Lock(session *concurrency.Session, id string) *etcd3Lock {
	return &etcd3Lock{
		Mutex:   concurrency.NewMutex(session, id),
		Session: session,
	}
}

func (lock *etcd3Lock) Acquire() error {
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	// this is a blocking call; if we receive DeadlineExceeded
	// the lock is most likely already taken
	if err := lock.Mutex.Lock(ctx); err != nil {
		if err == context.DeadlineExceeded {
			return tusd.ErrFileLocked
		} else {
			return err
		}
	}
	return nil
}

func (lock *etcd3Lock) Release() error {
	return lock.Mutex.Unlock(context.Background())
}

func (lock *etcd3Lock) CloseSession() error {
	return lock.Session.Close()
}