shkong/go-cqhttp
1
0
Fork 0
mirror of https://github.com/Mrs4s/go-cqhttp.git synced 2025-05-05 03:23:49 +08:00
Code Issues Projects Releases Wiki Activity

coolq: handle extract cover error

Browse Source 
For #1426
This commit is contained in:
wdvxdr 2022-03-18 23:50:21 +08:00
parent de4cfe0733
commit d34531790c
No known key found for this signature in database
GPG Key ID: 703F8C071DE7A1B6
1 changed files with 6 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
coolq/cqcode.go
View File

@ -1103,21 +1103,18 @@ func (bot *CQBot) ToElement(t string, d map[string]string, sourceType message.So
if cover, ok := d["cover"]; ok { if cover, ok := d["cover"]; ok {
data, _ = global.FindFile(cover, d["cache"], global.ImagePath) data, _ = global.FindFile(cover, d["cache"], global.ImagePath)
} else { } else {
_ = global.ExtractCover(v.File, v.File+".jpg") err = global.ExtractCover(v.File, v.File+".jpg")
if err != nil {
return nil, err
}
data, _ = os.ReadFile(v.File + ".jpg") data, _ = os.ReadFile(v.File + ".jpg")
} }
v.thumb = bytes.NewReader(data) v.thumb = bytes.NewReader(data)
video, _ := os.Open(v.File) video, _ := os.Open(v.File)
defer video.Close() defer video.Close()
_, err = video.Seek(4, io.SeekStart) _, _ = video.Seek(4, io.SeekStart)
if err != nil {
return nil, err
}
header := make([]byte, 4) header := make([]byte, 4)
_, err = video.Read(header) _, _ = video.Read(header)
if err != nil {
return nil, err
}
if !bytes.Equal(header, []byte{0x66, 0x74, 0x79, 0x70}) { // check file header ftyp if !bytes.Equal(header, []byte{0x66, 0x74, 0x79, 0x70}) { // check file header ftyp
_, _ = video.Seek(0, io.SeekStart) _, _ = video.Seek(0, io.SeekStart)
hash, _ := utils.ComputeMd5AndLength(video) hash, _ := utils.ComputeMd5AndLength(video)