mirror of
https://github.com/mariocandela/beelzebub.git
synced 2025-07-01 18:47:26 -04:00
first commit
This commit is contained in:
119
parser/configurationsParser.go
Normal file
119
parser/configurationsParser.go
Normal file
@ -0,0 +1,119 @@
|
||||
package parser
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
log "github.com/sirupsen/logrus"
|
||||
"gopkg.in/yaml.v3"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"path/filepath"
|
||||
)
|
||||
|
||||
type BeelzebubCoreConfigurations struct {
|
||||
Core struct {
|
||||
Logging Logging `yaml:"logging"`
|
||||
}
|
||||
}
|
||||
|
||||
type Logging struct {
|
||||
Debug bool `yaml:"debug"`
|
||||
DebugReportCaller bool `yaml:"debugReportCaller"`
|
||||
LogDisableTimestamp bool `yaml:"logDisableTimestamp"`
|
||||
LogsPath string `yaml:"logsPath,omitempty"`
|
||||
}
|
||||
|
||||
type BeelzebubServiceConfiguration struct {
|
||||
ApiVersion string `yaml:"apiVersion"`
|
||||
Protocol string `yaml:"protocol"`
|
||||
Address string `yaml:"address"`
|
||||
Commands []Command `yaml:"commands"`
|
||||
ServerVersion string `yaml:"serverVersion"`
|
||||
ServerName string `yaml:"serverName"`
|
||||
DeadlineTimeoutSeconds int `yaml:"deadlineTimeoutSeconds"`
|
||||
PasswordRegex string `yaml:"passwordRegex"`
|
||||
}
|
||||
|
||||
type Command struct {
|
||||
Regex string `yaml:"regex"`
|
||||
Handler string `yaml:"handler"`
|
||||
Headers []string `yaml:"headers"`
|
||||
StatusCode int `yaml:"statusCode"`
|
||||
}
|
||||
|
||||
type configurationsParser struct {
|
||||
configurationsCorePath string
|
||||
configurationsServicesDirectory string
|
||||
readFileBytes ReadFileBytes
|
||||
readDir ReadDir
|
||||
}
|
||||
|
||||
type ReadFileBytes func(filePath string) ([]byte, error)
|
||||
|
||||
type ReadDir func(dirName string) ([]string, error)
|
||||
|
||||
// Init Parser, return a configurationsParser and use the DI Pattern to inject the dependencies
|
||||
func Init(configurationsCorePath, configurationsServicesDirectory string) *configurationsParser {
|
||||
return &configurationsParser{
|
||||
configurationsCorePath: configurationsCorePath,
|
||||
configurationsServicesDirectory: configurationsServicesDirectory,
|
||||
readFileBytes: readFileBytes,
|
||||
readDir: readDir,
|
||||
}
|
||||
}
|
||||
|
||||
func (bp configurationsParser) ReadConfigurationsCore() (*BeelzebubCoreConfigurations, error) {
|
||||
buf, err := bp.readFileBytes(bp.configurationsCorePath)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("in file %s: %v", bp.configurationsCorePath, err)
|
||||
}
|
||||
|
||||
beelzebubConfiguration := &BeelzebubCoreConfigurations{}
|
||||
err = yaml.Unmarshal(buf, beelzebubConfiguration)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("in file %s: %v", bp.configurationsCorePath, err)
|
||||
}
|
||||
|
||||
return beelzebubConfiguration, nil
|
||||
}
|
||||
|
||||
func (bp configurationsParser) ReadConfigurationsServices() ([]BeelzebubServiceConfiguration, error) {
|
||||
services, err := bp.readDir(bp.configurationsServicesDirectory)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("in directory %s: %v", bp.configurationsServicesDirectory, err)
|
||||
}
|
||||
|
||||
var servicesConfiguration []BeelzebubServiceConfiguration
|
||||
for _, servicesName := range services {
|
||||
filePath := filepath.Join(bp.configurationsServicesDirectory, servicesName)
|
||||
buf, err := bp.readFileBytes(filePath)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("in file %s: %v", filePath, err)
|
||||
}
|
||||
beelzebubServiceConfiguration := &BeelzebubServiceConfiguration{}
|
||||
err = yaml.Unmarshal(buf, beelzebubServiceConfiguration)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("in file %s: %v", filePath, err)
|
||||
}
|
||||
log.Debug(beelzebubServiceConfiguration)
|
||||
servicesConfiguration = append(servicesConfiguration, *beelzebubServiceConfiguration)
|
||||
}
|
||||
|
||||
return servicesConfiguration, nil
|
||||
}
|
||||
|
||||
func readDir(dirName string) ([]string, error) {
|
||||
var filesName []string
|
||||
files, err := ioutil.ReadDir(dirName)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
for _, file := range files {
|
||||
filesName = append(filesName, file.Name())
|
||||
}
|
||||
return filesName, nil
|
||||
}
|
||||
|
||||
func readFileBytes(filePath string) ([]byte, error) {
|
||||
return os.ReadFile(filePath)
|
||||
}
|
Reference in New Issue
Block a user