HEW666/cli
1
0
Fork 0
forked from um/cli
Code Pull Requests Activity

refactor: remove logging in algorithms

Browse Source
This commit is contained in:
Unlock Music Dev 2022-11-19 07:25:40 +08:00
parent a2c55721cc
commit e2fc56ddb2
Signed by untrusted user: um-dev
GPG Key ID: 95202E10D3413A1D
8 changed files with 50 additions and 74 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
algo/common/common.go
View File

@ -1,14 +1,19 @@
package common
import "context"
type Decoder interface {
Validate() error
Decode() error
GetCoverImage() []byte
GetAudioData() []byte
GetAudioExt() string
GetMeta() Meta
}
type CoverImageGetter interface {
GetCoverImage(ctx context.Context) ([]byte, error)
}
type Meta interface {
GetArtists() []string
GetTitle() string

4
algo/common/raw.go
View File

@ -28,10 +28,6 @@ func (d RawDecoder) Decode() error {
return nil
}
func (d RawDecoder) GetCoverImage() []byte {
return nil
}
func (d RawDecoder) GetAudioData() []byte {
return d.file
}

4
algo/kgm/kgm.go
View File

@ -31,10 +31,6 @@ func NewDecoder(file []byte) common.Decoder {
}
}
func (d *Decoder) GetCoverImage() []byte {
return nil
}
func (d *Decoder) GetAudioData() []byte {
return d.audio
}

7
algo/kwm/kwm.go
View File

@ -4,10 +4,11 @@ import (
"bytes"
"encoding/binary"
"errors"
"github.com/unlock-music/cli/algo/common"
"strconv"
"strings"
"unicode"
"github.com/unlock-music/cli/algo/common"
)
var (
@ -31,10 +32,6 @@ type Decoder struct {
audio []byte
}
func (d *Decoder) GetCoverImage() []byte {
return nil
}
func (d *Decoder) GetAudioData() []byte {
return d.audio
}

81
algo/ncm/ncm.go
View File

@ -2,17 +2,18 @@ package ncm
import (
"bytes"
"context"
"encoding/base64"
"encoding/binary"
"encoding/json"
"errors"
"github.com/unlock-music/cli/algo/common"
"github.com/unlock-music/cli/internal/logging"
"github.com/unlock-music/cli/internal/utils"
"go.uber.org/zap"
"io/ioutil"
"fmt"
"io"
"net/http"
"strings"
"github.com/unlock-music/cli/algo/common"
"github.com/unlock-music/cli/internal/utils"
)
var (
@ -157,9 +158,9 @@ func (d *Decoder) readCoverData() error {
iCoverLen := binary.LittleEndian.Uint32(bCoverLen)
d.offsetAudio = coverLenStart + 4 + iCoverLen
if iCoverLen == 0 {
return errors.New("no any cover file found")
return nil
}
d.cover = d.file[coverLenStart+4 : 4+coverLenStart+iCoverLen]
d.cover = d.file[coverLenStart+4 : coverLenStart+4+iCoverLen]
return nil
}
@ -178,27 +179,25 @@ func (d *Decoder) readAudioData() error {
func (d *Decoder) Decode() error {
if err := d.readKeyData(); err != nil {
return err
return fmt.Errorf("read key data failed: %w", err)
}
d.buildKeyBox()
err := d.readMetaData()
if err == nil {
err = d.parseMeta()
if err := d.readMetaData(); err != nil {
return fmt.Errorf("read meta date failed: %w", err)
}
if err != nil {
logging.Log().Warn("parse ncm meta file failed", zap.Error(err))
if err := d.parseMeta(); err != nil {
return fmt.Errorf("parse meta failed: %w", err)
}
err = d.readCoverData()
if err != nil {
logging.Log().Warn("parse ncm cover file failed", zap.Error(err))
if err := d.readCoverData(); err != nil {
return fmt.Errorf("parse ncm cover file failed: %w", err)
}
return d.readAudioData()
}
func (d Decoder) GetAudioExt() string {
func (d *Decoder) GetAudioExt() string {
if d.meta != nil {
if format := d.meta.GetFormat(); format != "" {
return "." + d.meta.GetFormat()
@ -207,40 +206,38 @@ func (d Decoder) GetAudioExt() string {
return ""
}
func (d Decoder) GetAudioData() []byte {
func (d *Decoder) GetAudioData() []byte {
return d.audio
}
func (d Decoder) GetCoverImage() []byte {
func (d *Decoder) GetCoverImage(ctx context.Context) ([]byte, error) {
if d.cover != nil {
return d.cover
return d.cover, nil
}
{
imgURL := d.meta.GetAlbumImageURL()
if d.meta != nil && !strings.HasPrefix(imgURL, "http") {
return nil
}
resp, err := http.Get(imgURL)
if err != nil {
logging.Log().Warn("download image failed", zap.Error(err), zap.String("url", imgURL))
return nil
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
logging.Log().Warn("download image failed", zap.String("http", resp.Status),
zap.String("url", imgURL))
return nil
}
data, err := ioutil.ReadAll(resp.Body)
if err != nil {
logging.Log().Warn("download image failed", zap.Error(err), zap.String("url", imgURL))
return nil
}
return data
}
return nil, nil // no cover image
}
func (d Decoder) GetMeta() common.Meta {
// fetch cover image
req, err := http.NewRequestWithContext(ctx, http.MethodGet, imgURL, nil)
resp, err := http.DefaultClient.Do(req)
if err != nil {
return nil, fmt.Errorf("download image failed: %w", err)
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
return nil, fmt.Errorf("download image failed: unexpected http status %s", resp.Status)
}
data, err := io.ReadAll(resp.Body)
if err != nil {
return nil, fmt.Errorf("download image failed: %w", err)
}
return data, nil
}
func (d *Decoder) GetMeta() common.Meta {
return d.meta
}

4
algo/qmc/qmc.go
View File

@ -244,10 +244,6 @@ func (c *compactDecoder) Decode() error {
return err
}
func (c *compactDecoder) GetCoverImage() []byte {
return nil
}
func (c *compactDecoder) GetAudioData() []byte {
return c.buf.Bytes()
}

5
algo/tm/tm.go
View File

@ -3,6 +3,7 @@ package tm
import (
"bytes"
"errors"
"github.com/unlock-music/cli/algo/common"
)
@ -16,10 +17,6 @@ type Decoder struct {
audioExt string
}
func (d *Decoder) GetCoverImage() []byte {
return nil
}
func (d *Decoder) GetAudioData() []byte {
return d.audio
}

10
algo/xm/xm.go
View File

@ -3,9 +3,8 @@ package xm
import (
"bytes"
"errors"
"github.com/unlock-music/cli/algo/common"
"github.com/unlock-music/cli/internal/logging"
"go.uber.org/zap"
)
var (
@ -29,10 +28,6 @@ type Decoder struct {
audio []byte
}
func (d *Decoder) GetCoverImage() []byte {
return nil
}
func (d *Decoder) GetAudioData() []byte {
return d.audio
}
@ -69,9 +64,6 @@ func (d *Decoder) Validate() error {
return errors.New("detect unknown xm file type: " + string(d.file[4:8]))
}
if d.file[14] != 0 {
logging.Log().Warn("not a simple xm file", zap.Uint8("b[14]", d.file[14]))
}
d.headerLen = uint32(d.file[12]) | uint32(d.file[13])<<8 | uint32(d.file[14])<<16 // LittleEndian Unit24
if d.headerLen+16 > uint32(lenData) {
return ErrFileSize