mirror of
https://github.com/silicoflare/envy.git
synced 2026-05-26 19:57:59 +05:30
feat: add helper functions for file and db ops and integrate them in
commands
This commit is contained in:
82
cmd/utils/other.go
Normal file
82
cmd/utils/other.go
Normal file
@@ -0,0 +1,82 @@
|
||||
package utils
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"os"
|
||||
"path"
|
||||
|
||||
"github.com/pelletier/go-toml"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
func GetEnvy() (*toml.Tree, error) {
|
||||
cwd, err := os.Getwd()
|
||||
if err != nil {
|
||||
return nil, errors.New("Not able to fetch working directory: " + err.Error())
|
||||
}
|
||||
|
||||
_, err = os.Stat(path.Join(cwd, ".envy"))
|
||||
|
||||
// file exists
|
||||
if err == nil {
|
||||
envyFile, err := toml.LoadFile(path.Join(cwd, ".envy"))
|
||||
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 {
|
||||
return nil, errors.New("Some error occured:" + err.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 {
|
||||
return Project{}, errors.New("Some error occured: " + result.Error.Error())
|
||||
}
|
||||
}
|
||||
return dbProject, nil
|
||||
}
|
||||
|
||||
|
||||
func GetEnvironments(project uint) ([]Environment, error) {
|
||||
InitDb()
|
||||
|
||||
var dbEnvs []Environment
|
||||
result := DB.Where("project_id = ?", project).Find(&dbEnvs)
|
||||
|
||||
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) {
|
||||
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 {
|
||||
return Environment{}, errors.New("Some error occured: " + result.Error.Error())
|
||||
}
|
||||
}
|
||||
return dbEnv, nil
|
||||
}
|
||||
Reference in New Issue
Block a user