whatismyip/cmd/whatismyip.go

92 lines
2.2 KiB
Go
Raw Permalink Normal View History

2021-11-10 19:06:12 +00:00
package main
import (
"context"
"flag"
"fmt"
2024-03-23 16:41:34 +00:00
"net/http"
2021-11-10 19:06:12 +00:00
"os"
2024-04-12 17:26:48 +00:00
"slices"
"time"
2021-11-10 19:06:12 +00:00
"github.com/dcarrillo/whatismyip/internal/httputils"
"github.com/dcarrillo/whatismyip/internal/setting"
2024-04-12 17:26:48 +00:00
"github.com/dcarrillo/whatismyip/resolver"
2023-03-20 15:33:45 +00:00
"github.com/dcarrillo/whatismyip/server"
"github.com/gin-contrib/secure"
2024-04-12 17:26:48 +00:00
"github.com/patrickmn/go-cache"
2021-11-10 19:06:12 +00:00
"github.com/dcarrillo/whatismyip/router"
"github.com/gin-gonic/gin"
)
func main() {
o, err := setting.Setup(os.Args[1:])
if err == flag.ErrHelp || err == setting.ErrVersion {
fmt.Print(o)
os.Exit(0)
} else if err != nil {
2023-03-16 18:59:04 +00:00
fmt.Println(err)
os.Exit(1)
}
2024-04-12 17:26:48 +00:00
servers := []server.Server{}
2023-03-20 15:33:45 +00:00
engine := setupEngine()
2024-04-12 17:26:48 +00:00
if setting.App.Resolver.Domain != "" {
store := cache.New(1*time.Minute, 10*time.Minute)
dnsEngine := resolver.Setup(store)
nameServer := server.NewDNSServer(context.Background(), dnsEngine.Handler())
servers = append(servers, nameServer)
engine.Use(router.GetDNSDiscoveryHandler(store, setting.App.Resolver.Domain, setting.App.Resolver.RedirectPort))
}
2021-11-10 19:06:12 +00:00
router.SetupTemplate(engine)
router.Setup(engine)
2024-04-12 17:26:48 +00:00
servers = slices.Concat(servers, setupHTTPServers(context.Background(), engine.Handler()))
2021-11-10 19:06:12 +00:00
2024-03-23 16:41:34 +00:00
whatismyip := server.Setup(servers)
2023-03-20 15:33:45 +00:00
whatismyip.Run()
2021-11-10 19:06:12 +00:00
}
2023-03-20 15:33:45 +00:00
func setupEngine() *gin.Engine {
2021-11-10 19:06:12 +00:00
gin.DisableConsoleColor()
if os.Getenv(gin.EnvGinMode) == "" {
gin.SetMode(gin.ReleaseMode)
}
2023-03-20 15:33:45 +00:00
engine := gin.New()
2021-11-10 19:06:12 +00:00
engine.Use(gin.LoggerWithFormatter(httputils.GetLogFormatter))
engine.Use(gin.Recovery())
if setting.App.EnableSecureHeaders {
engine.Use(secure.New(secure.Config{
BrowserXssFilter: true,
ContentTypeNosniff: true,
FrameDeny: true,
}))
}
_ = engine.SetTrustedProxies(nil)
engine.TrustedPlatform = setting.App.TrustedHeader
2023-03-20 15:33:45 +00:00
return engine
}
2024-03-23 16:41:34 +00:00
func setupHTTPServers(ctx context.Context, handler http.Handler) []server.Server {
var servers []server.Server
if setting.App.BindAddress != "" {
tcpServer := server.NewTCPServer(ctx, &handler)
servers = append(servers, tcpServer)
}
if setting.App.TLSAddress != "" {
tlsServer := server.NewTLSServer(ctx, &handler)
servers = append(servers, tlsServer)
if setting.App.EnableHTTP3 {
quicServer := server.NewQuicServer(ctx, tlsServer)
servers = append(servers, quicServer)
}
}
return servers
}