add list functions to stdlib
This commit is contained in:
parent
49ea765e40
commit
b01415d786
4 changed files with 146 additions and 12 deletions
|
|
@ -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
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue