90 lines
1.7 KiB
Go
90 lines
1.7 KiB
Go
package commands
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/hamburghammer/gstat/args"
|
|
)
|
|
|
|
// Result gethers all the results for all commands
|
|
type Result struct {
|
|
args.Arguments
|
|
Collection collection
|
|
}
|
|
|
|
type collection struct {
|
|
Results []string
|
|
Errs []error
|
|
}
|
|
|
|
func (c collection) collectionEquals(otherCollection collection) bool {
|
|
if len(c.Results) != len(otherCollection.Results) {
|
|
return false
|
|
}
|
|
for i, s := range c.Results {
|
|
if s != otherCollection.Results[i] {
|
|
return false
|
|
}
|
|
}
|
|
|
|
if len(c.Errs) != len(otherCollection.Errs) {
|
|
return false
|
|
}
|
|
for i, s := range c.Errs {
|
|
if s != otherCollection.Errs[i] {
|
|
return false
|
|
}
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
// NewResult creates new result struct
|
|
func NewResult(a args.Arguments) Result {
|
|
return Result{Arguments: a}
|
|
}
|
|
|
|
// ExecCommands runs all commands
|
|
func (r Result) ExecCommands(executors []Executor) Result {
|
|
results := make([]string, 0, len(executors))
|
|
errors := make([]error, 0, len(executors))
|
|
|
|
for _, executor := range executors {
|
|
output, err := executor.Exec(r.Arguments)
|
|
s := string(output)
|
|
if err != nil {
|
|
errors = append(errors, err)
|
|
} else if s != "" {
|
|
results = append(results, rmFirstAndLastBracket(s))
|
|
}
|
|
}
|
|
|
|
r.Collection.Errs = errors
|
|
r.Collection.Results = results
|
|
return r
|
|
}
|
|
|
|
// ResultEquals checks for field equality
|
|
func (r Result) ResultEquals(otherResult Result) bool {
|
|
if !r.Arguments.Equals(otherResult.Arguments) {
|
|
return false
|
|
}
|
|
if !r.Collection.collectionEquals(otherResult.Collection) {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
func rmFirstAndLastBracket(s string) string {
|
|
s = strings.Replace(s, "{", "", 1)
|
|
s = reverse(strings.Replace(reverse(s), "}", "", 1))
|
|
return s
|
|
}
|
|
|
|
func reverse(s string) (result string) {
|
|
for _, v := range s {
|
|
result = string(v) + result
|
|
}
|
|
return
|
|
}
|