package proxy

import (
	"context"
	"ghproxy/config"
	"ghproxy/rate"
	"strings"

	"github.com/cloudwego/hertz/pkg/app"
)

func RoutingHandler(cfg *config.Config, limiter *rate.RateLimiter, iplimiter *rate.IPRateLimiter) app.HandlerFunc {
	return func(ctx context.Context, c *app.RequestContext) {

		var shoudBreak bool

		shoudBreak = rateCheck(cfg, c, limiter, iplimiter)
		if shoudBreak {
			return
		}

		var (
			rawPath string
		)

		rawPath = strings.TrimPrefix(string(c.Request.RequestURI()), "/") // 去掉前缀/

		var (
			user    string
			repo    string
			matcher string
		)

		user = c.Param("user")
		repo = c.Param("repo")
		matcher = c.GetString("matcher")

		logDump("%s %s %s %s %s Matched-Username: %s, Matched-Repo: %s", c.ClientIP(), c.Method(), rawPath, c.Request.Header.UserAgent(), c.Request.Header.GetProtocol(), user, repo)
		logDump("%s", c.Request.Header.Header())

		shoudBreak = listCheck(cfg, c, user, repo, rawPath)
		if shoudBreak {
			return
		}

		shoudBreak = authCheck(c, cfg, matcher, rawPath)
		if shoudBreak {
			return
		}

		// 处理blob/raw路径
		// 处理blob/raw路径
		if matcher == "blob" {
			rawPath = rawPath[10:]
			rawPath = "raw.githubusercontent.com" + rawPath
			rawPath = strings.Replace(rawPath, "/blob/", "/", 1)
			matcher = "raw"
		}

		// 为rawpath加入https:// 头
		rawPath = "https://" + rawPath

		logDebug("Matched: %v", matcher)

		switch matcher {
		case "releases", "blob", "raw", "gist", "api":
			ChunkedProxyRequest(ctx, c, rawPath, cfg, matcher)
		case "clone":
			GitReq(ctx, c, rawPath, cfg, "git")
		default:
			ErrorPage(c, NewErrorWithStatusLookup(500, "Matched But Not Matched"))
			logError("Matched But Not Matched Path: %s rawPath: %s matcher: %s", c.Path(), rawPath, matcher)
			return
		}
	}
}