3455f491ca
This commit adds and passes context around to (almost) every function. This allows cancelling (almost) every function call globally.
54 lines
1.6 KiB
Go
54 lines
1.6 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
wisski_distillery "github.com/FAU-CDI/wisski-distillery"
|
|
"github.com/FAU-CDI/wisski-distillery/internal/cli"
|
|
"github.com/FAU-CDI/wisski-distillery/internal/wisski"
|
|
"github.com/tkw1536/goprogram/exit"
|
|
"github.com/tkw1536/goprogram/status"
|
|
"github.com/tkw1536/goprogram/stream"
|
|
)
|
|
|
|
// Cron is the 'cron' command
|
|
var Rebuild wisski_distillery.Command = rebuild{}
|
|
|
|
type rebuild struct {
|
|
Parallel int `short:"p" long:"parallel" description:"run on (at most) this many instances in parallel. 0 for no limit." default:"1"`
|
|
Positionals struct {
|
|
Slug []string `positional-arg-name:"SLUG" required:"0" description:"slug of instance(s) to run rebuild"`
|
|
} `positional-args:"true"`
|
|
}
|
|
|
|
func (rebuild) Description() wisski_distillery.Description {
|
|
return wisski_distillery.Description{
|
|
Requirements: cli.Requirements{
|
|
NeedsDistillery: true,
|
|
},
|
|
Command: "rebuild",
|
|
Description: "Runs the rebuild script for several instances",
|
|
}
|
|
}
|
|
|
|
var errRebuildFailed = exit.Error{
|
|
Message: "Failed to run rebuild script for instance %q: exited with code %s",
|
|
ExitCode: exit.ExitGeneric,
|
|
}
|
|
|
|
func (rb rebuild) Run(context wisski_distillery.Context) error {
|
|
dis := context.Environment
|
|
|
|
// find the instances
|
|
wissKIs, err := dis.Instances().Load(context.Context, rb.Positionals.Slug...)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// and do the actual rebuild
|
|
return status.StreamGroup(context.IOStream, rb.Parallel, func(instance *wisski.WissKI, io stream.IOStream) error {
|
|
return instance.Barrel().Build(context.Context, io, true)
|
|
}, wissKIs, status.SmartMessage(func(item *wisski.WissKI) string {
|
|
return fmt.Sprintf("rebuild %q", item.Slug)
|
|
}))
|
|
}
|