package sqldb

import (
	"github.com/featurebasedb/featurebase/v3/dax"
	"github.com/featurebasedb/featurebase/v3/dax/models"
	"github.com/featurebasedb/featurebase/v3/errors"
	"github.com/featurebasedb/featurebase/v3/logger"
)

func NewDirectiveVersion(log logger.Logger) dax.DirectiveVersion {
	if log == nil {
		log = logger.NopLogger
	}
	return &directiveVersion{
		log: log,
	}
}

type directiveVersion struct {
	log logger.Logger
}

func (d *directiveVersion) Increment(tx dax.Transaction, delta uint64) (uint64, error) {
	dt, ok := tx.(*DaxTransaction)
	if !ok {
		return 0, dax.NewErrInvalidTransaction("*sqldb.DaxTransaction")
	}
	// table is pre-populated w/ a single record w/ ID=1 during schema migration
	dv := &models.DirectiveVersion{}
	err := dt.C.RawQuery("UPDATE directive_versions SET version = version + ? WHERE id = ? RETURNING id, version", delta, 1).First(dv)
	if err != nil {
		return 0, errors.Wrap(err, "updating directive_version")
	}
	return uint64(dv.Version), nil
}