diff --git a/business/jxstore/common/common.go b/business/jxstore/common/common.go index e11d5cebc..12b9136e7 100644 --- a/business/jxstore/common/common.go +++ b/business/jxstore/common/common.go @@ -280,7 +280,9 @@ func AddVendorOrgCode(ctx *jxcontext.Context, vendorOrgCode *model.VendorOrgCode return err } if len(list) > 0 { - return fmt.Errorf("库里有这个账号了，[%v]", vendorOrgCode.VendorOrgCode) + vendorOrgCode.ID = list[0].ID + _, err = dao.UpdateEntity(db, vendorOrgCode, "token", "updated_at", "app_key", "app_secret", "comment", "vendor_org_code") + return err } var userName string diff --git a/controllers/tiktok_store.go b/controllers/tiktok_store.go index b2517fdf9..bdfa95fd0 100644 --- a/controllers/tiktok_store.go +++ b/controllers/tiktok_store.go @@ -1,7 +1,11 @@ package controllers import ( + "crypto/aes" + "crypto/cipher" + "encoding/base64" "encoding/json" + "errors" "git.rosy.net.cn/baseapi/platformapi/tiktok_shop/tiktok_api" "git.rosy.net.cn/baseapi/utils" "git.rosy.net.cn/jx-callback/business/jxstore/common" @@ -11,47 +15,101 @@ import ( "github.com/astaxie/beego/server/web" "io/ioutil" "net/http" - "time" + "strings" ) type TiktokShopController struct { web.Controller } +func AesDecrypt(msgSecret, appSecret string) (string, error) { + var appSecretArr = []byte(strings.ReplaceAll(appSecret, "-", "")) + bytesPass, err := base64.StdEncoding.DecodeString(msgSecret) + if err != nil { + return "", errors.New("解密失败！！！") + } + sourceMsg, err := DoAesDecrypt(bytesPass, appSecretArr) + if err != nil { + return "", errors.New("解密失败！！！") + } + return string(sourceMsg), nil +} + +func DoAesDecrypt(encryptedMsg, key []byte) ([]byte, error) { + block, err := aes.NewCipher(key) + if err != nil { + return nil, err + } + //AES分组长度为128位，所以blockSize=16，单位字节 + blockSize := block.BlockSize() + blockMode := cipher.NewCBCDecrypter(block, key[:blockSize]) //初始向量的长度必须等于块block的长度16字节 + origData := make([]byte, len(encryptedMsg)) + blockMode.CryptBlocks(origData, encryptedMsg) + origData = PKCS5UnPadding(origData) + return origData, nil +} + +func PKCS5UnPadding(origData []byte) []byte { + length := len(origData) + unfilledNum := int(origData[length-1]) + return origData[:(length - unfilledNum)] +} + +type Code struct { + MsgType int64 `json:"msg_type"` + Msg string `json:"msg"` +} + +type MsgDetail struct { + ActionType int `json:"action_type"` + AppId int64 `json:"app_id"` + ShopId int `json:"shop_id"` + Code string `json:"code"` + CodeGenerateTime string `json:"code_generate_time"` +} + func (c *TiktokShopController) TokenMsg() { req := c.Ctx.Request - var code []string + var code string switch req.Method { case http.MethodPost: data, err := ioutil.ReadAll(req.Body) if err != nil { - globals.SugarLogger.Debugf("MethodPost=====err%s", utils.Format4Output(err, false)) c.Data["json"] = tiktok_api.CallbackResponse{Code: tiktok_api.CallbackFailCode, Msg: tiktok_api.CallbackFail} c.ServeJSON() return } - globals.SugarLogger.Debugf("data=====err%s", string(data)) - if err := json.Unmarshal(data, &code); err != nil { - globals.SugarLogger.Debugf("Unmarshal=====err%s", utils.Format4Output(err, false)) + codeToken, err := AesDecrypt(string(data), globals.TiktokShopAppSecret) + if err != nil { + c.Data["json"] = tiktok_api.CallbackResponse{Code: tiktok_api.CallbackFailCode, Msg: tiktok_api.CallbackFail} + c.ServeJSON() + } + + var codeMsg Code + if err := json.Unmarshal([]byte(codeToken), &codeMsg); err != nil { c.Data["json"] = tiktok_api.CallbackResponse{Code: tiktok_api.CallbackFailCode, Msg: tiktok_api.CallbackFail} c.ServeJSON() return } + + var msg MsgDetail + if err := json.Unmarshal([]byte(codeMsg.Msg), &msg); err != nil { + c.Data["json"] = tiktok_api.CallbackResponse{Code: tiktok_api.CallbackFailCode, Msg: tiktok_api.CallbackFail} + c.ServeJSON() + return + } + code = msg.Code case http.MethodGet: codeValues := req.URL.Query() - globals.SugarLogger.Debugf("codeValues=====%s", utils.Format4Output(codeValues, false)) codeList := codeValues.Get("code") - globals.SugarLogger.Debugf("codeList=====%s", utils.Format4Output(codeList, false)) - globals.SugarLogger.Debugf("codeValues.Get(code)===%s", utils.Format4Output(codeValues.Get("Code"), false)) + if err := json.Unmarshal([]byte(string(codeList)), &code); err != nil { - globals.SugarLogger.Debugf("codeList=====err%s", utils.Format4Output(err, false)) c.Data["json"] = tiktok_api.CallbackResponse{Code: tiktok_api.CallbackFailCode, Msg: tiktok_api.CallbackFail} c.ServeJSON() return } } - globals.SugarLogger.Debugf("code ==================%s", utils.Format4Output(code, false)) if len(code) == 0 { c.Data["json"] = tiktok_api.CallbackResponse{Code: tiktok_api.CallbackFailCode, Msg: tiktok_api.CallbackFail} @@ -59,41 +117,37 @@ func (c *TiktokShopController) TokenMsg() { return } - for _, v := range code { - result, err := api.TiktokStore.CreateToken(v) - if err != nil { - c.Data["json"] = tiktok_api.CallbackResponse{Code: tiktok_api.CallbackFailCode, Msg: tiktok_api.CallbackFail} - c.ServeJSON() - return - } - globals.SugarLogger.Debugf("result=====%s", utils.Format4Output(result, false)) + result, err := api.TiktokStore.CreateToken(code) + if err != nil { + c.Data["json"] = tiktok_api.CallbackResponse{Code: tiktok_api.CallbackFailCode, Msg: tiktok_api.CallbackFail} + c.ServeJSON() + return + } - result.ExpiresIn += time.Now().Unix() - data, err := json.Marshal(result) - if err != nil { - c.Data["json"] = tiktok_api.CallbackResponse{Code: tiktok_api.CallbackFailCode, Msg: tiktok_api.CallbackFail} - c.ServeJSON() - return - } + data, err := json.Marshal(result) + if err != nil { + c.Data["json"] = tiktok_api.CallbackResponse{Code: tiktok_api.CallbackFailCode, Msg: tiktok_api.CallbackFail} + c.ServeJSON() + return + } - param := &model.VendorOrgCode{ - VendorID: model.VendorIDDD, - VendorOrgCode: utils.Int64ToStr(result.ShopId), - Comment: "抖音授权", - VendorType: "platform", - IsJxCat: 1, - IsOpen: 1, - EmpowerURL: "https://fuwu.jinritemai.com/detail?from=open_partner_svcList&service_id=24070", - StoreBrandName: result.ShopName, - Token: string(data), - AppKey: "", // web.AppConfig.DefaultString("tiktokShopAppId", "7136048270014416392"), - AppSecret: "", // web.AppConfig.DefaultString("tiktokShopAppSecret", "c397aa9f-3927-47c4-8cfe-4d84e02602e0") - } - if err := common.AddVendorOrgCode(nil, param); err != nil { - c.Data["json"] = tiktok_api.CallbackResponse{Code: tiktok_api.CallbackFailCode, Msg: tiktok_api.CallbackFail} - c.ServeJSON() - return - } + param := &model.VendorOrgCode{ + VendorID: model.VendorIDDD, + VendorOrgCode: utils.Int64ToStr(result.ShopId), + Comment: result.ShopName, + VendorType: "platform", + IsJxCat: 1, + IsOpen: 1, + EmpowerURL: "https://fuwu.jinritemai.com/detail?from=open_partner_svcList&service_id=24070", + StoreBrandName: result.ShopName, + Token: string(data), + AppKey: "", // web.AppConfig.DefaultString("tiktokShopAppId", "7136048270014416392"), + AppSecret: "", // web.AppConfig.DefaultString("tiktokShopAppSecret", "c397aa9f-3927-47c4-8cfe-4d84e02602e0") + } + if err := common.AddVendorOrgCode(nil, param); err != nil { + c.Data["json"] = tiktok_api.CallbackResponse{Code: tiktok_api.CallbackFailCode, Msg: tiktok_api.CallbackFail} + c.ServeJSON() + return } c.Data["json"] = tiktok_api.CallbackResponse{Code: tiktok_api.CallbackSuccessCode, Msg: tiktok_api.CallbackSuccess} c.ServeJSON() diff --git a/globals/globals.go b/globals/globals.go index 158606505..3131c4774 100644 --- a/globals/globals.go +++ b/globals/globals.go @@ -82,6 +82,7 @@ var ( IsAddEvent bool OutputDebugMsgLevel int + TiktokShopAppSecret string ) func init() { @@ -164,6 +165,7 @@ func Init() { JdLoginName = web.AppConfig.DefaultString("jdLoginName", "") IsAddEvent = web.AppConfig.DefaultBool("addEvent", false) TictokpayNotifyURL = web.AppConfig.DefaultString("tiktokNotifyUrl", "") + TiktokShopAppSecret = web.AppConfig.DefaultString("tiktokShopAppSecret", "") IsStoreSkuAct = !IsProductEnv()