package dao

import (
	"context"

	"prompts-core/consts/public"
	"prompts-core/model/entity"

	"gitea.com/red-future/common/db/gfdb"
)

var ComposeTask = &composeTaskDao{}

type composeTaskDao struct{}

func (d *composeTaskDao) Insert(ctx context.Context, m *entity.ComposeTask) (id int64, err error) {
	r, err := gfdb.DB(ctx).Model(ctx, public.TableNameComposeTask).Data(m).Insert()
	if err != nil {
		return 0, err
	}
	return r.LastInsertId()
}

func (d *composeTaskDao) GetByTaskId(ctx context.Context, taskId string) (m *entity.ComposeTask, err error) {
	r, err := gfdb.DB(ctx).Model(ctx, public.TableNameComposeTask).
		Where("deleted_at IS NULL").
		Where(entity.ComposeTaskCol.TaskId, taskId).
		One()
	if err != nil {
		return nil, err
	}
	if r.IsEmpty() {
		return nil, nil
	}
	err = r.Struct(&m)
	return
}

func (d *composeTaskDao) UpdateByTaskId(ctx context.Context, taskId string, data map[string]any) (rows int64, err error) {
	data[entity.ComposeTaskCol.Updater] = ""
	r, err := gfdb.DB(ctx).Model(ctx, public.TableNameComposeTask).
		Where(entity.ComposeTaskCol.TaskId, taskId).
		Data(data).
		Update()
	if err != nil {
		return 0, err
	}
	return r.RowsAffected()
}