⬥ ID3 encoding/decoding library in Go ⬥
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

205 lines
3.8 KiB

package v1
import (
"bytes"
"fmt"
"io"
"strconv"
"github.com/Unbewohnte/id3ed/util"
)
var errDoesNotUseEnhancedID3v1 error = fmt.Errorf("does not use enhanced ID3v1 tag")
// Checks if rs contains a regular ID3v1 TAG
func containsTAG(rs io.ReadSeeker) bool {
_, err := rs.Seek(-int64(TAGSIZE), io.SeekEnd)
if err != nil {
return false
}
identifier, err := util.Read(rs, 3)
if err != nil {
return false
}
if string(identifier) != IDENTIFIER {
return false
}
return true
}
// Checks if enhanced tag is used
func containsEnhancedTAG(rs io.ReadSeeker) bool {
_, err := rs.Seek(-int64(TAGSIZE+ENHANCEDSIZE), io.SeekEnd)
if err != nil {
return false
}
identifier, err := util.Read(rs, 4)
if err != nil {
return false
}
if !bytes.Equal(identifier, []byte(ENHANCEDIDENTIFIER)) {
return false
}
return true
}
// Tries to read enhanced ID3V1 tag from rs
func readEnhancedTag(rs io.ReadSeeker) (EnhancedID3v1Tag, error) {
if !containsEnhancedTAG(rs) {
// rs does not contain enhanced TAG, there is nothing to read
return EnhancedID3v1Tag{}, errDoesNotUseEnhancedID3v1
}
var enhanced EnhancedID3v1Tag
// set reader into the position
_, err := rs.Seek(-int64(TAGSIZE+ENHANCEDSIZE), io.SeekEnd)
if err != nil {
return enhanced, fmt.Errorf("could not seek: %s", err)
}
// songname
songName, err := util.ReadToString(rs, 60)
if err != nil {
return EnhancedID3v1Tag{}, err
}
enhanced.SongName = songName
artist, err := util.ReadToString(rs, 60)
if err != nil {
return enhanced, err
}
enhanced.Artist = artist
// album
album, err := util.ReadToString(rs, 60)
if err != nil {
return enhanced, err
}
enhanced.Album = album
// speed
speedByte, err := util.Read(rs, 1)
if err != nil {
return enhanced, err
}
var speed string = EnhancedSpeed[int(speedByte[0])]
enhanced.Speed = speed
// genre
genre, err := util.ReadToString(rs, 30)
if err != nil {
return enhanced, err
}
enhanced.Genre = genre
// time
startTime, err := util.ReadToString(rs, 6)
if err != nil {
return enhanced, err
}
enhanced.StartTime = startTime
endtime, err := util.ReadToString(rs, 6)
if err != nil {
return enhanced, err
}
enhanced.EndTime = endtime
return enhanced, nil
}
// Retrieves ID3v1 field values from rs.
func Readv1Tag(rs io.ReadSeeker) (*ID3v1Tag, error) {
var tag ID3v1Tag
// check if need to read enhanced tag
if containsEnhancedTAG(rs) {
enhanced, _ := readEnhancedTag(rs)
tag.HasEnhancedTag = true
tag.EnhancedTag = enhanced
}
if !containsTAG(rs) {
// no TAG to read
return nil, ErrDoesNotUseID3v1
}
// Songname
songname, err := util.ReadToString(rs, 30)
if err != nil {
return nil, err
}
tag.SongName = songname
// Artist
artist, err := util.ReadToString(rs, 30)
if err != nil {
return nil, err
}
tag.Artist = artist
// Album name
album, err := util.ReadToString(rs, 30)
if err != nil {
return nil, err
}
tag.Album = album
// Year
yearStr, err := util.ReadToString(rs, 4)
if err != nil {
return nil, err
}
year, err := strconv.Atoi(yearStr)
if err != nil {
return nil, fmt.Errorf("could not convert yearbytes into int: %s", err)
}
tag.Year = year
// Comment and Track
comment, err := util.Read(rs, 30)
if err != nil {
return nil, err
}
tag.Comment = util.ToStringLossy(comment)
tag.Track = 0
var track int = 0
// check if 29th byte is null byte (v1.0 or v1.1)
if comment[28] == 0 {
// it is v1.1, track number exists
track = int(comment[29])
tag.Track = uint8(track)
comment = comment[0:28]
tag.Comment = util.ToStringLossy(comment)
}
// Genre
genreByte, err := util.Read(rs, 1)
if err != nil {
return nil, err
}
genreInt := int(genreByte[0])
genre, exists := id3v1genres[int(genreInt)]
if !exists {
genre = ""
}
tag.Genre = genre
if track == 0 {
tag.version = V1_0
} else {
tag.version = V1_1
}
return &tag, nil
}