youngee_b_api/service/selection.go

package service

import (
	"context"
	"encoding/json"
	"errors"
	"fmt"
	"reflect"
	"time"
	"youngee_m_api/db"
	"youngee_m_api/model/common_model"
	"youngee_m_api/model/gorm_model"
	"youngee_m_api/model/http_model"
	"youngee_m_api/pack"
	"youngee_m_api/util"

	"github.com/gin-gonic/gin"
	"github.com/sirupsen/logrus"

	"github.com/caixw/lib.go/conv"
)

var Selection *selection

type selection struct {
}

func (*selection) Create(ctx context.Context, request http_model.CreateSelectionRequest) (*http_model.CreateSelectionData, error) {
	enterpriseId := request.EnterpriseId
	// 1. 检查该企业id和商品id有无选品
	//selectionInfo, err := db.GetSelectionByEnterpiseIdAndProductId(ctx, enterpriseId, conv.MustInt(request.ProductId, 0))
	//if err != nil {
	//	return nil, err
	//}
	//if selectionInfo != nil {
	//	return nil, errors.New("该商品下选品已存在")
	//}

	// 2. 数据准备
	// a) 生成选品id
	selectionId := util.GetSelectionID()
	// b) 查找关联商品信息
	product, err := db.GetProductByID(ctx, conv.MustInt64(request.ProductId, 0))
	if err != nil {
		return nil, err
	}
	productPhotos, err := db.GetProductPhotoByProductID(ctx, conv.MustInt64(request.ProductId, 0))
	productInfoToJson, _ := json.Marshal(product)
	productPhotosToJson, _ := json.Marshal(productPhotos)
	// c) 选品名称
	selectionName := product.BrandName + "-" + product.ProductName

	// 3. 创建选品
	//taskDdl := time.Time{} //赋零值
	//taskDdl, _ = time.ParseInLocation("2006-01-02 15:04:05", "2026-01-01 08:00:00", time.Local)
	t := time.Now()
	newSelection := gorm_model.YounggeeSelectionInfo{
		SelectionStatus:  1,
		SelectionID:      selectionId,
		SelectionName:    selectionName,
		ProductID:        conv.MustInt(request.ProductId, 0),
		EnterpriseID:     enterpriseId,
		Platform:         conv.MustInt(request.Platform, 0),
		ProductSnap:      string(productInfoToJson),
		ProductPhotoSnap: string(productPhotosToJson),
		CreatedAt:        &t,
		UpdatedAt:        &t,
		CommissionRate:   "0",
		EstimatedCost:    "0",
		TaskReward:       "0",
		SettlementAmount: "0",
	}
	//Selection := gorm_model.YounggeeSelectionInfo{}
	err = db.CreateSelection(ctx, newSelection)
	if err != nil {
		return nil, err
	}

	res := &http_model.CreateSelectionData{
		SelectionId: selectionId,
	}
	return res, nil
}

func (*selection) Update(ctx context.Context, request http_model.UpdateSelectionRequest, enterpriseId string) (*http_model.UpdateSelectionData, error) {
	// 1. 检查该企业id和商品id有无选品
	// println("awdawdawdawdwad", request.EnterpriseId)
	fmt.Println("更新的选品信息")

	selectionInfo, err := db.GetSelectionById(ctx, request.SelectionID)
	if err != nil {
		return nil, err
	}
	if selectionInfo == nil {
		return nil, errors.New("选品不存在")
	}

	// 2. 数据准备
	// a) 查找关联商品信息
	product, err := db.GetProductByID(ctx, conv.MustInt64(request.ProductId, 0))
	if err != nil {
		return nil, err
	}
	productPhotos, err := db.GetProductPhotoByProductID(ctx, conv.MustInt64(request.ProductId, 0))
	productInfoToJson, _ := json.Marshal(product)
	productPhotosToJson, _ := json.Marshal(productPhotos)
	// b) 选品名称
	selectionName := product.BrandName + "-" + product.ProductName
	// d) 任务截止时间
	fmt.Println("taskDdl:", request.TaskDdl)
	taskDdl := time.Time{} //赋零值
	taskDdl, _ = time.ParseInLocation("2006-01-02 15:04:05", request.TaskDdl, time.Local)
	// f) 更新选品状态
	if request.SelectionStatus != 2 && request.SelectionStatus != 7 {
		request.SelectionStatus = 1
	}
	t := time.Now()
	println("taskDDl: ", &taskDdl)
	updateSelection := gorm_model.YounggeeSelectionInfo{
		SelectionID:      request.SelectionID,
		SelectionStatus:  request.SelectionStatus,
		SelectionName:    selectionName,
		EnterpriseID:     enterpriseId,
		ProductID:        conv.MustInt(request.ProductId, 0),
		ContentType:      conv.MustInt(request.ContentType, 0),
		TaskMode:         conv.MustInt(request.TaskMode, 0),
		Platform:         conv.MustInt(request.Platform, 0),
		SampleMode:       conv.MustInt(request.SampleMode, 0),
		ProductUrl:       request.ProductUrl,
		SampleNum:        conv.MustInt(request.SampleNum, 0),
		RemainNum:        conv.MustInt(request.SampleNum, 0),
		CommissionRate:   conv.MustString(request.CommissionRate, "0"),
		TaskReward:       conv.MustString(request.TaskReward, "0"),
		SettlementAmount: conv.MustString(request.SettlementAmount, "0"),
		EstimatedCost:    selectionInfo.EstimatedCost,
		SampleCondition:  request.SampleCondition,
		RewardCondition:  request.RewardCondition,
		TaskDdl:          &taskDdl,
		Detail:           request.Detail,
		ProductSnap:      string(productInfoToJson),
		ProductPhotoSnap: string(productPhotosToJson),
		CreatedAt:        selectionInfo.CreatedAt,
		UpdatedAt:        &t,
	}
	if request.SelectionStatus == 2 {
		updateSelection.SubmitAt = &t
	}
	if request.Status == 1 {
		updateSelection.Status = 1
	}
	// 合并传入参数和数据表中原记录，若传入参数字段值为空，则将字段赋值为原记录中值
	result := util.MergeStructValue(&updateSelection, selectionInfo)
	// 利用反射机制将interface类型转换为结构体类型
	v := reflect.ValueOf(&result).Elem()
	if v.Kind() == reflect.Struct {
		updateSelection = v.Interface().(gorm_model.YounggeeSelectionInfo)
		//fmt.Println(p)
	}
	// c) 计算预估成本（如果有）
	/*
		var estimatedCost float64
		if conv.MustInt(updateSelection.TaskMode, 0) == 1 {
			estimatedCost = conv.MustFloat64(updateSelection.TaskReward, 0) * conv.MustFloat64(updateSelection.SampleNum, 0)
		}
		estimatedCostToString, _ := conv.String(estimatedCost)
		updateSelection.EstimatedCost = estimatedCostToString
	*/

	// 3. 更新选品
	err = db.UpdateSelection(ctx, updateSelection)
	if err != nil {
		return nil, err
	}

	println("brief")

	// 4. 更新选品brief和示例
	if request.SecBrief != nil {
		// 删除已有brief
		err = db.DeleteSecBriefBySelectionId(ctx, selectionInfo.SelectionID)
		if err != nil {
			return nil, err
		}
		// 插入新的brief
		for _, v := range request.SecBrief {
			brief := gorm_model.YounggeeSecBrief{
				SelectionID: selectionInfo.SelectionID,
				FileUid:     v.PhotoUid,
				FileName:    v.Name,
				FileUrl:     v.PhotoUrl,
				CreatedAt:   time.Now(),
			}
			err = db.CreateSecBrief(ctx, brief)
			if err != nil {
				return nil, err
			}
		}
	}

	if request.SecExample != nil {
		// 删除已有示例
		err = db.DeleteSecExampleBySelectionId(ctx, selectionInfo.SelectionID)
		if err != nil {
			return nil, err
		}
		// 插入新的示例
		for _, v := range request.SecExample {
			Example := gorm_model.YounggeeSecExample{
				SelectionID: selectionInfo.SelectionID,
				FileUid:     v.PhotoUid,
				FileName:    v.Name,
				FileUrl:     v.PhotoUrl,
				CreatedAt:   time.Now(),
			}
			err = db.CreateSecExample(ctx, Example)
			if err != nil {
				return nil, err
			}
		}
	}

	println("更新带货任务的免费领样策略")
	fmt.Println("SampleMode: ", request.SampleMode)
	// 更新带货任务的免费领样策略
	if request.FreeStrategys != nil {
		// 1. 删除已有的免费领样策略
		err = db.DeleteFreeStrategyBySelectionId(ctx, request.SelectionID)
		if err != nil {
			return nil, err
		}
		// 2. 接收并创建新的免费领样策略
		if request.SampleMode == 1 {
			var Frees []gorm_model.FreeStrategy
			for _, v := range request.FreeStrategys {
				Free := gorm_model.FreeStrategy{
					SelectionId:    selectionInfo.SelectionID,
					StrategyId:     v.StrategyId,
					FansNum:        v.FansNum,
					SaleNum:        v.SaleNum,
					StrategyStatus: 1,
					EnrollNum:      0,
					ChooseNum:      0,
				}
				Frees = append(Frees, Free)
			}
			err = db.CreateFreeStrategy(ctx, Frees)
			if err != nil {
				return nil, err
			}
		}
	}

	println("更新带货任务的悬赏策略")
	fmt.Println("TaskMode: ", request.TaskMode)
	// 更新带货任务的悬赏策略
	if request.RewardStrategys != nil {
		// 1. 删除已有的悬赏策略
		err = db.DeleteRewardStrategyBySelectionId(ctx, request.SelectionID)
		if err != nil {
			return nil, err
		}
		if request.TaskMode == 1 {
			var Rewards []gorm_model.RewardStrategy
			for _, v := range request.RewardStrategys {
				Reward := gorm_model.RewardStrategy{
					SelectionId:    selectionInfo.SelectionID,
					Reward:         v.Reward,
					SaleActual:     v.SaleActual,
					PerReward:      v.PerReward,
					StrategyStatus: 1,
				}
				Rewards = append(Rewards, Reward)
			}
			err = db.CreateRewardStrategy(ctx, Rewards)
			if err != nil {
				return nil, err
			}
		}
	}

	res := &http_model.UpdateSelectionData{
		SelectionId: updateSelection.SelectionID,
	}
	return res, nil
}

func (*selection) Pay(ctx context.Context, request http_model.PaySelectionRequest, enterpriseId string) (*http_model.PaySelectionData, error) {
	// 校验
	// 1. 账户余额是否足够
	enterprise, err := db.GetEnterpriseByEnterpriseID(ctx, enterpriseId)
	if err != nil {
		return nil, err
	}
	if enterprise.AvailableBalance < request.PayMoney {
		return nil, errors.New("账户余额不足")
	}
	// 2. 选品项目状态是否正确
	selectionInfo, err := db.GetSelectionById(ctx, request.SelectionId)
	if err != nil {
		return nil, err
	}
	if selectionInfo == nil {
		return nil, errors.New("选品不存在")
	}
	if selectionInfo.SelectionStatus != 4 {
		return nil, errors.New("选品状态有误")
	}

	// 支付
	err = db.PaySelection(ctx, enterpriseId, request.PayMoney, request.SelectionId)
	if err != nil {
		return nil, err
	}

	res := &http_model.PaySelectionData{
		SelectionId: request.SelectionId,
	}
	return res, nil
}

func (*selection) Submit(ctx context.Context, request http_model.SubmitSelectionRequest) (*http_model.SubmitSelectionData, error) {
	t := time.Now()
	updateSelection := gorm_model.YounggeeSelectionInfo{
		SelectionID:     request.SelectionId,
		SelectionStatus: request.SelectionStatus,
		SubmitAt:        &t,
	}

	err := db.UpdateSelection(ctx, updateSelection)
	if err != nil {
		return nil, err
	}

	res := &http_model.SubmitSelectionData{}
	return res, nil
}

func (s *selection) GetAllSelection(ctx context.Context, enterpriseID string, pageSize, pageNum int64, conditions *common_model.SelectionConditions) (*http_model.SelectionData, error) {
	SelectionList, total, err := db.GetSelectionList(ctx, enterpriseID, pageSize, pageNum, conditions)
	if err != nil {
		logrus.WithContext(ctx).Errorf("[selectionDB service] call GetAllSelection error,err:%+v", err)
		return nil, err
	}
	SelectionListData := new(http_model.SelectionData)
	SelectionListData.SelectionInfo = pack.MGormSelectionToHttpSelectionPreview(SelectionList)
	SelectionListData.Total = conv.MustString(total, "")
	return SelectionListData, nil
}

func (s *selection) GetSelectionDetail(ctx *gin.Context, selectionId string, enterpriseId string) (*http_model.SelectionDetail, error) {
	selectionDetail := http_model.SelectionDetail{}
	selectionInfo, err := db.GetSelectionById(ctx, selectionId)
	if err != nil {
		logrus.WithContext(ctx).Errorf("[selectionDB service] call GetSelectionInfo error,err:%+v", err)
		return nil, err
	}
	selectionBriefInfo, err := db.GetSelectionBriefInfo(ctx, selectionId)
	if err != nil {
		logrus.WithContext(ctx).Errorf("[selectionDB service] call GetSelectionBriefInfo error,err:%+v", err)
		return nil, err
	}
	selectionExampleInfo, err := db.GetSelectionExampleInfo(ctx, selectionId)
	if err != nil {
		logrus.WithContext(ctx).Errorf("[selectionDB service] call GetSelectionExampleInfo error,err:%+v", err)
		return nil, err
	}
	productInfo, err := db.GetProductInfoBySelectionId(ctx, selectionId)
	if err != nil {
		logrus.WithContext(ctx).Errorf("[selectionDB service] call GetProductInfo error,err:%+v", err)
		return nil, err
	}
	productPhotoInfo, err := db.GetProductPhotoInfoBySelectionId(ctx, selectionId)
	if err != nil {
		logrus.WithContext(ctx).Errorf("[selectionDB service] call GetProductPhotoInfo error,err:%+v", err)
		return nil, err
	}
	// 查找免费领样策略
	freeStrategy, err := db.GetFreeStrategyBySelectionId(ctx, selectionId)
	if err != nil {
		logrus.WithContext(ctx).Errorf("[selectionDB service] call GetFreeStrategy error,err:%+v", err)
		return nil, err
	}
	// 查找悬赏策略
	rewardStrategy, err := db.GetRewardStrategyBySelectionId(ctx, selectionId)
	if err != nil {
		logrus.WithContext(ctx).Errorf("[selectionDB service] call GetRewardStrategy error,err:%+v", err)
		return nil, err
	}

	selectionDetail.SelectionBrief = selectionBriefInfo
	selectionDetail.SelectionInfo = selectionInfo
	selectionDetail.SelectionExample = selectionExampleInfo
	selectionDetail.ProductInfo = productInfo
	selectionDetail.ProductPhotoInfo = productPhotoInfo
	selectionDetail.FreeStrategy = freeStrategy
	selectionDetail.RewardStrategy = rewardStrategy
	return &selectionDetail, nil
}

func (*selection) Review(ctx context.Context, request http_model.ReviewSelectionRequest) (*http_model.ReviewSelectionData, error) {
	// 根据选品id查询选品信息
	selectionInfo, err := db.GetSelectionById(ctx, request.SelectionId)
	if err != nil {
		logrus.WithContext(ctx).Errorf("[selectionDB service] call GetSelectionById error,err:%+v", err)
		return nil, err
	}

	// 计算预估成本
	var estimatedCost float64 = 0.0
	var estimatedCostToString string = ""

	selection1, _ := db.GetRewardStrategyBySelectionId(ctx, request.SelectionId)

	if conv.MustInt(selectionInfo.TaskMode, 0) == 1 {
		// estimatedCost = conv.MustFloat64(selectionInfo.TaskReward, 0) * conv.MustFloat64(selectionInfo.SampleNum, 0)
		estimatedCost = conv.MustFloat64(selection1[0].Reward, 0)
		estimatedCostToString, _ = conv.String(estimatedCost)
	}

	// 若审核通过则更新选品阶段为待支付，否则更新为失效并赋值失效原因
	t := time.Now()
	AutoTaskID, err := db.GetLastAutoTaskID()
	updateSelection := gorm_model.YounggeeSelectionInfo{}
	if request.IsPass == 1 {
		if conv.MustInt(selectionInfo.TaskMode, 0) == 1 {
			//悬赏任务
			updateSelection = gorm_model.YounggeeSelectionInfo{
				SelectionID:     request.SelectionId,
				SelectionStatus: 4,
				PassAt:          &t,
				EstimatedCost:   estimatedCostToString,
				AutoTaskID:      int(AutoTaskID),
			}
		} else {
			// 纯佣带货
			updateSelection = gorm_model.YounggeeSelectionInfo{
				SelectionID:     request.SelectionId,
				SelectionStatus: 6,
				PassAt:          &t,
				EstimatedCost:   estimatedCostToString,
			}
		}
	} else {
		updateSelection = gorm_model.YounggeeSelectionInfo{
			SelectionID:     request.SelectionId,
			SelectionStatus: 7,
			FailReason:      2,
			FinishAt:        &t,
			EstimatedCost:   estimatedCostToString,
		}
	}
	err = db.UpdateSelection(ctx, updateSelection)
	if err != nil {
		logrus.WithContext(ctx).Errorf("[selectionDB service] call UpdateSelection error,err:%+v", err)
		return nil, err
	}
	res := &http_model.ReviewSelectionData{}
	return res, nil
}

func (*selection) End(ctx context.Context, request http_model.EndSelectionRequest) (*http_model.EndSelectionData, error) {
	selectionInfo, err := db.GetSelectionById(ctx, request.SelectionId)
	if err != nil {
		logrus.WithContext(ctx).Errorf("[selectionDB service] call GetSelectionById error,err:%+v", err)
		return nil, err
	}
	selectionstatus := conv.MustInt(selectionInfo.SelectionStatus, 0)
	sampleMode := conv.MustInt(selectionInfo.SampleMode, 0)
	beforedeliverynum := conv.MustInt(selectionInfo.BeforeDeliveryNum, 0)
	t := time.Now()
	updateSelection := gorm_model.YounggeeSelectionInfo{}
	if selectionstatus == 4 || (selectionstatus == 6 && sampleMode == 1 && beforedeliverynum == 0) || (selectionstatus == 6 && sampleMode == 2) {
		updateSelection = gorm_model.YounggeeSelectionInfo{
			SelectionID:     request.SelectionId,
			SelectionStatus: 8,
			FinishAt:        &t,
		}
	}
	err = db.UpdateSelection(ctx, updateSelection)
	if err != nil {
		logrus.WithContext(ctx).Errorf("[selectionDB service] call UpdateSelection error,err:%+v", err)
		return nil, err
	}
	res := &http_model.EndSelectionData{}
	return res, nil

}