2021-07-18 03:21:26 +00:00
|
|
|
package config
|
|
|
|
|
|
|
|
import (
|
2021-07-18 09:49:31 +00:00
|
|
|
"errors"
|
2021-07-18 03:21:26 +00:00
|
|
|
"fmt"
|
|
|
|
"io/fs"
|
|
|
|
"io/ioutil"
|
2021-07-19 06:04:37 +00:00
|
|
|
"os"
|
|
|
|
"path"
|
|
|
|
"path/filepath"
|
2021-07-19 07:36:38 +00:00
|
|
|
"strings"
|
2021-07-18 03:21:26 +00:00
|
|
|
|
|
|
|
"github.com/joho/godotenv"
|
2021-07-19 06:04:37 +00:00
|
|
|
"github.com/sirupsen/logrus"
|
2021-07-18 03:21:26 +00:00
|
|
|
)
|
|
|
|
|
2021-07-19 07:36:38 +00:00
|
|
|
var ABRA_DIR = os.ExpandEnv("$HOME/.abra")
|
2021-07-19 06:04:37 +00:00
|
|
|
var ABRA_SERVER_FOLDER = path.Join(ABRA_DIR, "servers")
|
2021-07-18 03:21:26 +00:00
|
|
|
|
2021-07-19 06:04:37 +00:00
|
|
|
// Type aliases to make code hints easier to understand
|
2021-07-18 09:49:31 +00:00
|
|
|
type AppEnv = map[string]string
|
2021-07-19 06:04:37 +00:00
|
|
|
type AppName = string
|
2021-07-18 09:49:31 +00:00
|
|
|
|
|
|
|
type App struct {
|
2021-07-19 06:04:37 +00:00
|
|
|
Name AppName
|
2021-07-18 09:49:31 +00:00
|
|
|
Type string
|
|
|
|
Domain string
|
|
|
|
Env AppEnv
|
|
|
|
}
|
2021-07-19 06:04:37 +00:00
|
|
|
type AppFile struct {
|
2021-07-19 07:36:38 +00:00
|
|
|
Path string
|
|
|
|
Server string
|
2021-07-18 09:49:31 +00:00
|
|
|
}
|
|
|
|
|
2021-07-19 06:04:37 +00:00
|
|
|
type AppFiles = map[AppName]AppFile
|
|
|
|
|
2021-07-19 11:47:46 +00:00
|
|
|
func LoadAppFiles() (AppFiles, error) {
|
2021-07-19 07:36:38 +00:00
|
|
|
appFiles := make(AppFiles)
|
2021-07-19 11:47:46 +00:00
|
|
|
servers, err := getAllFoldersInDirectory(ABRA_SERVER_FOLDER)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2021-07-19 07:36:38 +00:00
|
|
|
for _, server := range servers {
|
|
|
|
serverDir := path.Join(ABRA_SERVER_FOLDER, server)
|
2021-07-19 11:47:46 +00:00
|
|
|
files, err := getAllFilesInDirectory(serverDir)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2021-07-19 07:36:38 +00:00
|
|
|
for _, file := range files {
|
|
|
|
appName := strings.TrimSuffix(file.Name(), ".env")
|
|
|
|
appFilePath := path.Join(ABRA_SERVER_FOLDER, server, file.Name())
|
|
|
|
appFiles[appName] = AppFile{
|
|
|
|
Path: appFilePath,
|
|
|
|
Server: server,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2021-07-19 11:47:46 +00:00
|
|
|
return appFiles, nil
|
2021-07-19 07:36:38 +00:00
|
|
|
}
|
2021-07-19 06:04:37 +00:00
|
|
|
|
|
|
|
// GetApp loads an apps settings, reading it from file, in preparation to use it
|
|
|
|
//
|
|
|
|
// ONLY use when ready to use the env file to keep IO down and
|
|
|
|
// because this exits with code 1 if the file cannot be found or is malformed
|
2021-07-19 07:36:38 +00:00
|
|
|
// FIXME: Cannot read files with multiline vars
|
2021-07-19 06:04:37 +00:00
|
|
|
func GetApp(apps AppFiles, name AppName) (App, error) {
|
|
|
|
appFile, exists := apps[name]
|
|
|
|
if !exists {
|
|
|
|
return App{}, fmt.Errorf("cannot find app file with name '%s'", name)
|
|
|
|
}
|
|
|
|
app, err := readAppFile(appFile, name)
|
2021-07-18 09:49:31 +00:00
|
|
|
if err != nil {
|
2021-07-19 11:47:46 +00:00
|
|
|
return App{}, err
|
2021-07-18 09:49:31 +00:00
|
|
|
}
|
2021-07-19 06:04:37 +00:00
|
|
|
return app, nil
|
|
|
|
}
|
|
|
|
|
2021-07-19 13:36:00 +00:00
|
|
|
// TODO: maybe better names than read and get
|
|
|
|
|
2021-07-19 06:04:37 +00:00
|
|
|
func readAppFile(appFile AppFile, name AppName) (App, error) {
|
2021-07-19 11:47:46 +00:00
|
|
|
env, err := readEnv(appFile.Path)
|
|
|
|
if err != nil {
|
|
|
|
return App{}, fmt.Errorf("env file for '%s' couldn't be read: %s", name, err.Error())
|
|
|
|
}
|
2021-07-18 09:49:31 +00:00
|
|
|
app, err := makeApp(env, name)
|
|
|
|
if err != nil {
|
|
|
|
return App{}, fmt.Errorf("env file for '%s' has issues: %s", name, err.Error())
|
|
|
|
}
|
|
|
|
return app, nil
|
|
|
|
}
|
|
|
|
|
2021-07-19 11:47:46 +00:00
|
|
|
func readEnv(filePath string) (AppEnv, error) {
|
2021-07-19 06:04:37 +00:00
|
|
|
var envFile AppEnv
|
|
|
|
envFile, err := godotenv.Read(filePath)
|
|
|
|
if err != nil {
|
2021-07-19 11:47:46 +00:00
|
|
|
return nil, err
|
2021-07-19 06:04:37 +00:00
|
|
|
}
|
2021-07-19 11:47:46 +00:00
|
|
|
return envFile, nil
|
2021-07-19 06:04:37 +00:00
|
|
|
}
|
|
|
|
|
2021-07-18 09:49:31 +00:00
|
|
|
func makeApp(env AppEnv, name string) (App, error) {
|
|
|
|
// Checking for domain and type as they are required - apps wont work without them
|
|
|
|
domain, ok := env["DOMAIN"]
|
|
|
|
if !ok {
|
|
|
|
return App{}, errors.New("missing DOMAIN variable")
|
|
|
|
}
|
|
|
|
apptype, ok := env["TYPE"]
|
|
|
|
if !ok {
|
|
|
|
return App{}, errors.New("missing TYPE variable")
|
|
|
|
}
|
|
|
|
return App{
|
|
|
|
Name: name,
|
|
|
|
Domain: domain,
|
|
|
|
Type: apptype,
|
|
|
|
Env: env,
|
|
|
|
}, nil
|
|
|
|
}
|
|
|
|
|
2021-07-19 11:47:46 +00:00
|
|
|
func ReadServerNames() ([]string, error) {
|
|
|
|
serverNames, err := getAllFoldersInDirectory(ABRA_SERVER_FOLDER)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return serverNames, nil
|
2021-07-19 07:36:38 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// getAllFilesInDirectory returns filenames of all files in directory
|
2021-07-19 11:47:46 +00:00
|
|
|
func getAllFilesInDirectory(directory string) ([]fs.FileInfo, error) {
|
2021-07-19 07:36:38 +00:00
|
|
|
var realFiles []fs.FileInfo
|
|
|
|
files, err := ioutil.ReadDir(directory)
|
2021-07-18 03:21:26 +00:00
|
|
|
if err != nil {
|
2021-07-19 11:47:46 +00:00
|
|
|
return nil, err
|
2021-07-18 03:21:26 +00:00
|
|
|
}
|
2021-07-19 07:36:38 +00:00
|
|
|
for _, file := range files {
|
|
|
|
// Follow any symlinks
|
|
|
|
filePath := path.Join(directory, file.Name())
|
|
|
|
realPath, err := filepath.EvalSymlinks(filePath)
|
|
|
|
if err != nil {
|
|
|
|
logrus.Warningf("broken symlink in your abra config folders: '%s'", filePath)
|
|
|
|
} else {
|
|
|
|
realFile, err := os.Stat(realPath)
|
|
|
|
if err != nil {
|
2021-07-19 11:47:46 +00:00
|
|
|
return nil, err
|
2021-07-19 07:36:38 +00:00
|
|
|
}
|
|
|
|
if !realFile.IsDir() {
|
|
|
|
realFiles = append(realFiles, file)
|
|
|
|
}
|
|
|
|
}
|
2021-07-18 03:21:26 +00:00
|
|
|
|
2021-07-19 07:36:38 +00:00
|
|
|
}
|
2021-07-19 11:47:46 +00:00
|
|
|
return realFiles, nil
|
2021-07-19 06:04:37 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// getAllFoldersInDirectory returns both folder and symlink paths
|
2021-07-19 11:47:46 +00:00
|
|
|
func getAllFoldersInDirectory(directory string) ([]string, error) {
|
2021-07-19 06:04:37 +00:00
|
|
|
var folders []string
|
|
|
|
files, err := ioutil.ReadDir(directory)
|
2021-07-18 03:21:26 +00:00
|
|
|
if err != nil {
|
2021-07-19 11:47:46 +00:00
|
|
|
return nil, err
|
2021-07-19 06:04:37 +00:00
|
|
|
}
|
|
|
|
if len(files) == 0 {
|
2021-07-19 11:47:46 +00:00
|
|
|
return nil, fmt.Errorf("directory is empty: '%s'", directory)
|
2021-07-18 03:21:26 +00:00
|
|
|
}
|
|
|
|
for _, file := range files {
|
2021-07-19 07:36:38 +00:00
|
|
|
// Check if file is directory or symlink
|
2021-07-18 03:21:26 +00:00
|
|
|
if file.IsDir() || file.Mode()&fs.ModeSymlink != 0 {
|
2021-07-19 06:04:37 +00:00
|
|
|
filePath := path.Join(directory, file.Name())
|
|
|
|
realDir, err := filepath.EvalSymlinks(filePath)
|
|
|
|
if err != nil {
|
|
|
|
logrus.Warningf("broken symlink in your abra config folders: '%s'", filePath)
|
|
|
|
} else if stat, err := os.Stat(realDir); err == nil && stat.IsDir() {
|
|
|
|
// path is a directory
|
|
|
|
folders = append(folders, file.Name())
|
|
|
|
}
|
|
|
|
}
|
2021-07-18 03:21:26 +00:00
|
|
|
}
|
2021-07-19 11:47:46 +00:00
|
|
|
return folders, nil
|
2021-07-18 03:21:26 +00:00
|
|
|
}
|