63 lines
1.4 KiB
Go
63 lines
1.4 KiB
Go
|
package script
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"gopkg.in/yaml.v3"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Script struct {
|
||
|
|
Name string `yaml:"name"`
|
||
|
|
Description string `yaml:"description"`
|
||
|
|
Steps []Step `yaml:"steps"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type Step struct {
|
||
|
|
Name string `yaml:"name"`
|
||
|
|
Run string `yaml:"run"`
|
||
|
|
Register string `yaml:"register"`
|
||
|
|
Condition string `yaml:"if"`
|
||
|
|
IgnoreError bool `yaml:"ignore_error"`
|
||
|
|
Wait *Wait `yaml:"wait"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type Wait struct {
|
||
|
|
Run string `yaml:"run"`
|
||
|
|
Until string `yaml:"until"`
|
||
|
|
Timeout string `yaml:"timeout"`
|
||
|
|
Interval string `yaml:"interval"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func Parse(data []byte) (*Script, error) {
|
||
|
|
var s Script
|
||
|
|
if err := yaml.Unmarshal(data, &s); err != nil {
|
||
|
|
return nil, fmt.Errorf("invalid script YAML: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if s.Name == "" {
|
||
|
|
return nil, fmt.Errorf("script must have a 'name' field")
|
||
|
|
}
|
||
|
|
if len(s.Steps) == 0 {
|
||
|
|
return nil, fmt.Errorf("script must have at least one step")
|
||
|
|
}
|
||
|
|
|
||
|
|
for i, step := range s.Steps {
|
||
|
|
if step.Name == "" {
|
||
|
|
return nil, fmt.Errorf("step %d must have a 'name' field", i+1)
|
||
|
|
}
|
||
|
|
if step.Run == "" && step.Wait == nil {
|
||
|
|
return nil, fmt.Errorf("step %d (%s) must have either 'run' or 'wait'", i+1, step.Name)
|
||
|
|
}
|
||
|
|
if step.Wait != nil {
|
||
|
|
if step.Wait.Run == "" {
|
||
|
|
return nil, fmt.Errorf("step %d (%s): wait must have a 'run' field", i+1, step.Name)
|
||
|
|
}
|
||
|
|
if step.Wait.Until == "" {
|
||
|
|
return nil, fmt.Errorf("step %d (%s): wait must have an 'until' field", i+1, step.Name)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return &s, nil
|
||
|
|
}
|