Implement safe mode

jsonte/functions/docs.go

@@ -3,7 +3,6 @@ package functions
 import (
 	"fmt"
 	"github.com/Bedrock-OSS/go-burrito/burrito"
-	"io/ioutil"
 	"os"
 	"strings"
 )
@@ -46,7 +45,7 @@ func GenerateDocs() error {
 		if err != nil {
 			return burrito.WrapErrorf(err, "Failed to create %s directory", group.Name)
 		}
-		err = ioutil.WriteFile(fmt.Sprintf("%s-functions/index.md", group.Name), []byte(fmt.Sprintf(`---
+		err = os.WriteFile(fmt.Sprintf("%s-functions/index.md", group.Name), []byte(fmt.Sprintf(`---
 layout: page
 title: %[1]s
 parent: JSON Templating Engine
@@ -69,7 +68,7 @@ has_children: true
 				if fn.IsUnsafe {
 					summary += "\n\n**This method is marked as unsafe. It can be disabled in certain environments.**"
 				}
-				err = ioutil.WriteFile(fmt.Sprintf("%s-functions/%s.md", group.Name, fn.Name), []byte(fmt.Sprintf(`---
+				err = os.WriteFile(fmt.Sprintf("%s-functions/%s.md", group.Name, fn.Name), []byte(fmt.Sprintf(`---
 layout: page
 grand_parent: JSON Templating Engine
 parent: %[2]s

jsonte/functions/function_definition.go

@@ -33,6 +33,8 @@ var initialized = false
 var cacheAll = false
 var cacheAllBucket = "all-cache"
 
+var SafeMode = false
+
 func Init() {
 	if !initialized {
 		RegisterMathFunctions()
@@ -207,6 +209,9 @@ func callFunctionImpl(name string, fns []JsonFunction, args []types.JsonType, re
 		return nil, burrito.WrapErrorf(nil, "Ambiguous function call for \"%s\". Matched: %s", name, strings.Join(matched, ", "))
 	} else {
 		fn := matching[0]
+		if SafeMode && fn.IsUnsafe {
+			return nil, burrito.WrappedErrorf("Function \"%s\" is marked as unsafe and is disabled in safe mode", name)
+		}
 		key := ""
 		if cacheAll {
 			keyObject := types.AsObject(utils.ToNavigableMap(

jsonte/functions/minecraft_functions.go

@@ -508,7 +508,7 @@ func listLatestFiles(p string, m utils.NavigableMap[string, string], version *ty
 	result := map[string]string{}
 	keys := m.Keys()
 	for i := len(keys) - 1; i >= 0; i-- {
-		if !version.IsEmpty() {
+		if version != nil && !version.IsEmpty() {
 			ver, err := types.ParseSemverString(keys[i])
 			if err != nil {
 				continue

jsonte/init.go

@@ -9,11 +9,14 @@ import (
 )
 
 // LibraryInit initializes jsonte in library mode
-func LibraryInit(level zapcore.Level) {
+// level is the logging level
+// safeMode is whether to run in safe mode, which disables unsafe functions
+func LibraryInit(level zapcore.Level, safeMode bool) {
 	utils.InitLogging(level)
 	color.NoColor = true
 	types.Init()
 	functions.Init()
+	functions.SafeMode = safeMode
 }
 
 // FetchCache caches the vanilla packs