add list functions to stdlib

This commit is contained in:
Aidan 2020-06-21 12:29:20 -07:00
parent 49ea765e40
commit b01415d786
No known key found for this signature in database
GPG key ID: 327711E983899316
4 changed files with 146 additions and 12 deletions

View file

@ -23,6 +23,67 @@ import (
)
func GenFuncTable() ast.FuncTable {
var stdlib FuncTable
var stdlib = ast.FuncTable{
"...": &ast.Function{
Function: expand,
Name: "...",
TimesCalled: 0,
Args: 1,
},
"append": &ast.Function{
Function: l_append,
Name: "append",
TimesCalled: 0,
Args: -1,
},
}
return stdlib
}
// TODO: put list funcs in their own file
/* EXPAND
* Takes 1 element: a list
* retuns a sequence of elements (list contents)
* in event a not-list is passed in, returns the arg.
*/
func expand(input *ast.Token, vars ast.VarTable, funcs ast.FuncTable) *ast.Token {
if input.Tag != ast.LIST {
log.Log(log.DEBUG, "expand called on not a list", "expand")
return input
}
return input.Inner.(*ast.Token)
}
/* L_APPEND (append from repl)
* Appends N elements to the end of a list
* Arg one is a list, next args are appended
* if no args are a list, a list is made from all args
*/
func l_append(input *ast.Token, vars ast.VarTable, funcs ast.FuncTable) *ast.Token {
src := input
if input.Tag != ast.LIST {
// TODO: Position, if I can figure out what to do with it
src = &ast.Token{Tag: ast.LIST, Inner: input}
return src
}
// deref inner first
i := src.Inner.(*ast.Token)
iter := &i
if *iter == nil {
src.Inner = input.Next
return src
}
for (*iter).Next != nil {
iter = &((*iter).Next)
}
(*iter).Next = input.Next
return src
}