opms_backend/opms_parent/app/service/contract/ctr_contract.go

package service

import (
	"context"
	"database/sql"
	"fmt"

	basedao "dashoo.cn/micro/app/dao/base"
	dao "dashoo.cn/micro/app/dao/contract"
	custdao "dashoo.cn/micro/app/dao/cust"
	projdao "dashoo.cn/micro/app/dao/proj"
	sysdao "dashoo.cn/micro/app/dao/sys"
	model "dashoo.cn/micro/app/model/contract"
	proj "dashoo.cn/micro/app/model/proj"

	"dashoo.cn/opms_libary/micro_srv"
	"dashoo.cn/opms_libary/myerrors"
	"dashoo.cn/opms_libary/request"
	"github.com/gogf/gf/database/gdb"
	"github.com/gogf/gf/os/gtime"
	"github.com/gogf/gf/util/gvalid"
)

type CtrContractService struct {
	Dao             *dao.CtrContractDao
	ProjBusinessDao *projdao.ProjBusinessDao
	CustomerDao     *custdao.CustCustomerDao
	UserDao         *sysdao.SysUserDao
	CtrProductDao   *dao.CtrContractProductDao
	ProductDao      *basedao.BaseProductDao

	Tenant   string
	userInfo request.UserInfo
}

func NewCtrContractService(ctx context.Context) (*CtrContractService, error) {
	tenant, err := micro_srv.GetTenant(ctx)
	if err != nil {
		return nil, fmt.Errorf("获取组合码异常：%s", err.Error())
	}
	// 获取用户信息
	userInfo, err := micro_srv.GetUserInfo(ctx)
	if err != nil {
		return nil, fmt.Errorf("获取用户信息异常：%s", err.Error())
	}
	return &CtrContractService{
		Dao:             dao.NewCtrContractDao(tenant),
		ProjBusinessDao: projdao.NewProjBusinessDao(tenant),
		CustomerDao:     custdao.NewCustCustomerDao(tenant),
		UserDao:         sysdao.NewSysUserDao(tenant),
		CtrProductDao:   dao.NewCtrContractProductDao(tenant),
		ProductDao:      basedao.NewBaseProductDao(tenant),
		Tenant:          tenant,
		userInfo:        userInfo,
	}, nil
}

func (s CtrContractService) Get(ctx context.Context, id int) (*model.CtrContractGetRsp, error) {
	ent, err := s.Dao.Where("Id = ?", id).One()
	if err != nil {
		return nil, err
	}
	if ent == nil {
		return nil, myerrors.NewMsgError(nil, "合同不存在")
	}
	product, err := s.CtrProductDao.Where("contract_id = ?", id).All()
	if err != nil {
		return nil, err
	}
	if product == nil {
		product = []*model.CtrContractProduct{}
	}
	return &model.CtrContractGetRsp{
		CtrContract: *ent,
		Product:     product,
	}, nil
}

func (s CtrContractService) List(ctx context.Context, req *model.CtrContractListReq) (int, []*model.CtrContract, error) {
	dao := &s.Dao.CtrContractDao
	if req.SearchText != "" {
		likestr := fmt.Sprintf("%%%s%%", req.SearchText)
		dao = dao.Where("(contract_code LIKE ? || contract_name LIKE ? || cust_name LIKE ? || nbo_name LIKE ?)", likestr, likestr, likestr, likestr)
	}
	if req.ContractCode != "" {
		likestr := fmt.Sprintf("%%%s%%", req.ContractCode)
		dao = dao.Where("contract_code like ?", likestr)
	}
	if req.ContractName != "" {
		likestr := fmt.Sprintf("%%%s%%", req.ContractName)
		dao = dao.Where("contract_name like ?", likestr)
	}
	if req.CustId != 0 {
		dao = dao.Where("cust_id = ?", req.CustId)
	}
	if req.CustName != "" {
		likestr := fmt.Sprintf("%%%s%%", req.CustName)
		dao = dao.Where("cust_name like ?", likestr)
	}
	if req.NboId != 0 {
		dao = dao.Where("nbo_id = ?", req.NboId)
	}
	if req.NboName != "" {
		likestr := fmt.Sprintf("%%%s%%", req.NboName)
		dao = dao.Where("nbo_name like ?", likestr)
	}
	if req.ApproStatus != "" {
		dao = dao.Where("appro_status = ?", req.ApproStatus)
	}
	if req.ContractType != "" {
		dao = dao.Where("contract_type = ?", req.ContractType)
	}
	// if req.ContractStartTime != nil {
	// 	dao = dao.Where("contract_start_time > ?", req.ContractStartTime)
	// }
	// if req.ContractEndTime != nil {
	// 	dao = dao.Where("contract_end_time < ?", req.ContractEndTime)
	// }
	if req.InchargeId != 0 {
		dao = dao.Where("incharge_id = ?", req.InchargeId)
	}
	if req.InchargeName != "" {
		likestr := fmt.Sprintf("%%%s%%", req.InchargeName)
		dao = dao.Where("incharge_name like ?", likestr)
	}
	if req.SignatoryId != 0 {
		dao = dao.Where("signatory_id = ?", req.SignatoryId)
	}
	if req.SignatoryName != "" {
		likestr := fmt.Sprintf("%%%s%%", req.SignatoryName)
		dao = dao.Where("signatory_name like ?", likestr)
	}
	if req.DistributorId != 0 {
		dao = dao.Where("distributor_id = ?", req.DistributorId)
	}
	if req.DistributorName != "" {
		likestr := fmt.Sprintf("%%%s%%", req.DistributorName)
		dao = dao.Where("distributor_name like ?", likestr)
	}
	if req.BeginTime != "" {
		dao = dao.Where("created_time > ?", req.BeginTime)
	}
	if req.EndTime != "" {
		dao = dao.Where("created_time < ?", req.EndTime)
	}

	total, err := dao.Count()
	if err != nil {
		return 0, nil, err
	}
	if req.PageNum != 0 {
		dao = dao.Page(req.GetPage())
	}
	orderby := "created_time desc"
	if req.OrderBy != "" {
		orderby = req.OrderBy
	}
	dao = dao.Order(orderby)

	ents := []*model.CtrContract{}
	err = dao.Structs(&ents)
	if err != nil && err != sql.ErrNoRows {
		return 0, nil, err
	}
	return total, ents, err
}

func (s CtrContractService) BindProduct(tx *gdb.TX, id int, product []model.CtrAddProduct) error {
	var amount float64
	for _, p := range product {
		amount += p.TranPrice
	}

	_, err := s.CtrProductDao.Delete("contract_id = ?", id)
	if err != nil {
		return err
	}

	tocreate := []model.CtrContractProduct{}
	for _, p := range product {
		product, err := s.ProductDao.Where("id = ?", p.ProdId).One()
		if err != nil {
			return err
		}
		if product == nil {
			return myerrors.NewMsgError(nil, fmt.Sprintf("产品: %d 不存在", p.ProdId))
		}
		tocreate = append(tocreate, model.CtrContractProduct{
			ContractId:    id,
			ProdId:        p.ProdId,
			ProdCode:      product.ProdCode,
			ProdName:      product.ProdName,
			ProdClass:     product.ProdClass,
			ProdNum:       p.ProdNum,
			MaintTerm:     p.MaintTerm,
			SugSalesPrice: p.SugSalesPrice,
			TranPrice:     p.TranPrice,
			ContractPrive: amount,
			Remark:        p.Remark,
			CreatedBy:     int(s.userInfo.Id),
			CreatedName:   s.userInfo.NickName,
			CreatedTime:   gtime.Now(),
			UpdatedBy:     int(s.userInfo.Id),
			UpdatedName:   s.userInfo.NickName,
			UpdatedTime:   gtime.Now(),
			DeletedTime:   gtime.Now(),
		})
	}

	if len(tocreate) != 0 {
		_, err = tx.Insert("ctr_contract_product", tocreate)
		if err != nil {
			return err
		}
	}
	return nil
}

func (s CtrContractService) Add(ctx context.Context, req *model.CtrContractAddReq) (int, error) {
	validErr := gvalid.CheckStruct(ctx, req, nil)
	if validErr != nil {
		return 0, myerrors.NewMsgError(nil, validErr.Current().Error())
	}

	c, err := s.Dao.Where("contract_code = ?", req.ContractCode).One()
	if err != nil {
		return 0, err
	}
	if c != nil {
		return 0, myerrors.NewMsgError(nil, fmt.Sprintf("合同编号:%s 已存在", req.ContractCode))
	}
	c, err = s.Dao.Where("contract_name = ?", req.ContractName).One()
	if err != nil {
		return 0, err
	}
	if c != nil {
		return 0, myerrors.NewMsgError(nil, fmt.Sprintf("合同名称:%s 已存在", req.ContractName))
	}
	nbo, err := s.ProjBusinessDao.Where("id = ?", req.NboId).One()
	if err != nil {
		return 0, err
	}
	if nbo == nil {
		return 0, myerrors.NewMsgError(nil, "项目不存在")
	}
	var contractAmount float64
	for _, p := range req.Product {
		contractAmount += p.TranPrice
	}

	ctr := model.CtrContract{
		ContractCode:      req.ContractCode,
		ContractName:      req.ContractName,
		CustId:            nbo.CustId,
		CustName:          nbo.CustName,
		NboId:             nbo.Id,
		NboName:           nbo.NboName,
		ApproStatus:       "",
		ContractType:      req.ContractType,
		ContractAmount:    contractAmount,
		InvoiceAmount:     0,
		CollectedAmount:   0,
		ContractStartTime: req.ContractStartTime,
		ContractEndTime:   req.ContractEndTime,
		InchargeId:        req.InchargeId,
		InchargeName:      req.InchargeName,
		SignatoryId:       req.SignatoryId,
		SignatoryName:     req.SignatoryName,
		DistributorId:     req.DistributorId,
		DistributorName:   req.DistributorName,
		Remark:            req.Remark,
		CreatedBy:         int(s.userInfo.Id),
		CreatedName:       s.userInfo.NickName,
		CreatedTime:       gtime.Now(),
		UpdatedBy:         int(s.userInfo.Id),
		UpdatedName:       s.userInfo.NickName,
		UpdatedTime:       gtime.Now(),
		DeletedTime:       gtime.Now(),
	}

	var id int
	txerr := s.Dao.DB.Transaction(ctx, func(ctx context.Context, tx *gdb.TX) error {
		ctrid, err := tx.InsertAndGetId("ctr_contract", ctr)
		if err != nil {
			return err
		}
		err = s.BindProduct(tx, int(ctrid), req.Product)
		if err != nil {
			return err
		}
		id = int(ctrid)
		return nil
	})
	return id, txerr
}

func (s CtrContractService) Update(ctx context.Context, req *model.CtrContractUpdateReq) error {
	validErr := gvalid.CheckStruct(ctx, req, nil)
	if validErr != nil {
		return myerrors.NewMsgError(nil, validErr.Current().Error())
	}

	ent, err := s.Dao.Where("id = ?", req.Id).One()
	if err != nil {
		return err
	}
	if ent == nil {
		return myerrors.NewMsgError(nil, fmt.Sprintf("合同不存在: %d", req.Id))
	}

	if req.ContractCode != "" {
		exist, err := s.Dao.Where("contract_code = ?", req.ContractCode).One()
		if err != nil {
			return err
		}
		if exist != nil && exist.Id != req.Id {
			return myerrors.NewMsgError(nil, fmt.Sprintf("合同编号:%s 已存在", req.ContractCode))
		}
	}
	if req.ContractName != "" {
		exist, err := s.Dao.Where("contract_name = ?", req.ContractName).One()
		if err != nil {
			return err
		}
		if exist != nil && exist.Id != req.Id {
			return myerrors.NewMsgError(nil, fmt.Sprintf("合同名称:%s 已存在", req.ContractName))
		}
	}

	var nbo *proj.ProjBusiness
	if req.NboId != 0 {
		nbo, err = s.ProjBusinessDao.Where("id = ?", req.NboId).One()
		if err != nil {
			return err
		}
		if nbo == nil {
			return myerrors.NewMsgError(nil, "项目不存在")
		}
	}

	toupdate := map[string]interface{}{}
	if req.ContractCode != "" {
		toupdate["contract_code"] = req.ContractCode
	}
	if req.ContractName != "" {
		toupdate["contract_name"] = req.ContractName
	}
	if req.NboId != 0 {
		toupdate["cust_id"] = nbo.CustId
		toupdate["cust_name"] = nbo.CustName
		toupdate["nbo_id"] = nbo.Id
		toupdate["nbo_name"] = nbo.NboName
	}
	// if req.ApproStatus != "" {
	// 	toupdate["appro_status"] = req.ApproStatus
	// }
	if req.ContractType != "" {
		toupdate["contract_type"] = req.ContractType
	}
	// if req.ContractAmount != 0 {
	// 	toupdate["contract_amount"] = req.ContractAmount
	// }
	// if req.InvoiceAmount != 0 {
	// 	toupdate["invoice_amount"] = req.InvoiceAmount
	// }
	// if req.CollectedAmount != 0 {
	// 	toupdate["collected_amount"] = req.CollectedAmount
	// }
	if req.ContractStartTime != nil {
		toupdate["contract_start_time"] = req.ContractStartTime
	}
	if req.ContractEndTime != nil {
		toupdate["contract_end_time"] = req.ContractEndTime
	}
	if req.InchargeId != 0 {
		toupdate["incharge_id"] = req.InchargeId
	}
	if req.InchargeName != "" {
		toupdate["incharge_name"] = req.InchargeName
	}
	if req.SignatoryId != 0 {
		toupdate["signatory_id"] = req.SignatoryId
	}
	if req.SignatoryName != "" {
		toupdate["signatory_name"] = req.SignatoryName
	}
	if req.DistributorId != 0 {
		toupdate["distributor_id"] = req.DistributorId
	}
	if req.DistributorName != "" {
		toupdate["distributor_name"] = req.DistributorName
	}
	if req.Remark != nil {
		toupdate["remark"] = *req.Remark
	}
	if req.Product != nil {
		var contractAmount float64
		for _, p := range *req.Product {
			contractAmount += p.TranPrice
		}
		toupdate["contract_amount"] = contractAmount
	}
	if len(toupdate) != 0 {
		toupdate["updated_by"] = int(s.userInfo.Id)
		toupdate["updated_name"] = s.userInfo.NickName
		toupdate["updated_time"] = gtime.Now()

		txerr := s.Dao.DB.Transaction(ctx, func(ctx context.Context, tx *gdb.TX) error {
			_, err = tx.Update("ctr_contract", toupdate, "id = ?", req.Id)
			if err != nil {
				return err
			}
			if req.Product != nil {
				err = s.BindProduct(tx, req.Id, *req.Product)
				if err != nil {
					return err
				}
			}
			return nil
		})
		if txerr != nil {
			return txerr
		}
	}
	return nil
}

func (s CtrContractService) Transfer(ctx context.Context, req *model.CtrContractTransferReq) error {
	ent, err := s.Dao.Where("id = ?", req.Id).One()
	if err != nil {
		return err
	}
	if ent == nil {
		return myerrors.NewMsgError(nil, fmt.Sprintf("合同不存在: %d", req.Id))
	}

	txerr := s.Dao.DB.Transaction(ctx, func(ctx context.Context, tx *gdb.TX) error {
		toupdate := map[string]interface{}{
			"incharge_id":   req.InchargeId,
			"incharge_name": req.InchargeName,
		}
		_, err := tx.Update("ctr_contract", toupdate, "id = ?", req.Id)
		if err != nil {
			return err
		}
		return nil
	})
	if txerr != nil {
		return txerr
	}
	return nil
}

func (s CtrContractService) UpdateCollectedAmount(tx *gdb.TX, id int) error {
	ctr := model.CtrContract{}
	err := tx.GetStruct(&ctr, "select * from ctr_contract where id = ?", id)
	if err == sql.ErrNoRows {
		return myerrors.NewMsgError(err, fmt.Sprintf("合同不存在 %d", id))
	}
	if err != nil {
		return err
	}

	v, err := tx.GetValue("select sum(collection_amount) from ctr_contract_collection where contract_id=? and appro_status='20' and deleted_time is null", id)
	if err != nil {
		return err
	}
	amount := v.Float64()

	_, err = tx.Update("ctr_contract",
		map[string]interface{}{
			"collected_amount": amount,
		}, "id = ?", id)
	if err != nil {
		return err
	}
	return nil
}

func (s CtrContractService) Delete(ctx context.Context, id []int) error {
	if len(id) == 0 {
		return nil
	}
	_, err := s.Dao.Where("Id  IN (?)", id).Delete()
	return err
}