From bc5999ba6fe9da281b2157a712567dc38af2a854 Mon Sep 17 00:00:00 2001
From: cn-hideyoshi <645165778@qq.com>
Date: Wed, 11 Dec 2024 18:52:08 +0800
Subject: [PATCH 1/2] oa file

---
 weixin/base.go   |  1 +
 weixin/oa_sdk.go | 35 +++++++++++++++++++++++++++++++++++
 2 files changed, 36 insertions(+)

diff --git a/weixin/base.go b/weixin/base.go
index 3a30e6c..1da7e23 100644
--- a/weixin/base.go
+++ b/weixin/base.go
@@ -21,6 +21,7 @@ const (
 	CreateOaMenu         string = "https://api.weixin.qq.com/cgi-bin/menu/create"
 	QueryOaMenu          string = "https://api.weixin.qq.com/cgi-bin/get_current_selfmenu_info"
 	DeleteOaMenu         string = "https://api.weixin.qq.com/cgi-bin/menu/delete"
+	UploadOaMedia        string = "https://api.weixin.qq.com/cgi-bin/media/upload"
 )
 
 const (
diff --git a/weixin/oa_sdk.go b/weixin/oa_sdk.go
index 6e9e43d..5551c45 100644
--- a/weixin/oa_sdk.go
+++ b/weixin/oa_sdk.go
@@ -6,7 +6,9 @@ import (
 	"fmt"
 	"github.com/tidwall/gjson"
 	"io"
+	"mime/multipart"
 	"net/http"
+	"time"
 )
 
 type OaSdk struct {
@@ -98,6 +100,39 @@ func (o *OaSdk) GetUserInfoByOpenid(openid string) (*UserInfo, error) {
 	return user, nil
 }
 
+func (o *OaSdk) UploadFileByByte(file []byte, ext string) (string, string, error) {
+	body := &bytes.Buffer{}
+	writer := multipart.NewWriter(body)
+	formFile, err := writer.CreateFormFile("file", fmt.Sprintf("%d%s", time.Now().Unix(), ext))
+	if err != nil {
+		return "", "", err
+	}
+	_, err = formFile.Write(file)
+	if err != nil {
+		return "", "", err
+	}
+	err = writer.Close()
+	if err != nil {
+		return "", "", err
+	}
+	res, err := http.Post(UploadOaMedia, writer.FormDataContentType(), body)
+	if err != nil {
+		return "", "", err
+	}
+	resBody, err := io.ReadAll(res.Body)
+	if err != nil {
+		return "", "", err
+	}
+	g := gjson.ParseBytes(resBody)
+	errCode := g.Get("errcode").Int()
+	if errCode != 0 {
+		return "", "", fmt.Errorf("%d:%s", errCode, g.Get("errmsg").String())
+	}
+	fileType := g.Get("type").String()
+	fileId := g.Get("media_id").String()
+	return fileType, fileId, nil
+}
+
 func (o *OaSdk) CreateMenu(buttons []*OaMenuButton) error {
 	accessToken, err := o.getAccessToken()
 	if err != nil {

From d7d4c2e3f4e3f9e4c2c106db1c4d463b0a0f22de Mon Sep 17 00:00:00 2001
From: cn-hideyoshi <645165778@qq.com>
Date: Wed, 11 Dec 2024 19:23:29 +0800
Subject: [PATCH 2/2] upload media

---
 weixin/oa_sdk.go | 23 ++++++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

diff --git a/weixin/oa_sdk.go b/weixin/oa_sdk.go
index 5551c45..0adb6fe 100644
--- a/weixin/oa_sdk.go
+++ b/weixin/oa_sdk.go
@@ -115,7 +115,28 @@ func (o *OaSdk) UploadFileByByte(file []byte, ext string) (string, string, error
 	if err != nil {
 		return "", "", err
 	}
-	res, err := http.Post(UploadOaMedia, writer.FormDataContentType(), body)
+	token, err := o.getAccessToken()
+	if err != nil {
+		return "", "", err
+	}
+	postType := ""
+	if ext == ".jpg" || ext == ".png" || ext == ".gif" || ext == ".jpeg" {
+		postType = "image"
+	} else if ext == ".mp4" {
+		postType = "video"
+	} else if ext == ".mp3" {
+		postType = "voice"
+	}
+
+	if postType == "" {
+		return "", "", fmt.Errorf("不支持的文件类型")
+	}
+
+	res, err := http.Post(
+		fmt.Sprintf("%s?access_token=%s&type=%s", UploadOaMedia, token, postType),
+		writer.FormDataContentType(),
+		body,
+	)
 	if err != nil {
 		return "", "", err
 	}