Fix Tm Decoder

This commit is contained in:
Emmm Monster 2020-12-26 15:03:34 +08:00
parent 7c0d6604a2
commit 1aa5731225
No known key found for this signature in database
GPG Key ID: C98279C83FB50DB9

View File

@ -6,10 +6,14 @@ import (
"github.com/umlock-music/cli/algo/common" "github.com/umlock-music/cli/algo/common"
) )
var magicHeader = []byte{0x00, 0x00, 0x00, 0x20, 0x66, 0x74, 0x79, 0x70} var replaceHeader = []byte{0x00, 0x00, 0x00, 0x20, 0x66, 0x74, 0x79, 0x70}
var magicHeader = []byte{0x51, 0x51, 0x4D, 0x55} //0x15, 0x1D, 0x1A, 0x21
type Decoder struct { type Decoder struct {
file []byte file []byte
audio []byte
headerMatch bool
audioExt string
} }
func (d *Decoder) GetCoverImage() []byte { func (d *Decoder) GetCoverImage() []byte {
@ -17,11 +21,11 @@ func (d *Decoder) GetCoverImage() []byte {
} }
func (d *Decoder) GetAudioData() []byte { func (d *Decoder) GetAudioData() []byte {
return d.file[8:] return d.audio
} }
func (d *Decoder) GetAudioExt() string { func (d *Decoder) GetAudioExt() string {
return "" return d.audioExt
} }
func (d *Decoder) GetMeta() common.Meta { func (d *Decoder) GetMeta() common.Meta {
@ -36,12 +40,20 @@ func (d *Decoder) Validate() error {
if len(d.file) < 8 { if len(d.file) < 8 {
return errors.New("invalid file size") return errors.New("invalid file size")
} }
if !bytes.Equal(magicHeader, d.file[:8]) { if !bytes.Equal(magicHeader, d.file[:4]) {
return errors.New("not a valid tm file") return errors.New("not a valid tm file")
} }
d.headerMatch = true
return nil return nil
} }
func (d *Decoder) Decode() error { func (d *Decoder) Decode() error {
d.audio = d.file
if d.headerMatch {
for i := 0; i < 8; i++ {
d.audio[i] = replaceHeader[i]
}
d.audioExt = "m4a"
}
return nil return nil
} }