add http auth

server/cmd/main.go

@@ -21,6 +21,7 @@ type args struct {
 	Path     string `arg:"-d" help:"database path"`
 	LogPath  string `arg:"-l" help:"log path"`
 	RDB      bool   `arg:"-r" help:"start in read-only DB mode"`
+	HttpAuth bool   `arg:"-a" help:"Http auth on all requests"`
 	DontKill bool   `arg:"-k" help:"dont kill program on signal"`
 	UI       bool   `arg:"-u" help:"run page torrserver in browser"`
 }
@@ -43,6 +44,7 @@ func main() {
 	}
 
 	settings.Path = params.Path
+	settings.HttpAuth = params.HttpAuth
 	log.Init(params.LogPath)
 
 	dnsResolve()

server/settings/settings.go

@@ -4,6 +4,7 @@ var (
 	tdb      *TDB
 	Path     string
 	ReadOnly bool
+	HttpAuth bool
 )
 
 func InitSets(readOnly bool) {

server/web/api/route.go

@@ -14,7 +14,7 @@ type requestI struct {
 	Action string `json:"action,omitempty"`
 }
 
-func SetupRoute(route *gin.Engine) {
+func SetupRoute(route *gin.RouterGroup) {
 	route.GET("/echo", echo)
 	route.GET("/shutdown", shutdown)
 

server/web/pages/route.go

@@ -6,7 +6,7 @@ import (
 	"server/web/pages/template"
 )
 
-func SetupRoute(route *gin.Engine) {
+func SetupRoute(route *gin.RouterGroup) {
 	route.GET("/", mainPage)
 	route.GET("/stat", statPage)
 }

server/web/server.go

@@ -3,10 +3,12 @@ package web
 import (
 	"github.com/gin-contrib/cors"
 	"github.com/gin-gonic/gin"
+
 	"server/log"
 	"server/torr"
 	"server/version"
 	"server/web/api"
+	"server/web/auth"
 	"server/web/pages"
 )
 
@@ -26,8 +28,15 @@ func Start(port string) {
 
 	route := gin.New()
 	route.Use(gin.Recovery(), cors.Default())
-	api.SetupRoute(route)
-	pages.SetupRoute(route)
+
+	routeAuth := auth.SetupAuth(route)
+	if routeAuth != nil {
+		api.SetupRoute(routeAuth)
+		pages.SetupRoute(routeAuth)
+	} else {
+		api.SetupRoute(&route.RouterGroup)
+		pages.SetupRoute(&route.RouterGroup)
+	}
 	log.TLogln("Start web", port)
 	waitChan <- route.Run(":" + port)
 }