1set/starlet

package starlet

import (
    libatom "github.com/1set/starlet/lib/atom"
    libb64 "github.com/1set/starlet/lib/base64"
    libcsv "github.com/1set/starlet/lib/csv"
    libfile "github.com/1set/starlet/lib/file"
    libgoid "github.com/1set/starlet/lib/goidiomatic"
    libhash "github.com/1set/starlet/lib/hashlib"
    libhttp "github.com/1set/starlet/lib/http"
    libjson "github.com/1set/starlet/lib/json"
    liblog "github.com/1set/starlet/lib/log"
    libnet "github.com/1set/starlet/lib/net"
    libpath "github.com/1set/starlet/lib/path"
    librand "github.com/1set/starlet/lib/random"
    libre "github.com/1set/starlet/lib/re"
    librt "github.com/1set/starlet/lib/runtime"
    libstat "github.com/1set/starlet/lib/stats"
    libstr "github.com/1set/starlet/lib/string"
    stdmath "go.starlark.net/lib/math"
    stdtime "go.starlark.net/lib/time"
    "go.starlark.net/resolve"
    "go.starlark.net/starlark"
    stdstruct "go.starlark.net/starlarkstruct"
)

var allBuiltinModules = ModuleLoaderMap{
    libgoid.ModuleName: func() (starlark.StringDict, error) {
        return libgoid.LoadModule()
    },
    "math": func() (starlark.StringDict, error) {
        return starlark.StringDict{
            "math": stdmath.Module,
        }, nil
    },
    "struct": func() (starlark.StringDict, error) {
        return starlark.StringDict{
            "struct": starlark.NewBuiltin("struct", stdstruct.Make),
        }, nil
    },
    "time": func() (starlark.StringDict, error) {
        return starlark.StringDict{
            "time": stdtime.Module,
        }, nil
    },
    // add third-party modules
    libatom.ModuleName: libatom.LoadModule,
    libb64.ModuleName:  libb64.LoadModule,
    libcsv.ModuleName:  libcsv.LoadModule,
    libfile.ModuleName: libfile.LoadModule,
    libhash.ModuleName: libhash.LoadModule,
    libhttp.ModuleName: libhttp.LoadModule,
    libnet.ModuleName:  libnet.LoadModule,
    libjson.ModuleName: libjson.LoadModule,
    liblog.ModuleName:  liblog.LoadModule,
    libpath.ModuleName: libpath.LoadModule,
    librand.ModuleName: librand.LoadModule,
    libre.ModuleName:   libre.LoadModule,
    librt.ModuleName:   librt.LoadModule,
    libstr.ModuleName:  libstr.LoadModule,
    libstat.ModuleName: libstat.LoadModule,
}

// GetAllBuiltinModuleNames returns a list of all builtin module names.
func GetAllBuiltinModuleNames() []string {
    return allBuiltinModules.Keys()
}

// GetAllBuiltinModules returns a list of all builtin modules.
func GetAllBuiltinModules() ModuleLoaderList {
    return allBuiltinModules.Values()
}

// GetBuiltinModuleMap returns a map of all builtin modules.
func GetBuiltinModuleMap() ModuleLoaderMap {
    return allBuiltinModules.Clone()
}

// GetBuiltinModule returns the builtin module with the given name.
func GetBuiltinModule(name string) ModuleLoader {
    return allBuiltinModules[name]
}

// EnableRecursionSupport enables recursion support in Starlark environments for loading modules.
func EnableRecursionSupport() {
    resolve.AllowRecursion = true
}

// DisableRecursionSupport disables recursion support in Starlark environments for loading modules.
func DisableRecursionSupport() {
    resolve.AllowRecursion = false
}

// EnableGlobalReassign enables global reassignment in Starlark environments for loading modules.
func EnableGlobalReassign() {
    resolve.AllowGlobalReassign = true
}

// DisableGlobalReassign disables global reassignment in Starlark environments for loading modules.
func DisableGlobalReassign() {
    resolve.AllowGlobalReassign = false
}