package service

import "model-engine/db"

func GetTasks() (tasks []*db.ModelTask, err error) {
	models, err := db.NewModelSearch().SetEnabled(true).FindAll()
	if err != nil {
		return nil, err
	}
	if len(models) == 0 {
		return nil, nil
	}

	modelIds := make([]string, 0, len(models))
	for _, model := range models {
		modelIds = append(modelIds, model.ID)
	}

	return db.NewModelTaskSearch().
		SetModelIDs(modelIds).
		SetEnabled(true).
		SetUnexpired().
		FindAll()
}