esctl/pkg/es/task.go

/*
Copyright © 2026 Thomas von Dein

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package es

import (
	"context"
	"fmt"
	"log/slog"
	"strconv"
	"time"

	"codeberg.org/scip/esctl/pkg/cfg"
	"codeberg.org/scip/esctl/pkg/printer"
)

func TaskList(conf *cfg.Config) error {
	res, err := conf.DefaultCluster.ES().Cat.Tasks().
		Do(context.Background())

	if err != nil {
		return fmt.Errorf("failed to get index templates: %s", esErrorString(err))
	}

	slog.Debug("res", "tasks", res)

	table := printer.NewTable(conf, 8, len(res))
	table.Addheaders("task id", "action", "start time", "run time", "node", "type")
	for idx, task := range res {
		var ts time.Time

		epc, err := strconv.Atoi(*task.StartTime)
		if err != nil {
			ts = time.Now()
		} else {
			ts = time.Unix(int64(epc/1000), 0)
		}

		table.Entries[idx] = []any{
			*task.TaskId,
			*task.Action,
			ts,
			*task.RunningTime,
			*task.Node,
			*task.Type,
		}
	}

	return table.Print()
}

func TaskCancel(conf *cfg.Config, taskid string) error {
	_, err := conf.DefaultCluster.ES().Tasks.Cancel().
		TaskId(taskid).
		Do(context.Background())

	if err != nil {
		return fmt.Errorf("failed to cancel task: %s", esErrorString(err))
	}

	return nil
}