ghproxy/api/api.go

package api

import (
	"context"
	"ghproxy/config"

	"github.com/WJQSERVER-STUDIO/go-utils/logger"
	"github.com/cloudwego/hertz/pkg/app"
	"github.com/cloudwego/hertz/pkg/app/server"
)

var (
	logw       = logger.Logw
	logDump    = logger.LogDump
	logDebug   = logger.LogDebug
	logInfo    = logger.LogInfo
	logWarning = logger.LogWarning
	logError   = logger.LogError
)

func NoCacheMiddleware() app.HandlerFunc {
	return func(ctx context.Context, c *app.RequestContext) {
		// 设置禁止缓存的响应头
		c.Response.Header.Set("Cache-Control", "no-store, no-cache, must-revalidate")
		c.Response.Header.Set("Pragma", "no-cache")
		c.Response.Header.Set("Expires", "0")
		c.Next(ctx) // 继续处理请求
	}
}

func InitHandleRouter(cfg *config.Config, r *server.Hertz, version string) {
	apiRouter := r.Group("/api", NoCacheMiddleware())
	{
		apiRouter.GET("/size_limit", func(ctx context.Context, c *app.RequestContext) {
			SizeLimitHandler(cfg, c, ctx)
		})
		apiRouter.GET("/whitelist/status", func(ctx context.Context, c *app.RequestContext) {
			WhiteListStatusHandler(cfg, c, ctx)
		})
		apiRouter.GET("/blacklist/status", func(ctx context.Context, c *app.RequestContext) {
			BlackListStatusHandler(cfg, c, ctx)
		})
		apiRouter.GET("/cors/status", func(ctx context.Context, c *app.RequestContext) {
			CorsStatusHandler(cfg, c, ctx)
		})
		apiRouter.GET("/healthcheck", func(ctx context.Context, c *app.RequestContext) {
			HealthcheckHandler(c, ctx)
		})
		apiRouter.GET("/version", func(ctx context.Context, c *app.RequestContext) {
			VersionHandler(c, ctx, version)
		})
		apiRouter.GET("/rate_limit/status", func(ctx context.Context, c *app.RequestContext) {
			RateLimitStatusHandler(cfg, c, ctx)
		})
		apiRouter.GET("/rate_limit/limit", func(ctx context.Context, c *app.RequestContext) {
			RateLimitLimitHandler(cfg, c, ctx)
		})
		apiRouter.GET("/smartgit/status", func(ctx context.Context, c *app.RequestContext) {
			SmartGitStatusHandler(cfg, c, ctx)
		})

	}
	logInfo("API router Init success")
}

func SizeLimitHandler(cfg *config.Config, c *app.RequestContext, ctx context.Context) {
	sizeLimit := cfg.Server.SizeLimit
	logInfo("%s %s %s %s %s", c.ClientIP(), c.Method(), string(c.Path()), c.Request.Header.UserAgent(), c.Request.Header.GetProtocol())
	c.Response.Header.Set("Content-Type", "application/json")
	c.JSON(200, (map[string]interface{}{
		"MaxResponseBodySize": sizeLimit,
	}))
}

func WhiteListStatusHandler(cfg *config.Config, c *app.RequestContext, ctx context.Context) {
	logInfo("%s %s %s %s %s", c.ClientIP(), c.Method(), string(c.Path()), c.Request.Header.UserAgent(), c.Request.Header.GetProtocol())
	c.Response.Header.Set("Content-Type", "application/json")
	c.JSON(200, (map[string]interface{}{
		"Whitelist": cfg.Whitelist.Enabled,
	}))
}

func BlackListStatusHandler(cfg *config.Config, c *app.RequestContext, ctx context.Context) {
	logInfo("%s %s %s %s %s", c.ClientIP(), c.Method(), string(c.Path()), c.Request.Header.UserAgent(), c.Request.Header.GetProtocol())
	c.Response.Header.Set("Content-Type", "application/json")
	c.JSON(200, (map[string]interface{}{
		"Blacklist": cfg.Blacklist.Enabled,
	}))
}

func CorsStatusHandler(cfg *config.Config, c *app.RequestContext, ctx context.Context) {
	logInfo("%s %s %s %s %s", c.ClientIP(), c.Method(), string(c.Path()), c.Request.Header.UserAgent(), c.Request.Header.GetProtocol())
	c.Response.Header.Set("Content-Type", "application/json")
	c.JSON(200, (map[string]interface{}{
		"Cors": cfg.Server.Cors,
	}))
}

func HealthcheckHandler(c *app.RequestContext, ctx context.Context) {
	logInfo("%s %s %s %s %s", c.ClientIP(), c.Method(), string(c.Path()), c.Request.Header.UserAgent(), c.Request.Header.GetProtocol())
	c.Response.Header.Set("Content-Type", "application/json")
	c.JSON(200, (map[string]interface{}{
		"Status": "OK",
	}))
}

func VersionHandler(c *app.RequestContext, ctx context.Context, version string) {
	logInfo("%s %s %s %s %s", c.ClientIP(), c.Method(), string(c.Path()), c.Request.Header.UserAgent(), c.Request.Header.GetProtocol())
	c.Response.Header.Set("Content-Type", "application/json")
	c.JSON(200, (map[string]interface{}{
		"Version": version,
	}))
}

func RateLimitStatusHandler(cfg *config.Config, c *app.RequestContext, ctx context.Context) {
	logInfo("%s %s %s %s %s", c.ClientIP(), c.Method(), string(c.Path()), c.Request.Header.UserAgent(), c.Request.Header.GetProtocol())
	c.Response.Header.Set("Content-Type", "application/json")
	c.JSON(200, (map[string]interface{}{
		"RateLimit": cfg.RateLimit.Enabled,
	}))
}

func RateLimitLimitHandler(cfg *config.Config, c *app.RequestContext, ctx context.Context) {
	logInfo("%s %s %s %s %s", c.ClientIP(), c.Method(), string(c.Path()), c.Request.Header.UserAgent(), c.Request.Header.GetProtocol())
	c.Response.Header.Set("Content-Type", "application/json")
	c.JSON(200, (map[string]interface{}{
		"RatePerMinute": cfg.RateLimit.RatePerMinute,
	}))
}

func SmartGitStatusHandler(cfg *config.Config, c *app.RequestContext, ctx context.Context) {
	logInfo("%s %s %s %s %s", c.ClientIP(), c.Method(), string(c.Path()), c.Request.Header.UserAgent(), c.Request.Header.GetProtocol())
	c.Response.Header.Set("Content-Type", "application/json")
	c.JSON(200, (map[string]interface{}{
		"enabled": cfg.GitClone.Mode == "cache",
	}))
}