From 7f98182e94636cc5147ec4db19eb8e0c52a55296 Mon Sep 17 00:00:00 2001
From: Sebastian Frank <frank@basiskonfiguration.de>
Date: Sat, 30 Nov 2019 14:08:42 +0100
Subject: [PATCH] proccessResults also for put/post

---
 handler.go | 78 +++++++++++++++++++++++++++++++-----------------------
 hooks.go   | 11 ++++++++
 2 files changed, 56 insertions(+), 33 deletions(-)

diff --git a/handler.go b/handler.go
index 07e065f..cbda5dd 100644
--- a/handler.go
+++ b/handler.go
@@ -5,8 +5,6 @@ import (
 	"strconv"
 	"strings"
 
-	"github.com/davecgh/go-spew/spew"
-
 	"gitbase.de/gopackage/mgocrud"
 	"github.com/gin-gonic/gin"
 	mgo "gopkg.in/mgo.v2"
@@ -55,21 +53,16 @@ func (api *API) collectionGetOneHandler(m mgocrud.ModelInterface) gin.HandlerFun
 			return
 		}
 
-		if i, ok := m.(interface {
-			ProcessResults(*Context, interface{}) error
-		}); ok {
-			// custom select manipulation
-			err := i.ProcessResults(&Context{
-				API:     api,
-				Context: c,
-				DB:      db,
-			}, newM)
-			if err != nil {
-				c.JSON(500, gin.H{
-					"error": err.Error(),
-				})
-				return
-			}
+		err = processResults(&Context{
+			API:     api,
+			Context: c,
+			DB:      db,
+		}, m, newM)
+		if err != nil {
+			c.JSON(500, gin.H{
+				"error": err.Error(),
+			})
+			return
 		}
 
 		c.JSON(200, newM)
@@ -200,21 +193,16 @@ func (api *API) collectionGetHandler(m mgocrud.ModelInterface) gin.HandlerFunc {
 			return
 		}
 
-		if i, ok := m.(interface {
-			ProcessResults(*Context, interface{}) error
-		}); ok {
-			// custom select manipulation
-			err := i.ProcessResults(&Context{
-				API:     api,
-				Context: c,
-				DB:      db,
-			}, results)
-			if err != nil {
-				c.JSON(500, gin.H{
-					"error": err.Error(),
-				})
-				return
-			}
+		err = processResults(&Context{
+			API:     api,
+			Context: c,
+			DB:      db,
+		}, m, results)
+		if err != nil {
+			c.JSON(500, gin.H{
+				"error": err.Error(),
+			})
+			return
 		}
 
 		if setIdent {
@@ -287,6 +275,18 @@ func (api *API) collectionPostHandler(m mgocrud.ModelInterface) gin.HandlerFunc
 			return
 		}
 
+		err = processResults(&Context{
+			API:     api,
+			Context: c,
+			DB:      db,
+		}, m, readM)
+		if err != nil {
+			c.JSON(500, gin.H{
+				"error": err.Error(),
+			})
+			return
+		}
+
 		c.JSON(200, readM)
 	}
 }
@@ -384,7 +384,7 @@ func (api *API) collectionPutHandler(m mgocrud.ModelInterface) gin.HandlerFunc {
 				orgMStruct.Field(i).Set(field)
 			}
 		}
-		spew.Dump(changes)
+		//spew.Dump(changes)
 		//spew.Dump(orgM)
 
 		if err := validateObject(&Context{
@@ -427,6 +427,18 @@ func (api *API) collectionPutHandler(m mgocrud.ModelInterface) gin.HandlerFunc {
 			return
 		}
 
+		err = processResults(&Context{
+			API:     api,
+			Context: c,
+			DB:      db,
+		}, m, newM)
+		if err != nil {
+			c.JSON(500, gin.H{
+				"error": err.Error(),
+			})
+			return
+		}
+
 		c.JSON(200, newM)
 
 	}
diff --git a/hooks.go b/hooks.go
index 581400f..38c29ab 100644
--- a/hooks.go
+++ b/hooks.go
@@ -83,3 +83,14 @@ func savedObject(c *Context, m mgocrud.ModelInterface, changes bson.M) error {
 
 	return nil
 }
+
+func processResults(c *Context, m mgocrud.ModelInterface, results interface{}) error {
+	if i, ok := m.(interface {
+		ProcessResults(*Context, interface{}) error
+	}); ok {
+		// custom select manipulation
+		err := i.ProcessResults(c, results)
+		return err
+	}
+	return nil
+}