shkong/MiraiGo
1
0
Fork 0
mirror of https://github.com/Mrs4s/MiraiGo.git synced 2025-05-05 03:23:50 +08:00
Code Issues Projects Releases Wiki Activity

feat send notice with pic

Browse Source
This commit is contained in:
wdvxdr 2021-02-21 15:24:51 +08:00
parent 995e6ceb24
commit cb034c17b5
No known key found for this signature in database
GPG Key ID: 55FF1414A69CEBA6
2 changed files with 113 additions and 65 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

65
client/client.go
View File

@ -4,16 +4,10 @@ import (
"bytes" "bytes"
"crypto/md5" "crypto/md5"
"fmt" "fmt"
"html"
"io" "io"
"io/ioutil"
"math" "math"
"math/rand" "math/rand"
"mime/multipart"
"net" "net"
"net/http"
"net/textproto"
"net/url"
"runtime/debug" "runtime/debug"
"sort" "sort"
"strconv" "strconv"
@ -384,65 +378,6 @@ func (c *QQClient) GetGroupHonorInfo(groupCode int64, honorType HonorType) (*Gro
return &ret, nil return &ret, nil
} }
func (c *QQClient) uploadGroupNoticePic(img []byte) (string, error) {
buf := new(bytes.Buffer)
var w = multipart.NewWriter(buf)
err := w.WriteField("bkn", strconv.Itoa(c.getCSRFToken()))
if err != nil {
return "", errors.Wrap(err, "write multipart<bkn> failed")
}
err = w.WriteField("source", "troopNotice")
if err != nil {
return "", errors.Wrap(err, "write multipart<source> failed")
}
err = w.WriteField("m", "0")
if err != nil {
return "", errors.Wrap(err, "write multipart<m> failed")
}
h := make(textproto.MIMEHeader)
h.Set("Content-Disposition", `form-data; name="pic_up"; filename="temp_uploadFile.png"`)
h.Set("Content-Type", "image/png")
fw, err := w.CreatePart(h)
if err != nil {
return "", errors.Wrap(err, "create multipart field<pic_up> failed")
}
_, err = fw.Write(img)
if err != nil {
return "", errors.Wrap(err, "write multipart<pic_up> failed")
}
err = w.Close()
if err != nil {
return "", errors.Wrap(err, "close multipart failed")
}
req, err := http.NewRequest("POST", "https://web.qun.qq.com/cgi-bin/announce/upload_img", buf)
if err != nil {
return "", errors.Wrap(err, "new request error")
}
req.Header.Set("Content-Type", w.FormDataContentType())
req.Header.Set("Cookie", c.getCookies())
resp, err := http.DefaultClient.Do(req)
if err != nil {
return "", errors.Wrap(err, "post error")
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
return "", errors.Wrap(err, "read body error")
}
str := html.UnescapeString(string(body))
//fmt.Println(str)
return str, nil
}
func (c *QQClient) AddGroupNoticeSimple(groupCode int64, text string) error {
body := fmt.Sprintf(`qid=%v&bkn=%v&text=%v&pinned=0&type=1&settings={"is_show_edit_card":0,"tip_window_type":1,"confirm_required":1}`, groupCode, c.getCSRFToken(), url.QueryEscape(text))
_, err := utils.HttpPostBytesWithCookie("https://web.qun.qq.com/cgi-bin/announce/add_qun_notice?bkn="+fmt.Sprint(c.getCSRFToken()), []byte(body), c.getCookiesWithDomain("qun.qq.com"))
if err != nil {
return errors.Wrap(err, "request error")
}
return nil
}
func (c *QQClient) GetWordSegmentation(text string) ([]string, error) { func (c *QQClient) GetWordSegmentation(text string) ([]string, error) {
rsp, err := c.sendAndWait(c.buildWordSegmentationPacket([]byte(text))) rsp, err := c.sendAndWait(c.buildWordSegmentationPacket([]byte(text)))
if err != nil { if err != nil {

113
client/group_notice.go Normal file
View File

@ -0,0 +1,113 @@
package client
import (
"bytes"
"fmt"
"html"
"io/ioutil"
"mime/multipart"
"net/http"
"net/textproto"
"net/url"
"strconv"
"github.com/Mrs4s/MiraiGo/utils"
"github.com/pkg/errors"
)
type noticePicUpResponse struct {
ErrorCode int `json:"ec"`
ErrorMessage string `json:"em"`
ID string `json:"id"`
}
type noticeImage struct {
Height string `json:"h"`
Width string `json:"w"`
ID string `json:"id"`
}
func (c *QQClient) uploadGroupNoticePic(img []byte) (*noticeImage, error) {
buf := new(bytes.Buffer)
var w = multipart.NewWriter(buf)
err := w.WriteField("bkn", strconv.Itoa(c.getCSRFToken()))
if err != nil {
return nil, errors.Wrap(err, "write multipart<bkn> failed")
}
err = w.WriteField("source", "troopNotice")
if err != nil {
return nil, errors.Wrap(err, "write multipart<source> failed")
}
err = w.WriteField("m", "0")
if err != nil {
return nil, errors.Wrap(err, "write multipart<m> failed")
}
h := make(textproto.MIMEHeader)
h.Set("Content-Disposition", `form-data; name="pic_up"; filename="temp_uploadFile.png"`)
h.Set("Content-Type", "image/png")
fw, err := w.CreatePart(h)
if err != nil {
return nil, errors.Wrap(err, "create multipart field<pic_up> failed")
}
_, err = fw.Write(img)
if err != nil {
return nil, errors.Wrap(err, "write multipart<pic_up> failed")
}
err = w.Close()
if err != nil {
return nil, errors.Wrap(err, "close multipart failed")
}
req, err := http.NewRequest("POST", "https://web.qun.qq.com/cgi-bin/announce/upload_img", buf)
if err != nil {
return nil, errors.Wrap(err, "new request error")
}
req.Header.Set("Content-Type", w.FormDataContentType())
req.Header.Set("Cookie", c.getCookies())
resp, err := http.DefaultClient.Do(req)
if err != nil {
return nil, errors.Wrap(err, "post error")
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
return nil, errors.Wrap(err, "read body error")
}
res := noticePicUpResponse{}
err = json.Unmarshal(body, &res)
if err != nil {
return nil, errors.Wrap(err, "failed to unmarshal json")
}
if res.ErrorCode != 0 {
return nil, errors.New(res.ErrorMessage)
}
ret := &noticeImage{}
err = json.UnmarshalFromString(html.UnescapeString(res.ID), &ret)
if err != nil {
return nil, errors.Wrap(err, "failed to unmarshal json")
}
return ret, nil
}
// AddGroupNoticeSimple 发群公告
func (c *QQClient) AddGroupNoticeSimple(groupCode int64, text string) error {
body := fmt.Sprintf(`qid=%v&bkn=%v&text=%v&pinned=0&type=1&settings={"is_show_edit_card":0,"tip_window_type":1,"confirm_required":1}`, groupCode, c.getCSRFToken(), url.QueryEscape(text))
_, err := utils.HttpPostBytesWithCookie("https://web.qun.qq.com/cgi-bin/announce/add_qun_notice?bkn="+fmt.Sprint(c.getCSRFToken()), []byte(body), c.getCookiesWithDomain("qun.qq.com"))
if err != nil {
return errors.Wrap(err, "request error")
}
return nil
}
// AddGroupNoticeWithPic 发群公告带图片
func (c *QQClient) AddGroupNoticeWithPic(groupCode int64, text string, pic []byte) error {
img, err := c.uploadGroupNoticePic(pic)
if err != nil {
return err
}
body := fmt.Sprintf(`qid=%v&bkn=%v&text=%v&pinned=0&type=1&settings={"is_show_edit_card":0,"tip_window_type":1,"confirm_required":1}&pic=%v&imgWidth=%v&imgHeight=%v`, groupCode, c.getCSRFToken(), url.QueryEscape(text), img.ID, img.Width, img.Height)
_, err = utils.HttpPostBytesWithCookie("https://web.qun.qq.com/cgi-bin/announce/add_qun_notice?bkn="+fmt.Sprint(c.getCSRFToken()), []byte(body), c.getCookiesWithDomain("qun.qq.com"))
if err != nil {
return errors.Wrap(err, "request error")
}
return nil
}