里見 灯花
158 lines
3.5 KiB
Go
158 lines
3.5 KiB
Go
package main
|
|
|
|
import (
|
|
"embed"
|
|
"flag"
|
|
"fmt"
|
|
"io"
|
|
"io/fs"
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
|
|
"ghproxy/api"
|
|
"ghproxy/auth"
|
|
"ghproxy/config"
|
|
"ghproxy/proxy"
|
|
"ghproxy/rate"
|
|
|
|
"github.com/WJQSERVER-STUDIO/go-utils/logger"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
var (
|
|
cfg *config.Config
|
|
router *gin.Engine
|
|
configfile = "/data/ghproxy/config/config.toml"
|
|
cfgfile string
|
|
version string
|
|
dev string
|
|
runMode string
|
|
limiter *rate.RateLimiter
|
|
iplimiter *rate.IPRateLimiter
|
|
)
|
|
|
|
var (
|
|
//go:embed pages/*
|
|
pagesFS embed.FS
|
|
)
|
|
|
|
var (
|
|
logw = logger.Logw
|
|
logInfo = logger.LogInfo
|
|
logWarning = logger.LogWarning
|
|
logError = logger.LogError
|
|
)
|
|
|
|
func readFlag() {
|
|
flag.StringVar(&cfgfile, "cfg", configfile, "config file path")
|
|
}
|
|
|
|
func loadConfig() {
|
|
var err error
|
|
cfg, err = config.LoadConfig(cfgfile)
|
|
if err != nil {
|
|
log.Fatalf("Failed to load config: %v", err)
|
|
}
|
|
fmt.Println("Config File Path: ", cfgfile)
|
|
fmt.Printf("Loaded config: %v\n", cfg)
|
|
}
|
|
|
|
func setupLogger(cfg *config.Config) {
|
|
var err error
|
|
err = logger.Init(cfg.Log.LogFilePath, cfg.Log.MaxLogSize)
|
|
if err != nil {
|
|
log.Fatalf("Failed to initialize logger: %v", err)
|
|
}
|
|
logInfo("Config File Path: ", cfgfile)
|
|
logInfo("Loaded config: %v\n", cfg)
|
|
logInfo("Init Completed")
|
|
}
|
|
|
|
func loadlist(cfg *config.Config) {
|
|
auth.Init(cfg)
|
|
}
|
|
|
|
func setupApi(cfg *config.Config, router *gin.Engine, version string) {
|
|
api.InitHandleRouter(cfg, router, version)
|
|
}
|
|
|
|
func setupRateLimit(cfg *config.Config) {
|
|
if cfg.RateLimit.Enabled {
|
|
if cfg.RateLimit.RateMethod == "ip" {
|
|
iplimiter = rate.NewIPRateLimiter(cfg.RateLimit.RatePerMinute, cfg.RateLimit.Burst, 1*time.Minute)
|
|
} else if cfg.RateLimit.RateMethod == "total" {
|
|
limiter = rate.New(cfg.RateLimit.RatePerMinute, cfg.RateLimit.Burst, 1*time.Minute)
|
|
} else {
|
|
logError("Invalid RateLimit Method: %s", cfg.RateLimit.RateMethod)
|
|
}
|
|
logInfo("Rate Limit Loaded")
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
readFlag()
|
|
flag.Parse()
|
|
loadConfig()
|
|
setupLogger(cfg)
|
|
loadlist(cfg)
|
|
setupRateLimit(cfg)
|
|
|
|
if cfg.Server.Debug {
|
|
dev = "true"
|
|
version = "dev"
|
|
}
|
|
if dev == "true" {
|
|
gin.SetMode(gin.DebugMode)
|
|
runMode = "dev"
|
|
} else {
|
|
gin.SetMode(gin.ReleaseMode)
|
|
runMode = "release"
|
|
}
|
|
|
|
gin.LoggerWithWriter(io.Discard)
|
|
router = gin.New()
|
|
router.Use(gin.Recovery())
|
|
//H2C默认值为true，而后遵循cfg.Server.EnableH2C的设置
|
|
if cfg.Server.EnableH2C == "on" {
|
|
router.UseH2C = true
|
|
} else if cfg.Server.EnableH2C == "" {
|
|
router.UseH2C = true
|
|
} else {
|
|
router.UseH2C = false
|
|
}
|
|
|
|
setupApi(cfg, router, version)
|
|
|
|
if cfg.Pages.Enabled {
|
|
indexPagePath := fmt.Sprintf("%s/index.html", cfg.Pages.StaticDir)
|
|
faviconPath := fmt.Sprintf("%s/favicon.ico", cfg.Pages.StaticDir)
|
|
router.GET("/", func(c *gin.Context) {
|
|
c.File(indexPagePath)
|
|
logInfo("IP:%s UA:%s METHOD:%s HTTPv:%s", c.ClientIP(), c.Request.UserAgent(), c.Request.Method, c.Request.Proto)
|
|
})
|
|
router.StaticFile("/favicon.ico", faviconPath)
|
|
} else if !cfg.Pages.Enabled {
|
|
pages, err := fs.Sub(pagesFS, "pages")
|
|
if err != nil {
|
|
log.Fatalf("Failed when processing pages: %s", err)
|
|
}
|
|
router.GET("/", gin.WrapH(http.FileServer(http.FS(pages))))
|
|
router.GET("/favicon.ico", gin.WrapH(http.FileServer(http.FS(pages))))
|
|
}
|
|
|
|
router.NoRoute(func(c *gin.Context) {
|
|
proxy.NoRouteHandler(cfg, limiter, iplimiter, runMode)(c)
|
|
})
|
|
}
|
|
|
|
func main() {
|
|
err := router.Run(fmt.Sprintf("%s:%d", cfg.Server.Host, cfg.Server.Port))
|
|
if err != nil {
|
|
logError("Failed to start server: %v\n", err)
|
|
}
|
|
defer logger.Close()
|
|
fmt.Println("Program Exit")
|
|
}
|