package workflow

import (
	"errors"
)

func addTask(tk *Task) error {
	_, err := wfdb.Insert(tk)
	if err != nil {
		return err
	}
	return nil
}

func updateTask(tk *Task) error {
	_, err := wfdb.Id(tk.Id).Update(tk)
	if err != nil {
		return err
	}
	return nil
}

func deleteTask(tkid string) error {
	tk := Task{}
	_, err := wfdb.Id(tkid).Delete(&tk)
	if err != nil {
		return err
	}
	return nil
}

func selectTask(tkid string) (*Task, error) {
	tk := Task{}
	bo, err := wfdb.Id(tkid).Get(&tk)
	if err != nil {
		return nil, err
	}
	if !bo {
		return nil, errors.New("tkid not found")
	}
	return &tk, err
}

func checkTaskQuery(tkid string, user string) (*Task, error) {
	tk := Task{}
	bo, err := wfdb.Where("id = ? and user = ?", tkid, user).Get(&tk)
	if err != nil {
		return nil, err
	}
	if !bo {
		return nil, errors.New("tkid not found")
	}
	return &tk, err
}

func selectTaskStepinsId(stepinsid string) ([]Task, error) {
	tk := []Task{}
	err := wfdb.Where("stepins_id = ? and status = ?", stepinsid, 0).Find(&tk)
	if err != nil {
		return nil, err
	}
	return tk, err
}

func removeTaskStepinsIdAll(flowinsid string) error {
	tk := Task{}
	_, err := wfdb.Where("flowins_id = ? and status = ?", flowinsid, 0).Delete(&tk)
	if err != nil {
		return err
	}
	return nil
}

func removeTaskNodeId(flowinsid string, nodeids ...string) error {
	if len(nodeids) > 0 {
		for _, nodesid := range nodeids {
			tk := Task{}
			_, err := wfdb.Where("flowins_id = ? and status = ? and node_id = ?", flowinsid, 0, nodesid).Delete(&tk)
			if err != nil {
				return err
			}
		}
	}

	return nil
}

func removeTaskFlowId(flowinsid string) error {
	tk := Task{}
	_, err := wfdb.Where("flowins_id = ? and status = ?", flowinsid, 0).Delete(&tk)
	if err != nil {
		return err
	}
	return nil
}

func removeTaskStepinsId(stepinsid string) error {
	tk := Task{}
	_, err := wfdb.Where("stepins_id = ? and status = ?", stepinsid, 0).Delete(&tk)
	if err != nil {
		return err
	}
	return nil
}