59 lines
955 B
Go
59 lines
955 B
Go
|
package shsh
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Operation func(*Token) *Token
|
||
|
|
|
||
|
|
type Function struct {
|
||
|
|
func operation
|
||
|
|
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
|
||
|
|
for iter = &args; *iter; iter = &(*iter.next) {
|
||
|
|
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
|
||
|
|
return target.operation(args)
|
||
|
|
}
|
||
|
|
|
||
|
|
func GetFunction(string arg) *Function {
|
||
|
|
target, ok := GlobalFuncTable[arg]
|
||
|
|
if !ok {
|
||
|
|
// TODO: hook into stdlib exec call
|
||
|
|
;
|
||
|
|
}
|
||
|
|
|
||
|
|
return target
|
||
|
|
}
|