response_middleware.go

package main

import (
	"errors"
	"net/http"
)

var RESPONSE_PROCESSORS map[string]TykResponseHandler = map[string]TykResponseHandler{
	"header_injector":         HeaderInjector{},
	"response_body_transform": ResponseTransformMiddleware{},
}

type TykResponseHandler interface {
	HandleResponse(http.ResponseWriter, *http.Response, *http.Request, *SessionState) error
	New(interface{}, *APISpec) (TykResponseHandler, error)
}

func GetResponseProcessorByName(name string) (TykResponseHandler, error) {
	processor, ok := RESPONSE_PROCESSORS[name]
	if !ok {
		return nil, errors.New("Not found")
	}

	return processor, nil

}

type ResponseChain struct{}

func (r ResponseChain) Go(chain *[]TykResponseHandler, rw http.ResponseWriter, res *http.Response, req *http.Request, ses *SessionState) error {

	if chain == nil {
		return nil
	}

	for _, rh := range *chain {
		mwErr := rh.HandleResponse(rw, res, req, ses)
		if mwErr != nil {
			return mwErr
		}
	}

	return nil
}