usetaglib.cpp

   1 /* Copyright (C) 2015 Alexander Schmidt <alex@treefish.org>
   2  *
   3  * This program is free software: you can redistribute it and/or modify
   4  * it under the terms of the GNU General Public License as published by
   5  * the Free Software Foundation, either version 3 of the License, or
   6  * (at your option) any later version.
   7  *
   8  * This program is distributed in the hope that it will be useful,
   9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11  * GNU General Public License for more details.
  12  *
  13  * You should have received a copy of the GNU General Public License
  14  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  15  */
  16
  17 #include <iostream>
  18 #include <iomanip>
  19 #include <iostream>
  20 #include <taglib/fileref.h>
  21 #include <taglib/tag.h>
  22 #include <taglib/tpropertymap.h>
  23 #include <taglib/tstringlist.h>
  24 #include <getopt.h>
  25 #include <vector>
  26
  27 using namespace std;
  28
  29 enum action {LIST, REPLACE, INSERT, ERASE, AUDIO};
  30 typedef pair<action,string> actionpair;
  31 typedef vector<actionpair> actionqueue;
  32 typedef pair<string,string> tagpair;
  33
  34 TagLib::StringList argToStringList (const string &rawtagarg)
  35 {
  36   TagLib::StringList newlist;
  37
  38   size_t delpos = 0;
  39   while (1) {
  40     size_t nextdelpos = rawtagarg.find('=', delpos+1);
  41     if (nextdelpos == -1)
  42       break;
  43     newlist.append(rawtagarg.substr(delpos,nextdelpos-delpos));
  44     delpos = nextdelpos+1;
  45   }
  46   newlist.append(rawtagarg.substr(delpos,string::npos));
  47
  48   return newlist;
  49 }
  50
  51 tagpair splitToTagPair (const string &rawarg)
  52 {
  53   const size_t delpos = rawarg.find('=');
  54   return make_pair(rawarg.substr(0, delpos), rawarg.substr(delpos+1, string::npos));
  55 }
  56
  57 void action_eraseTag (const TagLib::FileRef f, const string &key)
  58 {
  59   TagLib::PropertyMap propmap = f.file()->properties();
  60   propmap.erase(key);
  61   f.file()->setProperties(propmap);
  62 }
  63
  64 void action_replaceTag (const TagLib::FileRef f, const tagpair &tagPair)
  65 {
  66   TagLib::PropertyMap propmap = f.file()->properties();
  67   propmap.replace(tagPair.first, argToStringList(tagPair.second));
  68   f.file()->setProperties(propmap);
  69 }
  70
  71 void action_insertTag (const TagLib::FileRef f, const tagpair &tagPair)
  72 {
  73   TagLib::PropertyMap propmap = f.file()->properties();
  74   propmap.insert(tagPair.first, argToStringList(tagPair.second));
  75   f.file()->setProperties(propmap);
  76 }
  77
  78 int action_printTags (const TagLib::FileRef f)
  79 {
  80   if(f.tag()) {
  81     TagLib::PropertyMap tags = f.file()->properties();
  82     unsigned int longest = 0;
  83     for(TagLib::PropertyMap::ConstIterator i = tags.begin(); i != tags.end(); ++i) {
  84       if (i->first.size() > longest) {
  85         longest = i->first.size();
  86       }
  87     }
  88     cout << "-- TAG (properties) --" << endl;
  89     for(TagLib::PropertyMap::ConstIterator i = tags.begin(); i != tags.end(); ++i) {
  90       for(TagLib::StringList::ConstIterator j = i->second.begin(); j != i->second.end(); ++j) {
  91         cout << i->first << "=" << *j << endl;
  92       }
  93     }
  94     return 0;
  95   }
  96   else
  97     return 1;
  98 }
  99
 100 int action_printAudio (const TagLib::FileRef f)
 101 {
 102   if(f.audioProperties()) {
 103     TagLib::AudioProperties *properties = f.audioProperties();
 104     int seconds = properties->length() % 60;
 105     int minutes = (properties->length() - seconds) / 60;
 106     cout << "-- AUDIO --" << endl;
 107     cout << "BITRATE=" << properties->bitrate() << endl;
 108     cout << "SAMPLERATE=" << properties->sampleRate() << endl;
 109     cout << "CHANNELS=" << properties->channels() << endl;
 110     cout << "LENGTH=" << minutes << ":" << setfill('0') << setw(2) << seconds << endl;
 111     return 0;
 112   }
 113   else
 114     return 1;
 115 }
 116
 117 void printHelp ()
 118 {
 119   cout << "Usage: usetaglib [ACTION]... [FILE]..." << endl;
 120   cout << "List and edit tags on mediafiles in formats supported by libtag." << endl;
 121   cout << endl;
 122   cout << "-h, --help   Show this help" << endl;
 123   cout << endl;
 124   cout << "ACTIONS" << endl;
 125   cout << setfill(' ') << setw(45) << left << "  -l, --list"
 126        << "list all tags (implicit if no action specified)"<< endl;
 127   cout << setfill(' ') << setw(45) << left << "  -e, --erase=TAGNAME"
 128        << "erase tag with name TAGNAME"<< endl;
 129   cout << setfill(' ') << setw(45) << left << "  -r, --replace=TAGNAME=TAGVAL[=TAGVAL...]"
 130        << "replace tag TAGNAME with list of values TAGVAL"<< endl;
 131   cout << setfill(' ') << setw(45) << left << "  -i, --insert=TAGNAME=TAGVAL[=TAGVAL...]"
 132        << "insert list of values TAGVAL for tag TAGNAME"<< endl;
 133 }
 134
 135 int main(int argc, char *argv[])
 136 {
 137   int c;
 138   actionqueue requestedActions;
 139
 140   while (1)
 141     {
 142       static struct option long_options[] =
 143         {
 144           {"help",      no_argument,       0, 'h'},
 145           {"list",      no_argument,       0, 'l'},
 146           {"listaudio", no_argument,       0, 'a'},
 147           {"insert",    required_argument, 0, 'i'},
 148           {"erase",     required_argument, 0, 'e'},
 149           {"replace",   required_argument, 0, 'r'},
 150           {0, 0, 0, 0}
 151         };
 152
 153       int option_index = 0;
 154       c = getopt_long (argc, argv, "hlai:e:r:",
 155                        long_options, &option_index);
 156
 157       if (c == -1)
 158         break;
 159
 160       switch (c)
 161         {
 162         case 0:
 163           if (long_options[option_index].flag != 0)
 164             break;
 165
 166         case 'h':
 167           printHelp();
 168           return 0;
 169           break;
 170
 171         case 'l':
 172           requestedActions.push_back( make_pair(LIST, "") );
 173           break;
 174
 175         case 'a':
 176           requestedActions.push_back( make_pair(AUDIO, "") );
 177           break;
 178
 179         case 'i':
 180           requestedActions.push_back( make_pair(INSERT, optarg) );
 181           break;
 182
 183         case 'e':
 184           requestedActions.push_back( make_pair(ERASE, optarg) );
 185           break;
 186
 187         case 'r':
 188           requestedActions.push_back( make_pair(REPLACE, optarg) );
 189           break;
 190
 191         case '?':
 192           break;
 193
 194         default:
 195           abort ();
 196         }
 197     }
 198
 199   if ( optind == argc ) {
 200     printHelp();
 201     return 0;
 202   }
 203
 204   if (requestedActions.size() == 0)
 205     requestedActions.push_back( make_pair(LIST, "") );
 206
 207   for(int i = optind; i < argc; i++) {
 208     cout << "******************** \"" << argv[i] << "\" ********************" << endl;
 209     TagLib::FileRef f(argv[i]);
 210
 211     if(f.isNull())
 212       continue;
 213
 214     for (actionqueue::iterator actit = requestedActions.begin(); actit != requestedActions.end(); ++actit) {
 215       switch (actit->first) {
 216       case LIST:
 217         action_printTags(f);
 218         break;
 219
 220       case AUDIO:
 221         action_printAudio(f);
 222         break;
 223
 224       case ERASE:
 225         action_eraseTag(f, actit->second);
 226         break;
 227
 228       case REPLACE:
 229         action_replaceTag(f, splitToTagPair(actit->second));
 230         break;
 231
 232       case INSERT:
 233         action_insertTag(f, splitToTagPair(actit->second));
 234         break;
 235       }
 236     }
 237
 238     f.file()->save();
 239   }
 240
 241   return 0;
 242 }