2024-04-10 10:45:58 +00:00
|
|
|
package cmdlang
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"errors"
|
2024-04-10 12:19:11 +00:00
|
|
|
"fmt"
|
2024-04-10 10:45:58 +00:00
|
|
|
"github.com/lmika/gopkgs/fp/slices"
|
|
|
|
"strconv"
|
2024-04-10 12:19:11 +00:00
|
|
|
"strings"
|
2024-04-10 10:45:58 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
type evaluator struct {
|
|
|
|
}
|
|
|
|
|
2024-04-10 11:58:06 +00:00
|
|
|
func (e evaluator) evaluate(ctx context.Context, ec *evalCtx, n *astPipeline) (object, error) {
|
|
|
|
res, err := e.evalCmd(ctx, ec, n.First)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
if len(n.Rest) == 0 {
|
|
|
|
return res, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// Command is a pipeline, so build it out
|
|
|
|
for _, rest := range n.Rest {
|
|
|
|
out, err := e.evalCmd(ctx, ec.withCurrentStream(asStream(res)), rest)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
res = out
|
|
|
|
}
|
|
|
|
return res, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (e evaluator) evalCmd(ctx context.Context, ec *evalCtx, ast *astCmd) (object, error) {
|
2024-04-10 10:45:58 +00:00
|
|
|
cmd, err := ec.lookupCmd(ast.Name)
|
|
|
|
if err != nil {
|
2024-04-10 11:58:06 +00:00
|
|
|
return nil, err
|
2024-04-10 10:45:58 +00:00
|
|
|
}
|
|
|
|
|
2024-04-10 12:19:11 +00:00
|
|
|
args, err := slices.MapWithError(ast.Args, func(a astCmdArg) (object, error) {
|
2024-04-10 10:45:58 +00:00
|
|
|
return e.evalArg(ctx, ec, a)
|
|
|
|
})
|
|
|
|
if err != nil {
|
2024-04-10 11:58:06 +00:00
|
|
|
return nil, err
|
2024-04-10 10:45:58 +00:00
|
|
|
}
|
|
|
|
|
2024-04-11 10:58:59 +00:00
|
|
|
invArgs := invocationArgs{ec: ec, args: args}
|
|
|
|
|
2024-04-11 10:47:59 +00:00
|
|
|
if ec.currentStream != nil {
|
|
|
|
if si, ok := cmd.(streamInvokable); ok {
|
2024-04-11 10:58:59 +00:00
|
|
|
return si.invokeWithStream(ctx, ec.currentStream, invArgs)
|
2024-04-11 10:47:59 +00:00
|
|
|
} else {
|
|
|
|
if err := ec.currentStream.close(); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2024-04-11 10:58:59 +00:00
|
|
|
return cmd.invoke(ctx, invArgs)
|
2024-04-10 10:45:58 +00:00
|
|
|
}
|
|
|
|
|
2024-04-10 12:19:11 +00:00
|
|
|
func (e evaluator) evalArg(ctx context.Context, ec *evalCtx, n astCmdArg) (object, error) {
|
|
|
|
switch {
|
|
|
|
case n.Literal != nil:
|
|
|
|
return e.evalLiteral(ctx, ec, n.Literal)
|
2024-04-11 10:58:59 +00:00
|
|
|
case n.Var != nil:
|
|
|
|
v, ok := ec.getVar(*n.Var)
|
|
|
|
if !ok {
|
|
|
|
return nil, fmt.Errorf("unknown variable %s", *n.Var)
|
|
|
|
}
|
|
|
|
return v, nil
|
2024-04-10 12:19:11 +00:00
|
|
|
case n.Sub != nil:
|
|
|
|
return e.evalSub(ctx, ec, n.Sub)
|
|
|
|
}
|
|
|
|
return nil, errors.New("unhandled arg type")
|
2024-04-10 10:45:58 +00:00
|
|
|
}
|
|
|
|
|
2024-04-10 12:19:11 +00:00
|
|
|
func (e evaluator) evalLiteral(ctx context.Context, ec *evalCtx, n *astLiteral) (object, error) {
|
2024-04-10 10:45:58 +00:00
|
|
|
switch {
|
|
|
|
case n.Str != nil:
|
|
|
|
uq, err := strconv.Unquote(*n.Str)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
2024-04-10 12:19:11 +00:00
|
|
|
return strObject(uq), nil
|
2024-04-10 10:45:58 +00:00
|
|
|
case n.Ident != nil:
|
2024-04-10 12:19:11 +00:00
|
|
|
return strObject(*n.Ident), nil
|
2024-04-10 10:45:58 +00:00
|
|
|
}
|
|
|
|
return "", errors.New("unhandled literal type")
|
|
|
|
}
|
2024-04-10 12:19:11 +00:00
|
|
|
|
|
|
|
func (e evaluator) evalSub(ctx context.Context, ec *evalCtx, n *astPipeline) (object, error) {
|
|
|
|
pipelineRes, err := e.evaluate(ctx, ec, n)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
|
|
|
|
switch v := pipelineRes.(type) {
|
|
|
|
case stream:
|
|
|
|
// TODO: use proper lists here, not a string join
|
|
|
|
sb := strings.Builder{}
|
|
|
|
if err := forEach(v, func(o object) error {
|
|
|
|
// TODO: use o.String()
|
|
|
|
sb.WriteString(fmt.Sprint(o))
|
|
|
|
return nil
|
|
|
|
}); err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
|
|
|
|
return strObject(sb.String()), nil
|
|
|
|
}
|
|
|
|
return pipelineRes, nil
|
|
|
|
}
|