1
0
mirror of https://github.com/silenceper/wechat.git synced 2026-02-15 18:22:26 +08:00

Compare commits

...

6 Commits

Author SHA1 Message Date
_oah
d678af24dd fix issue 2024-07-18 23:12:33 +08:00
_oah
7efb8f96e9 Merge remote-tracking branch 'origin/feature_add_stream_upload' into feature_add_stream_upload 2024-07-18 23:01:04 +08:00
_oah
807a8a11e5 Merge branch 'feature_add_stream_upload' into v2
# Conflicts:
#	officialaccount/material/material.go
#	util/http.go
2024-07-18 08:47:41 +08:00
_oah
3b888ddbf5 Merge remote-tracking branch 'origin/feature_add_stream_upload' into feature_add_stream_upload
# Conflicts:
#	officialaccount/material/media.go
2024-07-16 23:35:37 +08:00
_oah
afaa30759a 新增支持流的形式上传方式 2024-07-16 23:32:16 +08:00
_oah
f3024864cf 新增支持流的形式上传方式 2024-07-11 22:35:02 +08:00
8 changed files with 109 additions and 60 deletions

View File

@@ -53,9 +53,12 @@ func (content *Content) CheckImage(media string) error {
if err != nil { if err != nil {
return err return err
} }
var directory = media
response, err := util.PostFile( response, err := util.PostFile(
"media", "media",
media, nil,
"",
directory,
fmt.Sprintf(checkImageURL, accessToken), fmt.Sprintf(checkImageURL, accessToken),
) )
if err != nil { if err != nil {

View File

@@ -102,7 +102,8 @@ func (security *Security) ImageCheckV1(filename string) (err error) {
} }
uri := fmt.Sprintf(imageCheckURL, accessToken) uri := fmt.Sprintf(imageCheckURL, accessToken)
response, err := util.PostFile("media", filename, uri) var directory = filename
response, err := util.PostFile("media", nil, "", directory, uri)
if err != nil { if err != nil {
return return
} }

View File

@@ -212,7 +212,8 @@ func (csm *Manager) UploadHeadImg(kfAccount, fileName string) (err error) {
} }
uri := fmt.Sprintf("%s?access_token=%s&kf_account=%s", customerServiceUploadHeadImg, accessToken, kfAccount) uri := fmt.Sprintf("%s?access_token=%s&kf_account=%s", customerServiceUploadHeadImg, accessToken, kfAccount)
var response []byte var response []byte
response, err = util.PostFile("media", fileName, uri) var directory = fileName
response, err = util.PostFile("media", nil, "", directory, uri)
if err != nil { if err != nil {
return return
} }

View File

@@ -6,6 +6,7 @@ import (
"fmt" "fmt"
"io" "io"
"os" "os"
"path"
"github.com/silenceper/wechat/v2/officialaccount/context" "github.com/silenceper/wechat/v2/officialaccount/context"
"github.com/silenceper/wechat/v2/util" "github.com/silenceper/wechat/v2/util"
@@ -163,7 +164,7 @@ type resAddMaterial struct {
} }
// AddMaterialFromReader 上传永久性素材(处理视频需要单独上传),从 io.Reader 中读取 // AddMaterialFromReader 上传永久性素材(处理视频需要单独上传),从 io.Reader 中读取
func (material *Material) AddMaterialFromReader(mediaType MediaType, filename string, reader io.Reader) (mediaID string, url string, err error) { func (material *Material) AddMaterialFromReader(mediaType MediaType, directory string, reader io.Reader) (mediaID string, url string, err error) {
if mediaType == MediaTypeVideo { if mediaType == MediaTypeVideo {
err = errors.New("永久视频素材上传使用 AddVideo 方法") err = errors.New("永久视频素材上传使用 AddVideo 方法")
return return
@@ -176,7 +177,7 @@ func (material *Material) AddMaterialFromReader(mediaType MediaType, filename st
uri := fmt.Sprintf("%s?access_token=%s&type=%s", addMaterialURL, accessToken, mediaType) uri := fmt.Sprintf("%s?access_token=%s&type=%s", addMaterialURL, accessToken, mediaType)
var response []byte var response []byte
response, err = util.PostFileFromReader("media", filename, uri, reader) response, err = util.PostFileFromReader("media", directory, uri, reader)
if err != nil { if err != nil {
return return
} }
@@ -195,14 +196,14 @@ func (material *Material) AddMaterialFromReader(mediaType MediaType, filename st
} }
// AddMaterial 上传永久性素材(处理视频需要单独上传) // AddMaterial 上传永久性素材(处理视频需要单独上传)
func (material *Material) AddMaterial(mediaType MediaType, filename string) (mediaID string, url string, err error) { func (material *Material) AddMaterial(mediaType MediaType, directory string) (mediaID string, url string, err error) {
f, err := os.Open(filename) f, err := os.Open(directory)
if err != nil { if err != nil {
return return
} }
defer func() { _ = f.Close() }() defer func() { _ = f.Close() }()
return material.AddMaterialFromReader(mediaType, filename, f) return material.AddMaterialFromReader(mediaType, directory, f)
} }
type reqVideo struct { type reqVideo struct {
@@ -211,7 +212,7 @@ type reqVideo struct {
} }
// AddVideoFromReader 永久视频素材文件上传,从 io.Reader 中读取 // AddVideoFromReader 永久视频素材文件上传,从 io.Reader 中读取
func (material *Material) AddVideoFromReader(filename, title, introduction string, reader io.Reader) (mediaID string, url string, err error) { func (material *Material) AddVideoFromReader(directory, title, introduction string, reader io.Reader) (mediaID string, url string, err error) {
var accessToken string var accessToken string
accessToken, err = material.GetAccessToken() accessToken, err = material.GetAccessToken()
if err != nil { if err != nil {
@@ -229,12 +230,14 @@ func (material *Material) AddVideoFromReader(filename, title, introduction strin
if err != nil { if err != nil {
return return
} }
// 获取文件名
fileName := path.Base(directory)
fields := []util.MultipartFormField{ fields := []util.MultipartFormField{
{ {
IsFile: true, IsFile: true,
Fieldname: "media", Fieldname: "media",
Filename: filename, Filename: fileName,
Directory: directory,
FileReader: reader, FileReader: reader,
}, },
{ {
@@ -265,14 +268,14 @@ func (material *Material) AddVideoFromReader(filename, title, introduction strin
} }
// AddVideo 永久视频素材文件上传 // AddVideo 永久视频素材文件上传
func (material *Material) AddVideo(filename, title, introduction string) (mediaID string, url string, err error) { func (material *Material) AddVideo(directory, title, introduction string) (mediaID string, url string, err error) {
f, err := os.Open(filename) f, err := os.Open(directory)
if err != nil { if err != nil {
return "", "", err return "", "", err
} }
defer func() { _ = f.Close() }() defer func() { _ = f.Close() }()
return material.AddVideoFromReader(filename, title, introduction, f) return material.AddVideoFromReader(directory, title, introduction, f)
} }
type reqDeleteMaterial struct { type reqDeleteMaterial struct {

View File

@@ -3,6 +3,9 @@ package material
import ( import (
"encoding/json" "encoding/json"
"fmt" "fmt"
"io"
"net/http"
"strings"
"github.com/silenceper/wechat/v2/util" "github.com/silenceper/wechat/v2/util"
) )
@@ -38,16 +41,32 @@ type Media struct {
} }
// MediaUpload 临时素材上传 // MediaUpload 临时素材上传
func (material *Material) MediaUpload(mediaType MediaType, filename string) (media Media, err error) { func (material *Material) MediaUpload(mediaType MediaType, url string) (media Media, err error) {
var accessToken string var accessToken string
accessToken, err = material.GetAccessToken() if accessToken, err = material.GetAccessToken(); err != nil {
if err != nil {
return return
} }
uri := fmt.Sprintf("%s?access_token=%s&type=%s", mediaUploadURL, accessToken, mediaType) uri := fmt.Sprintf("%s?access_token=%s&type=%s", mediaUploadURL, accessToken, mediaType)
// 使用strings.LastIndex函数找到最后一个斜杠的位置
lastSlashIndex := strings.LastIndex(url, "/")
// 从最后一个斜杠的位置截取到最后,获取文件名
filename := url[lastSlashIndex+1:]
// 获取资源
resp, err := http.Get(url)
if err != nil {
err = fmt.Errorf("get image error: %v", err)
return
}
// 读取响应到内存
var imageData []byte
imageData, err = io.ReadAll(resp.Body)
defer resp.Body.Close()
if err != nil {
err = fmt.Errorf("read image error: %v", err)
return
}
var response []byte var response []byte
response, err = util.PostFile("media", filename, uri) response, err = util.PostFile("media", imageData, filename, "", uri)
if err != nil { if err != nil {
return return
} }
@@ -56,7 +75,7 @@ func (material *Material) MediaUpload(mediaType MediaType, filename string) (med
return return
} }
if media.ErrCode != 0 { if media.ErrCode != 0 {
err = fmt.Errorf("MediaUpload error : errcode=%v , errmsg=%v", media.ErrCode, media.ErrMsg) err = fmt.Errorf("MediaUpload error : errcode=%v, errmsg=%v", media.ErrCode, media.ErrMsg)
return return
} }
return return
@@ -91,7 +110,8 @@ func (material *Material) ImageUpload(filename string) (url string, err error) {
uri := fmt.Sprintf("%s?access_token=%s", mediaUploadImageURL, accessToken) uri := fmt.Sprintf("%s?access_token=%s", mediaUploadImageURL, accessToken)
var response []byte var response []byte
response, err = util.PostFile("media", filename, uri) var directory = filename
response, err = util.PostFile("media", nil, "", directory, uri)
if err != nil { if err != nil {
return return
} }

View File

@@ -61,15 +61,15 @@ func (tpl *Template) Send(msg *TemplateMessage) (msgID int64, err error) {
if err != nil { if err != nil {
return return
} }
var ( uri := fmt.Sprintf("%s?access_token=%s", templateSendURL, accessToken)
uri = fmt.Sprintf("%s?access_token=%s", templateSendURL, accessToken) var response []byte
response []byte response, err = util.PostJSON(uri, msg)
) if err != nil {
if response, err = util.PostJSON(uri, msg); err != nil {
return return
} }
var result resTemplateSend var result resTemplateSend
if err = json.Unmarshal(response, &result); err != nil { err = json.Unmarshal(response, &result)
if err != nil {
return return
} }
if result.ErrCode != 0 { if result.ErrCode != 0 {
@@ -103,11 +103,10 @@ func (tpl *Template) List() (templateList []*TemplateItem, err error) {
if err != nil { if err != nil {
return return
} }
var ( uri := fmt.Sprintf("%s?access_token=%s", templateListURL, accessToken)
uri = fmt.Sprintf("%s?access_token=%s", templateListURL, accessToken) var response []byte
response []byte response, err = util.HTTPGet(uri)
) if err != nil {
if response, err = util.HTTPGet(uri); err != nil {
return return
} }
var res resTemplateList var res resTemplateList
@@ -122,23 +121,22 @@ type resTemplateAdd struct {
} }
// Add 添加模板. // Add 添加模板.
func (tpl *Template) Add(shortID string, keyNameList []string) (templateID string, err error) { func (tpl *Template) Add(shortID string) (templateID string, err error) {
var accessToken string var accessToken string
accessToken, err = tpl.GetAccessToken() accessToken, err = tpl.GetAccessToken()
if err != nil { if err != nil {
return return
} }
var ( var msg = struct {
msg = struct { ShortID string `json:"template_id_short"`
ShortID string `json:"template_id_short"` }{ShortID: shortID}
KeyNameList []string `json:"keyword_name_list"` uri := fmt.Sprintf("%s?access_token=%s", templateAddURL, accessToken)
}{ShortID: shortID, KeyNameList: keyNameList} var response []byte
uri = fmt.Sprintf("%s?access_token=%s", templateAddURL, accessToken) response, err = util.PostJSON(uri, msg)
response []byte if err != nil {
)
if response, err = util.PostJSON(uri, msg); err != nil {
return return
} }
var result resTemplateAdd var result resTemplateAdd
err = util.DecodeWithError(response, &result, "AddTemplate") err = util.DecodeWithError(response, &result, "AddTemplate")
return result.TemplateID, err return result.TemplateID, err
@@ -151,14 +149,14 @@ func (tpl *Template) Delete(templateID string) (err error) {
if err != nil { if err != nil {
return return
} }
var ( var msg = struct {
msg = struct { TemplateID string `json:"template_id"`
TemplateID string `json:"template_id"` }{TemplateID: templateID}
}{TemplateID: templateID}
uri = fmt.Sprintf("%s?access_token=%s", templateDelURL, accessToken) uri := fmt.Sprintf("%s?access_token=%s", templateDelURL, accessToken)
response []byte var response []byte
) response, err = util.PostJSON(uri, msg)
if response, err = util.PostJSON(uri, msg); err != nil { if err != nil {
return return
} }
return util.DecodeWithCommonError(response, "DeleteTemplate") return util.DecodeWithCommonError(response, "DeleteTemplate")

View File

@@ -13,6 +13,7 @@ import (
"mime/multipart" "mime/multipart"
"net/http" "net/http"
"os" "os"
"path"
"golang.org/x/crypto/pkcs12" "golang.org/x/crypto/pkcs12"
) )
@@ -146,25 +147,43 @@ func PostJSONWithRespContentType(uri string, obj interface{}) ([]byte, string, e
return responseData, contentType, err return responseData, contentType, err
} }
// PostFile 上传文件 // PostFile 支持流或文件形式上传
func PostFile(fieldName, filename, uri string) ([]byte, error) { func PostFile(fieldName string, data []byte, fileName string, directory string, uri string) ([]byte, error) {
var fileContent []byte
var isFile bool
// 判断
if len(data) == 0 && directory != "" {
fileName = path.Base(directory)
isFile = true
} else if len(data) > 0 && fileName != "" {
fileContent = data
isFile = false
} else {
return nil, fmt.Errorf("Invalid argument: required stream data and filename or only directory")
}
fields := []MultipartFormField{ fields := []MultipartFormField{
{ {
IsFile: true, IsFile: isFile,
Fieldname: fieldName, Fieldname: fieldName,
Filename: filename, Value: fileContent,
Filename: fileName,
Directory: directory,
}, },
} }
return PostMultipartForm(fields, uri) return PostMultipartForm(fields, uri)
} }
// PostFileFromReader 上传文件,从 io.Reader 中读取 // PostFileFromReader 上传文件,从 io.Reader 中读取
func PostFileFromReader(filedName, fileName, uri string, reader io.Reader) ([]byte, error) { func PostFileFromReader(filedName, directory, uri string, reader io.Reader) ([]byte, error) {
// 获取文件名
fileName := path.Base(directory)
fields := []MultipartFormField{ fields := []MultipartFormField{
{ {
IsFile: true, IsFile: true,
Fieldname: filedName, Fieldname: filedName,
Filename: fileName, Filename: fileName,
Directory: directory,
FileReader: reader, FileReader: reader,
}, },
} }
@@ -177,6 +196,7 @@ type MultipartFormField struct {
Fieldname string Fieldname string
Value []byte Value []byte
Filename string Filename string
Directory string
FileReader io.Reader FileReader io.Reader
} }
@@ -197,7 +217,7 @@ func PostMultipartForm(fields []MultipartFormField, uri string) (respBody []byte
} }
if field.FileReader == nil { if field.FileReader == nil {
fh, e := os.Open(field.Filename) fh, e := os.Open(field.Directory)
if e != nil { if e != nil {
err = fmt.Errorf("error opening file , err=%v", e) err = fmt.Errorf("error opening file , err=%v", e)
return return
@@ -213,7 +233,7 @@ func PostMultipartForm(fields []MultipartFormField, uri string) (respBody []byte
} }
} }
} else { } else {
partWriter, e := bodyWriter.CreateFormField(field.Fieldname) partWriter, e := bodyWriter.CreateFormFile(field.Fieldname, field.Filename)
if e != nil { if e != nil {
err = e err = e
return return

View File

@@ -48,7 +48,8 @@ func (r *Client) UploadImg(filename string) (*UploadImgResponse, error) {
return nil, err return nil, err
} }
var response []byte var response []byte
if response, err = util.PostFile("media", filename, fmt.Sprintf(uploadImgURL, accessToken)); err != nil { var directory = filename
if response, err = util.PostFile("media", nil, "", directory, fmt.Sprintf(uploadImgURL, accessToken)); err != nil {
return nil, err return nil, err
} }
result := &UploadImgResponse{} result := &UploadImgResponse{}
@@ -68,7 +69,8 @@ func (r *Client) UploadTempFile(filename string, mediaType string) (*UploadTempF
return nil, err return nil, err
} }
var response []byte var response []byte
if response, err = util.PostFile("media", filename, fmt.Sprintf(uploadTempFile, accessToken, mediaType)); err != nil { var directory = filename
if response, err = util.PostFile("media", nil, "", directory, fmt.Sprintf(uploadTempFile, accessToken, mediaType)); err != nil {
return nil, err return nil, err
} }
result := &UploadTempFileResponse{} result := &UploadTempFileResponse{}
@@ -89,7 +91,8 @@ func (r *Client) UploadAttachment(filename string, mediaType string, attachmentT
return nil, err return nil, err
} }
var response []byte var response []byte
if response, err = util.PostFile("media", filename, fmt.Sprintf(uploadAttachment, accessToken, mediaType, attachmentType)); err != nil { var directory = filename
if response, err = util.PostFile("media", nil, "", directory, fmt.Sprintf(uploadTempFile, accessToken, mediaType)); err != nil {
return nil, err return nil, err
} }
result := &UploadAttachmentResponse{} result := &UploadAttachmentResponse{}