1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
package askcli
import (
"bytes"
"context"
"encoding/json"
"fmt"
"io"
"strings"
)
func (d *Dispatcher) handleInfo(ctx context.Context, args []string, stdout, stderr io.Writer) (int, error) {
tasks, code, err := d.infoTasks(ctx, args, stderr)
if err != nil {
writeInfoError(stderr, err)
return code, nil
}
if d.jsonOutput {
data, err := json.Marshal(tasks)
if err != nil {
fmt.Fprintf(stderr, "error: failed to marshal JSON: %v\n", err)
return 1, nil
}
_, _ = stdout.Write(data)
_, _ = io.WriteString(stdout, "\n")
} else {
allUUIDs := append([]string{tasks[0].UUID}, tasks[0].Depends...)
aliases, err := ensureTaskAliasesForUUIDs(allUUIDs)
if err != nil {
fmt.Fprintf(stderr, "error: failed to load task aliases: %v\n", err)
return 1, nil
}
_, _ = io.WriteString(stdout, FormatTaskInfo(tasks[0], displayTaskAlias(tasks[0].UUID, aliases), aliases))
}
return 0, nil
}
func (d *Dispatcher) infoTasks(ctx context.Context, args []string, stderr io.Writer) ([]TaskExport, int, error) {
if len(args) >= 2 {
_, tasks, code, err := d.resolveTaskSelector(ctx, args[1], stderr)
return tasks, code, err
}
return d.startedInfoTasks(ctx, stderr)
}
func (d *Dispatcher) startedInfoTasks(ctx context.Context, stderr io.Writer) ([]TaskExport, int, error) {
tasks, code, err := d.exportTasks(ctx, []string{"started", "export"}, stderr)
if err != nil {
return nil, code, err
}
switch len(tasks) {
case 0:
return nil, 1, fmt.Errorf("no started task found")
case 1:
return tasks, 0, nil
default:
return nil, 1, fmt.Errorf("multiple started tasks found; pass an ID or UUID explicitly")
}
}
func (d *Dispatcher) exportTasks(ctx context.Context, args []string, stderr io.Writer) ([]TaskExport, int, error) {
var outBuf bytes.Buffer
code, err := d.runner.Run(ctx, args, nil, &outBuf, stderr)
if code != 0 {
return nil, code, err
}
tasks, err := ParseTaskExport(&outBuf)
if err != nil {
return nil, 1, err
}
if len(tasks) == 0 && len(args) > 0 && strings.HasPrefix(args[0], "uuid:") {
return nil, 1, fmt.Errorf("task not found")
}
return tasks, 0, nil
}
func writeInfoError(stderr io.Writer, err error) {
msg := err.Error()
if strings.HasPrefix(msg, "error:") {
_, _ = io.WriteString(stderr, msg+"\n")
return
}
fmt.Fprintf(stderr, "error: %v\n", err)
}
|
