scpt/scpt.go

155 lines
3.4 KiB
Go
Raw Normal View History

2021-03-01 17:11:22 +00:00
package scpt
import (
"errors"
"fmt"
"github.com/alecthomas/participle"
"github.com/antonmedv/expr"
2021-03-01 17:11:22 +00:00
"io"
"reflect"
2021-03-01 17:11:22 +00:00
"strings"
)
// Vars stores any variables set during script runtime
2021-03-01 17:11:22 +00:00
var Vars = map[string]interface{}{}
// FuncMap is a map of strings mapped to suitable script functions
2021-03-01 17:11:22 +00:00
type FuncMap map[string]func(map[string]interface{}) (interface{}, error)
// Funcs stores the functions allowed for use in a script
2021-03-01 17:11:22 +00:00
var Funcs = FuncMap{
"display-dialog": displayDialog,
"do-shell-script": doShellScript,
"print": print,
2021-03-01 17:11:22 +00:00
}
// AddFuncs adds all functions from the provided FuncMap into
// the Funcs variable
2021-03-01 17:11:22 +00:00
func AddFuncs(fnMap FuncMap) {
for name, fn := range fnMap {
Funcs[name] = fn
}
}
// AddVars adds all functions from the provided map into
// the Vars variable
2021-03-01 17:11:22 +00:00
func AddVars(varMap map[string]interface{}) {
for name, val := range varMap {
Vars[name] = val
}
}
// Parse uses participle to parse a script from r into a new AST
2021-03-01 17:11:22 +00:00
func Parse(r io.Reader) (*AST, error) {
parser, err := participle.Build(&AST{})
if err != nil {
return nil, err
}
ast := &AST{}
2021-03-01 17:22:00 +00:00
err = parser.Parse(r, ast)
2021-03-01 17:11:22 +00:00
if err != nil {
return nil, err
}
return ast, nil
}
// ParseValue parses a Value struct into a go value
func ParseValue(val *Value) (interface{}, error) {
2021-03-01 17:11:22 +00:00
if val.String != nil {
return strings.Trim(*val.String, `"`), nil
2021-03-01 17:11:22 +00:00
} else if val.Bool != nil {
return *val.Bool, nil
2021-03-01 17:11:22 +00:00
} else if val.Float != nil {
return *val.Float, nil
2021-03-01 17:11:22 +00:00
} else if val.Integer != nil {
return *val.Integer, nil
2021-03-01 17:11:22 +00:00
} else if val.SubCmd != nil {
return val.SubCmd, nil
2021-03-01 17:11:22 +00:00
} else if val.VarVal != nil {
return Vars[*val.VarVal], nil
} else if val.Expr != nil {
left, _ := ParseValue(val.Expr.Left)
if isStr(left) {
left = requoteStr(left.(string))
}
right, _ := ParseValue(val.Expr.Right)
if isStr(right) {
right = requoteStr(right.(string))
}
exp := fmt.Sprintf(
"%v %s %v",
left,
val.Expr.Op,
right,
)
fmt.Println(exp)
program, err := expr.Compile(strings.ReplaceAll(exp, "^", "**"))
if err != nil {
return nil, err
}
out, err := expr.Run(program, Vars)
if err != nil {
return nil, err
}
return out, nil
}
return nil, nil
}
// Add quotes to an unquoted string
func requoteStr(s string) string {
return `"` + s + `"`
}
// Check if i is a string
func isStr(i interface{}) bool {
if reflect.TypeOf(i).String() == "string" {
return true
2021-03-01 17:11:22 +00:00
}
return false
2021-03-01 17:11:22 +00:00
}
// UnwrapArgs takes a slice of Arg structs and returns a map
// storing the argument name and its value. If the argument has
// no name, its key will be an empty string
func UnwrapArgs(args []*Arg) (map[string]interface{}, error) {
2021-03-01 17:11:22 +00:00
argMap := map[string]interface{}{}
for _, arg := range args {
val, err := ParseValue(arg.Value)
if err != nil {
return nil, err
}
if IsFuncCall(val) {
argMap[arg.Key], err = CallFunction(val.(*FuncCall))
if err != nil {
return nil, err
}
continue
}
argMap[arg.Key] = val
2021-03-01 17:11:22 +00:00
}
return argMap, nil
2021-03-01 17:11:22 +00:00
}
// IsFuncCall checks if val is a FuncCall struct
func IsFuncCall(val interface{}) bool {
if strings.Contains(reflect.TypeOf(val).String(), ".FuncCall") {
return true
}
return false
2021-03-01 17:11:22 +00:00
}
// CallFunction executes a given function call in the form of
// a FuncCall struct
func CallFunction(call *FuncCall) (interface{}, error) {
argMap, err := UnwrapArgs(call.Args)
if err != nil {
return nil, err
2021-03-01 17:11:22 +00:00
}
fn, ok := Funcs[call.Name]
2021-03-01 17:11:22 +00:00
if !ok {
return nil, errors.New("no such function: " + call.Name)
2021-03-01 17:11:22 +00:00
}
return fn(argMap)
2021-03-01 17:11:22 +00:00
}