mirror of
https://github.com/WJQSERVER-STUDIO/ghproxy.git
synced 2026-02-03 00:01:10 +08:00
120 lines
2.6 KiB
Go
120 lines
2.6 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
|
|
"ghproxy/api"
|
|
"ghproxy/auth"
|
|
"ghproxy/config"
|
|
"ghproxy/logger"
|
|
"ghproxy/proxy"
|
|
"ghproxy/rate"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
var (
|
|
cfg *config.Config
|
|
router *gin.Engine
|
|
configfile = "/data/ghproxy/config/config.toml"
|
|
cfgfile string
|
|
limiter *rate.RateLimiter
|
|
version string
|
|
)
|
|
|
|
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 {
|
|
limiter = rate.New(cfg.RateLimit.RatePerMinute, cfg.RateLimit.Burst, 1*time.Minute)
|
|
logInfo("Rate Limit Loaded")
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
readFlag()
|
|
flag.Parse()
|
|
loadConfig()
|
|
setupLogger(cfg)
|
|
loadlist(cfg)
|
|
setupRateLimit(cfg)
|
|
|
|
gin.SetMode(gin.ReleaseMode)
|
|
|
|
router = gin.Default()
|
|
if cfg.Server.EnableH2C {
|
|
router.UseH2C = true
|
|
}
|
|
|
|
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 {
|
|
router.GET("/", func(c *gin.Context) {
|
|
c.String(http.StatusForbidden, "403 Forbidden Access")
|
|
logWarning("403 > Path:/ IP:%s UA:%s METHOD:%s HTTPv:%s", c.ClientIP(), c.Request.UserAgent(), c.Request.Method, c.Request.Proto)
|
|
})
|
|
}
|
|
|
|
router.NoRoute(func(c *gin.Context) {
|
|
proxy.NoRouteHandler(cfg, limiter)(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")
|
|
}
|