mw_service_register/api/panel/category.go

package panel

import (
	"errors"
	"github.com/gin-gonic/gin"
	"github.com/han-joker/moo-layout/api/moo/dbm"
	"github.com/han-joker/moo-layout/api/moo/logm"
	"github.com/han-joker/moo-layout/api/panel/models"
	"github.com/han-joker/moo-layout/api/tables"
	"net/http"
	"strconv"
)

func CategoryPost(c *gin.Context) {
	req := models.CategoryReq{}
	if err := c.ShouldBind(&req); err != nil {
		c.JSON(200, gin.H{
			"error": err.Error(),
		})
		return
	}

	row := &tables.Category{
		Name:    req.Name,
		Description: req.Description,
		ParentID: req.ParentID,
		Sorter: req.Sorter,
		Status: req.Status,
	}

	if err := dbm.Get().Create(&row).Error; err != nil {
		c.JSON(200, gin.H{
			"error": err.Error(),
		})
		return
	}

	c.JSON(200, gin.H{
		"error":  nil,
		"row": row,
	})
}

func CategoryPut(c *gin.Context) {
	id, err := strconv.Atoi(c.Query("id"))
	if err != nil {
		logm.Get().Info(err.Error())
		c.JSON(http.StatusOK, gin.H{
			"error": err.Error(),
		})
		return
	}

	row := tables.Category{}
	if err := dbm.Get().Where("id=?", id).First(&row).Error; err != nil {
		logm.Get().Info(err.Error())
		c.JSON(http.StatusOK, gin.H{
			"error": err.Error(),
		})
		return
	}

	req := models.CategoryReq{}
	if err := c.ShouldBind(&req); err != nil {
		logm.Get().Info(err.Error())
		c.JSON(http.StatusOK, gin.H{
			"error": err.Error(),
		})
		return
	}
	row.Name = req.Name
	row.Description = req.Description
	row.Status = req.Status
	row.Sorter = req.Sorter
	row.ParentID = req.ParentID

	if err := dbm.Get().Save(&row).Error; err != nil {
		logm.Get().Info(err.Error())
		c.JSON(http.StatusOK, gin.H{
			"error": err.Error(),
		})
		return
	}

	c.JSON(http.StatusOK, gin.H{
		"error":  nil,
		"row": row,
	})
}

func CategoryGets(c *gin.Context) {
	query := models.CategoryQuery{}
	if err := c.ShouldBind(&query); err != nil {
	}

	rows := []tables.Category{}
	if err := dbm.Get().Order("sorter asc").Find(&rows).Error; err != nil {
		logm.Get().Info(err.Error())
		c.JSON(http.StatusOK, gin.H{
			"error": err.Error(),
		})
		return
	}

	nested := make([]models.CategoryNested, 0)
	for _, row := range rows {
		if row.ID == query.Filters.ID {
			nested = append(nested, models.CategoryNested{
				Category: row,
			})
			break
		}
	}
	if len(nested) == 0 {
		nested = categoryChildren(rows, query.Filters.ID)
	} else if len(nested) == 1 {
		nested[0].Children = categoryChildren(rows, query.Filters.ID)
	}

	c.JSON(http.StatusOK, gin.H{
		"error": nil,
		"rows":  nested,
	})
}

func categoryChildren(rows []tables.Category, parentID uint) []models.CategoryNested {
	children := make([]models.CategoryNested, 0)

	for _, row := range rows {
		if row.ParentID == parentID {
			item := models.CategoryNested{
				Category: row,
			}
			item.Children = categoryChildren(rows, row.ID)
			children = append(children, item)
		}
	}

	return children
}

func CategoryGet(c *gin.Context) {

	id, err := strconv.Atoi(c.Query("id"))
	if err != nil {
		logm.Get().Info(err.Error())
		c.JSON(http.StatusOK, gin.H{
			"error":err.Error(),
		})
		return
	}

	row := tables.Category{}
	if err := dbm.Get().Where("id=?", id).First(&row).Error; err != nil {
		logm.Get().Info(err.Error())
		c.JSON(http.StatusOK, gin.H{
			"error": err.Error(),
		})
		return
	}

	c.JSON(http.StatusOK, gin.H{
		"error":  nil,
		"row": row,
	})
}

func CategoryDelete(c *gin.Context) {
	id, err := strconv.Atoi(c.Query("id"))
	if err != nil {
		logm.Get().Info(err.Error())
		c.JSON(http.StatusOK, gin.H{
			"error": err.Error(),
		})
		return
	}

	row := tables.Category{}
	if err := dbm.Get().Where("id=?", id).First(&row).Error; err != nil {
		logm.Get().Info(err.Error())
		c.JSON(http.StatusOK, gin.H{
			"error": err.Error(),
		})
		return
	}

	childrenSize := int64(0)
	if err := dbm.Get().Model(&tables.Category{}).Where("parent_id=?", row.ID).Count(&childrenSize).Error; err != nil {
		logm.Get().Info(err.Error())
		c.JSON(http.StatusOK, gin.H{
			"error": err.Error(),
		})
		return
	}
	if childrenSize > 0 {
		err := errors.New("has children")
		logm.Get().Info(err.Error())
		c.JSON(http.StatusOK, gin.H{
			"error": err.Error(),
		})
		return
	}

	if err := dbm.Get().Delete(&row).Error; err != nil {
		logm.Get().Info(err.Error())
		c.JSON(http.StatusOK, gin.H{
			"error": err.Error(),
		})
		return
	}

	c.JSON(http.StatusOK, gin.H{
		"error":   nil,
		"row": row,
	})
	return

}

func CategoryCount(c *gin.Context) {

	query := models.CategoryQuery{}
	if err := c.ShouldBind(&query); err != nil {
	}



	rows := []tables.Category{}
	if err := dbm.Get().Find(&rows).Error; err != nil {
		logm.Get().Info(err.Error())
		c.JSON(http.StatusOK, gin.H{
			"error": err.Error(),
		})
		return
	}
	count := categoryCount(rows, query)

	c.JSON(http.StatusOK, gin.H{
		"error":   nil,
		"count": count,
	})
}

func categoryCount(rows []tables.Category, query models.CategoryQuery) int64 {
	count := int64(0)

	for _, row := range rows {
		statusCond := true
		if query.Filters.Status > 0 {
			statusCond = row.Status == query.Filters.Status
		}

		if row.ParentID == query.Filters.ParentID && statusCond {
			count += 1
			count += categoryCount(rows, models.CategoryQuery{
				Filters: models.CategoryQueryFilters{
					ParentID: row.ID,
					Status:   query.Filters.Status,
				},
			})
		}
	}

	return count
}