2019-11-28 00:39:08 -08:00
|
|
|
package shsh
|
|
|
|
|
|
|
|
|
|
type Operation func(*Token) *Token
|
|
|
|
|
|
|
|
|
|
type Function struct {
|
2019-11-28 20:07:28 -08:00
|
|
|
function Operation
|
2019-11-28 00:39:08 -08:00
|
|
|
timesCalled int
|
|
|
|
|
args int
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
type FuncTable map[string]*Function
|
|
|
|
|
|
|
|
|
|
var (
|
|
|
|
|
GlobalFuncTable *FuncTable
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
// TODO: Currently only checks arg list length
|
|
|
|
|
func ParseFunction(target *Function, args *Token) bool {
|
|
|
|
|
// HANDLE EXEC
|
|
|
|
|
if target.args < 0 {
|
|
|
|
|
return true
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
i := target.args
|
2019-11-28 20:07:28 -08:00
|
|
|
for iter := &args; *iter != nil; iter = &((*iter).next) {
|
2019-11-28 00:39:08 -08:00
|
|
|
i -= 1
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if i != 0 {
|
|
|
|
|
// log error here?
|
|
|
|
|
return false
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return true
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func CallFunction(target *Function, args *Token) *Token {
|
|
|
|
|
if !ParseFunction(target, args) {
|
|
|
|
|
return args
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
target.timesCalled += 1
|
2019-11-28 20:07:28 -08:00
|
|
|
return target.function(args)
|
2019-11-28 00:39:08 -08:00
|
|
|
}
|
|
|
|
|
|
2019-11-28 20:07:28 -08:00
|
|
|
func GetFunction(arg string) *Function {
|
|
|
|
|
target, ok := (*GlobalFuncTable)[arg]
|
2019-11-28 00:39:08 -08:00
|
|
|
if !ok {
|
|
|
|
|
// TODO: hook into stdlib exec call
|
|
|
|
|
;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return target
|
|
|
|
|
}
|