expr/func-import.go

// Copyright (c) 2024 Celestino Amoroso (celestino.amoroso@gmail.com).
// All rights reserved.

// func-import.go
package expr

import (
	"io"
	"os"
)

const ENV_EXPR_PATH = "EXPR_PATH"

func importFunc(ctx ExprContext, name string, args []any) (result any, err error) {
	return importGeneral(ctx, name, args)
}

func importAllFunc(ctx ExprContext, name string, args []any) (result any, err error) {
	enable(ctx, control_export_all)
	return importGeneral(ctx, name, args)
}

func importGeneral(ctx ExprContext, name string, args []any) (result any, err error) {
	var dirList []string

	dirList = addEnvImportDirs(dirList)
	dirList = addPresetImportDirs(ctx, dirList)
	result, err = doImport(ctx, name, dirList, NewArrayIterator(args))
	return
}

func addPresetImportDirs(ctx ExprContext, dirList []string) []string {
	return addPresetDirs(ctx, ControlImportPath, dirList)
}

func doImport(ctx ExprContext, name string, dirList []string, it Iterator) (result any, err error) {
	var v any
	var sourceFilepath string

	for v, err = it.Next(); err == nil; v, err = it.Next() {
		if err = checkStringParamExpected(name, v, it.Index()); err != nil {
			break
		}
		if sourceFilepath, err = makeFilepath(v.(string), dirList); err != nil {
			break
		}
		var file *os.File
		if file, err = os.Open(sourceFilepath); err == nil {
			defer file.Close()
			var expr *ast
			scanner := NewScanner(file, DefaultTranslations())
			parser := NewParser(ctx)
			if expr, err = parser.parseGeneral(scanner, true, true); err == nil {
				result, err = expr.eval(ctx, false)
			}
			if err != nil {
				break
			}
		} else {
			break
		}
	}
	if err != nil {
		if err == io.EOF {
			err = nil
		} else {
			result = nil
		}
	}
	return
}

func ImportImportFuncs(ctx ExprContext) {
	ctx.RegisterFunc("import", NewGolangFunctor(importFunc), TypeAny, []ExprFuncParam{
		NewFuncParamFlag(paramFilepath, PfRepeat),
	})
	ctx.RegisterFunc("importAll", NewGolangFunctor(importAllFunc), TypeAny, []ExprFuncParam{
		NewFuncParamFlag(paramFilepath, PfRepeat),
	})
}

func init() {
	registerImport("import", ImportImportFuncs, "Functions import() and include()")
}