unlock-music-cli/algo/common/raw.go

package common

import (
	"errors"
	"fmt"
	"io"

	"unlock-music.dev/cli/internal/sniff"
)

type RawDecoder struct {
	rd io.ReadSeeker

	audioExt string
}

func NewRawDecoder(p *DecoderParams) Decoder {
	return &RawDecoder{rd: p.Reader}
}

func (d *RawDecoder) Validate() error {
	header := make([]byte, 16)
	if _, err := io.ReadFull(d.rd, header); err != nil {
		return fmt.Errorf("read file header failed: %v", err)
	}
	if _, err := d.rd.Seek(0, io.SeekStart); err != nil {
		return fmt.Errorf("seek file failed: %v", err)
	}

	var ok bool
	d.audioExt, ok = sniff.AudioExtension(header)
	if !ok {
		return errors.New("raw: sniff audio type failed")
	}
	return nil
}

func (d *RawDecoder) Read(p []byte) (n int, err error) {
	return d.rd.Read(p)
}

func init() {
	RegisterDecoder("mp3", true, NewRawDecoder)
	RegisterDecoder("flac", true, NewRawDecoder)
	RegisterDecoder("ogg", true, NewRawDecoder)
	RegisterDecoder("m4a", true, NewRawDecoder)
	RegisterDecoder("wav", true, NewRawDecoder)
	RegisterDecoder("wma", true, NewRawDecoder)
	RegisterDecoder("aac", true, NewRawDecoder)
}