Unbewohnte
3 years ago
13 changed files with 104 additions and 392 deletions
Binary file not shown.
@ -1,85 +0,0 @@ |
|||||||
package v1 |
|
||||||
|
|
||||||
import ( |
|
||||||
"fmt" |
|
||||||
"os" |
|
||||||
"path/filepath" |
|
||||||
"testing" |
|
||||||
) |
|
||||||
|
|
||||||
var TESTV11TAG = &ID3v11Tag{ |
|
||||||
SongName: "testsong", |
|
||||||
Artist: "testartist", |
|
||||||
Album: "testalbum", |
|
||||||
Year: 727, |
|
||||||
Comment: "testcomment", |
|
||||||
Track: 5, |
|
||||||
Genre: "Blues", |
|
||||||
} |
|
||||||
|
|
||||||
func TestGetID3v11Tags(t *testing.T) { |
|
||||||
testfile, err := os.Open(filepath.Join(TESTDATAPATH, "testreadv1.mp3")) |
|
||||||
if err != nil { |
|
||||||
t.Errorf("could not open file for testing: %s", err) |
|
||||||
} |
|
||||||
mp3tags, err := Getv11Tag(testfile) |
|
||||||
if err != nil { |
|
||||||
t.Errorf("GetID3v11Tag failed: %s", err) |
|
||||||
} |
|
||||||
|
|
||||||
if mp3tags.Artist != "Artist" { |
|
||||||
fmt.Printf("%v", mp3tags.Artist) |
|
||||||
t.Errorf("GetID3v11Tag failed: expected artist %s; got %s", "Artist", mp3tags.Artist) |
|
||||||
} |
|
||||||
|
|
||||||
if mp3tags.Track != 8 { |
|
||||||
t.Errorf("GetID3v11Tag failed: expected track %d; got %d", 8, mp3tags.Track) |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
// WILL ADD NEW "TAG" WITHOUT REMOVING THE OLD ONE !!!
|
|
||||||
func TestWriteID3v11Tags(t *testing.T) { |
|
||||||
f, err := os.OpenFile(filepath.Join(TESTDATAPATH, "testwritev1.mp3"), os.O_CREATE|os.O_RDWR, os.ModePerm) |
|
||||||
if err != nil { |
|
||||||
t.Errorf("%s", err) |
|
||||||
} |
|
||||||
defer f.Close() |
|
||||||
|
|
||||||
tag := TESTV11TAG |
|
||||||
|
|
||||||
err = tag.Write(f) |
|
||||||
if err != nil { |
|
||||||
t.Errorf("WriteID3v1Tags failed: %s", err) |
|
||||||
} |
|
||||||
|
|
||||||
readTags, err := Getv11Tag(f) |
|
||||||
if err != nil { |
|
||||||
t.Errorf("%s", err) |
|
||||||
} |
|
||||||
|
|
||||||
if readTags.Album != "testalbum" { |
|
||||||
t.Errorf("WriteID3v11Tag failed: expected album %s; got %s", "testalbum", readTags.Album) |
|
||||||
} |
|
||||||
|
|
||||||
if readTags.Year != 727 { |
|
||||||
t.Errorf("WriteID3v11Tag failed: expected year %d; got %d", 727, readTags.Year) |
|
||||||
} |
|
||||||
|
|
||||||
if readTags.Track != 10 { |
|
||||||
t.Errorf("WriteID3v11Tag failed: expected track %d; got %d", 10, readTags.Track) |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
func TestWriteID3v11ToFile(t *testing.T) { |
|
||||||
f, err := os.OpenFile(filepath.Join(TESTDATAPATH, "testwritev1.mp3"), os.O_CREATE|os.O_RDWR, os.ModePerm) |
|
||||||
if err != nil { |
|
||||||
t.Errorf("%s", err) |
|
||||||
} |
|
||||||
|
|
||||||
tag := TESTV11TAG |
|
||||||
|
|
||||||
err = tag.WriteToFile(f) |
|
||||||
if err != nil { |
|
||||||
t.Errorf("WriteID3v1ToFile failed: %s", err) |
|
||||||
} |
|
||||||
} |
|
@ -1,15 +0,0 @@ |
|||||||
package v1 |
|
||||||
|
|
||||||
type ID3v11Tag struct { |
|
||||||
SongName string |
|
||||||
Artist string |
|
||||||
Album string |
|
||||||
Year int |
|
||||||
Comment string |
|
||||||
Track int |
|
||||||
Genre string |
|
||||||
} |
|
||||||
|
|
||||||
func (*ID3v11Tag) Version() int { |
|
||||||
return 11 |
|
||||||
} |
|
@ -1,129 +0,0 @@ |
|||||||
package v1 |
|
||||||
|
|
||||||
import ( |
|
||||||
"bytes" |
|
||||||
"encoding/binary" |
|
||||||
"fmt" |
|
||||||
"io" |
|
||||||
"os" |
|
||||||
|
|
||||||
"github.com/Unbewohnte/id3ed/util" |
|
||||||
) |
|
||||||
|
|
||||||
// Writes given ID3v1.1 tag to dst
|
|
||||||
// NOTE: will not remove already existing ID3v1.1 tag if it`s present,
|
|
||||||
// use ⁕WriteToFile⁕ method if you`re working with REAL mp3 files !!!
|
|
||||||
func (tag *ID3v11Tag) Write(dst io.WriteSeeker) error { |
|
||||||
dst.Seek(0, io.SeekEnd) |
|
||||||
|
|
||||||
// TAG
|
|
||||||
_, err := dst.Write([]byte(ID3v1IDENTIFIER)) |
|
||||||
if err != nil { |
|
||||||
return fmt.Errorf("could not write to dst: %s", err) |
|
||||||
} |
|
||||||
|
|
||||||
// Song name
|
|
||||||
err = util.WriteToExtent(dst, []byte(tag.SongName), 30) |
|
||||||
if err != nil { |
|
||||||
return fmt.Errorf("could not write to dst: %s", err) |
|
||||||
} |
|
||||||
|
|
||||||
// Artist
|
|
||||||
err = util.WriteToExtent(dst, []byte(tag.Artist), 30) |
|
||||||
if err != nil { |
|
||||||
return fmt.Errorf("could not write to dst: %s", err) |
|
||||||
} |
|
||||||
|
|
||||||
// Album
|
|
||||||
err = util.WriteToExtent(dst, []byte(tag.Album), 30) |
|
||||||
if err != nil { |
|
||||||
return fmt.Errorf("could not write to dst: %s", err) |
|
||||||
} |
|
||||||
|
|
||||||
// Year
|
|
||||||
err = util.WriteToExtent(dst, []byte(fmt.Sprint(tag.Year)), 4) |
|
||||||
if err != nil { |
|
||||||
return fmt.Errorf("could not write to dst: %s", err) |
|
||||||
} |
|
||||||
|
|
||||||
// Comment
|
|
||||||
err = util.WriteToExtent(dst, []byte(tag.Comment), 28) |
|
||||||
if err != nil { |
|
||||||
return fmt.Errorf("could not write to dst: %s", err) |
|
||||||
} |
|
||||||
|
|
||||||
_, err = dst.Write([]byte{0}) |
|
||||||
if err != nil { |
|
||||||
return fmt.Errorf("could not write to dst: %s", err) |
|
||||||
} |
|
||||||
|
|
||||||
// Track
|
|
||||||
trackBytes := make([]byte, 1) |
|
||||||
binary.PutVarint(trackBytes, int64(tag.Track)) |
|
||||||
_, err = dst.Write(trackBytes) |
|
||||||
if err != nil { |
|
||||||
return fmt.Errorf("could not write to dst: %s", err) |
|
||||||
} |
|
||||||
|
|
||||||
//Genre
|
|
||||||
genreCode := util.GetKey(id3v1genres, tag.Genre) |
|
||||||
if genreCode == -1 { |
|
||||||
// if no genre found - encode genre code as 255
|
|
||||||
genreCode = ID3v1INVALIDGENRE |
|
||||||
} |
|
||||||
genrebyte := make([]byte, 1) |
|
||||||
binary.PutVarint(genrebyte, int64(genreCode)) |
|
||||||
|
|
||||||
err = util.WriteToExtent(dst, genrebyte, 1) |
|
||||||
if err != nil { |
|
||||||
return fmt.Errorf("could not write to dst: %s", err) |
|
||||||
} |
|
||||||
|
|
||||||
return nil |
|
||||||
} |
|
||||||
|
|
||||||
// Checks for existing ID3v1.1 tag in file, if present - removes it and replaces with provided tag
|
|
||||||
func (tag *ID3v11Tag) WriteToFile(f *os.File) error { |
|
||||||
defer f.Close() |
|
||||||
|
|
||||||
// check for existing ID3v1.1 tag
|
|
||||||
_, err := f.Seek(-int64(ID3v1SIZE), io.SeekEnd) |
|
||||||
if err != nil { |
|
||||||
return fmt.Errorf("could not seek: %s", err) |
|
||||||
} |
|
||||||
|
|
||||||
identifier, err := util.Read(f, 3) |
|
||||||
if err != nil { |
|
||||||
// return err
|
|
||||||
return err |
|
||||||
|
|
||||||
} |
|
||||||
|
|
||||||
if !bytes.Equal(identifier, []byte(ID3v1IDENTIFIER)) { |
|
||||||
// no existing identifier, just write given tag
|
|
||||||
err = tag.Write(f) |
|
||||||
if err != nil { |
|
||||||
return fmt.Errorf("could not write to writer: %s", err) |
|
||||||
} |
|
||||||
return nil |
|
||||||
} |
|
||||||
|
|
||||||
// does contain ID3v1.1 tag. Removing it
|
|
||||||
fStats, err := f.Stat() |
|
||||||
if err != nil { |
|
||||||
return fmt.Errorf("cannot get file stats: %s", err) |
|
||||||
} |
|
||||||
|
|
||||||
err = f.Truncate(fStats.Size() - int64(ID3v1SIZE)) |
|
||||||
if err != nil { |
|
||||||
return fmt.Errorf("could not truncate file %s", err) |
|
||||||
} |
|
||||||
|
|
||||||
// writing new tag
|
|
||||||
err = tag.Write(f) |
|
||||||
if err != nil { |
|
||||||
return fmt.Errorf("could not write to writer: %s", err) |
|
||||||
} |
|
||||||
|
|
||||||
return nil |
|
||||||
} |
|
@ -1,80 +0,0 @@ |
|||||||
package v1 |
|
||||||
|
|
||||||
import ( |
|
||||||
"bytes" |
|
||||||
"fmt" |
|
||||||
"io" |
|
||||||
"strconv" |
|
||||||
|
|
||||||
"github.com/Unbewohnte/id3ed/util" |
|
||||||
) |
|
||||||
|
|
||||||
// Retrieves ID3v1 field values of provided io.ReadSeeker (usually a file)
|
|
||||||
func Getv1Tag(rs io.ReadSeeker) (*ID3v1Tag, error) { |
|
||||||
// set reader to the last 128 bytes
|
|
||||||
_, err := rs.Seek(-int64(ID3v1SIZE), io.SeekEnd) |
|
||||||
if err != nil { |
|
||||||
return nil, fmt.Errorf("could not seek: %s", err) |
|
||||||
} |
|
||||||
|
|
||||||
tag, err := util.Read(rs, 3) |
|
||||||
if err != nil { |
|
||||||
return nil, err |
|
||||||
} |
|
||||||
|
|
||||||
if !bytes.Equal(tag, []byte(ID3v1IDENTIFIER)) { |
|
||||||
// no TAG, given file does not use ID3v1
|
|
||||||
return nil, fmt.Errorf("does not use ID3v1: expected %s; got %s", ID3v1IDENTIFIER, tag) |
|
||||||
} |
|
||||||
|
|
||||||
songname, err := util.ReadToString(rs, 30) |
|
||||||
if err != nil { |
|
||||||
return nil, err |
|
||||||
} |
|
||||||
|
|
||||||
artist, err := util.ReadToString(rs, 30) |
|
||||||
if err != nil { |
|
||||||
return nil, err |
|
||||||
} |
|
||||||
|
|
||||||
album, err := util.ReadToString(rs, 30) |
|
||||||
if err != nil { |
|
||||||
return nil, err |
|
||||||
} |
|
||||||
|
|
||||||
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) |
|
||||||
} |
|
||||||
|
|
||||||
comment, err := util.ReadToString(rs, 30) |
|
||||||
if err != nil { |
|
||||||
return nil, err |
|
||||||
} |
|
||||||
|
|
||||||
genreByte, err := util.Read(rs, 1) |
|
||||||
if err != nil { |
|
||||||
return nil, err |
|
||||||
} |
|
||||||
genreInt, err := util.ByteToInt(genreByte[0]) |
|
||||||
if err != nil { |
|
||||||
return nil, fmt.Errorf("cannot convert bytes to int: %s", err) |
|
||||||
} |
|
||||||
genre, exists := id3v1genres[int(genreInt)] |
|
||||||
if !exists { |
|
||||||
genre = "" |
|
||||||
} |
|
||||||
|
|
||||||
return &ID3v1Tag{ |
|
||||||
SongName: songname, |
|
||||||
Artist: artist, |
|
||||||
Album: album, |
|
||||||
Year: year, |
|
||||||
Comment: comment, |
|
||||||
Genre: genre, |
|
||||||
}, nil |
|
||||||
} |
|
@ -1,34 +1,6 @@ |
|||||||
package v2 |
package v2 |
||||||
|
|
||||||
// type ID3v2Tag struct {
|
type ID3v2Tag struct { |
||||||
// Header Header
|
Header Header |
||||||
// Frames []Frame
|
Frames []Frame |
||||||
// }
|
} |
||||||
|
|
||||||
// type V2TagReader interface {
|
|
||||||
// ReadFrames(io.ReadSeeker) ([]*Frame, error)
|
|
||||||
// GetHeader(io.ReadSeeker) (*Header, error)
|
|
||||||
// HasPadding(io.ReadSeeker) (bool, error)
|
|
||||||
// }
|
|
||||||
|
|
||||||
// type V2TagWriter interface {
|
|
||||||
// Write(*os.File) error
|
|
||||||
// }
|
|
||||||
|
|
||||||
// func Get(f *os.File) (*ID3v2Tag, error) {
|
|
||||||
// var tag ID3v2Tag
|
|
||||||
|
|
||||||
// header, err := GetHeader(f)
|
|
||||||
// if err != nil {
|
|
||||||
// return nil, err
|
|
||||||
// }
|
|
||||||
// frames, err := GetFrames(f)
|
|
||||||
// if err != nil {
|
|
||||||
// return nil, err
|
|
||||||
// }
|
|
||||||
|
|
||||||
// tag.Header = header
|
|
||||||
// tag.Frames = frames
|
|
||||||
|
|
||||||
// return &tag, nil
|
|
||||||
// }
|
|
||||||
|
Loading…
Reference in new issue