avatar_route.go

package api

import (
	"context"
	"net/http"
	"strconv"

	"github.com/pressly/chi"
	chiRender "github.com/pressly/chi/render"
	"github.com/titouanfreville/popcubeapi/datastores"
	"github.com/titouanfreville/popcubeapi/models"
)

const (
	avatarNameKey key = "avatarName"
	avatarLinkKey key = "avatarLink"
	oldAvatarKey  key = "oldAvatar"
)

func initAvatarRoute(router chi.Router) {
	router.Route("/avatar", func(r chi.Router) {
		r.Use(tokenAuth.Verifier)
		r.Use(Authenticator)
		// swagger:route GET /avatar Avatars getAllAvatar
		//
		// Get avatars
		//
		// This will get all the avatars available in the organisation.
		//
		// 	Responses:
		//    200: avatarArraySuccess
		// 	  503: databaseError
		// 	  default: genericError
		r.Get("/", getAllAvatar)
		// swagger:route POST /avatar Avatars newAvatar
		//
		// New avatar
		//
		// This will create an avatar for organisation avatars library.
		//
		// 	Responses:
		//    201: avatarObjectSuccess
		// 	  422: wrongEntity
		// 	  503: databaseError
		// 	  default: genericError
		r.Post("/", newAvatar)
		// swagger:route GET /avatar/all Avatars getAllAvatar1
		//
		// Get avatars
		//
		// This will get all the avatars available in the organisation.
		//
		// 	Responses:
		//    200: avatarArraySuccess
		// 	  503: databaseError
		// 	  default: genericError
		r.Get("/all", getAllAvatar)
		// swagger:route POST /avatar/new Avatars newAvatar1
		//
		// New avatar
		//
		// This will create an avatar for organisation avatars library.
		//
		// 	Responses:
		//    201: avatarObjectSuccess
		// 	  422: wrongEntity
		// 	  503: databaseError
		// 	  default: genericError
		r.Post("/new", newAvatar)
		r.Route("/link/", func(r chi.Router) {
			r.Route("/:avatarLink", func(r chi.Router) {
				r.Use(avatarContext)
				// swagger:route GET /avatar/link/{avatarLink} Avatars getAvatarFromLink
				//
				// Get avatar from link
				//
				// This will return the avatar object corresponding to provided link
				//
				// 	Responses:
				//    200: avatarObjectSuccess
				// 	  503: databaseError
				// 	  default: genericError
				r.Get("/", getAvatarFromLink)
			})
		})
		r.Route("/name/", func(r chi.Router) {
			r.Route("/:avatarName", func(r chi.Router) {
				r.Use(avatarContext)
				// swagger:route GET /avatar/name/{avatarName} Avatars getAvatarFromName
				//
				// Get avatar from name
				//
				// This will return the avatar object corresponding to provided name
				//
				// 	Responses:
				//    200: avatarObjectSuccess
				// 	  503: databaseError
				// 	  default: genericError
				r.Get("/", getAvatarFromName)
			})
		})
		r.Route("/:avatarID", func(r chi.Router) {
			r.Use(avatarContext)
			// swagger:route PUT /avatar/{avatarID} Avatars updateAvatar
			//
			// Update avatar
			//
			// This will return the new avatar object
			//
			// 	Responses:
			//    200: avatarObjectSuccess
			// 	  422: wrongEntity
			// 	  503: databaseError
			// 	  default: genericError
			r.Put("/update", updateAvatar)
			// swagger:route PUT /avatar/{avatarID} Avatars updateAvatar
			//
			// Update avatar
			//
			// This will return the new avatar object
			//
			// 	Responses:
			//    200: avatarObjectSuccess
			// 	  422: wrongEntity
			// 	  503: databaseError
			// 	  default: genericError
			r.Delete("/delete", deleteAvatar)
		})
	})
}

func avatarContext(next http.Handler) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		avatarID, err := strconv.ParseUint(chi.URLParam(r, "avatarID"), 10, 64)
		name := chi.URLParam(r, "avatarName")
		link := chi.URLParam(r, "avatarLink")
		oldAvatar := models.Avatar{}
		ctx := context.WithValue(r.Context(), avatarNameKey, name)
		ctx = context.WithValue(ctx, avatarLinkKey, link)
		if err == nil {
			oldAvatar = datastores.Store().Avatar().GetByID(avatarID, dbStore.db)
		}
		ctx = context.WithValue(ctx, oldAvatarKey, oldAvatar)
		next.ServeHTTP(w, r.WithContext(ctx))
	})
}

func getAllAvatar(w http.ResponseWriter, r *http.Request) {
	store := datastores.Store()
	db := dbStore.db
	if err := db.DB().Ping(); err == nil {
		result := store.Avatar().GetAll(db)
		render.JSON(w, 200, result)
	} else {
		render.JSON(w, error503.StatusCode, error503)
	}
}

func getAvatarFromName(w http.ResponseWriter, r *http.Request) {
	store := datastores.Store()

	db := dbStore.db
	if err := db.DB().Ping(); err == nil {
		name := r.Context().Value(avatarNameKey).(string)
		avatar := store.Avatar().GetByName(name, db)
		render.JSON(w, 200, avatar)
	} else {
		render.JSON(w, error503.StatusCode, error503)
	}
}

func getAvatarFromLink(w http.ResponseWriter, r *http.Request) {
	store := datastores.Store()

	db := dbStore.db
	if err := db.DB().Ping(); err == nil {
		link := r.Context().Value(avatarLinkKey).(string)
		avatar := store.Avatar().GetByLink(link, db)
		render.JSON(w, 200, avatar)
	} else {
		render.JSON(w, error503.StatusCode, error503)
	}
}

func newAvatar(w http.ResponseWriter, r *http.Request) {
	var data struct {
		Avatar *models.Avatar
		OmitID interface{} `json:"id,omitempty"`
	}
	store := datastores.Store()

	db := dbStore.db
	request := r.Body
	err := chiRender.Bind(request, &data)
	if err != nil || data.Avatar == nil {
		render.JSON(w, error422.StatusCode, error422)
	} else {
		if err := db.DB().Ping(); err == nil {
			err := store.Avatar().Save(data.Avatar, db)
			if err == nil {
				render.JSON(w, 201, data.Avatar)
			} else {
				render.JSON(w, err.StatusCode, err)
			}
		} else {
			render.JSON(w, error503.StatusCode, error503)
		}
	}
}

func updateAvatar(w http.ResponseWriter, r *http.Request) {
	var data struct {
		Avatar *models.Avatar
		OmitID interface{} `json:"id,omitempty"`
	}
	store := datastores.Store()

	db := dbStore.db
	request := r.Body
	err := chiRender.Bind(request, &data)
	avatar := r.Context().Value(oldAvatarKey).(models.Avatar)
	if err != nil || data.Avatar == nil {
		render.JSON(w, error422.StatusCode, error422)
	} else {
		if err := db.DB().Ping(); err == nil {
			err := store.Avatar().Update(&avatar, data.Avatar, db)
			if err == nil {
				render.JSON(w, 200, avatar)
			} else {
				render.JSON(w, err.StatusCode, err)
			}
		} else {
			render.JSON(w, error503.StatusCode, error503)
		}
	}
}

func deleteAvatar(w http.ResponseWriter, r *http.Request) {
	avatar := r.Context().Value(oldAvatarKey).(models.Avatar)
	store := datastores.Store()

	message := deleteMessageModel{
		Object: avatar,
	}
	db := dbStore.db
	if err := db.DB().Ping(); err == nil {
		err := store.Avatar().Delete(&avatar, db)
		if err == nil {
			message.Success = true
			message.Message = "Avatar well removed."
			render.JSON(w, 200, message)
		} else {
			message.Success = false
			message.Message = err.Message
			render.JSON(w, err.StatusCode, message.Message)
		}
	} else {
		render.JSON(w, 503, error503)
	}
}