fn-serverless/api/server/routes_create_update.go

package server

import (
	"context"
	"net/http"
	"path"
	"strings"

	"github.com/fnproject/fn/api"
	"github.com/fnproject/fn/api/models"
	"github.com/gin-gonic/gin"
)

var bad routeResponse

/* handleRouteCreateOrUpdate is used to handle POST PUT and PATCH for routes.
   Post will only create route if its not there and create app if its not.
       create only
	   Post does not skip validation of zero values
   Put will create app if its not there and if route is there update if not it will create new route.
       update if exists or create if not exists
	   Put does not skip validation of zero values
   Patch will not create app if it does not exist since the route needs to exist as well...
       update only
	   Patch accepts partial updates / skips validation of zero values.
*/
func (s *Server) handleRoutesPostPutPatch(c *gin.Context) {
	ctx := c.Request.Context()
	method := strings.ToUpper(c.Request.Method)

	var wroute models.RouteWrapper
	err := bindRoute(c, method, &wroute)
	if err != nil {
		handleErrorResponse(c, err)
		return
	}
	if method != http.MethodPatch {
		err = s.ensureApp(ctx, &wroute, method)
		if err != nil {
			handleErrorResponse(c, err)
			return
		}
	}
	resp, err := s.ensureRoute(ctx, method, &wroute)
	if err != nil {
		handleErrorResponse(c, err)
		return
	}

	s.cachedelete(resp.Route.AppName, resp.Route.Path)
	c.JSON(http.StatusOK, resp)
}

func (s *Server) submitRoute(ctx context.Context, wroute *models.RouteWrapper) error {
	err := s.setDefaultsAndValidate(wroute)
	if err != nil {
		return err
	}
	r, err := s.Datastore.InsertRoute(ctx, wroute.Route)
	if err != nil {
		return err
	}
	wroute.Route = r
	return nil
}

func (s *Server) changeRoute(ctx context.Context, wroute *models.RouteWrapper) error {
	err := wroute.Validate(true)
	if err != nil {
		return err
	}
	r, err := s.Datastore.UpdateRoute(ctx, wroute.Route)
	if err != nil {
		return err
	}
	wroute.Route = r
	return nil
}

// ensureApp will only execute if it is on put
func (s *Server) ensureRoute(ctx context.Context, method string, wroute *models.RouteWrapper) (routeResponse, error) {
	resp := routeResponse{"Route successfully created", nil}
	up := routeResponse{"Route successfully updated", nil}

	switch method {
	case http.MethodPost:
		err := s.submitRoute(ctx, wroute)
		if err != nil {
			return bad, err
		}
		resp.Route = wroute.Route
		return resp, nil
	case http.MethodPut:
		_, err := s.Datastore.GetRoute(ctx, wroute.Route.AppName, wroute.Route.Path)
		if err != nil && err == models.ErrRoutesNotFound {
			err := s.submitRoute(ctx, wroute)
			if err != nil {
				return bad, err
			}
			resp.Route = wroute.Route
			return resp, nil
		} else {
			err := s.changeRoute(ctx, wroute)
			if err != nil {
				return bad, err
			}
			up.Route = wroute.Route
			return up, nil
		}
	case http.MethodPatch:
		err := s.changeRoute(ctx, wroute)
		if err != nil {
			return bad, err
		}
		up.Route = wroute.Route
		return up, nil
	}
	return bad, nil
}

// ensureApp will only execute if it is on post or put. Patch is not allowed to create apps.
func (s *Server) ensureApp(ctx context.Context, wroute *models.RouteWrapper, method string) error {
	app, err := s.Datastore.GetApp(ctx, wroute.Route.AppName)
	if err != nil && err != models.ErrAppsNotFound {
		return err
	} else if app == nil {
		// Create a new application
		newapp := &models.App{Name: wroute.Route.AppName}
		if err = newapp.Validate(); err != nil {
			return err
		}

		err = s.FireBeforeAppCreate(ctx, newapp)
		if err != nil {
			return err
		}

		_, err = s.Datastore.InsertApp(ctx, newapp)
		if err != nil {
			return err
		}

		err = s.FireAfterAppCreate(ctx, newapp)
		if err != nil {
			return err
		}

	}
	return nil
}

// bindRoute binds the RouteWrapper to the json from the request.
func bindRoute(c *gin.Context, method string, wroute *models.RouteWrapper) error {
	err := c.BindJSON(wroute)
	if err != nil {
		return models.ErrInvalidJSON
	}

	if wroute.Route == nil {
		return models.ErrRoutesMissingNew
	}
	wroute.Route.AppName = c.MustGet(api.AppName).(string)

	if method == http.MethodPut || method == http.MethodPatch {
		p := path.Clean(c.MustGet(api.Path).(string))

		if wroute.Route.Path != "" && wroute.Route.Path != p {
			return models.ErrRoutesPathImmutable
		}
		wroute.Route.Path = p
	}
	if method == http.MethodPost {
		if wroute.Route.Path == "" {
			return models.ErrRoutesValidationMissingPath
		}
	}
	return nil
}

func (s *Server) setDefaultsAndValidate(wroute *models.RouteWrapper) error {
	wroute.Route.SetDefaults()
	return wroute.Validate(false)
}