box-cost/boxcost/api/plan.go

package api

import (
	"box-cost/db/model"
	"box-cost/db/repo"
	"box-cost/log"
	"errors"
	"fmt"
	"time"

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

// 生产计划管理
func ProductPlan(r *GinRouter) {

	// 创建生产计划
	r.POST("/plan/create", CreateProductPlan)

	// 获取生产计划详情
	r.GET("/plan/detail/:id", GetProductPlan)

	// 获取生产计划列表
	r.GET("/plan/list", GetProductPlans)

	// 更新生产计划
	r.POST("/plan/update", UpdateProductPlan)

	// 删除生产计划
	r.POST("/plan/delete/:id", DelProductPlan)

	// 下载部件打印单
	// r.GET("/bill/plan/download", DownLoadPlan)

	// 生产成本表
	// r.GET("/plan/cost/download", DownLoadPlanCost)

}

type SupplierPlanCost struct {
	*model.ProductPlan
	SupplierId string
}

// func DownLoadPlanCost(c *gin.Context, apictx *ApiSession) (interface{}, error) {
// 	_planId := c.Query("id")
// 	supplierId := c.Query("supplierId")
// 	planId, err := primitive.ObjectIDFromHex(_planId)
// 	if err != nil {
// 		return nil, errors.New("planId错误")
// 	}
// 	supplierPlanCost := &SupplierPlanCost{}

// 	plan := model.ProductPlan{}
// 	found, err := repo.RepoSeachDoc(apictx.CreateRepoCtx(), &repo.DocSearchOptions{
// 		CollectName: repo.CollectionProductPlan,
// 		Query:       repo.Map{"_id": planId},
// 	}, &plan)
// 	if !found || err != nil {
// 		return nil, errors.New("数据未找到")
// 	}
// 	supplierPlanCost.ProductPlan = &plan
// 	supplierPlanCost.SupplierId = supplierId

// 	f := excelize.NewFile()
// 	index := f.NewSheet("Sheet1")
// 	f.SetActiveSheet(index)
// 	f.SetDefaultFont("宋体")
// 	planCostExcel := NewPlanCostExcel(f)
// 	planCostExcel.Title = fmt.Sprintf("生产成本表(%s)%d盒", plan.Name, plan.Total)
// 	planCostExcel.Content = supplierPlanCost
// 	planCostExcel.Draws()

// 	c.Header("Content-Type", "application/octet-stream")
// 	c.Header("Content-Disposition", "attachment; filename="+"planCost.xlsx")
// 	c.Header("Content-Transfer-Encoding", "binary")

// 	err = f.Write(c.Writer)
// 	if err != nil {
// 		return nil, err
// 	}
// 	return nil, nil

// }

// func DownLoadPlan(c *gin.Context, apictx *ApiSession) (interface{}, error) {
// 	_planId := c.Query("id")
// 	compId := c.Query("compId")
// 	planId, err := primitive.ObjectIDFromHex(_planId)
// 	if err != nil {
// 		return nil, errors.New("planId错误")
// 	}
// 	plan := model.ProductPlan{}
// 	found, err := repo.RepoSeachDoc(apictx.CreateRepoCtx(), &repo.DocSearchOptions{
// 		CollectName: repo.CollectionProductPlan,
// 		Query:       repo.Map{"_id": planId},
// 	}, &plan)
// 	if !found || err != nil {
// 		return nil, errors.New("数据未找到")
// 	}
// 	// 获取部件单据
// 	curComp := &model.PackComponent{}
// 	for _, comp := range plan.Pack.Components {
// 		if comp.Id == compId {
// 			curComp = comp
// 		}
// 	}
// 	if curComp.Id == "" {
// 		return nil, errors.New("该组件不存在")
// 	}

// 	// 获取bill
// 	if len(curComp.Stages) == 0 {
// 		return nil, errors.New("该组件数据不存在")
// 	}
// 	f := excelize.NewFile()
// 	index := f.NewSheet("Sheet1")
// 	f.SetActiveSheet(index)
// 	f.SetDefaultFont("宋体")
// 	companyName := getCompanyName(apictx)

// 	// 获取不同类型的单据id
// 	set := make(map[int]map[primitive.ObjectID]struct{})
// 	for _, stage := range curComp.Stages {
// 		_billId := stage.BillId
// 		billId, _ := primitive.ObjectIDFromHex(_billId)
// 		if !billId.IsZero() {
// 			set[stage.Type] = map[primitive.ObjectID]struct{}{billId: struct{}{}}
// 		}
// 	}

// 	// row := 0
// 	// for k, bill := range set {
// 	// 	for billId, _ := range bill {
// 	// 		var billExcel IExcel
// 	// 		collectName := ""
// 	// 		if k == 1 {
// 	// 			collectName = repo.CollectionBillPurchase
// 	// 			billExcel = NewPurchaseBill(f)
// 	// 		}
// 	// 		if k == 2 {
// 	// 			collectName = repo.CollectionBillProduce
// 	// 			billExcel = NewProduceBill(f)
// 	// 		}
// 	// 		if k == 2 {
// 	// 			collectName = repo.CollectionBillProduct
// 	// 			billExcel = NewProductBill(f)
// 	// 		}
// 	// 		billExcel.SetRow(row)

// 			// purchase := model.PurchaseBill{}
// 			// found, _ := repo.RepoSeachDoc(apictx.CreateRepoCtx(), &repo.DocSearchOptions{
// 			// 	CollectName: repo.CollectionBillPurchase,
// 			// 	Query:       repo.Map{"_id": billId},
// 			// }, &purchase)
// 			// fmt.Println(purchase)
// 			// if found {
// 			// 	var billExcel IPurchBill
// 			// 	if purchase.Process != nil {
// 			// 		billExcel = NewProcessBill(f)
// 			// 	}

// 			// 	if len(purchase.Paper) > 0 {
// 			// 		billExcel = NewPurchaseBill(f)
// 			// 	}
// 			// 	// purchaseExcel := NewPurchaseBill(f)
// 			// 	billExcel.SetContent(&purchase)
// 			// 	billExcel.SetTitle(fmt.Sprintf("%s原材料采购单", companyName))
// 			// 	billExcel.SetOffset(offset)
// 			// 	// billExcel.Title = fmt.Sprintf("%s原材料采购单", companyName)
// 			// 	//设置对应的数据
// 			// 	// purchaseExcel.Offset = offset
// 			// 	billExcel.Draws()

// 			// 	offset += 15
// 			// }
// //
// 		// }

// 		// 采购单
// 		// if k == 1 {

// 		// }
// 		// // 加工单
// 		// if k == 2 {
// 		// 	for billId, _ := range bill {

// 		// 	}

// 		// }
// 		// // 成品采购单
// 		// if k == 3 {
// 		// 	for billId, _ := range bill {

// 		// 	}

// 		// }
// 	}

// 	c.Header("Content-Type", "application/octet-stream")
// 	c.Header("Content-Disposition", "attachment; filename="+"bill.xlsx")
// 	c.Header("Content-Transfer-Encoding", "binary")

// 	err = f.Write(c.Writer)
// 	if err != nil {
// 		return nil, err
// 	}

// 	return nil, nil
// }

// 创建生产计划
func CreateProductPlan(c *gin.Context, apictx *ApiSession) (interface{}, error) {

	var plan model.ProductPlan

	err := c.ShouldBindJSON(&plan)
	if err != nil {
		fmt.Println(err)
		return nil, errors.New("参数错误！")
	}
	ctx := apictx.CreateRepoCtx()

	if plan.Name == "" {
		return nil, errors.New("生产计划名为空")
	}
	if plan.Total == 0 {
		return nil, errors.New("生产计划数应不为0")
	}

	plan.Status = "process" // 进行中
	plan.CreateTime = time.Now()
	plan.UpdateTime = time.Now()

	result, err := repo.RepoAddDoc(ctx, repo.CollectionProductPlan, &plan)
	return result, err
}

// 获取生产计划信息
func GetProductPlan(c *gin.Context, apictx *ApiSession) (interface{}, error) {
	planId := c.Param("id")
	id, err := primitive.ObjectIDFromHex(planId)
	if err != nil {
		return nil, errors.New("非法id")
	}
	var plan model.ProductPlan
	option := &repo.DocSearchOptions{
		CollectName: repo.CollectionProductPlan,
		Query:       repo.Map{"_id": id},
	}

	found, err := repo.RepoSeachDoc(apictx.CreateRepoCtx(), option, &plan)
	if !found || err != nil {
		log.Info(err)
		return nil, errors.New("数据未找到")
	}

	billStates := map[string]string{}
	if plan.Pack != nil && plan.Pack.Components != nil {
		for _, comp := range plan.Pack.Components {
			if comp.Stages != nil {
				for _, stage := range comp.Stages {

					if len(stage.BillId) > 0 {
						collectName := ""
						// 材料
						if stage.Type == 1 {
							collectName = repo.CollectionBillPurchase
						}
						// 工艺
						if stage.Type == 2 {
							collectName = repo.CollectionBillProduce
						}
						// 成品
						if stage.Type == 3 {
							collectName = repo.CollectionBillProduct
						}

						ok, state := repo.RepoSeachDocMap(apictx.CreateRepoCtx(), &repo.DocSearchOptions{CollectName: collectName, Query: repo.Map{"_id": stage.BillId}, Project: []string{"status"}})
						if ok {
							billStates[stage.BillId] = state["status"].(string)
						}
					}

				}
			}
		}
	}

	return map[string]interface{}{
		"plan":       plan,
		"billStates": billStates,
	}, nil
}

// 获取生产计划列表
func GetProductPlans(c *gin.Context, apictx *ApiSession) (interface{}, error) {

	page, size, query := UtilQueryPageSize(c)
	if _packId, ok := query["packId"]; ok {
		packId, _ := primitive.ObjectIDFromHex(_packId.(string))
		query["pack._id"] = packId
		delete(query, "packId")
	}

	option := &repo.PageSearchOptions{
		CollectName: repo.CollectionProductPlan,
		Query:       query,
		Page:        page,
		Size:        size,
		Sort:        bson.M{"createTime": -1},
		Project:     []string{"_id", "thumbnail", "name", "updateTime", "createTime", "createUser", "total", "totalPrice", "status"},
	}
	return repo.RepoPageSearch(apictx.CreateRepoCtx(), option)
}

// 更新生产计划
func UpdateProductPlan(c *gin.Context, apictx *ApiSession) (interface{}, error) {
	var plan model.ProductPlan
	err := c.ShouldBindJSON(&plan)
	if err != nil {
		return nil, errors.New("参数错误")
	}
	if plan.Id.Hex() == "" {
		return nil, errors.New("id的为空")
	}
	plan.UpdateTime = time.Now()
	return repo.RepoUpdateSetDoc(apictx.CreateRepoCtx(), repo.CollectionProductPlan, plan.Id.Hex(), &plan)
}

// 删除生产计划
func DelProductPlan(c *gin.Context, apictx *ApiSession) (interface{}, error) {
	planId := c.Param("id")
	if planId == "" {
		return nil, errors.New("id为空")
	}

	return repo.RepoDeleteDoc(apictx.CreateRepoCtx(), repo.CollectionProductPlan, planId)
}