package smithy

import (
	"go_dreamfactory/comm"
	"go_dreamfactory/pb"
	"go_dreamfactory/utils"

	"go.mongodb.org/mongo-driver/mongo"
)

func (this *apiComp) CustomerCheck(session comm.IUserSession, req *pb.SmithyCustomerReq) (errdata *pb.ErrorData) {
	return
}

func (this *apiComp) Customer(session comm.IUserSession, req *pb.SmithyCustomerReq) (errdata *pb.ErrorData) {
	customerCount := this.module.ModuleTools.GetGlobalConf().SmithyMaxNpc
	if customerCount <= 0 {
		code = pb.ErrorCode_ConfigNoFound
		return
	}
	cus, err := this.module.modelTrade.getDBCustomer(session.GetUserId())
	if err != nil {
		if err == mongo.ErrNoDocuments {
			c, err := this.module.modelTrade.addCustomer(session.GetUserId(), customerCount) //3个顾客
			if err != nil {
				errdata = &pb.ErrorData{
					Code:    pb.ErrorCode_DBError,
					Title:   pb.ErrorCode_DBError.ToString(),
					Message: err.Error(),
				}
				return
			}
			cus = c
		} else {
			errdata = &pb.ErrorData{
				Code:    pb.ErrorCode_DBError,
				Title:   pb.ErrorCode_DBError.ToString(),
				Message: err.Error(),
			}
			return
		}
	}

	if utils.IsFirstTody(cus.LastRefreshTime) {
		this.module.modelTrade.DelByUId(session.GetUserId())
		c, err := this.module.modelTrade.addCustomer(session.GetUserId(), customerCount)
		if err != nil {
			errdata = &pb.ErrorData{
				Code:    pb.ErrorCode_DBError,
				Title:   pb.ErrorCode_DBError.ToString(),
				Message: err.Error(),
			}
			return
		}
		cus = c
	}

	rsp := &pb.SmithyCustomerResp{
		Customers: cus.Customers,
	}

	session.SendMsg(string(this.module.GetType()), "customer", rsp)
	return
}