middleware.go

package apib

import (
	"bytes"
	"io/ioutil"
	"net/url"
	"regexp"
	"strings"

	"github.com/labstack/echo/v4"
)

func ApibGenerator(next echo.HandlerFunc) echo.HandlerFunc {
	return func(c echo.Context) error {
		if recording && currentGroup != "" && currentName != "" {
			bodyBytes, _ := ioutil.ReadAll(c.Request().Body)
			c.Request().Body = ioutil.NopCloser(bytes.NewBuffer(bodyBytes))
			request := Request{
				URI:        c.Request().URL.RequestURI(),
				Method:     c.Request().Method,
				Body:       string(bodyBytes),
				Headers:    make(map[string]string),
				Params:     make(map[string][]string),
				FormParams: make(map[string]string),
			}
			for k, v := range c.Request().Header {
				request.Headers[k] = v[0]
			}

			queryParams := []string{}
			for k, v := range c.QueryParams() {
				request.Params[k] = v
				queryParams = append(queryParams, k)
			}
			if len(queryParams) > 0 {
				rx := regexp.MustCompile(`^([^?]+)(\?.+)$`)
				request.URI = rx.ReplaceAllString(request.URI, "$1{?"+strings.Join(queryParams, ",")+"}")
			}

			for k, v := range c.Request().Form {
				request.FormParams[k] = v[0]
			}

			if len(params(c)) > 0 {
				for k, v := range params(c) {
					var escaped string
					if k == "*" { // API Blueprint doesn't allow it as a param name
						escaped = url.PathEscape("catch_all")
					} else {
						escaped = url.PathEscape(k)
					}
					request.Params[escaped] = []string{v}
					request.URI = strings.Replace(request.URI, "/"+v, "/{"+escaped+"}", 1)
				}
			}

			call := Call{
				Group:   currentGroup,
				Name:    currentName,
				Request: request,
			}
			res := c.Response()
			rw := res.Writer
			w := NewWriter(rw)
			c.Response().Writer = w
			err := next(c)
			var response Response
			if err != nil {
				return err
			}
			response = Response{
				Body:       w.Body(),
				StatusCode: c.Response().Status,
				Headers:    make(map[string]string),
			}

			for k, v := range c.Response().Header() {
				response.Headers[k] = v[0]
			}
			call.Response = response
			if calls[currentGroup] == nil {
				calls[currentGroup] = Resources{}
			}
			rss := Resource{
				Params:      request.Params,
				ExtraParams: currentParams,
				Call:        call,
				Attributes:  request.FormParams,
			}
			// rss := calls[currentGroup][request.URI]
			// if rss.Calls == nil {
			// 	rss.Calls = []Call{}
			// 	rss.Params = make(map[string][]string)
			// }
			// if rss.ExtraParams == nil {
			// 	rss.ExtraParams = []*Param{}
			// }
			// rss.Params = request.Params
			// rss.Calls = append(rss.Calls, call)
			// if len(currentParams) > 0 {
			// 	rss.ExtraParams = append(rss.ExtraParams, currentParams...)
			// }
			calls[currentGroup] = append(calls[currentGroup], rss)
			currentName = ""
			currentGroup = ""
			return nil
		}
		return next(c)
	}
}