httpserver

var BaseMiddlewareArr = []Middleware{
	GlobalMiddlewarePanic,
	GlobalMiddlewareAddRequestId(requestIdFiled),
	GlobalMiddlewareCorsMiddleware(nil),
	GlobalMiddlewareOptionRequest,
}

var GlobalMiddlewareAddRequestId = func(requestIdField string) Middleware {
	return func(ctx *Ctx, next func(*Ctx)) {

		c := context.WithValue(ctx.request.Context(), requestIdField, utils.NewRequestId())
		ctx.request = ctx.request.WithContext(c)

		next(ctx)
	}
}

var GlobalMiddlewareCorsMiddleware = func(domainWhites []string) Middleware {
	return func(ctx *Ctx, next func(*Ctx)) {
		allow := false

		if origin := ctx.request.Header.Get("Origin"); origin != "" {
			if domainWhites == nil {
				allow = true
			} else {
				for _, white := range domainWhites {
					if white == origin {
						allow = true
						break
					}
				}
			}
			if allow {
				ctx.writer.Header().Set("Access-Control-Allow-Origin", origin)
			}
		}
		ctx.writer.Header().Set("Access-Control-Allow-Credentials", "true")
		ctx.writer.Header().Set("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE")
		ctx.writer.Header().Set("Access-Control-Allow-Headers", "Content-type,Authorization,x-requested-with,Request-Service")

		next(ctx)
	}
}

var GlobalMiddlewareQPSLimiter = func(r rate.Limit, b int) Middleware {
	limiter := rate.NewLimiter(r, b)
	return func(ctx *Ctx, next func(ctx *Ctx)) {
		if !limiter.Allow() {
			ctx.writer.WriteHeader(http.StatusNoContent)
			return
		}

		next(ctx)
	}
}

var HealthMiddlewareArr = []Middleware{
	GlobalMiddlewarePanic,
	GlobalMiddlewareAddRequestId(requestIdFiled),
	GlobalMiddlewareCorsMiddleware(nil),
	GlobalMiddlewareOptionRequest,
	GlobalMiddlewareQPSLimiter(limiterR, limiterInit),
}

var WithOpsRequestId = func(requestIdField string) ServerOps {
	return func(server *HttpServer) {
		server.config.requestId = requestIdField != ""
		server.config.requestIdField = requestIdField
	}
}