chtwrs/chirpnest
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

test

Browse Source
This commit is contained in:
shoopea 2019-06-09 18:20:08 +08:00
parent 71c28649cc
commit b2c5eaa754
1 changed files with 16 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
bot.go
View File

@ -288,18 +288,25 @@ func botMsgLoad(m *tb.Message) {
TGCmdQueue <- c TGCmdQueue <- c
r, err := b.GetFile(&f) r, err := b.GetFile(&f)
logOnError(err, "botMsgLoad : GetFile") logOnError(err, "botMsgLoad : GetFile")
_, err := ioutil.ReadAll(r.Body) buf := new(bytes.Buffer)
c = TGCommand{ n, err := buf.ReadFrom(r.Reader)
Type: commandReplyMsg, logOnError(err, "botMsgLoad : ReadFrom")
Text: "File downloaded ?", if n > 0 {
FromMsgID64: int64(m.ID), data := buf.Bytes()
FromChatID64: m.Chat.ID, s := fmt.Sprintf("Read %d/%d bytes.", n, len(data))
c = TGCommand{
Type: commandReplyMsg,
Text: s,
FromMsgID64: int64(m.ID),
FromChatID64: m.Chat.ID,
}
TGCmdQueue <- c
} }
TGCmdQueue <- c
if f.OnDisk() { if f.OnDisk() {
c = TGCommand{ c = TGCommand{
Type: commandReplyMsg, Type: commandReplyMsg,
Text: "Yes !", Text: "File now on disk !",
FromMsgID64: int64(m.ID), FromMsgID64: int64(m.ID),
FromChatID64: m.Chat.ID, FromChatID64: m.Chat.ID,
} }
@ -307,7 +314,7 @@ func botMsgLoad(m *tb.Message) {
} else { } else {
c = TGCommand{ c = TGCommand{
Type: commandReplyMsg, Type: commandReplyMsg,
Text: "No !", Text: "Still not on disk ?",
FromMsgID64: int64(m.ID), FromMsgID64: int64(m.ID),
FromChatID64: m.Chat.ID, FromChatID64: m.Chat.ID,
} }