mirror of
https://github.com/mariocandela/beelzebub.git
synced 2025-07-01 18:47:26 -04:00
120 lines
3.6 KiB
Go
120 lines
3.6 KiB
Go
![]() |
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)
|
||
|
}
|