diff options
Diffstat (limited to 'fixID3.cpp')
| -rw-r--r-- | fixID3.cpp | 110 |
1 files changed, 110 insertions, 0 deletions
diff --git a/fixID3.cpp b/fixID3.cpp new file mode 100644 index 0000000..aa4055f --- /dev/null +++ b/fixID3.cpp @@ -0,0 +1,110 @@ +/* Copyright (C) 2008 Jason A. Donenfeld <Jason@zx2c4.com> */ + +/* Use at your own risk. */ + +#include <iostream> +#include <mpegfile.h> +#include <tstring.h> +#include <tag.h> +#include <id3v2tag.h> +#include <textidentificationframe.h> + +using namespace TagLib; +using namespace std; + +int main(int argc, char *argv[]) +{ + for(int i = 1; i < argc; i++) { + MPEG::File *file = new MPEG::File(argv[i]); + if(argc > 2) + cout << "--File " << i << " of " << argc - 1 << "--" << endl; + if(!file->isValid()) { + cout << "[!] Could not open " << argv[i] << "." << endl; + continue; + } + Tag *tag = file->tag(); + if(tag->isEmpty()) { + cout << "[!] No tags in " << argv[i] << "." << endl; + continue; + } + cout << "[*] Reading " << argv[i] << ":" << endl; + String artist = tag->artist().stripWhiteSpace(); + String album = tag->album().stripWhiteSpace(); + String title = tag->title().stripWhiteSpace(); + TagLib::uint track = tag->track(); + TagLib::uint year = tag->year(); + String genre = tag->genre().stripWhiteSpace(); + // COMMENTS ARE NOT SAVED. I HATE ID3 COMMENTS, SO THEY'RE STRIPPED OUT + String disc; + String bpm; + String composer; + String albumArtist; + String compilation; + if(file->ID3v2Tag()) { + if(!file->ID3v2Tag()->frameListMap()["TPOS"].isEmpty()) + disc = file->ID3v2Tag()->frameListMap()["TPOS"].front()->toString().stripWhiteSpace(); + if(!file->ID3v2Tag()->frameListMap()["TBPM"].isEmpty()) + bpm = file->ID3v2Tag()->frameListMap()["TBPM"].front()->toString().stripWhiteSpace(); + if(!file->ID3v2Tag()->frameListMap()["TCOM"].isEmpty()) + composer = file->ID3v2Tag()->frameListMap()["TCOM"].front()->toString().stripWhiteSpace(); + if(!file->ID3v2Tag()->frameListMap()["TPE2"].isEmpty()) + albumArtist = file->ID3v2Tag()->frameListMap()["TPE2"].front()->toString().stripWhiteSpace(); + if(!file->ID3v2Tag()->frameListMap()["TCMP"].isEmpty()) + compilation = file->ID3v2Tag()->frameListMap()["TCMP"].front()->toString().stripWhiteSpace(); + } + cout + << "\tArtist: " << artist << endl + << "\tComposer: " << composer << endl + << "\tAlbum: " << album << endl + << "\tAlbum Artist: " << albumArtist << endl + << "\tTitle: " << title << endl + << "\tTrack: " << track << endl + << "\tDisc: " << disc << endl + << "\tCompilation: " << compilation << endl + << "\tYear: " << year << endl + << "\tGenre: " << genre << endl; + cout << "[*] Stripping tags and saving." << endl; + file->strip(MPEG::File::AllTags); + file->save(); + delete file; + cout << "[*] Re-opening and re-writing new tags." << endl; + file = new MPEG::File(argv[i]); + tag = file->tag(); + tag->setArtist(artist); + tag->setAlbum(album); + tag->setTitle(title); + tag->setTrack(track); + tag->setYear(year); + tag->setGenre(genre); + ::ID3v2::TextIdentificationFrame *frame; + if(!disc.isEmpty()) { + frame = new ::ID3v2::TextIdentificationFrame("TPOS", ::ID3v2::FrameFactory::instance()->defaultTextEncoding()); + frame->setText(disc); + file->ID3v2Tag()->addFrame(frame); + } + if(!bpm.isEmpty()) { + frame = new ::ID3v2::TextIdentificationFrame("TBPM", ::ID3v2::FrameFactory::instance()->defaultTextEncoding()); + frame->setText(bpm); + file->ID3v2Tag()->addFrame(frame); + } + if(!composer.isEmpty()) { + frame = new ::ID3v2::TextIdentificationFrame("TCOM", ::ID3v2::FrameFactory::instance()->defaultTextEncoding()); + frame->setText(composer); + file->ID3v2Tag()->addFrame(frame); + } + if(!albumArtist.isEmpty()) { + frame = new ::ID3v2::TextIdentificationFrame("TPE2", ::ID3v2::FrameFactory::instance()->defaultTextEncoding()); + frame->setText(albumArtist); + file->ID3v2Tag()->addFrame(frame); + } + if(!compilation.isEmpty()) { + frame = new ::ID3v2::TextIdentificationFrame("TCMP", ::ID3v2::FrameFactory::instance()->defaultTextEncoding()); + frame->setText(compilation); + file->ID3v2Tag()->addFrame(frame); + } + cout << "[*] Saving file." << endl; + file->save(); + delete file; + cout << "[*] Success." << endl; + } +} |