1
0
mirror of https://github.com/silenceper/wechat.git synced 2026-02-14 17:52:27 +08:00

Compare commits

..

3 Commits

Author SHA1 Message Date
Lien Li
2e0708845b add-show_splash_ad-for-qrcodeparam (#794) 2024-08-26 10:20:39 +08:00
oah8
c1770130a0 add stream_upload_source (#786)
Co-authored-by: _oah <57302072@qq.com>
2024-08-22 14:21:58 +08:00
一叶知秋
c22a036b7f fix: 增加一个用户信息接口 (#793) 2024-08-22 12:08:44 +08:00
6 changed files with 100 additions and 41 deletions

2
cache/redis.go vendored
View File

@@ -16,6 +16,7 @@ type Redis struct {
// RedisOpts redis 连接属性
type RedisOpts struct {
Host string `yml:"host" json:"host"`
Username string `yaml:"username" json:"username"`
Password string `yml:"password" json:"password"`
Database int `yml:"database" json:"database"`
MaxIdle int `yml:"max_idle" json:"max_idle"`
@@ -28,6 +29,7 @@ func NewRedis(ctx context.Context, opts *RedisOpts) *Redis {
conn := redis.NewUniversalClient(&redis.UniversalOptions{
Addrs: []string{opts.Host},
DB: opts.Database,
Username: opts.Username,
Password: opts.Password,
IdleTimeout: time.Second * time.Duration(opts.IdleTimeout),
MinIdleConns: opts.MaxIdle,

View File

@@ -54,6 +54,8 @@ type QRCoder struct {
IsHyaline bool `json:"is_hyaline,omitempty"`
// envVersion 要打开的小程序版本。正式版为 "release",体验版为 "trial",开发版为 "develop"
EnvVersion string `json:"env_version,omitempty"`
// ShowSplashAd 控制通过该小程序码进入小程序是否展示封面广告1、默认为true展示封面广告2、传入为false时不展示封面广告
ShowSplashAd bool `json:"show_splash_ad,omitempty"`
}
// fetchCode 请求并返回二维码二进制数据

View File

@@ -164,7 +164,7 @@ type resAddMaterial struct {
}
// AddMaterialFromReader 上传永久性素材(处理视频需要单独上传),从 io.Reader 中读取
func (material *Material) AddMaterialFromReader(mediaType MediaType, directory string, reader io.Reader) (mediaID string, url string, err error) {
func (material *Material) AddMaterialFromReader(mediaType MediaType, filePath string, reader io.Reader) (mediaID string, url string, err error) {
if mediaType == MediaTypeVideo {
err = errors.New("永久视频素材上传使用 AddVideo 方法")
return
@@ -177,9 +177,9 @@ func (material *Material) AddMaterialFromReader(mediaType MediaType, directory s
uri := fmt.Sprintf("%s?access_token=%s&type=%s", addMaterialURL, accessToken, mediaType)
// 获取文件名
filename := path.Base(directory)
filename := path.Base(filePath)
var response []byte
response, err = util.PostFileFromReader("media", directory, filename, uri, reader)
response, err = util.PostFileFromReader("media", filePath, filename, uri, reader)
if err != nil {
return
}
@@ -214,7 +214,7 @@ type reqVideo struct {
}
// AddVideoFromReader 永久视频素材文件上传,从 io.Reader 中读取
func (material *Material) AddVideoFromReader(directory, title, introduction string, reader io.Reader) (mediaID string, url string, err error) {
func (material *Material) AddVideoFromReader(filePath, title, introduction string, reader io.Reader) (mediaID string, url string, err error) {
var accessToken string
accessToken, err = material.GetAccessToken()
if err != nil {
@@ -232,12 +232,12 @@ func (material *Material) AddVideoFromReader(directory, title, introduction stri
if err != nil {
return
}
fileName := path.Base(directory)
fileName := path.Base(filePath)
fields := []util.MultipartFormField{
{
IsFile: true,
Fieldname: "media",
Directory: directory,
FilePath: filePath,
Filename: fileName,
FileReader: reader,
},

View File

@@ -3,6 +3,8 @@ package material
import (
"encoding/json"
"fmt"
"io"
"github.com/silenceper/wechat/v2/util"
)
@@ -37,7 +39,7 @@ type Media struct {
}
// MediaUpload 临时素材上传
func (material *Material) MediaUpload(mediaType MediaType, url string) (media Media, err error) {
func (material *Material) MediaUpload(mediaType MediaType, filename string) (media Media, err error) {
var accessToken string
accessToken, err = material.GetAccessToken()
if err != nil {
@@ -45,10 +47,38 @@ func (material *Material) MediaUpload(mediaType MediaType, url string) (media Me
}
uri := fmt.Sprintf("%s?access_token=%s&type=%s", mediaUploadURL, accessToken, mediaType)
filename, byteData, err := util.GetFileSourceByURL(url)
var response []byte
response, err = util.PostFile("media", 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
}
// MediaUploadFromReader 临时素材上传
func (material *Material) MediaUploadFromReader(mediaType MediaType, filename string, reader io.Reader) (media Media, err error) {
var accessToken string
accessToken, err = material.GetAccessToken()
if err != nil {
return
}
uri := fmt.Sprintf("%s?access_token=%s&type=%s", mediaUploadURL, accessToken, mediaType)
var byteData []byte
byteData, err = io.ReadAll(reader)
if err != nil {
return
}
var response []byte
response, err = util.PostFileByStream("media", filename, uri, byteData)
if err != nil {

View File

@@ -8,13 +8,13 @@ import (
"encoding/pem"
"encoding/xml"
"fmt"
"golang.org/x/crypto/pkcs12"
"io"
"log"
"mime/multipart"
"net/http"
"os"
"path"
"golang.org/x/crypto/pkcs12"
)
// URIModifier URI修改器
@@ -159,44 +159,25 @@ func PostFileByStream(fieldName, fileName, uri string, byteData []byte) ([]byte,
return PostMultipartForm(fields, uri)
}
// GetFileSourceByURL 从给定的 URL 获取文件名和文件字节数据
func GetFileSourceByURL(url string) (filename string, byteData []byte, err error) {
// 获取文件名
filename = path.Base(url)
// 获取资源
resp, err := http.Get(url)
if err != nil {
err = fmt.Errorf("get resp error: %v", err)
return
}
byteData, err = io.ReadAll(resp.Body)
defer func() { _ = resp.Body.Close() }()
if err != nil {
err = fmt.Errorf("read resp error: %v", err)
return
}
return
}
// PostFile 上传文件
func PostFile(fieldName, directory, uri string) ([]byte, error) {
func PostFile(fieldName, filePath, uri string) ([]byte, error) {
fields := []MultipartFormField{
{
IsFile: true,
Fieldname: fieldName,
Directory: directory,
FilePath: filePath,
},
}
return PostMultipartForm(fields, uri)
}
// PostFileFromReader 上传文件,从 io.Reader 中读取
func PostFileFromReader(filedName, directory, fileName, uri string, reader io.Reader) ([]byte, error) {
func PostFileFromReader(filedName, filePath, fileName, uri string, reader io.Reader) ([]byte, error) {
fields := []MultipartFormField{
{
IsFile: true,
Fieldname: filedName,
Directory: directory,
FilePath: filePath,
Filename: fileName,
FileReader: reader,
},
@@ -209,7 +190,7 @@ type MultipartFormField struct {
IsFile bool
Fieldname string
Value []byte
Directory string
FilePath string
Filename string
FileReader io.Reader
}
@@ -231,7 +212,7 @@ func PostMultipartForm(fields []MultipartFormField, uri string) (respBody []byte
}
if field.FileReader == nil {
fh, e := os.Open(field.Directory)
fh, e := os.Open(field.FilePath)
if e != nil {
err = fmt.Errorf("error opening file , err=%v", e)
return

View File

@@ -2,6 +2,8 @@ package material
import (
"fmt"
"io"
"github.com/silenceper/wechat/v2/util"
)
@@ -58,8 +60,7 @@ func (r *Client) UploadImg(filename string) (*UploadImgResponse, error) {
// UploadTempFile 上传临时素材
// @see https://developer.work.weixin.qq.com/document/path/90253
// @mediaType 媒体文件类型分别有图片image、语音voice、视频video普通文件file
// 临时素材一般都是存储在oss上的可以直接传递url
func (r *Client) UploadTempFile(url string, mediaType string) (*UploadTempFileResponse, error) {
func (r *Client) UploadTempFile(filename string, mediaType string) (*UploadTempFileResponse, error) {
var (
accessToken string
err error
@@ -67,7 +68,49 @@ func (r *Client) UploadTempFile(url string, mediaType string) (*UploadTempFileRe
if accessToken, err = r.GetAccessToken(); err != nil {
return nil, err
}
filename, byteData, err := util.GetFileSourceByURL(url)
var response []byte
if response, err = util.PostFile("media", filename, fmt.Sprintf(uploadTempFile, accessToken, mediaType)); err != nil {
return nil, err
}
result := &UploadTempFileResponse{}
err = util.DecodeWithError(response, result, "UploadTempFile")
return result, err
}
// UploadAttachment 上传附件资源
// @see https://developer.work.weixin.qq.com/document/path/95098
// @mediaType 媒体文件类型分别有图片image、视频video、普通文件file
// @attachment_type 附件类型不同的附件类型用于不同的场景。1朋友圈2:商品图册
func (r *Client) UploadAttachment(filename string, mediaType string, attachmentType int) (*UploadAttachmentResponse, error) {
var (
accessToken string
err error
)
if accessToken, err = r.GetAccessToken(); err != nil {
return nil, err
}
var response []byte
if response, err = util.PostFile("media", filename, fmt.Sprintf(uploadAttachment, accessToken, mediaType, attachmentType)); err != nil {
return nil, err
}
result := &UploadAttachmentResponse{}
err = util.DecodeWithError(response, result, "UploadAttachment")
return result, err
}
// UploadTempFileFromReader 上传临时素材
// @see https://developer.work.weixin.qq.com/document/path/90253
// @mediaType 媒体文件类型分别有图片image、语音voice、视频video普通文件file
func (r *Client) UploadTempFileFromReader(filename, mediaType string, reader io.Reader) (*UploadTempFileResponse, error) {
var (
accessToken string
err error
)
if accessToken, err = r.GetAccessToken(); err != nil {
return nil, err
}
var byteData []byte
byteData, err = io.ReadAll(reader)
if err != nil {
return nil, err
}
@@ -80,11 +123,11 @@ func (r *Client) UploadTempFile(url string, mediaType string) (*UploadTempFileRe
return result, err
}
// UploadAttachment 上传附件资源
// UploadAttachmentFromReader 上传附件资源
// @see https://developer.work.weixin.qq.com/document/path/95098
// @mediaType 媒体文件类型分别有图片image、视频video、普通文件file
// @attachment_type 附件类型不同的附件类型用于不同的场景。1朋友圈2:商品图册
func (r *Client) UploadAttachment(url string, mediaType string, attachmentType int) (*UploadAttachmentResponse, error) {
func (r *Client) UploadAttachmentFromReader(filename, mediaType string, reader io.Reader, attachmentType int) (*UploadAttachmentResponse, error) {
var (
accessToken string
err error
@@ -92,7 +135,8 @@ func (r *Client) UploadAttachment(url string, mediaType string, attachmentType i
if accessToken, err = r.GetAccessToken(); err != nil {
return nil, err
}
filename, byteData, err := util.GetFileSourceByURL(url)
var byteData []byte
byteData, err = io.ReadAll(reader)
if err != nil {
return nil, err
}