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 |
||||
|
||||
// type ID3v2Tag struct {
|
||||
// Header Header
|
||||
// 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
|
||||
// }
|
||||
type ID3v2Tag struct { |
||||
Header Header |
||||
Frames []Frame |
||||
} |
||||
|
Loading…
Reference in new issue