package dao

import (
	pgTransaction "github.com/linmadan/egglib-go/transaction/pg"
	"gitlab.fjmaimaimai.com/allied-creation/performance/pkg/domain"
	"gitlab.fjmaimaimai.com/allied-creation/performance/pkg/infrastructure/pg/models"
)

type EvaluationProjectDao struct {
	transactionContext *pgTransaction.TransactionContext
}

func NewEvaluationProjectDao(options map[string]interface{}) *EvaluationProjectDao {
	var transactionContext *pgTransaction.TransactionContext
	if value, ok := options["transactionContext"]; ok {
		transactionContext = value.(*pgTransaction.TransactionContext)
	}
	return &EvaluationProjectDao{
		transactionContext: transactionContext,
	}
}

func (d *EvaluationProjectDao) UpdateSummaryState(id int64, status domain.ProjectSummaryState) error {
	db := d.transactionContext.PgTx
	_, err := db.Model(&models.EvaluationProject{}).
		Where("id=?", id).
		Set("summary_state=?", int(status)).
		Update()
	return err
}