import template code

app/server/server.go

@@ -0,0 +1,241 @@
+package server
+
+import (
+	"os"
+	"os/signal"
+	"syscall"
+
+	"os/user"
+	"path/filepath"
+	"strconv"
+
+	"helmet/app/config"
+	"helmet/app/handler"
+	"helmet/app/logger"
+	"helmet/app/operator"
+	"helmet/app/service"
+	"helmet/pkg/x509crt"
+)
+
+type Server struct {
+	conf     *config.Config
+	oper     *operator.Operator
+	svc      *service.Service
+	hand     *handler.Handler
+	log      *logger.Logger
+	x509cert []byte
+	x509key  []byte
+}
+
+func NewServer() (*Server, error) {
+	var err error
+	srv := &Server{}
+	srv.log = logger.NewLogger("server")
+	return srv, err
+}
+
+func (srv *Server) Config() *config.Config {
+	return srv.conf
+}
+
+func (srv *Server) Configure() error {
+	var err error
+	srv.conf, err = config.NewConfig()
+	if err != nil {
+		return err
+	}
+	err = srv.conf.Read()
+	if err != nil {
+		return err
+	}
+	err = srv.conf.Validate()
+	if err != nil {
+		return err
+	}
+	return err
+}
+
+func (srv *Server) Build() error {
+	var err error
+	srv.log.Infof("Build server")
+
+	if srv.conf.AsDaemon {
+		logDir := filepath.Dir(srv.conf.LogPath)
+		srv.log.Infof("Create %s dir", logDir)
+		err = os.MkdirAll(logDir, 0750)
+		if err != nil {
+			return err
+		}
+		runDir := filepath.Dir(srv.conf.RunPath)
+		srv.log.Infof("Create %s dir", runDir)
+		err = os.MkdirAll(runDir, 0750)
+		if err != nil {
+			return err
+		}
+	}
+
+	// Create X509 certs
+	srv.x509cert, srv.x509key, err = x509crt.CreateX509SelfSignedCert(srv.conf.Hostname)
+	if err != nil {
+		return err
+	}
+	// Create operator
+	operatorConfig := &operator.OperatorConfig{
+		Auths: srv.conf.Auths,
+		//Database: srv.db,
+	}
+	srv.oper, err = operator.NewOperator(operatorConfig)
+	if err != nil {
+		return err
+	}
+	// Create ghandler
+	handlerConfig := &handler.HandlerConfig{
+		Operator: srv.oper,
+	}
+	srv.hand = handler.NewHandler(handlerConfig)
+
+	// Create gservice
+	serviceConfig := &service.ServiceConfig{
+		PortNum:  srv.conf.Service.Port,
+		Hostname: srv.conf.Hostname,
+		Handler:  srv.hand,
+		Operator: srv.oper,
+		X509Cert: srv.x509cert,
+		X509Key:  srv.x509key,
+	}
+	srv.svc = service.NewService(serviceConfig)
+
+	return err
+}
+
+func (srv *Server) Run() error {
+	var err error
+
+	yamlConfig, err := srv.conf.YAML()
+	if err != nil {
+		return err
+	}
+	srv.log.Debugf("Server configuration:\n%s\n", yamlConfig)
+
+	currUser, err := user.Current()
+	if err != nil {
+		return err
+	}
+	srv.log.Infof("Running server as user %s", currUser.Username)
+
+	sigs := make(chan os.Signal, 1)
+	done := make(chan error, 1)
+
+	// Run service
+	startService := func(svc *service.Service, done chan error) {
+		err = svc.Run()
+		if err != nil {
+			srv.log.Errorf("Service error: %v", err)
+			done <- err
+		}
+	}
+	go startService(srv.svc, done)
+
+	signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
+	var signal os.Signal
+	select {
+	case signal = <-sigs:
+		srv.log.Infof("Services stopped by signal: %v", signal)
+		srv.svc.Stop()
+	}
+	return err
+}
+
+func (srv *Server) PseudoFork() error {
+	const successExit int = 0
+	var keyEnv string = "IMX0LTSELMRF8K"
+	var err error
+
+	_, isChild := os.LookupEnv(keyEnv)
+	switch {
+	case !isChild:
+		os.Setenv(keyEnv, "TRUE")
+
+		procAttr := syscall.ProcAttr{}
+		cwd, err := os.Getwd()
+		if err != nil {
+			return err
+		}
+		var sysFiles = make([]uintptr, 3)
+		sysFiles[0] = uintptr(syscall.Stdin)
+		sysFiles[1] = uintptr(syscall.Stdout)
+		sysFiles[2] = uintptr(syscall.Stderr)
+
+		procAttr.Files = sysFiles
+		procAttr.Env = os.Environ()
+		procAttr.Dir = cwd
+
+		_, err = syscall.ForkExec(os.Args[0], os.Args, &procAttr)
+		if err != nil {
+			return err
+		}
+		os.Exit(successExit)
+	case isChild:
+		_, err = syscall.Setsid()
+		if err != nil {
+			return err
+		}
+	}
+	os.Unsetenv(keyEnv)
+	return err
+}
+
+func (srv *Server) Daemonize() error {
+	var err error
+	if srv.conf.AsDaemon {
+		// Restart process process
+		err = srv.PseudoFork()
+		if err != nil {
+			return err
+		}
+		// Redirect stdin
+		nullFile, err := os.OpenFile("/dev/null", os.O_RDWR, 0)
+		if err != nil {
+			return err
+		}
+		err = syscall.Dup2(int(nullFile.Fd()), int(os.Stdin.Fd()))
+		if err != nil {
+			return err
+		}
+		// Redirect stderr and stout
+		logdir := filepath.Dir(srv.conf.LogPath)
+		err = os.MkdirAll(logdir, 0750)
+		if err != nil {
+			return err
+		}
+		logFile, err := os.OpenFile(srv.conf.LogPath, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0640)
+		if err != nil {
+			return err
+		}
+		err = syscall.Dup2(int(logFile.Fd()), int(os.Stdout.Fd()))
+		if err != nil {
+			return err
+		}
+		err = syscall.Dup2(int(logFile.Fd()), int(os.Stderr.Fd()))
+		if err != nil {
+			return err
+		}
+		// Write process ID
+		rundir := filepath.Dir(srv.conf.RunPath)
+		err = os.MkdirAll(rundir, 0750)
+		if err != nil {
+			return err
+		}
+		pidFile, err := os.OpenFile(srv.conf.RunPath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0640)
+		if err != nil {
+			return err
+		}
+		defer pidFile.Close()
+		currPid := os.Getpid()
+		_, err = pidFile.WriteString(strconv.Itoa(currPid))
+		if err != nil {
+			return err
+		}
+	}
+	return err
+}