first commit

main.go

@@ -0,0 +1,77 @@
+package main
+
+import (
+	"beelzebub/parser"
+	"beelzebub/protocols"
+	log "github.com/sirupsen/logrus"
+	"io"
+	"os"
+)
+
+var quit = make(chan struct{})
+
+func main() {
+	configurationsParser := parser.Init("./configurations/beelzebub.yaml", "./configurations/services/")
+
+	coreConfigurations, err := configurationsParser.ReadConfigurationsCore()
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	fileLogs := configureLogging(coreConfigurations.Core.Logging)
+	defer fileLogs.Close()
+
+	beelzebubServicesConfiguration, err := configurationsParser.ReadConfigurationsServices()
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	// Protocol strategies
+	secureShellStrategy := &protocols.SecureShellStrategy{}
+	hypertextTransferProtocolStrategy := &protocols.HypertextTransferProtocolStrategy{}
+
+	// Protocol manager
+	manager := protocols.ProtocolManager{}
+	serviceManager := manager.InitServiceManager()
+
+	for _, beelzebubServiceConfiguration := range beelzebubServicesConfiguration {
+		switch beelzebubServiceConfiguration.Protocol {
+		case "http":
+			serviceManager.SetProtocolStrategy(hypertextTransferProtocolStrategy)
+			break
+		case "ssh":
+			serviceManager.SetProtocolStrategy(secureShellStrategy)
+			break
+		default:
+			log.Fatalf("Protocol %s not managed", beelzebubServiceConfiguration.Protocol)
+			continue
+		}
+
+		err := serviceManager.InitService(beelzebubServiceConfiguration)
+		if err != nil {
+			log.Errorf("Error during init protocol: %s, %s", beelzebubServiceConfiguration.Protocol, err.Error())
+		}
+	}
+	<-quit
+}
+
+func configureLogging(configurations parser.Logging) *os.File {
+	file, err := os.OpenFile(configurations.LogsPath, os.O_APPEND|os.O_CREATE|os.O_RDWR, 0666)
+	if err != nil {
+		log.Fatalf("error opening file: %v", err)
+	}
+	mw := io.MultiWriter(os.Stdout, file)
+
+	log.SetOutput(mw)
+
+	log.SetFormatter(&log.JSONFormatter{
+		DisableTimestamp: configurations.LogDisableTimestamp,
+	})
+	log.SetReportCaller(configurations.DebugReportCaller)
+	if configurations.Debug {
+		log.SetLevel(log.DebugLevel)
+	} else {
+		log.SetLevel(log.InfoLevel)
+	}
+	return file
+}