silicoflare/envy
1
0
Fork 0
mirror of https://github.com/silicoflare/envy.git synced 2026-05-26 19:57:59 +05:30
Code Issues Packages Projects Releases Wiki Activity

refactor: move db related functions to db.go

Browse Source
This commit is contained in:
Suraj B M
2025-09-28 10:55:48 +05:30
parent f96a917fea
commit 9eaa34d7fe
2 changed files with 57 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

57
cmd/utils/db.go
View File

@@ -1,6 +1,7 @@
package utils package utils
import ( import (
"errors"
"path" "path"
"gorm.io/driver/sqlite" "gorm.io/driver/sqlite"
@@ -40,3 +41,59 @@ func MigrateDb() {
panic(err) panic(err)
} }
} }
func GetProjects(name string) ([]Project, error) {
InitDb()
var dbProjects []Project
result := DB.Where("name = ?", name).Find(&dbProjects)
if result.Error != nil {
return nil, errors.New("Some error occured: " + result.Error.Error())
}
return dbProjects, nil
}
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
}

45
cmd/utils/other.go
View File

@@ -9,7 +9,6 @@ import (
"strings" "strings"
"github.com/pelletier/go-toml" "github.com/pelletier/go-toml"
"gorm.io/gorm"
) )
func StopIfErr(err error) { func StopIfErr(err error) {
@@ -71,50 +70,6 @@ func WriteEnvy(tree *toml.Tree) error {
return nil return nil
} }
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
}
func ParseEnv(env string) string { func ParseEnv(env string) string {
re := regexp.MustCompile(`^\s*(#?)\s*([A-Z0-9_]+)\s*=\s*(?:"([^"\n]+)"|([^"\n]+))$`) re := regexp.MustCompile(`^\s*(#?)\s*([A-Z0-9_]+)\s*=\s*(?:"([^"\n]+)"|([^"\n]+))$`)