servicemanager/internal/envreader/envreader.go

   1 // -
   2 //   ========================LICENSE_START=================================
   3 //   O-RAN-SC
   4 //   %%
   5 //   Copyright (C) 2024: OpenInfra Foundation Europe
   6 //   %%
   7 //   Licensed under the Apache License, Version 2.0 (the "License");
   8 //   you may not use this file except in compliance with the License.
   9 //   You may obtain a copy of the License at
  10 //
  11 //        http://www.apache.org/licenses/LICENSE-2.0
  12 //
  13 //   Unless required by applicable law or agreed to in writing, software
  14 //   distributed under the License is distributed on an "AS IS" BASIS,
  15 //   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16 //   See the License for the specific language governing permissions and
  17 //   limitations under the License.
  18 //   ========================LICENSE_END===================================
  19 //
  20
  21 package envreader
  22
  23 import (
  24         "os"
  25         "path/filepath"
  26         "runtime"
  27         "strconv"
  28
  29         "github.com/joho/godotenv"
  30         log "github.com/sirupsen/logrus"
  31 )
  32
  33 func ReadDotEnv() (map[string]string, map[string]int, error) {
  34         log.SetFormatter(&log.TextFormatter{
  35                 FullTimestamp: true,
  36         })
  37
  38         env := os.Getenv("SERVICE_MANAGER_ENV")
  39         log.Infof("read SERVICE_MANAGER_ENV: %s", env)
  40
  41         if env == "" {
  42                 env = "development"
  43         }
  44
  45         // Root folder of this project
  46         _, file, _, _ := runtime.Caller(0)
  47         basePath := filepath.Join(filepath.Dir(file), "../..")
  48         basePath += "/"
  49
  50         var myEnv map[string]string
  51         envFile := basePath + ".env." + env
  52         myEnv, err := godotenv.Read(envFile)
  53
  54         if err != nil {
  55                 log.Warnf("error reading .env file: %s", err)
  56
  57                 envFile = basePath + ".env"
  58                 myEnv, err = godotenv.Read(envFile)
  59                 if err != nil {
  60                         log.Fatalf("error reading .env file: %s", err)
  61                         return nil, nil, err
  62                 }
  63         }
  64         log.Infof("imported .env: %s", envFile)
  65
  66         loglevel, err := log.ParseLevel(myEnv["LOG_LEVEL"])
  67         if err != nil {
  68                 log.Fatalf("error loading LOG_LEVEL from .env file: %s", err)
  69                 return nil, nil, err
  70         }
  71         log.SetLevel(loglevel)
  72
  73         log.Infof("KONG_DOMAIN %s", myEnv["KONG_DOMAIN"])
  74         log.Infof("KONG_PROTOCOL %s", myEnv["KONG_PROTOCOL"])
  75         log.Infof("KONG_IPV4 %s", myEnv["KONG_IPV4"])
  76         log.Infof("KONG_DATA_PLANE_PORT %s", myEnv["KONG_DATA_PLANE_PORT"])
  77         log.Infof("KONG_CONTROL_PLANE_PORT %s", myEnv["KONG_CONTROL_PLANE_PORT"])
  78         log.Infof("CAPIF_PROTOCOL %s", myEnv["CAPIF_PROTOCOL"])
  79         log.Infof("CAPIF_IPV4 %s", myEnv["CAPIF_IPV4"])
  80         log.Infof("CAPIF_PORT %s", myEnv["CAPIF_PORT"])
  81         log.Infof("LOG_LEVEL %s", myEnv["LOG_LEVEL"])
  82         log.Infof("SERVICE_MANAGER_PORT %s", myEnv["SERVICE_MANAGER_PORT"])
  83         log.Infof("TEST_SERVICE_IPV4 %s", myEnv["TEST_SERVICE_IPV4"])
  84         log.Infof("TEST_SERVICE_PORT %s", myEnv["TEST_SERVICE_PORT"])
  85
  86         var myPorts = make(map[string]int)
  87
  88         myPorts["KONG_DATA_PLANE_PORT"], err = strconv.Atoi(myEnv["KONG_DATA_PLANE_PORT"])
  89         if err != nil {
  90                 log.Fatalf("error loading KONG_DATA_PLANE_PORT from .env file: %s", err)
  91                 return nil, nil, err
  92         }
  93
  94         myPorts["KONG_CONTROL_PLANE_PORT"], err = strconv.Atoi(myEnv["KONG_CONTROL_PLANE_PORT"])
  95         if err != nil {
  96                 log.Fatalf("error loading KONG_CONTROL_PLANE_PORT from .env file: %s", err)
  97                 return nil, nil, err
  98         }
  99
 100         myPorts["CAPIF_PORT"], err = strconv.Atoi(myEnv["CAPIF_PORT"])
 101         if err != nil {
 102                 log.Fatalf("error loading CAPIF_PORT from .env file: %s", err)
 103                 return nil, nil, err
 104         }
 105
 106         myPorts["SERVICE_MANAGER_PORT"], err = strconv.Atoi(myEnv["SERVICE_MANAGER_PORT"])
 107         if err != nil {
 108                 log.Fatalf("error loading SERVICE_MANAGER_PORT from .env file: %s", err)
 109                 return nil, nil, err
 110         }
 111
 112         // TEST_SERVICE_PORT is required for unit testing, but not required for production
 113         if myEnv["TEST_SERVICE_PORT"] != "" {
 114                 myPorts["TEST_SERVICE_PORT"], err = strconv.Atoi(myEnv["TEST_SERVICE_PORT"])
 115                 if err != nil {
 116                         log.Fatalf("error loading TEST_SERVICE_PORT from .env file: %s", err)
 117                         return nil, nil, err
 118                 }
 119         }
 120
 121         return myEnv, myPorts, err
 122 }