Implement repeat loops

This commit is contained in:
Elara 2021-03-01 20:18:52 -08:00
parent 745920b139
commit 877f85ef78
5 changed files with 53 additions and 15 deletions

26
ast.go
View File

@ -100,6 +100,22 @@ func executeCmd(cmd *Command) error {
} }
} }
} }
} else if cmd.RptLoops != nil {
// For each repeat loop
for _, RptLoop := range cmd.RptLoops {
for i:=0;i<*RptLoop.Times;i++ {
if RptLoop.IndexVar != nil {
Vars[*RptLoop.IndexVar] = i
}
for _, InnerCmd := range RptLoop.InnerCmds {
// Execute command recursively
err := executeCmd(InnerCmd)
if err != nil {
return fmt.Errorf("%s: %s", InnerCmd.Pos, err)
}
}
}
}
} }
return nil return nil
} }
@ -110,6 +126,7 @@ type Command struct {
Tokens []lexer.Token Tokens []lexer.Token
Vars []*Var `( @@` Vars []*Var `( @@`
Ifs []*If `| @@` Ifs []*If `| @@`
RptLoops []*RptLoop`| @@`
Calls []*FuncCall `| @@)` Calls []*FuncCall `| @@)`
} }
@ -117,7 +134,14 @@ type Command struct {
type If struct { type If struct {
Pos lexer.Position Pos lexer.Position
Condition *Value `"if" @@ "{"` Condition *Value `"if" @@ "{"`
InnerCmds []*Command `@@*"}"` InnerCmds []*Command `@@* "}"`
}
type RptLoop struct {
Pos lexer.Position
Times *int `"repeat" @Number "times" "{"`
IndexVar *string `(@Ident "in")?`
InnerCmds []*Command `@@* "}"`
} }
// FuncCall stores any function calls encountered while parsing a script // FuncCall stores any function calls encountered while parsing a script

View File

@ -81,3 +81,11 @@ func doShellScript(args map[string]interface{}) (interface{}, error) {
return nil, errors.New("script not provided") return nil, errors.New("script not provided")
} }
} }
func toString(args map[string]interface{}) (interface{}, error) {
val, ok := args[""]
if !ok {
return nil, errors.New("no value provided")
}
return fmt.Sprint(val), nil
}

View File

@ -10,7 +10,7 @@ var scptLexer = lexer.Must(stateful.NewSimple([]stateful.Rule{
{"Ident", `[a-zA-Z]\w*`, nil}, {"Ident", `[a-zA-Z]\w*`, nil},
{"String", `"[^"]*"`, nil}, {"String", `"[^"]*"`, nil},
{"Number", `(?:\d*\.)?\d+`, nil}, {"Number", `(?:\d*\.)?\d+`, nil},
{"Punct", `[-[!@#$&()_{}\|:;"',.?/]|]`, nil}, {"Punct", `[-[!@$&()_{}\|:;"',.?/]|]`, nil},
{"Whitespace", `[ \t\r\n]+`, nil}, {"Whitespace", `[ \t\r\n]+`, nil},
{"Comment", `#[^\n]+`, nil}, {"Comment", `#[^\n]+`, nil},
{"Operator", `(>=|<=|>|<|==|!=)|[-+*/^%]`, nil}, {"Operator", `(>=|<=|>|<|==|!=)|[-+*/^%]`, nil},

View File

@ -36,6 +36,7 @@ type FuncMap map[string]func(map[string]interface{}) (interface{}, error)
var Funcs = FuncMap{ var Funcs = FuncMap{
"display-dialog": displayDialog, "display-dialog": displayDialog,
"do-shell-script": doShellScript, "do-shell-script": doShellScript,
"string": toString,
} }
// AddFuncs adds all functions from the provided FuncMap into // AddFuncs adds all functions from the provided FuncMap into

View File

@ -14,3 +14,8 @@ if {3 == 3} {
} }
} }
do-shell-script "echo rpt"
repeat 5 times { z in
do-shell-script {"echo " + (string $z)}
}