mirror of
https://github.com/dcarrillo/whatismyip.git
synced 2025-11-04 13:59:09 +00:00
110 lines
2.8 KiB
Go
110 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"slices"
|
|
"time"
|
|
|
|
"github.com/dcarrillo/whatismyip/internal/httputils"
|
|
"github.com/dcarrillo/whatismyip/internal/metrics"
|
|
"github.com/dcarrillo/whatismyip/internal/setting"
|
|
"github.com/dcarrillo/whatismyip/resolver"
|
|
"github.com/dcarrillo/whatismyip/server"
|
|
"github.com/dcarrillo/whatismyip/service"
|
|
"github.com/gin-contrib/secure"
|
|
"github.com/patrickmn/go-cache"
|
|
|
|
"github.com/dcarrillo/whatismyip/router"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func main() {
|
|
o, err := setting.Setup(os.Args[1:])
|
|
if err != nil {
|
|
if err == flag.ErrHelp || err == setting.ErrVersion {
|
|
fmt.Print(o)
|
|
os.Exit(0)
|
|
}
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
servers := []server.Server{}
|
|
engine := setupEngine()
|
|
|
|
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))
|
|
}
|
|
|
|
var geoSvc *service.Geo
|
|
if setting.App.GeodbPath.City != "" || setting.App.GeodbPath.ASN != "" {
|
|
if geoSvc, err = service.NewGeo(context.Background(), setting.App.GeodbPath.City, setting.App.GeodbPath.ASN); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
router.SetupTemplate(engine)
|
|
router.Setup(engine, geoSvc)
|
|
servers = slices.Concat(servers, setupHTTPServers(context.Background(), engine.Handler()))
|
|
|
|
if setting.App.PrometheusAddress != "" {
|
|
prometheusServer := server.NewPrometheusServer(context.Background())
|
|
servers = append(servers, prometheusServer)
|
|
}
|
|
|
|
whatismyip := server.Setup(servers, geoSvc)
|
|
whatismyip.Run()
|
|
}
|
|
|
|
func setupEngine() *gin.Engine {
|
|
gin.DisableConsoleColor()
|
|
if os.Getenv(gin.EnvGinMode) == "" {
|
|
gin.SetMode(gin.ReleaseMode)
|
|
}
|
|
engine := gin.New()
|
|
engine.Use(gin.LoggerWithFormatter(httputils.GetLogFormatter), gin.Recovery())
|
|
if setting.App.PrometheusAddress != "" {
|
|
metrics.Enable()
|
|
engine.Use(metrics.GinMiddleware())
|
|
}
|
|
if setting.App.EnableSecureHeaders {
|
|
engine.Use(secure.New(secure.Config{
|
|
BrowserXssFilter: true,
|
|
ContentTypeNosniff: true,
|
|
FrameDeny: true,
|
|
}))
|
|
}
|
|
_ = engine.SetTrustedProxies(nil)
|
|
engine.TrustedPlatform = setting.App.TrustedHeader
|
|
|
|
return engine
|
|
}
|
|
|
|
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
|
|
}
|