stash/internal/api/resolver_query_job.go

package api

import (
	"context"
	"strconv"

	"github.com/stashapp/stash/internal/manager"
	"github.com/stashapp/stash/pkg/job"
)

func (r *queryResolver) JobQueue(ctx context.Context) ([]*Job, error) {
	queue := manager.GetInstance().JobManager.GetQueue()

	var ret []*Job
	for _, j := range queue {
		ret = append(ret, jobToJobModel(j))
	}

	return ret, nil
}

func (r *queryResolver) FindJob(ctx context.Context, input FindJobInput) (*Job, error) {
	jobID, err := strconv.Atoi(input.ID)
	if err != nil {
		return nil, err
	}
	j := manager.GetInstance().JobManager.GetJob(jobID)
	if j == nil {
		return nil, nil
	}

	return jobToJobModel(*j), nil
}

func jobToJobModel(j job.Job) *Job {
	subTasks := make([]string, len(j.Details))
	for i, t := range j.Details {
		subTasks[i] = sanitiseWebsocketString(t)
	}

	var jobError *string
	if j.Error != nil {
		s := sanitiseWebsocketString(*j.Error)
		jobError = &s
	}

	ret := &Job{
		ID:          strconv.Itoa(j.ID),
		Status:      JobStatus(j.Status),
		Description: sanitiseWebsocketString(j.Description),
		SubTasks:    subTasks,
		StartTime:   j.StartTime,
		EndTime:     j.EndTime,
		AddTime:     j.AddTime,
		Error:       jobError,
	}

	if j.Progress != -1 {
		ret.Progress = &j.Progress
	}

	return ret
}