From f5385cceb919b1487584b8a90a6ea660f9e62aeb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=8B=8F=E5=B0=B9=E5=B2=9A?= <770236076@qq.com>
Date: Tue, 20 Apr 2021 16:50:27 +0800
Subject: [PATCH] aa

---
 platformapi/mtpsapi/shop_test.go          |  2 +-
 platformapi/mtunionapi/mtunionapi.go      | 18 +++++++++++++++++-
 platformapi/mtunionapi/mtunionapi_test.go | 10 +++++++++-
 3 files changed, 27 insertions(+), 3 deletions(-)

diff --git a/platformapi/mtpsapi/shop_test.go b/platformapi/mtpsapi/shop_test.go
index 39971ee7..61b72888 100644
--- a/platformapi/mtpsapi/shop_test.go
+++ b/platformapi/mtpsapi/shop_test.go
@@ -18,7 +18,7 @@ func TestShopQuery(t *testing.T) {
 	// if err == nil {
 	// 	t.Fatal("应该报错找不到门店")
 	// }
-	shopInfo, err := api.ShopQuery("667018")
+	shopInfo, err := api.ShopQuery("668089")
 	if err != nil {
 		t.Fatal(err)
 	}
diff --git a/platformapi/mtunionapi/mtunionapi.go b/platformapi/mtunionapi/mtunionapi.go
index 96dd6879..1cab563d 100644
--- a/platformapi/mtunionapi/mtunionapi.go
+++ b/platformapi/mtunionapi/mtunionapi.go
@@ -138,7 +138,7 @@ func (a *API) AccessStorePage(fullURL string, bizParams map[string]interface{},
 
 //https://union.meituan.com/v2/apiDetail?id=8
 //https://runion.meituan.com/generateLink
-func (a *API) GenerateLink(actID int, userID string) (url string, err error) {
+func (a *API) MiniCode(actID int, userID string) (url string, err error) {
 	result, err := a.AccessAPI("miniCode", false, map[string]interface{}{
 		"sid":   strings.ToLower(userID),
 		"actId": actID,
@@ -153,6 +153,22 @@ func (a *API) GenerateLink(actID int, userID string) (url string, err error) {
 	return url, err
 }
 
+func (a *API) GenerateLink(actID,linkType int, userID string) (url string, err error) {
+	result, err := a.AccessAPI("generateLink", false, map[string]interface{}{
+		"sid":   strings.ToLower(userID),
+		"actId": actID,
+		"linkType": linkType,
+	})
+	if err == nil  {
+		if utils.MustInterface2Int64(result["status"]) != 0 {
+			return  "",fmt.Errorf(result["des"].(string))
+		}else {
+			return result["data"].(string),err
+		}
+	}
+	return url, err
+}
+
 type ActivityListResult struct {
 	ID        int    `json:"id"`
 	ActName   string `json:"actName"`
diff --git a/platformapi/mtunionapi/mtunionapi_test.go b/platformapi/mtunionapi/mtunionapi_test.go
index 9cab524f..43981449 100644
--- a/platformapi/mtunionapi/mtunionapi_test.go
+++ b/platformapi/mtunionapi/mtunionapi_test.go
@@ -27,8 +27,16 @@ func init() {
 	`)
 }
 
+func TestMiniCode(t *testing.T) {
+	result, err := api.MiniCode(2, strings.ToLower("EFA9876238FC11EB9101525400C36BDA"))
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log(utils.Format4Output(result, false))
+}
+
 func TestGenerateLink(t *testing.T) {
-	result, err := api.GenerateLink(2, strings.ToLower("EFA9876238FC11EB9101525400C36BDA"))
+	result, err := api.GenerateLink(2, 1,strings.ToLower("EFA9876238FC11EB9101525400C36BDA"))
 	if err != nil {
 		t.Fatal(err)
 	}