mirror of
https://github.com/silicoflare/envy.git
synced 2026-05-26 19:57:59 +05:30
feat: add env parsing before creating environment
This commit is contained in:
@@ -2,21 +2,24 @@ package utils
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"os"
|
||||
"path"
|
||||
"regexp"
|
||||
"strings"
|
||||
|
||||
"github.com/pelletier/go-toml"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
func StopIfErr(err error) {
|
||||
if err != nil {
|
||||
func StopIfErr(err error) {
|
||||
if err != nil {
|
||||
ErrPrint(err.Error())
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
||||
|
||||
func GetEnvy() (*toml.Tree, error) {
|
||||
func GetEnvy() (*toml.Tree, error) {
|
||||
cwd, err := os.Getwd()
|
||||
if err != nil {
|
||||
return nil, errors.New("Not able to fetch working directory: " + err.Error())
|
||||
@@ -27,63 +30,87 @@ func GetEnvy() (*toml.Tree, error) {
|
||||
// file exists
|
||||
if err == nil {
|
||||
envyFile, err := toml.LoadFile(path.Join(cwd, ".envy"))
|
||||
if err != nil {
|
||||
if err != nil {
|
||||
return nil, errors.New("Error occured while reading .envy file:" + err.Error())
|
||||
}
|
||||
return envyFile, nil
|
||||
}
|
||||
|
||||
// file does not exist
|
||||
if os.IsNotExist(err) {
|
||||
return nil, errors.New(".envy file doesn't exist in the current directory.")
|
||||
} else {
|
||||
if os.IsNotExist(err) {
|
||||
return nil, errors.New(".envy file doesn't exist in the current directory")
|
||||
} else {
|
||||
return nil, errors.New("Some error occured:" + err.Error())
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
func GetProject(name string) (Project, error) {
|
||||
func GetProject(name string) (Project, error) {
|
||||
InitDb()
|
||||
|
||||
var dbProject Project
|
||||
result := DB.Where("name = ?", name).First(&dbProject)
|
||||
|
||||
if result.Error != nil {
|
||||
if errors.Is(result.Error, gorm.ErrRecordNotFound) {
|
||||
return Project{}, errors.New("Project with the name " + name + " not found in the database.")
|
||||
} else {
|
||||
if result.Error != nil {
|
||||
if errors.Is(result.Error, gorm.ErrRecordNotFound) {
|
||||
return Project{}, errors.New("Project with the name " + name + " not found in the database")
|
||||
} else {
|
||||
return Project{}, errors.New("Some error occured: " + result.Error.Error())
|
||||
}
|
||||
}
|
||||
return dbProject, nil
|
||||
}
|
||||
|
||||
|
||||
func GetEnvironments(project uint) ([]Environment, error) {
|
||||
func GetEnvironments(project uint) ([]Environment, error) {
|
||||
InitDb()
|
||||
|
||||
var dbEnvs []Environment
|
||||
result := DB.Where("project_id = ?", project).Find(&dbEnvs)
|
||||
|
||||
if result.Error != nil {
|
||||
if result.Error != nil {
|
||||
return nil, errors.New("Some error occured: " + result.Error.Error())
|
||||
}
|
||||
return dbEnvs, nil
|
||||
}
|
||||
|
||||
|
||||
func GetEnvironment(project uint, environment string) (Environment, error) {
|
||||
func GetEnvironment(project uint, environment string) (Environment, error) {
|
||||
InitDb()
|
||||
|
||||
var dbEnv Environment
|
||||
result := DB.Where("project_id = ? AND name = ?", project, environment).First(&dbEnv)
|
||||
|
||||
if result.Error != nil {
|
||||
if errors.Is(result.Error, gorm.ErrRecordNotFound) {
|
||||
return Environment{}, errors.New("Environment with the name " + environment + " not found in the database.")
|
||||
} else {
|
||||
if result.Error != nil {
|
||||
if errors.Is(result.Error, gorm.ErrRecordNotFound) {
|
||||
return Environment{}, errors.New("Environment with the name " + environment + " not found in the database")
|
||||
} else {
|
||||
return Environment{}, errors.New("Some error occured: " + result.Error.Error())
|
||||
}
|
||||
}
|
||||
return dbEnv, nil
|
||||
}
|
||||
|
||||
func ParseEnv(env string) string {
|
||||
re := regexp.MustCompile(`^\s*(#?)\s*([A-Z0-9_]+)\s*=\s*(?:"([^"\n]+)"|([^"\n]+))$`)
|
||||
|
||||
var envs string
|
||||
|
||||
for _, line := range strings.Split(string(env), "\n") {
|
||||
matches := re.FindStringSubmatch(line)
|
||||
var key, value string
|
||||
var enabled bool
|
||||
if matches != nil {
|
||||
enabled = matches[1] == ""
|
||||
key = matches[2] // the key
|
||||
value = matches[3] // quoted value
|
||||
if value == "" {
|
||||
value = matches[4] // unquoted value
|
||||
}
|
||||
var comment = ""
|
||||
if enabled {
|
||||
comment = "# "
|
||||
}
|
||||
envs += fmt.Sprintf("%v%v=%v\n", comment, key, value)
|
||||
}
|
||||
}
|
||||
|
||||
return envs
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user