cmd/graph.go


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

package cmd

import (
	"os"

	"github.com/nektos/act/pkg/common"
	"github.com/nektos/act/pkg/model"
)

func drawGraph(plan *model.Plan) error {
	drawings := make([]*common.Drawing, 0)

	jobPen := common.NewPen(common.StyleSingleLine, 96)
	arrowPen := common.NewPen(common.StyleNoLine, 97)
	for i, stage := range plan.Stages {
		if i > 0 {
			drawings = append(drawings, arrowPen.DrawArrow())
		}

		ids := make([]string, 0)
		for _, r := range stage.Runs {
			ids = append(ids, r.String())
		}
		drawings = append(drawings, jobPen.DrawBoxes(ids...))
	}

	maxWidth := 0
	for _, d := range drawings {
		if d.GetWidth() > maxWidth {
			maxWidth = d.GetWidth()
		}
	}

	for _, d := range drawings {
		d.Draw(os.Stdout, maxWidth)
	}
	return nil
}