77 lines
1.5 KiB
Go
77 lines
1.5 KiB
Go
package tm
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
|
|
"github.com/unlock-music/cli/algo/common"
|
|
)
|
|
|
|
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 {
|
|
file []byte
|
|
audio []byte
|
|
headerMatch bool
|
|
audioExt string
|
|
}
|
|
|
|
func (d *Decoder) GetAudioData() []byte {
|
|
return d.audio
|
|
}
|
|
|
|
func (d *Decoder) GetAudioExt() string {
|
|
if d.audioExt != "" {
|
|
return "." + d.audioExt
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func (d *Decoder) GetMeta() common.Meta {
|
|
return nil
|
|
}
|
|
|
|
func (d *Decoder) Validate() error {
|
|
if len(d.file) < 8 {
|
|
return errors.New("invalid file size")
|
|
}
|
|
if !bytes.Equal(magicHeader, d.file[:4]) {
|
|
return errors.New("not a valid tm file")
|
|
}
|
|
d.headerMatch = true
|
|
return nil
|
|
}
|
|
|
|
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
|
|
}
|
|
|
|
//goland:noinspection GoUnusedExportedFunction
|
|
func NewDecoder(data []byte) common.Decoder {
|
|
return &Decoder{file: data}
|
|
}
|
|
|
|
func DecoderFuncWithExt(ext string) common.NewDecoderFunc {
|
|
return func(file []byte) common.Decoder {
|
|
return &Decoder{file: file, audioExt: ext}
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
// QQ Music IOS M4a
|
|
common.RegisterDecoder("tm2", false, DecoderFuncWithExt("m4a"))
|
|
common.RegisterDecoder("tm6", false, DecoderFuncWithExt("m4a"))
|
|
// QQ Music IOS Mp3
|
|
common.RegisterDecoder("tm0", false, common.NewRawDecoder)
|
|
common.RegisterDecoder("tm3", false, common.NewRawDecoder)
|
|
|
|
}
|