47 lines
1.2 KiB
Go
47 lines
1.2 KiB
Go
package getweb
|
|
|
|
import (
|
|
"git.lxtend.com/qqbot/message"
|
|
"git.lxtend.com/qqbot/model"
|
|
"git.lxtend.com/qqbot/util"
|
|
)
|
|
|
|
func init() {
|
|
// handler.RegisterHandler("上网", getweb, constants.LEVEL_TRUSTED)
|
|
// handler.RegisterHelpInform("上网", "getweb", "上网 <URL> 截取网页")
|
|
}
|
|
|
|
func getweb(msg model.Message) (reply *model.Reply) {
|
|
if len(msg.RawMsg) <= len("上网 ") {
|
|
return &model.Reply{}
|
|
}
|
|
url := msg.RawMsg[len("上网 "):]
|
|
url = formatURL(url)
|
|
// if err := util.ScreenshotURL(url, "./tmp/getweb/url.png", 1620, 1960, 0, 0, 0, 0, ""); err != nil {
|
|
if err := util.ScreenshotURL(url, util.GenTempFilePath("getweb/url.png"), 1620, 1960, 0, 0, 0, 0, ""); err != nil {
|
|
return &model.Reply{
|
|
ReplyMsg: err.Error(),
|
|
ReferOriginMsg: true,
|
|
FromMsg: msg,
|
|
}
|
|
}
|
|
imageMsg := message.ImageMessage{
|
|
Type: "image",
|
|
Data: message.ImageMessageData{
|
|
File: "file:///tmp/qqbot/getweb/url.png",
|
|
},
|
|
}
|
|
return &model.Reply{
|
|
ReplyMsg: imageMsg.ToCQString(),
|
|
ReferOriginMsg: true,
|
|
FromMsg: msg,
|
|
}
|
|
}
|
|
|
|
func formatURL(url string) string {
|
|
if url[:7] != "http://" && url[:8] != "https://" {
|
|
return "http://" + url
|
|
}
|
|
return url
|
|
}
|