main.go

package main

import (
	"errors"
	"flag"
	"fmt"
	"github.com/tidwall/gjson"
	"io"
	"os"
	"strings"
	"time"

	"github.com/robGoods/sams/dd"
)

var (
	showHelp     = flag.Bool("help", false, "show help")
	version      = flag.Bool("version", false, "查看版本号")
	authToken    = flag.String("authToken", "", "必选, Sam's App HTTP头部auth-token")
	barkId       = flag.String("barkId", "", "可选，通知用的`bark` id, 可选参数")
	floorId      = flag.Int("floorId", 1, "可选，1,普通商品 2,全球购保税 3,特殊订购自提 4,大件商品 5,厂家直供商品 6,特殊订购商品 7,失效商品")
	deliveryType = flag.Int("deliveryType", 2, "可选，1 急速达，2， 全程配送")
	longitude    = flag.String("longitude", "", "可选，HTTP头部longitude")
	latitude     = flag.String("latitude", "", "可选，HTTP头部latitude")
	deviceId     = flag.String("deviceId", "", "可选，HTTP头部device-id")
	trackInfo    = flag.String("trackInfo", "", "可选，HTTP头部track-info")
	promotionId  = flag.String("promotionId", "", "可选，优惠券id,多个用逗号隔开，山姆app优惠券列表接口中的'ruleId'字段")
	addressId    = flag.String("addressId", "", "可选，地址id")
	payMethod    = flag.Int("payMethod", 1, "可选，1,微信 2,支付宝")
	deliveryFee  = flag.Bool("deliveryFee", false, "可选，是否免运费下单")
	storeConf    = flag.String("storeConf", "", "可选，加载商店信息文件名")
	isSelected   = flag.Bool("isSelected", false, "可选，是否只选择勾选商品")
)

func main() {
	flag.Parse()
	if *version {
		fmt.Println("Rob Sam's 1.7.0 GNU General Public License v3.0")
		return
	}

	if *showHelp {
		flag.Usage()
		return
	}

	if *authToken == "" {
		flag.Usage()
		return
	}

	splitFn := func(c rune) bool {
		return c == ','
	}

	session := dd.DingdongSession{
		SettleDeliveryInfo: map[int]dd.SettleDeliveryInfo{},
		StoreList:          map[string]dd.Store{},
	}
	conf := dd.Config{
		AuthToken:    *authToken,                                //HTTP头部auth-token
		BarkId:       *barkId,                                   //通知用的bark id，下载bark后从app界面获取, 如果不需要可以填空字符串
		FloorId:      *floorId,                                  //1,普通商品 2,全球购保税 3,特殊订购自提 4,大件商品 5,厂家直供商品 6,特殊订购商品 7,失效商品
		DeliveryType: *deliveryType,                             //1 急速达，2， 全程配送
		Longitude:    *longitude,                                //HTTP头部longitude,可选参数
		Latitude:     *latitude,                                 //HTTP头部latitude,可选参数
		Deviceid:     *deviceId,                                 //HTTP头部device-id,可选参数
		Trackinfo:    *trackInfo,                                //HTTP头部track-info,可选参数
		PromotionId:  strings.FieldsFunc(*promotionId, splitFn), //优惠券id
		AddressId:    *addressId,                                //地址
		PayMethod:    *payMethod,                                //支付方式
		DeliveryFee:  *deliveryFee,
		StoreConf:    *storeConf,
		IsSelected:   *isSelected,
	}

	err := session.InitSession(conf)

	if err != nil {
		fmt.Println(err)
		return
	}

	for true {
	SaveDeliveryAddress:
		fmt.Println("########## 切换购物车收货地址 ###########")
		err = session.SaveDeliveryAddress()
		if err != nil {
			goto SaveDeliveryAddress
		} else {
			fmt.Println("切换成功!")
			fmt.Printf("%s %s %s %s %s \n", session.Address.Name, session.Address.DistrictName, session.Address.ReceiverAddress, session.Address.DetailAddress, session.Address.Mobile)
		}

		if session.Conf.StoreConf != "" {
			if _, err := os.Stat(session.Conf.StoreConf); err == nil {
				if file, err := os.Open(session.Conf.StoreConf); err == nil {
					fmt.Println("########## 预加载商店配置 ###########")
					var bytes []byte
					buf := make([]byte, 10)
					for {
						n, err := file.Read(buf)
						if err != nil && err != io.EOF {
							fmt.Println("read buf fail", err)
							return
						}
						//说明读取结束
						if n == 0 {
							break
						}
						bytes = append(bytes, buf[:n]...)
					}

					for index, store := range session.GetStoreList(gjson.ParseBytes(bytes)) {
						if _, ok := session.StoreList[store.StoreId]; !ok {
							session.StoreList[store.StoreId] = store
							fmt.Printf("[%v] Id：%s 名称：%s, 类型 ：%s\n", index, store.StoreId, store.StoreName, store.StoreType)
						}
					}
				} else {
					fmt.Println(err)
				}
			} else {
				fmt.Println(err)
			}
		}
	StoreLoop:
		fmt.Println("########## 获取地址附近可用商店 ###########")
		stores, err := session.CheckStore()
		if err != nil {
			fmt.Printf("%s", err)
			goto StoreLoop
		}

		for index, store := range stores {
			if oStore, ok := session.StoreList[store.StoreId]; !ok || oStore.StoreDeliveryTemplateId != store.StoreDeliveryTemplateId || oStore.AreaBlockId != store.AreaBlockId {
				session.StoreList[store.StoreId] = store
				fmt.Printf("[%v] Id：%s 名称：%s, 类型 ：%s\n", index, store.StoreId, store.StoreName, store.StoreType)
			}
		}
	CartLoop:
		fmt.Printf("########## 获取购物车中有效商品【%s】 ###########\n", time.Now().Format("15:04:05"))
		err = session.CheckCart()
		for _, v := range session.Cart.FloorInfoList {
			if v.FloorId == session.Conf.FloorId && v.DeliveryType == session.Conf.DeliveryType {
				session.GoodsList = make([]dd.Goods, 0)
				for _, goods := range v.NormalGoodsList {
					if goods.StockQuantity > 0 && goods.StockStatus && goods.IsPutOnSale && goods.IsAvailable {
						if goods.StockQuantity <= goods.Quantity {
							goods.Quantity = goods.StockQuantity
						}
						if goods.LimitNum > 0 && goods.Quantity > goods.LimitNum {
							goods.Quantity = goods.LimitNum
						}

						if goods.LimitNum > 0 && goods.Quantity > goods.ResiduePurchaseNum {
							goods.Quantity = goods.ResiduePurchaseNum
						}

						if goods.Quantity > 0 {
							session.GoodsList = append(session.GoodsList, goods.ToGoods())
						}
					}
				}

				for _, goods := range v.ShortageStockGoodsList {
					if goods.StockQuantity > 0 && goods.StockStatus && goods.IsPutOnSale && goods.IsAvailable {
						if goods.StockQuantity <= goods.Quantity {
							goods.Quantity = goods.StockQuantity
						}
						if goods.LimitNum > 0 && goods.Quantity > goods.LimitNum {
							goods.Quantity = goods.LimitNum
						}
						if goods.LimitNum > 0 && goods.Quantity > goods.ResiduePurchaseNum {
							goods.Quantity = goods.ResiduePurchaseNum
						}

						if goods.Quantity > 0 {
							session.GoodsList = append(session.GoodsList, goods.ToGoods())
						}
					}
				}

				for _, goods := range v.AllOutOfStockGoodsList {
					if goods.StockQuantity > 0 && goods.StockStatus && goods.IsPutOnSale && goods.IsAvailable {
						if goods.StockQuantity <= goods.Quantity {
							goods.Quantity = goods.StockQuantity
						}

						if goods.LimitNum > 0 && goods.Quantity > goods.LimitNum {
							goods.Quantity = goods.LimitNum
						}

						if goods.LimitNum > 0 && goods.Quantity > goods.ResiduePurchaseNum {
							goods.Quantity = goods.ResiduePurchaseNum
						}

						if goods.Quantity > 0 {
							session.GoodsList = append(session.GoodsList, goods.ToGoods())
						}
					}
				}

				session.FloorInfo = v
			} else {
				//无效商品
				//for index, goods := range v.NormalGoodsList {
				//	fmt.Printf("----[%v] %s 数量：%v 总价：%d\n", index, goods.SpuId, goods.StoreId, goods.Price)
				//}
			}
		}

		var selGoods = make([]dd.Goods, 0)
		for index, goods := range session.GoodsList {
			fmt.Printf("[%v] %s 数量：%v 总价：%d * %d, 是否勾选： %v \n", index, goods.GoodsName, goods.Quantity, goods.Price, goods.Quantity, goods.IsSelected)
			if goods.IsSelected && session.Conf.IsSelected {
				selGoods = append(selGoods, goods)
			}
		}

		if session.Conf.IsSelected {
			session.GoodsList = selGoods
		}

		if len(session.GoodsList) == 0 {
			if err != nil {
				fmt.Println(err)
			} else {
				fmt.Println("当前购物车中无有效商品")
			}
			if errors.Is(err, dd.LimitedErr1) {
				time.Sleep(1 * time.Second)
			}
			goto StoreLoop
		}
	GoodsLoop:
		fmt.Printf("########## 开始校验当前商品【%s】 ###########\n", time.Now().Format("15:04:05"))
		if _, err := session.CheckGoods(); err != nil {
			fmt.Println(err)
			time.Sleep(1 * time.Second)
			switch err {
			case dd.OOSErr:
				goto CartLoop
			default:
				goto CartLoop
			}
		}
		if settleInfo, err := session.CheckSettleInfo(); err == nil {
			fmt.Printf("运费： %s\n", settleInfo.DeliveryFee)
			if store, ok := session.StoreList[session.FloorInfo.StoreId]; ok && store.StoreDeliveryTemplateId != settleInfo.SettleDelivery.StoreDeliveryTemplateId {
				store.StoreDeliveryTemplateId = settleInfo.SettleDelivery.StoreDeliveryTemplateId
				store.AreaBlockId = settleInfo.SettleDelivery.AreaBlockId
				session.StoreList[session.FloorInfo.StoreId] = store
			}

			if session.Conf.DeliveryFee && settleInfo.DeliveryFee != "0" {
				goto CartLoop
			}
		} else {
			fmt.Printf("校验商品失败：%s\n", err)
			time.Sleep(1 * time.Second)
			switch err {
			case dd.CartGoodChangeErr:
				goto CartLoop
			case dd.LimitedErr:
				goto GoodsLoop
			case dd.NoMatchDeliverMode:
				goto SaveDeliveryAddress
			default:
				goto GoodsLoop
			}
		}
	CapacityLoop:
		fmt.Printf("########## 获取当前可用配送时间【%s】 ###########\n", time.Now().Format("15:04:05"))
		capacity, err := session.GetCapacity(session.StoreList[session.FloorInfo.StoreId].StoreDeliveryTemplateId)
		if err != nil {
			fmt.Println(err)
			switch err {
			case dd.CapacityErr:
				goto StoreLoop
			default:
				time.Sleep(1 * time.Second)
				//刷新可用配送时间， 会出现“服务器正忙,请稍后再试”， 可以忽略。
				goto CapacityLoop
			}
		}

		session.SettleDeliveryInfo = map[int]dd.SettleDeliveryInfo{}
		for _, caps := range capacity.CapCityResponseList {
			for _, v := range caps.List {
				if v.TimeISFull == false && v.Disabled == false {
					session.SettleDeliveryInfo[len(session.SettleDeliveryInfo)] = dd.SettleDeliveryInfo{
						ArrivalTimeStr:       fmt.Sprintf("%s %s - %s", caps.StrDate, v.StartTime, v.EndTime),
						ExpectArrivalTime:    v.StartRealTime,
						ExpectArrivalEndTime: v.EndRealTime,
					}
				}
			}
		}

		if len(session.SettleDeliveryInfo) > 0 {
			for _, v := range session.SettleDeliveryInfo {
				fmt.Printf("发现可用的配送时段::%s!\n", v.ArrivalTimeStr)
			}
		} else {
			fmt.Println("当前无可用配送时间段")
			time.Sleep(1 * time.Second)
			goto CapacityLoop
		}
	OrderLoop:
		for len(session.SettleDeliveryInfo) > 0 {
			for k, v := range session.SettleDeliveryInfo {
				fmt.Printf("########## 提交订单中【%s】 ###########\n", time.Now().Format("15:04:05"))
				fmt.Printf("配送时段: %s!\n", v.ArrivalTimeStr)
				if order, err := session.CommitPay(v); err == nil {
					fmt.Println("抢购成功，请前往app付款！")
					if session.Conf.BarkId != "" {
						for true {
							err = session.PushSuccess(fmt.Sprintf("Smas抢单成功，订单号：%s", order.OrderNo))
							if err == nil {
								break
							} else {
								fmt.Println(err)
							}
							time.Sleep(1 * time.Second)
						}
					}
					return
				} else {
					fmt.Printf("下单失败：%s\n", err)
					switch err {
					case dd.LimitedErr1:
						fmt.Println("立即重试...")
						goto OrderLoop
					case dd.CloudGoodsOverWightErr:
						maxKey := len(session.GoodsList) - 1
						for key, v := range session.GoodsList {
							if v.Quantity > 1 && v.Weight > session.GoodsList[maxKey].Weight {
								maxKey = key
							}
						}
						if maxKey >= 0 {
							if session.GoodsList[maxKey].Quantity > 1 {
								session.GoodsList[maxKey].Quantity -= 1
							} else {
								session.GoodsList = append(session.GoodsList[:maxKey], session.GoodsList[maxKey+1:]...)
							}
						}
						goto OrderLoop
					case dd.OOSErr, dd.PreGoodNotStartSellErr, dd.CartGoodChangeErr, dd.GoodsExceedLimitErr:
						goto CartLoop
					case dd.StoreHasClosedError, dd.GetDeliveryInfoErr:
						goto StoreLoop
					case dd.CloseOrderTimeExceptionErr, dd.DecreaseCapacityCountError, dd.NotDeliverCapCityErr:
						delete(session.SettleDeliveryInfo, k)
					default:
						goto CapacityLoop
					}
				}
			}
		}
		goto CapacityLoop
	}
}