middlewares

var CacheMiddleware = func(cache Cacher, ttl time.Duration, logger requests.Logger) requests.Middleware {
	return func(next requests.MiddlewareHandlerFunc) requests.MiddlewareHandlerFunc {
		return func(req *http.Request) (*http.Response, error) {

			if req.Method != http.MethodGet {
				return next(req)
			}

			cacheKey := generateCacheKey(req)

			cachedData, ok := cache.Get(cacheKey)
			if ok {
				logger.Debugf("Cache hit", map[string]interface{}{
					"url": req.URL.String(),
					"key": cacheKey,
				})

				return buildResponseFromCache(cachedData)
			}

			resp, err := next(req)
			if err != nil {
				return nil, err
			}

			if resp.StatusCode == http.StatusOK {
				if data, err := cacheResponse(resp); err == nil {

					cache.Set(cacheKey, data, ttl)
					logger.Debugf("Cached response", map[string]interface{}{
						"url": req.URL.String(),
						"key": cacheKey,
					})
				}
			}

			return resp, nil
		}
	}
}

var CookieMiddleware = func(cookies []*http.Cookie) requests.Middleware {
	return func(next requests.MiddlewareHandlerFunc) requests.MiddlewareHandlerFunc {
		return func(req *http.Request) (*http.Response, error) {
			for _, cookie := range cookies {
				req.AddCookie(cookie)
			}
			return next(req)
		}
	}
}

var HeaderMiddleware = func(headers http.Header) requests.Middleware {
	return func(next requests.MiddlewareHandlerFunc) requests.MiddlewareHandlerFunc {
		return func(req *http.Request) (*http.Response, error) {
			for key, values := range headers {
				for _, value := range values {
					req.Header.Add(key, value)
				}
			}
			return next(req)
		}
	}
}