60 lines
1.4 KiB
Go
60 lines
1.4 KiB
Go
// Copyright (c) 2024 Celestino Amoroso (celestino.amoroso@gmail.com).
|
|
// All rights reserved.
|
|
|
|
// builtin-fmt.go
|
|
package expr
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
|
|
"git.portale-stac.it/go-pkg/expr/kern"
|
|
)
|
|
|
|
func getStdout(ctx kern.ExprContext) io.Writer {
|
|
var w io.Writer
|
|
if wany, exists := ctx.GetVar(kern.ControlStdout); exists && wany != nil {
|
|
w, _ = wany.(io.Writer)
|
|
}
|
|
if w == nil {
|
|
w = os.Stdout
|
|
}
|
|
return w
|
|
}
|
|
|
|
func printFunc(ctx kern.ExprContext, name string, args map[string]any) (result any, err error) {
|
|
var n int = 0
|
|
if v, exists := args[kern.ParamItem]; exists && v != nil {
|
|
argv := v.([]any)
|
|
n, err = fmt.Fprint(getStdout(ctx), argv...)
|
|
}
|
|
result = int64(n)
|
|
return
|
|
}
|
|
|
|
func printLnFunc(ctx kern.ExprContext, name string, args map[string]any) (result any, err error) {
|
|
var n int = 0
|
|
if v, exists := args[kern.ParamItem]; exists && v != nil {
|
|
argv := v.([]any)
|
|
n, err = fmt.Fprintln(getStdout(ctx), argv...)
|
|
} else {
|
|
n, err = fmt.Fprintln(getStdout(ctx))
|
|
}
|
|
result = int64(n)
|
|
return
|
|
}
|
|
|
|
func ImportFmtFuncs(ctx kern.ExprContext) {
|
|
ctx.RegisterFunc("print", kern.NewGolangFunctor(printFunc), kern.TypeInt, []kern.ExprFuncParam{
|
|
NewFuncParamFlag(kern.ParamItem, PfRepeat),
|
|
})
|
|
ctx.RegisterFunc("println", kern.NewGolangFunctor(printLnFunc), kern.TypeInt, []kern.ExprFuncParam{
|
|
NewFuncParamFlag(kern.ParamItem, PfRepeat),
|
|
})
|
|
}
|
|
|
|
func init() {
|
|
RegisterBuiltinModule("fmt", ImportFmtFuncs, "String and console formatting functions")
|
|
}
|