3dshow-customer/3dshow/api/product.go

package api

import (
	"3dshow/db/model"
	"3dshow/db/repo"
	"3dshow/log"
	"errors"
	"fmt"
	"time"

	"github.com/gin-gonic/gin"
	"go.mongodb.org/mongo-driver/bson/primitive"
)

// 产品管理
func Product(r *GinRouter) {
	r.POSTJWT("/product/create", ProductAdd)
	r.GETJWT("/product/list", ProductList)
	r.POSTJWT("/product/update", ProductUpdate)
	r.GETJWT("/product/detail/:id", ProductDetail)
	r.GET("/product/share/detail/:id", ProductDetail)
	r.POSTJWT("/product/delete/:id", ProductDelete)
}

// 新增产品
func ProductAdd(c *gin.Context, apictx *ApiSession) (interface{}, error) {
	var form model.Product
	err := c.ShouldBindJSON(&form)
	if err != nil {
		fmt.Println(err)
		return nil, errors.New("参数错误！")
	}
	ctx := apictx.CreateRepoCtx()
	if form.SupplyId.Hex() == "" {
		return nil, errors.New("供应链id不能为空")
	}
	if form.Name == "" {
		return nil, errors.New("产品名不能为空")
	}
	form.CreateTime = time.Now()
	// 状态默认为下架
	if form.Status == 0 {
		form.Status = -1
	}

	result, err := repo.RepoAddDoc(ctx, repo.CollectionProduct, &form)
	return result, err

}

// 产品列表
// ??? 上架才能展示 前端传 status:1
func ProductList(c *gin.Context, apictx *ApiSession) (interface{}, error) {
	page, size, query := UtilQueryPageSize(c)
	// asc:升序1  desc:降序-1
	// 默认排序
	onsaleTimeSort := repo.Map{"onsaleTime": -1}
	if _, ok := query["sort"]; ok {
		if query["sort"].(string) == "asc" {
			onsaleTimeSort = repo.Map{"onsaleTime": 1}
		}
		// if sort.(string) == "desc"{
		// 	onsaleTimeSort = repo.Map{"onsaleTime": -1}
		// }

	}
	delete(query, "sort")
	// 查询数据
	if _, ok := query["supplyId"]; !ok {
		return nil, errors.New("供应链id不能为空")
	}
	supplyId, err := primitive.ObjectIDFromHex(query["supplyId"].(string))
	if err != nil {
		return nil, errors.New("供应链id错误")
	}
	query["supplyId"] = supplyId

	option := &repo.PageSearchOptions{
		CollectName: repo.CollectionProduct,
		Page:        page,
		Size:        size,
		Query:       query,
		// Project:     []string{},
		Sort: onsaleTimeSort,
	}
	pageResult, err := repo.RepoPageSearch(apictx.CreateRepoCtx(), option)
	if err != nil {
		return nil, err
	}
	// 查询收藏状态
	var collects []*model.Collect
	_userId := apictx.User.ID
	userId, err := primitive.ObjectIDFromHex(_userId)
	if err != nil {
		return nil, errors.New("非法用户")
	}
	option1 := &repo.PageSearchOptions{
		CollectName: repo.CollectionCollect,
		Query:       repo.Map{"userId": userId, "supplyId": supplyId},
		Project:     []string{"_id", "productId"},
	}
	err1 := repo.RepoDocsSearch(apictx.CreateRepoCtx(), option1, &collects)

	if len(pageResult.List) > 0 {
		for _, v := range pageResult.List {
			v["isCollect"] = false
			v["collectId"] = ""
			if len(collects) > 0 && err1 == nil {
				for _, col := range collects {
					if v["_id"].(primitive.ObjectID) == col.ProductId { // productId唯一
						v["isCollect"] = true
						v["collectId"] = col.Id.Hex()
						break
					}
				}
			}
		}
	}
	return pageResult, err
}

// 更新产品/编辑、下架
func ProductUpdate(c *gin.Context, apictx *ApiSession) (interface{}, error) {
	var form model.Product
	err := c.ShouldBindJSON(&form)
	if err != nil {
		return nil, errors.New("参数错误")
	}
	if form.Id.Hex() == "" {
		return nil, errors.New("更新的产品id不能为空")
	}
	form.UpdateTime = time.Now()
	return repo.RepoUpdateSetDoc(apictx.CreateRepoCtx(), repo.CollectionProduct, form.Id.Hex(), &form)
}

// 产品信息
type ProudctDetailRes struct {
	model.Product
	IsCollect bool   `json:"isCollect"`
	CollectId string `json:"collectId"`
}

func ProductDetail(c *gin.Context, apictx *ApiSession) (interface{}, error) {
	productId := c.Param("id")
	id, err := primitive.ObjectIDFromHex(productId)
	if err != nil {
		return nil, errors.New("非法id")
	}
	var product model.Product
	option := &repo.DocSearchOptions{
		CollectName: repo.CollectionProduct,
		Query:       repo.Map{"_id": id},
	}

	found, err := repo.RepoSeachDoc(apictx.CreateRepoCtx(), option, &product)
	if !found || err != nil {
		log.Info(err)
		return nil, errors.New("数据未找到")
	}
	// 是否收藏
	if apictx.User != nil {
		_userId := apictx.User.ID
		userId, err := primitive.ObjectIDFromHex(_userId)
		if err != nil {
			return nil, errors.New("非法用户")
		}
		var collect model.Collect
		found, _ = repo.RepoSeachDoc(apictx.CreateRepoCtx(), &repo.DocSearchOptions{
			CollectName: repo.CollectionCollect,
			Query:       repo.Map{"userId": userId, "productId": id},
		}, &collect)
		return ProudctDetailRes{Product: product, IsCollect: found, CollectId: collect.Id.Hex()}, nil
	}

	return product, nil
}

// 删除产品
// ???权限控制 admin 和 供应商本人
func ProductDelete(c *gin.Context, apictx *ApiSession) (interface{}, error) {
	productId := c.Param("id")
	_, err := primitive.ObjectIDFromHex(productId)
	if err != nil {
		return nil, errors.New("非法id")
	}

	return repo.RepoDeleteDoc(apictx.CreateRepoCtx(), repo.CollectionProduct, productId)
}