summaryrefslogtreecommitdiffstats
path: root/main.go
diff options
context:
space:
mode:
Diffstat (limited to 'main.go')
-rw-r--r--main.go190
1 files changed, 6 insertions, 184 deletions
diff --git a/main.go b/main.go
index 6089e74..1bb7110 100644
--- a/main.go
+++ b/main.go
@@ -5,20 +5,11 @@ package main
import (
"fmt"
"os"
- "os/exec"
- "os/signal"
- "path/filepath"
- "syscall"
- "time"
"github.com/Sirupsen/logrus"
"github.com/codegangsta/cli"
- "github.com/iotbzh/xds-server/lib/crosssdk"
- "github.com/iotbzh/xds-server/lib/folder"
- "github.com/iotbzh/xds-server/lib/model"
- "github.com/iotbzh/xds-server/lib/syncthing"
- "github.com/iotbzh/xds-server/lib/webserver"
"github.com/iotbzh/xds-server/lib/xdsconfig"
+ "github.com/iotbzh/xds-server/lib/xdsserver"
)
const (
@@ -39,192 +30,23 @@ var AppVersion = "?.?.?"
// Should be set by compilation -ldflags "-X main.AppSubVersion=xxx"
var AppSubVersion = "unknown-dev"
-// Context holds the XDS server context
-type Context struct {
- ProgName string
- Cli *cli.Context
- Config *xdsconfig.Config
- Log *logrus.Logger
- LogLevelSilly bool
- SThg *st.SyncThing
- SThgCmd *exec.Cmd
- SThgInotCmd *exec.Cmd
- MFolders *model.Folders
- SDKs *crosssdk.SDKs
- WWWServer *webserver.Server
- Exit chan os.Signal
-}
-
-// NewContext Create a new instance of XDS server
-func NewContext(cliCtx *cli.Context) *Context {
- var err error
-
- // Set logger level and formatter
- log := cliCtx.App.Metadata["logger"].(*logrus.Logger)
-
- logLevel := cliCtx.GlobalString("log")
- if logLevel == "" {
- logLevel = "error" // FIXME get from Config DefaultLogLevel
- }
- if log.Level, err = logrus.ParseLevel(logLevel); err != nil {
- fmt.Printf("Invalid log level : \"%v\"\n", logLevel)
- os.Exit(1)
- }
- log.Formatter = &logrus.TextFormatter{}
-
- sillyVal, sillyLog := os.LookupEnv("XDS_LOG_SILLY")
-
- // Define default configuration
- ctx := Context{
- ProgName: cliCtx.App.Name,
- Cli: cliCtx,
- Log: log,
- LogLevelSilly: (sillyLog && sillyVal == "1"),
- Exit: make(chan os.Signal, 1),
- }
-
- // register handler on SIGTERM / exit
- signal.Notify(ctx.Exit, os.Interrupt, syscall.SIGTERM)
- go handlerSigTerm(&ctx)
-
- return &ctx
-}
-
-// Handle exit and properly stop/close all stuff
-func handlerSigTerm(ctx *Context) {
- <-ctx.Exit
- if ctx.SThg != nil {
- ctx.Log.Infof("Stoping Syncthing... (PID %d)", ctx.SThgCmd.Process.Pid)
- ctx.SThg.Stop()
- ctx.Log.Infof("Stoping Syncthing-inotify... (PID %d)", ctx.SThgInotCmd.Process.Pid)
- ctx.SThg.StopInotify()
- }
- if ctx.WWWServer != nil {
- ctx.Log.Infof("Stoping Web server...")
- ctx.WWWServer.Stop()
- }
- os.Exit(0)
-}
-
-// Helper function to log message on both stdout and logger
-func logPrint(ctx *Context, format string, args ...interface{}) {
- fmt.Printf(format, args...)
- if ctx.Log.Out != os.Stdout {
- ctx.Log.Infof(format, args...)
- }
-}
-
// XDS Server application main routine
func xdsApp(cliCtx *cli.Context) error {
var err error
// Create XDS server context
- ctx := NewContext(cliCtx)
+ ctxSvr := xdsserver.NewXdsServer(cliCtx)
// Load config
- cfg, err := xdsconfig.Init(ctx.Cli, ctx.Log)
+ ctxSvr.Config, err = xdsconfig.Init(cliCtx, ctxSvr.Log)
if err != nil {
return cli.NewExitError(err, -2)
}
- ctx.Config = cfg
- // Logs redirected into a file when logfile option or logsDir config is set
- ctx.Config.LogVerboseOut = os.Stderr
- if ctx.Config.FileConf.LogsDir != "" {
- if ctx.Config.Options.LogFile != "stdout" {
- logFile := ctx.Config.Options.LogFile
-
- fdL, err := os.OpenFile(logFile, os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0666)
- if err != nil {
- msgErr := fmt.Sprintf("Cannot create log file %s", logFile)
- return cli.NewExitError(msgErr, int(syscall.EPERM))
- }
- ctx.Log.Out = fdL
-
- logPrint(ctx, "Logging file: %s\n", logFile)
- }
-
- logFileHTTPReq := filepath.Join(ctx.Config.FileConf.LogsDir, "xds-server-verbose.log")
- fdLH, err := os.OpenFile(logFileHTTPReq, os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0666)
- if err != nil {
- msgErr := fmt.Sprintf("Cannot create log file %s", logFileHTTPReq)
- return cli.NewExitError(msgErr, int(syscall.EPERM))
- }
- ctx.Config.LogVerboseOut = fdLH
-
- logPrint(ctx, "Logging file for HTTP requests: %s\n", logFileHTTPReq)
- }
-
- // Create syncthing instance when section "syncthing" is present in config.json
- if ctx.Config.FileConf.SThgConf != nil {
- ctx.SThg = st.NewSyncThing(ctx.Config, ctx.Log)
- }
-
- // Start local instance of Syncthing and Syncthing-notify
- if ctx.SThg != nil {
- ctx.Log.Infof("Starting Syncthing...")
- ctx.SThgCmd, err = ctx.SThg.Start()
- if err != nil {
- return cli.NewExitError(err, -4)
- }
- logPrint(ctx, "Syncthing started (PID %d)\n", ctx.SThgCmd.Process.Pid)
-
- ctx.Log.Infof("Starting Syncthing-inotify...")
- ctx.SThgInotCmd, err = ctx.SThg.StartInotify()
- if err != nil {
- return cli.NewExitError(err, -4)
- }
- logPrint(ctx, "Syncthing-inotify started (PID %d)\n", ctx.SThgInotCmd.Process.Pid)
-
- // Establish connection with local Syncthing (retry if connection fail)
- logPrint(ctx, "Establishing connection with Syncthing...\n")
- time.Sleep(2 * time.Second)
- maxRetry := 30
- retry := maxRetry
- err = nil
- for retry > 0 {
- if err = ctx.SThg.Connect(); err == nil {
- break
- }
- ctx.Log.Warningf("Establishing connection to Syncthing (retry %d/%d)", retry, maxRetry)
- time.Sleep(time.Second)
- retry--
- }
- if err != nil || retry == 0 {
- return cli.NewExitError(err, -4)
- }
-
- // FIXME: do we still need Builder notion ? if no cleanup
- if ctx.Config.Builder, err = xdsconfig.NewBuilderConfig(ctx.SThg.MyID); err != nil {
- return cli.NewExitError(err, -4)
- }
- ctx.Config.SupportedSharing[folder.TypeCloudSync] = true
- }
-
- // Init model folder
- ctx.MFolders = model.FoldersNew(ctx.Config, ctx.SThg)
-
- // Load initial folders config from disk
- if err := ctx.MFolders.LoadConfig(); err != nil {
- return cli.NewExitError(err, -5)
- }
-
- // Init cross SDKs
- ctx.SDKs, err = crosssdk.Init(ctx.Config, ctx.Log)
- if err != nil {
- return cli.NewExitError(err, -6)
- }
-
- // Create Web Server
- ctx.WWWServer = webserver.New(ctx.Config, ctx.MFolders, ctx.SDKs, ctx.Log, ctx.LogLevelSilly)
-
- // Run Web Server until exit requested (blocking call)
- if err = ctx.WWWServer.Serve(); err != nil {
- ctx.Log.Println(err)
- return cli.NewExitError(err, -7)
- }
+ // Run XDS Server (main loop)
+ errCode, err := ctxSvr.Run()
- return cli.NewExitError("Program exited ", -99)
+ return cli.NewExitError(err, errCode)
}
// main