main.go

package main

import (
	"flag"
	"fmt"
	"log"
	"net/http"
	_ "net/http/pprof" // !DEBUG
	"os"
	"os/signal"
	"syscall"

	"github.com/guerillagrow/gobox/controllers"
	//"github.com/guerillagrow/gobox/custom/xsession"
	"github.com/guerillagrow/gobox/lib/utils"
	"github.com/guerillagrow/gobox/models"
	_ "github.com/guerillagrow/gobox/routers"

	"github.com/guerillagrow/beego"
)

var VERSION string

func PprofServe() {

	go func() {
		// !DEBUG
		log.Println(http.ListenAndServe(":6060", nil))
	}()
}

func main() {

	utils.LocalVersion = VERSION
	sigs := make(chan os.Signal, 1)
	//done := make(chan bool, 1)

	models.ARG_ConfigFile = flag.String("c", "./conf/raspberrypi.json", "Config file")
	models.ARG_DBFile = flag.String("d", "./conf/main.db", "Database file")
	models.ARG_Debug = flag.Bool("debug", false, "Debug mode")

	ARG_VERSION := flag.Bool("version", false, "Show version")
	flag.Parse()

	if *ARG_VERSION {
		fmt.Println(VERSION)
		return
	}

	if *models.ARG_Debug == true {
		PprofServe()
	}

	models.Init() // Initialize models database etc.

	go func() {
		sig := <-sigs
		if sig == os.Interrupt || sig == os.Kill || sig == syscall.SIGTERM {
			models.GoBox.Stop()
			os.Exit(1)
		}
		//done <- true
	}()

	signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)

	cfgDebugFlag, cderr := models.BoxConfig.GetBool("debug_mode")

	if cderr == nil && cfgDebugFlag == true && !*models.ARG_Debug {
		models.ARG_Debug = &cfgDebugFlag
		PprofServe()
	}

	models.GoBox.Start()

	beego.BConfig.WebConfig.Session.SessionOn = true
	beego.BConfig.WebConfig.Session.SessionAutoSetCookie = true
	beego.BConfig.MaxMemory = 1 << 26
	//beego.BConfig.CopyRequestBody = true // required for json request bodies!

	/*sessionconf := &session.ManagerConfig{
		CookieName:      "gobox_sess",
		CookieLifeTime:  3600,
		Maxlifetime:     3600,
		Gclifetime:      3600 / 4,
		ProviderConfig:  "./tmp",
		EnableSetCookie: true,
	}
	beego.GlobalSessions, _ = session.NewManager("file", sessionconf) // !NOTE: Beego BUG: File based sessions are not saved correctly when using ajax requests
	go beego.GlobalSessions.GC()*/

	beego.ErrorController(&controllers.ErrorController{})
	beego.Run()

}