chickliu/Sillyv2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
cdle
2021-10-10 23:51:10 +08:00
parent 0562ba174c
commit 73af826bbb
2 changed files with 2 additions and 270 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.gitignore
+2 -1
View File
@@ -14,4 +14,5 @@ replies
qinglongs
sets.conf
yawx
xyo
xyo
wx
im/wx/init.go
-269
View File
@@ -1,269 +0,0 @@
package wx
import (
"crypto/md5"
"encoding/hex"
"encoding/json"
"fmt"
"io"
"os"
"regexp"
"strings"
"github.com/axgle/mahonia"
"github.com/beego/beego/v2/adapter/httplib"
"github.com/cdle/sillyGirl/core"
"github.com/gin-gonic/gin"
)
var wx = core.NewBucket("wx")
var api_url = wx.Get("api_url")
var robot_wxid = wx.Get("robot_wxid")
func sendTextMsg(pmsg *TextMsg) {
pmsg.Msg = TrimHiddenCharacter(pmsg.Msg)
if pmsg.Msg == "" {
return
}
pmsg.Event = "SendTextMsg"
pmsg.RobotWxid = robot_wxid
req := httplib.Post(api_url)
req.Header("Content-Type", "application/json")
data, _ := json.Marshal(pmsg)
enc := mahonia.NewEncoder("gbk")
d := enc.ConvertString(string(data))
d = regexp.MustCompile(`[\n\s]*\n[\s\n]*`).ReplaceAllString(d, "\n")
req.Body(d)
req.Response()
}
func TrimHiddenCharacter(originStr string) string {
srcRunes := []rune(originStr)
dstRunes := make([]rune, 0, len(srcRunes))
for _, c := range srcRunes {
if c >= 0 && c <= 31 && c != 10 {
continue
}
if c == 127 {
continue
}
dstRunes = append(dstRunes, c)
}
return string(dstRunes)
}
func sendOtherMsg(pmsg *OtherMsg) {
if pmsg.Event == "" {
pmsg.Event = "SendImageMsg"
}
pmsg.RobotWxid = robot_wxid
req := httplib.Post(api_url)
req.Header("Content-Type", "application/json")
data, _ := json.Marshal(pmsg)
req.Body(data)
req.Response()
}
type TextMsg struct {
Event string `json:"event"`
ToWxid string `json:"to_wxid"`
Msg string `json:"msg"`
RobotWxid string `json:"robot_wxid"`
GroupWxid string `json:"group_wxid"`
MemberWxid string `json:"member_wxid"`
}
type OtherMsg struct {
Success bool `json:"success"`
Message string `json:"message"`
Event string `json:"event"`
RobotWxid string `json:"robot_wxid"`
ToWxid string `json:"to_wxid"`
MemberWxid string `json:"member_wxid"`
MemberName string `json:"member_name"`
GroupWxid string `json:"group_wxid"`
Msg Msg `json:"msg"`
}
type Msg struct {
URL string `json:"url"`
Name string `json:"name"`
}
func init() {
core.Pushs["wx"] = func(i interface{}, s string) {
if robot_wxid != "" {
pmsg := TextMsg{
Msg: s,
ToWxid: fmt.Sprint(i),
}
sendTextMsg(&pmsg)
}
}
core.GroupPushs["wx"] = func(i, j interface{}, s string) {
to := fmt.Sprint(i) + "@chatroom"
pmsg := TextMsg{
ToWxid: to,
}
if j != nil && fmt.Sprint(j) != "" {
pmsg.MemberWxid = fmt.Sprint(j)
}
for _, v := range regexp.MustCompile(`\[CQ:image,file=([^\[\]]+)\]`).FindAllStringSubmatch(s, -1) {
s = strings.Replace(s, fmt.Sprintf(`[CQ:image,file=%s]`, v[1]), "", -1)
data, err := os.ReadFile(core.ExecPath + "/data/images/" + v[1])
if err == nil {
add := regexp.MustCompile("(https.*)").FindString(string(data))
if add != "" {
pmsg := OtherMsg{
ToWxid: to,
Msg: Msg{
URL: relay(add),
Name: name(add),
},
}
defer sendOtherMsg(&pmsg)
}
}
}
s = regexp.MustCompile(`\[CQ:([^\[\]]+)\]`).ReplaceAllString(s, "")
pmsg.Msg = s
sendTextMsg(&pmsg)
}
core.Server.POST("/wx/receive", func(c *gin.Context) {
data, _ := c.GetRawData()
jms := JsonMsg{}
json.Unmarshal(data, &jms)
c.JSON(200, map[string]string{"code": "-1"})
if jms.Event != "EventFriendMsg" && jms.Event != "EventGroupMsg" {
return
}
if jms.Type != 1 && jms.Type != 3 {
return
}
if jms.FinalFromWxid == jms.RobotWxid {
return
}
if robot_wxid != jms.RobotWxid {
robot_wxid = jms.RobotWxid
wx.Set("robot_wxid", robot_wxid)
}
core.Senders <- &Sender{
value: jms,
}
})
core.Server.GET("/relay", func(c *gin.Context) {
url := c.Query("url")
rsp, err := httplib.Get(url).Response()
if err == nil {
io.Copy(c.Writer, rsp.Body)
}
})
}
var myip = ""
func relay(url string) string {
ip, _ := httplib.Get("https://imdraw.com/ip").String()
if ip != "" {
myip = ip
}
return fmt.Sprintf("http://%s:%s/relay?url=%s", myip, wx.Get("relay_port", "8002"), url)
}
type Sender struct {
leixing int
mtype int
deleted bool
value JsonMsg
core.BaseSender
}
type JsonMsg struct {
Event string `json:"event"`
RobotWxid string `json:"robot_wxid"`
RobotName string `json:"robot_name"`
Type int `json:"type"`
FromWxid string `json:"from_wxid"`
FromName string `json:"from_name"`
FinalFromWxid string `json:"final_from_wxid"`
FinalFromName string `json:"final_from_name"`
ToWxid string `json:"to_wxid"`
Msg string `json:"msg"`
}
func (sender *Sender) GetContent() string {
return sender.value.Msg
}
func (sender *Sender) GetUserID() interface{} {
return sender.value.FinalFromWxid
}
func (sender *Sender) GetChatID() interface{} {
return strings.Replace(sender.value.FromWxid, "@chatroom", "", -1)
}
func (sender *Sender) GetImType() string {
return "wx"
}
func (sender *Sender) GetUsername() string {
return sender.value.FinalFromName
}
func (sender *Sender) GetReplySenderUserID() int {
if !sender.IsReply() {
return 0
}
return 0
}
func (sender *Sender) IsAdmin() bool {
return strings.Contains(wx.Get("masters"), fmt.Sprint(sender.GetUserID()))
}
func (sender *Sender) Reply(msgs ...interface{}) (int, error) {
to := sender.value.FromWxid
at := ""
if to == "" {
to = sender.value.FinalFromWxid
} else {
at = sender.value.FinalFromWxid
}
pmsg := TextMsg{
ToWxid: to,
MemberWxid: at,
}
for _, item := range msgs {
switch item.(type) {
case string:
pmsg.Msg = item.(string)
case []byte:
pmsg.Msg = string(item.([]byte))
case core.ImageUrl:
url := string(item.(core.ImageUrl))
pmsg := OtherMsg{
ToWxid: to,
MemberWxid: at,
Msg: Msg{
URL: relay(url),
Name: name(url),
},
}
sendOtherMsg(&pmsg)
}
}
if pmsg.Msg != "" {
sendTextMsg(&pmsg)
}
return 0, nil
}
func name(str string) string {
pr := "jpg"
ss := regexp.MustCompile(`\.([A-Za-z0-9]+)$`).FindStringSubmatch(str)
if len(ss) != 0 {
pr = ss[1]
}
md5 := md5V(str)
return md5 + "." + pr
}
func md5V(str string) string {
h := md5.New()
h.Write([]byte(str))
return hex.EncodeToString(h.Sum(nil))
}