mirror of
https://github.com/silenceper/wechat.git
synced 2026-02-12 00:32:26 +08:00
130 lines
3.3 KiB
Go
130 lines
3.3 KiB
Go
package material
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"io"
|
||
"net/http"
|
||
"strings"
|
||
|
||
"github.com/silenceper/wechat/v2/util"
|
||
)
|
||
|
||
// MediaType 媒体文件类型
|
||
type MediaType string
|
||
|
||
const (
|
||
// MediaTypeImage 媒体文件:图片
|
||
MediaTypeImage MediaType = "image"
|
||
// MediaTypeVoice 媒体文件:声音
|
||
MediaTypeVoice MediaType = "voice"
|
||
// MediaTypeVideo 媒体文件:视频
|
||
MediaTypeVideo MediaType = "video"
|
||
// MediaTypeThumb 媒体文件:缩略图
|
||
MediaTypeThumb MediaType = "thumb"
|
||
)
|
||
|
||
const (
|
||
mediaUploadURL = "https://api.weixin.qq.com/cgi-bin/media/upload"
|
||
mediaUploadImageURL = "https://api.weixin.qq.com/cgi-bin/media/uploadimg"
|
||
mediaGetURL = "https://api.weixin.qq.com/cgi-bin/media/get"
|
||
)
|
||
|
||
// Media 临时素材上传返回信息
|
||
type Media struct {
|
||
util.CommonError
|
||
|
||
Type MediaType `json:"type"`
|
||
MediaID string `json:"media_id"`
|
||
ThumbMediaID string `json:"thumb_media_id"`
|
||
CreatedAt int64 `json:"created_at"`
|
||
}
|
||
|
||
// MediaUpload 临时素材上传
|
||
func (material *Material) MediaUpload(mediaType MediaType, url string) (media Media, err error) {
|
||
var accessToken string
|
||
if accessToken, err = material.GetAccessToken(); err != nil {
|
||
return
|
||
}
|
||
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
|
||
response, err = util.PostFile("media", imageData, filename, "", uri)
|
||
if err != nil {
|
||
return
|
||
}
|
||
err = json.Unmarshal(response, &media)
|
||
if err != nil {
|
||
return
|
||
}
|
||
if media.ErrCode != 0 {
|
||
err = fmt.Errorf("MediaUpload error : errcode=%v, errmsg=%v", media.ErrCode, media.ErrMsg)
|
||
return
|
||
}
|
||
return
|
||
}
|
||
|
||
// GetMediaURL 返回临时素材的下载地址供用户自己处理
|
||
// NOTICE: URL 不可公开，因为含access_token 需要立即另存文件
|
||
func (material *Material) GetMediaURL(mediaID string) (mediaURL string, err error) {
|
||
var accessToken string
|
||
accessToken, err = material.GetAccessToken()
|
||
if err != nil {
|
||
return
|
||
}
|
||
mediaURL = fmt.Sprintf("%s?access_token=%s&media_id=%s", mediaGetURL, accessToken, mediaID)
|
||
return
|
||
}
|
||
|
||
// resMediaImage 图片上传返回结果
|
||
type resMediaImage struct {
|
||
util.CommonError
|
||
|
||
URL string `json:"url"`
|
||
}
|
||
|
||
// ImageUpload 图片上传
|
||
func (material *Material) ImageUpload(filename string) (url string, err error) {
|
||
var accessToken string
|
||
accessToken, err = material.GetAccessToken()
|
||
if err != nil {
|
||
return
|
||
}
|
||
|
||
uri := fmt.Sprintf("%s?access_token=%s", mediaUploadImageURL, accessToken)
|
||
var response []byte
|
||
var directory = filename
|
||
response, err = util.PostFile("media", nil, "", directory, uri)
|
||
if err != nil {
|
||
return
|
||
}
|
||
var image resMediaImage
|
||
err = json.Unmarshal(response, &image)
|
||
if err != nil {
|
||
return
|
||
}
|
||
if image.ErrCode != 0 {
|
||
err = fmt.Errorf("UploadImage error : errcode=%v , errmsg=%v", image.ErrCode, image.ErrMsg)
|
||
return
|
||
}
|
||
url = image.URL
|
||
return
|
||
}
|