95 lines
2.0 KiB
Go
95 lines
2.0 KiB
Go
package tart
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"os/exec"
|
|
"strings"
|
|
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
const tartCommandName = "tart"
|
|
|
|
var (
|
|
ErrTartNotFound = errors.New("tart command not found")
|
|
ErrTartFailed = errors.New("tart command returned non-zero exit code")
|
|
)
|
|
|
|
type VMInfo struct {
|
|
Name string
|
|
Source string
|
|
State string
|
|
Running bool
|
|
}
|
|
|
|
func Tart(
|
|
ctx context.Context,
|
|
logger *zap.SugaredLogger,
|
|
args ...string,
|
|
) (string, string, error) {
|
|
cmd := exec.CommandContext(ctx, tartCommandName, args...)
|
|
|
|
var stdout, stderr bytes.Buffer
|
|
|
|
cmd.Stdout = &stdout
|
|
cmd.Stderr = &stderr
|
|
|
|
logger.Debugf("running '%s %s'", tartCommandName, strings.Join(args, " "))
|
|
err := cmd.Run()
|
|
if err != nil {
|
|
if errors.Is(err, exec.ErrNotFound) {
|
|
return "", "", fmt.Errorf("%w: %s command not found in PATH, make sure Tart is installed",
|
|
ErrTartNotFound, tartCommandName)
|
|
}
|
|
|
|
if exitErr, ok := err.(*exec.ExitError); ok {
|
|
select {
|
|
case <-ctx.Done():
|
|
// Do not log an error because it's the user's intent to cancel this VM operation
|
|
default:
|
|
logger.Warnf(
|
|
"'%s %s' failed with exit code %d: %s",
|
|
tartCommandName, strings.Join(args, " "),
|
|
exitErr.ExitCode(), firstNonEmptyLine(stderr.String(), stdout.String()),
|
|
)
|
|
}
|
|
|
|
// Tart command failed, redefine the error to be the Tart-specific output
|
|
err = fmt.Errorf("%w: %q", ErrTartFailed, firstNonEmptyLine(stderr.String(), stdout.String()))
|
|
}
|
|
}
|
|
|
|
return stdout.String(), stderr.String(), err
|
|
}
|
|
|
|
func List(ctx context.Context, logger *zap.SugaredLogger) ([]VMInfo, error) {
|
|
output, _, err := Tart(ctx, logger, "list", "--format", "json")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var entries []VMInfo
|
|
|
|
if err := json.Unmarshal([]byte(output), &entries); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return entries, nil
|
|
}
|
|
|
|
func firstNonEmptyLine(outputs ...string) string {
|
|
for _, output := range outputs {
|
|
for _, line := range strings.Split(output, "\n") {
|
|
if line != "" {
|
|
return line
|
|
}
|
|
}
|
|
}
|
|
|
|
return ""
|
|
}
|