dbeaver/pro#5701 re-use the root .env file in Go examples

Author: n5a5

go/env/env.go

@@ -1,41 +1,56 @@
 package env
 
 import (
-	"encoding/json"
+	"bufio"
+	"fmt"
+	"log/slog"
 	"os"
 	"runtime"
+	"strings"
 
 	"github.com/dbeaver/cloudbeaver-graphql-examples/go/lib"
 )
 
 type Env struct {
-	ServerInfo ServerInfo `json:"server"`
-	Token      string     `json:"apiToken"`
+	APIToken   string
+	serverURL  string
+	serviceURI string
 }
 
-type ServerInfo struct {
-	ServerURL  string `json:"serverURL"`
-	ServiceURI string `json:"serviceURI"`
-}
-
-func Read() (Env, error) {
+func Read(path string) (Env, error) {
 	env := Env{}
-	bytes, err := os.ReadFile("env/env.json")
+	file, err := os.Open(path)
 	if err != nil {
-		return env, err
+		return env, lib.WrapError("error while opening the env file", err)
 	}
-	err = json.Unmarshal(bytes, &env)
-	if err != nil {
-		err = lib.WrapError("error while unmarshalling the env file", err)
+	defer lib.CloseOrWarn(file)
+	scanner := bufio.NewScanner(file)
+	for scanner.Scan() {
+		before, after, found := strings.Cut(scanner.Text(), "=")
+		if !found {
+			continue
+		}
+		before = strings.TrimSpace(before)
+		after = strings.TrimSpace(after)
+		switch before {
+		case "api_token":
+			env.APIToken = after
+		case "server_url":
+			env.serverURL = after
+		case "service_uri":
+			env.serviceURI = after
+		default:
+			slog.Warn(fmt.Sprintf("unknown env variable: %s", before))
+		}
 	}
 	return env, err
 }
 
 func (env *Env) GraphqlEndpoint() string {
-	return env.ServerInfo.ServerURL + "/" + env.ServerInfo.ServiceURI + "/gql"
+	return env.serverURL + "/" + env.serviceURI + "/gql"
 }
 
 func (env *Env) PurgeToken() {
-	env.Token = ""
+	env.APIToken = ""
 	runtime.GC()
 }

go/env/env.json.template

@@ -4,7 +4,6 @@ import (
 	"bytes"
 	"encoding/json"
 	"io"
-	"log/slog"
 	"net/http"
 
 	"github.com/dbeaver/cloudbeaver-graphql-examples/go/lib"
@@ -23,7 +22,7 @@ func (client Client) Execute(endpoint string, request Request) (Response, error)
 	if err != nil {
 		return Response{}, lib.WrapError("unable to execute POST request", err)
 	}
-	defer closeOrWarn(httpResponse.Body)
+	defer lib.CloseOrWarn(httpResponse.Body)
 	rawResponseBody, err := io.ReadAll(httpResponse.Body)
 	if err != nil {
 		return Response{}, lib.WrapError("unable to read GraphQL response body", err)
@@ -34,9 +33,3 @@ func (client Client) Execute(endpoint string, request Request) (Response, error)
 	}
 	return response, err
 }
-
-func closeOrWarn(closer io.Closer) {
-	if err := closer.Close(); err != nil {
-		slog.Warn("error while closing a Closer: " + err.Error())
-	}
-}

go/graphql/client.go

@@ -1,7 +1,17 @@
 package lib
 
-import "fmt"
+import (
+	"fmt"
+	"io"
+	"log/slog"
+)
 
 func WrapError(errorMessage string, original error) error {
 	return fmt.Errorf("%s\n\tCaused by: %w", errorMessage, original)
 }
+
+func CloseOrWarn(closer io.Closer) {
+	if err := closer.Close(); err != nil {
+		slog.Warn("error while closing a Closer: " + err.Error())
+	}
+}

go/lib/lib.go

@@ -27,14 +27,14 @@ func main() {
 
 func main0() error {
 	// Instantiate a client
-	env, err := env.Read()
+	env, err := readEnv()
 	if err != nil {
 		return lib.WrapError("error while reading variables", err)
 	}
 	apiClient := initClient(env.GraphqlEndpoint())
 
 	// Auth
-	err = apiClient.Auth(env.Token)
+	err = apiClient.Auth(env.APIToken)
 	env.PurgeToken()
 	if err != nil {
 		return err
@@ -67,6 +67,16 @@ func main0() error {
 	return nil
 }
 
+func readEnv() (env.Env, error) {
+	var envFilePath string
+	if len(os.Args) > 1 {
+		envFilePath = os.Args[1]
+	} else {
+		envFilePath = "../.env"
+	}
+	return env.Read(envFilePath)
+}
+
 func initClient(endpoint string) api.Client {
 	cookieJar, err := cookiejar.New(nil)
 	if err != nil {