This commit is contained in:
cdle
2021-10-07 15:47:36 +08:00
parent f0b4dc0fe4
commit 1be3c13381
+25 -27
View File
@@ -1,9 +1,7 @@
package tg package tg
import ( import (
"bytes"
"fmt" "fmt"
"image/png"
"os" "os"
"regexp" "regexp"
"strings" "strings"
@@ -12,7 +10,6 @@ import (
"github.com/astaxie/beego/httplib" "github.com/astaxie/beego/httplib"
"github.com/beego/beego/v2/core/logs" "github.com/beego/beego/v2/core/logs"
"github.com/cdle/sillyGirl/core" "github.com/cdle/sillyGirl/core"
"golang.org/x/image/webp"
tb "gopkg.in/tucnak/telebot.v2" tb "gopkg.in/tucnak/telebot.v2"
) )
@@ -114,30 +111,31 @@ func init() {
Handler(m) Handler(m)
} }
}) })
b.Handle(tb.OnSticker, func(m *tb.Message) { // b.Handle(tb.OnSticker, func(m *tb.Message) {
buf := new(bytes.Buffer) // buf := new(bytes.Buffer)
buf.ReadFrom(m.Sticker.FileReader) // buf.ReadFrom(m.Sticker.FileReader)
img, err := webp.Decode(buf) // img, err := webp.Decode(buf)
if err != nil { // if err != nil {
return
} // return
buf.Reset() // }
imgBuf := buf // buf.Reset()
err = png.Encode(imgBuf, img) // imgBuf := buf
if err != nil { // err = png.Encode(imgBuf, img)
return // if err != nil {
} // return
filename := fmt.Sprint(time.Now().UnixNano()) + ".image" // }
filepath := core.ExecPath + "/data/images/" + filename // filename := fmt.Sprint(time.Now().UnixNano()) + ".image"
f, err := os.Create(filepath) // filepath := core.ExecPath + "/data/images/" + filename
if err != nil { // f, err := os.Create(filepath)
return // if err != nil {
} // return
f.Write(imgBuf.Bytes()) // }
f.Close() // f.Write(imgBuf.Bytes())
m.Text = fmt.Sprintf(`[TG:image,file=%s]`, filename) + m.Caption // f.Close()
Handler(m) // m.Text = fmt.Sprintf(`[TG:image,file=%s]`, filename) + m.Caption
}) // Handler(m)
// })
b.Handle(tb.OnText, Handler) b.Handle(tb.OnText, Handler)
logs.Info("监听telegram机器人") logs.Info("监听telegram机器人")
b.Start() b.Start()