master
dustoair 3 years ago
parent 3f4aa9f7d4
commit a8997ac318

@ -21,3 +21,12 @@ func RecordWechatLog(ToUsername, FromUsername string, CreateTime uint32, MsgType
}
}
func GetLastWechatLog(ToUsername, MsgType string) WechatLog {
Record := WechatLog{
ToUsername: ToUsername,
MsgType: MsgType,
}
global.DB.Last(&Record)
return Record
}

@ -10,6 +10,6 @@ import (
func replyImage(msgContent MsgContent, timestamp, nonce string, w http.ResponseWriter) {
//third_part.SendPicMid(msgContent.FromUsername, utils.RandMapValue(global.ImageMap))
filename := global.WechatImagePath + "/" + utils.RandomFile(global.WechatImagePath)
go RecordWechatLog(msgContent.ToUsername, msgContent.FromUsername, msgContent.CreateTime, msgContent.MsgType, filename)
go RecordWechatLog(msgContent.FromUsername, msgContent.ToUsername, msgContent.CreateTime, msgContent.MsgType, filename)
third_part.SendPicFile(msgContent.FromUsername, filename)
}

@ -20,8 +20,16 @@ func replyText(msgContent MsgContent, timestamp, nonce string, w http.ResponseWr
} else if msgContent.Content == "history" {
cmdString = utils.SlicePrint(global.HistoryCmds)
} else if msgContent.Content == "删除图片" {
log.Println("删除图片")
return
lastImg := GetLastWechatLog(msgContent.FromUsername, "image")
err := utils.DeleteFile(lastImg.Content)
if err != nil {
log.Println("删除图片失败")
return
} else {
log.Println("删除图片")
return
}
} else {
cmdString = msgContent.Content
}

@ -15,3 +15,12 @@ func RandomFile(dirname string) string {
return fileList[rand.Intn(len(fileList))].Name()
}
func DeleteFile(filename string) error {
err := os.Remove(filename)
if err != nil {
log.Println(err)
return err
}
return nil
}

Loading…
Cancel
Save