main.go

package main

import (
	"context"
	"html/template"
	"log"
	"net/http"
	"os"
	"os/signal"
	"time"

	"github.com/gin-contrib/static"
	"github.com/gin-gonic/gin"

	"github.com/recoilme/slowpoke"

	"github.com/joho/godotenv"
	"github.com/recoilme/tgram/routers"
)

// Port - typegram port (and address)
var Port = ":8081"

// LoadEnv parse env file if present or load
func LoadEnv() {
	err := godotenv.Load("tgram.env")
	if err == nil {
		setifset := func(new, def string) string {
			if new == "" {
				return def
			}
			return new
		}
		Port = setifset(os.Getenv("TGRAMPORT"), ":8081") //port with ":", example -  :8081
		routers.Config.Title = setifset(os.Getenv("TGRAMTITLE"), "typegram")
		routers.Config.SiteName = setifset(os.Getenv("TGRAMNAME"), "Typegram")
		routers.Config.Description = setifset(os.Getenv("TGRAMDESC"), "open source publishing platform")
		routers.Config.Admin = setifset(os.Getenv("TGRAMADMIN"), "recoilme")
		routers.Config.AboutPage = setifset(os.Getenv("TGRAMABOUT"), "/@recoilme/1")
		routers.Config.Domain = setifset(os.Getenv("TGRAMDOMAIN"), "tgr.am")
		routers.Config.NBSecretPassword = setifset(os.Getenv("TGRAMPWD"), "A String Very Very Very Niubilty!!@##$!@#4")
		routers.Config.Type2TeleBot = setifset(os.Getenv("TGRAM2TELE"), "")
		routers.Config.SMTPHost = setifset(os.Getenv("TGRAMSMTPHOST"), "")
		routers.Config.SMTPPort = setifset(os.Getenv("TGRAMSMTPPORT"), "") //port with ":", example -  :587
		routers.Config.SMTPUser = setifset(os.Getenv("TGRAMSMTPUSER"), "")
		routers.Config.SMTPPassword = setifset(os.Getenv("TGRAMSMTPPASS"), "")
		routers.Config.FCMAuth = setifset(os.Getenv("TGRAMFCMAUTH"), "")

	}
}

func main() {

	LoadEnv()

	srv := &http.Server{
		Addr:    Port,
		Handler: InitRouter(),
	}

	go func() {
		// service connections
		if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
			log.Fatalf("listen: %s\n", err)
		}
	}()

	// Wait for interrupt signal to gracefully shutdown the server with
	// a timeout of 5 seconds.
	quit := make(chan os.Signal)
	signal.Notify(quit, os.Interrupt)
	<-quit
	log.Println("Shutdown Server ...")

	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()
	if err := srv.Shutdown(ctx); err != nil {
		log.Fatal("Server Shutdown:", err)
	}
	// Close db
	if err := slowpoke.CloseAll(); err != nil {
		log.Fatal("Database Shutdown:", err)
	}
	log.Println("Server exiting")

}

func globalRecover(c *gin.Context) {
	defer func(c *gin.Context) {

		if err := recover(); err != nil {
			if err := slowpoke.CloseAll(); err != nil {
				log.Println("Database Shutdown err:", err)
			}
			log.Println("Server recovery with err:", err)
			gin.RecoveryWithWriter(gin.DefaultErrorWriter)
			//c.AbortWithStatus(500)
		}
	}(c)
	c.Next()
}

// InitRouter - init router
func InitRouter() *gin.Engine {
	//gin.SetMode(gin.DebugMode)
	gin.SetMode(gin.ReleaseMode)
	r := gin.New()
	r.Use(globalRecover)
	//r.Use(gin.Recovery())
	//gin.DefaultWriter = ioutil.Discard

	r.Use(static.Serve("/m", static.LocalFile("./media", false)))
	r.Use(static.Serve("/i", static.LocalFile("./img", false))) //where i use it?
	r.Use(static.Serve("/a", static.LocalFile("./ava", false)))
	r.Use(static.Serve("/", static.LocalFile("./media/txt", false))) //for ssl cert

	r.SetFuncMap(template.FuncMap{
		"tostr":   routers.ToStr,
		"todate":  routers.ToDate,
		"getlead": routers.GetLead,
		"var":     routers.NewVar,
		"set":     routers.SetVar,
	})
	r.LoadHTMLGlob("views/*.html")

	r.Use(routers.CheckAuth())
	//r.Use(CORSMiddleware())
	r.GET("/", routers.Main)
	r.GET("/home", routers.Home)
	r.GET("/mid", routers.All)
	r.GET("/top", routers.Top)
	r.GET("/btm", routers.Btm)

	r.GET("/register", routers.Register)
	r.POST("/register", routers.Register)

	r.GET("/login", routers.Login)
	r.POST("/login", routers.Login)

	r.GET("/@:username/:aid", routers.Article)
	r.GET("/@:username", routers.Author)
	r.GET("/a/:avatar", routers.Avatar)

	r.GET("/favorites/@:username", routers.Favorites)

	r.GET("/policy", routers.Policy)
	r.GET("/terms", routers.Terms)

	// only for registered users
	r.Use(routers.GoToRegister())

	r.GET("/settings", routers.Settings)
	r.POST("/settings", routers.Settings)

	r.POST("/logout", routers.Logout)

	r.GET("/delete/a/:aid", routers.ArticleDelete)
	r.GET("/bad/@:author/:aid/:bad", routers.ArticleBad)

	r.GET("/editor/:aid", routers.Editor)
	r.POST("/editor/:aid", routers.Editor)

	r.GET("follow/:user/*action", routers.Follow)
	r.GET("unfollow/:user/*action", routers.Unfollow)

	r.GET("fav/:aid/*action", routers.Fav)
	r.GET("unfav/:aid/*action", routers.Unfav)

	r.GET("vote/:mode/@:author/:aid", routers.Vote)

	r.POST("/comments/@:username/:aid", routers.CommentNew)
	r.GET("/commentup/@:authorart/:authorcom/:aid/:cid", routers.CommentUp)
	r.GET("/commentdel/@:authorart/:authorcom/:aid/:cid", routers.CommentDel)

	r.GET("/upload", routers.Upload)
	r.POST("/upload", routers.Upload)

	r.GET("/export/type2tele", routers.Type2tele)
	r.POST("/export/type2tele", routers.Type2tele)

	return r
}

// CORSMiddleware - open for request from javascript
func CORSMiddleware() gin.HandlerFunc {
	return func(c *gin.Context) {
		c.Writer.Header().Set("Access-Control-Allow-Origin", "http://crazydoge.com/")
		c.Writer.Header().Set("Access-Control-Max-Age", "86400")
		c.Writer.Header().Set("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE, UPDATE")
		c.Writer.Header().Set("Access-Control-Allow-Headers", "Origin, Content-Type, Content-Length, Accept-Encoding, X-CSRF-Token, Authorization")
		c.Writer.Header().Set("Access-Control-Expose-Headers", "Content-Length")
		c.Writer.Header().Set("Access-Control-Allow-Credentials", "true")

		if c.Request.Method == "OPTIONS" {
			//fmt.Println("OPTIONS")
			c.AbortWithStatus(200)
		} else {
			c.Next()
		}
	}
}