Merge pull request #42 from JoseLucasapp/env-support

feat: .env support

Author: JoseLucasapp

code_examples/env/.env

@@ -0,0 +1,4 @@
+API_KEY=abc123
+ENV=production
+
+#this file is only for test, never push your .env file public

code_examples/env/env.zum

@@ -0,0 +1,5 @@
+dotenvLoad(".env");
+
+var key << dotenvGet("API_KEY");
+var environment << dotenvGet("ENV");
+show("{}, {}", key, environment);

evaluator/builtins.go

@@ -17,7 +17,7 @@ func init() {
 	}
 
 	extras := []string{
-		"date", "hashCode",
+		"date", "dotenvLoad", "dotenvGet", "hashCode",
 	}
 
 	http := []string{

object/builtins/builtins.go

@@ -39,6 +39,12 @@ var Builtins = []struct {
 	{
 		"dictValues", DictValuesBuiltin(),
 	},
+	{
+		"dotenvLoad", loadEnvBuiltin(),
+	},
+	{
+		"dotenvGet", getEnvBuiltin(),
+	},
 	{
 		"first", ArrayFirstBuiltin(),
 	},

object/builtins/env_builtin.go

@@ -0,0 +1,79 @@
+package builtins
+
+import (
+	"bufio"
+	"fmt"
+	"os"
+	"strings"
+	"zumbra/object"
+)
+
+var EnvVars = map[string]string{}
+
+func loadEnvBuiltin() *object.Builtin {
+	return &object.Builtin{
+		Fn: func(args ...object.Object) object.Object {
+			if len(args) != 1 {
+				return NewError("wrong number of arguments. got=%d, want=1", len(args))
+			}
+
+			path, ok := args[0].(*object.String)
+
+			if !ok {
+				return NewError("argument to `env` must be STRING, got %s", args[0].Type())
+			}
+
+			file, err := os.Open(path.Value)
+			if err != nil {
+				fmt.Println(fmt.Sprintf("failed to open file: %s", err))
+				return nil
+			}
+			defer file.Close()
+
+			scanner := bufio.NewScanner(file)
+
+			for scanner.Scan() {
+				line := scanner.Text()
+				if strings.TrimSpace(line) == "" || strings.HasPrefix(line, "#") {
+					continue
+				}
+				parts := strings.SplitN(line, "=", 2)
+				if len(parts) == 2 {
+					key := strings.TrimSpace(parts[0])
+					value := strings.TrimSpace(parts[1])
+					EnvVars[key] = value
+				}
+			}
+
+			if err := scanner.Err(); err != nil {
+				return NewError("failed to read file: %s", err)
+			}
+
+			return nil
+		},
+	}
+}
+
+func getEnvBuiltin() *object.Builtin {
+	return &object.Builtin{
+		Fn: func(args ...object.Object) object.Object {
+			if len(args) != 1 {
+				return NewError("wrong number of arguments. got=%d, want=1", len(args))
+			}
+
+			key, ok := args[0].(*object.String)
+
+			if !ok {
+				return NewError("argument to `getEnv` must be STRING, got %s", args[0].Type())
+			}
+
+			value, ok := EnvVars[key.Value]
+
+			if !ok {
+				return nil
+			}
+
+			return &object.String{Value: value}
+		},
+	}
+}

object/object.go

@@ -29,6 +29,7 @@ const (
 	FLOAT_OBJ             = "FLOAT"
 	DATE_OBJ              = "DATE"
 	RECORD_OBJ            = "RECORD"
+	ENV_OBJ               = "ENV"
 )
 
 type Object interface {