summaryrefslogtreecommitdiff
path: root/omptagger
blob: 56e48a9695447e567ee328f2423e49314f7e092d (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
#!/usr/bin/env ruby
#
# omptagger
# http://github.com/omp/omptagger
# Modify and display metadata of audio files.
#
# Copyright 2007-2010 David Vazgenovich Shakaryan <dvshakaryan@gmail.com>
# Distributed under the terms of the GNU General Public License v3.
# See http://www.gnu.org/licenses/gpl.txt for the full license text.
#
# Dependencies:
#   Ruby            http://www.ruby-lang.org/
#   RubyGems        http://rubygems.org/
#   rtaglib         http://rubygems.org/gems/rtaglib
#   ruby-filemagic  http://rubygems.org/gems/ruby-filemagic
#   TagLib          http://developer.kde.org/~wheeler/taglib.html

require 'getoptlong'
require 'rubygems'
require 'TagLib'
require 'filemagic'

class Hash
  def longest_key_length
    self.keys.inject(0) do |longest, key|
      key.length > longest ? key.length : longest
    end
  end
end

class Metadata
  def output(tag, val, padding = 0)
    puts '    ' + tag + ' ' * (2 + padding) + val
  end

  private :output

  def view
    puts '  Viewing tags:'

    if @tags.empty?
      puts '    No tags set.'
      throw :next
    end
  end

  def viewtag(tag)
    puts '  Viewing ' + tag + ' tag:'

    unless @tags.has_key? tag
      puts '    Tag not set.'
      throw :next
    end
  end
end

class VorbisComment < Metadata
  def view
    super

    @tags.sort.each do |tag, val|
      val.each do |val|
        output(tag, val, @tags.longest_key_length - tag.length)
      end
    end
  end

  def viewtag(tag)
    super

    @tags[tag].each do |val|
      output(tag, val)
    end
  end
end

class FLAC < VorbisComment
  def initialize(file)
    @filename = File.expand_path(file)
    @write = false

    @file = TagLib::FLAC::File.new(file)
    @tags = @file.xiphComment.fieldListMap.hash
  end
end

def help
  puts <<-end
Usage: omptagger [options] [files]

Options:
  --view        -v  View all tags
  --view-tag    -t  View a tag
  --add-tag     -a  Add a tag
  --set-tag     -s  Set a tag
  --remove      -r  Remove all tags
  --remove-tag  -d  Remove a tag
  --generate    -g  Generate tags based on filename
  --rename      -m  Generate filename based on tags
  --scheme      -n  Specify a file naming scheme
  --list        -l  Display available scheme keys
  --help        -h  Display help information
  --pretend     -p  Do not finalise changes

Notes:
  * Actions are executed in the order they are specified.
  * The default file naming scheme is Artist - Title.
  end

  exit
end

def list
  puts <<-end
Keys   Vorbis Comments   ID3v2 Frames
----   ---------------   -------------
 %a    Artist            Artist
 %b    Album             Album
 %d    Date              Year
 %n    TrackNumber       Track
 %t    Title             Title
----   ---------------   -------------
 %*    Wildcard          Wildcard
 %%    Per cent sign     Per cent sign
  end

  exit
end

actions = Array.new
options = Array.new
scheme  = '%a - %t'

GetoptLong.new(
  ['--view',       '-v', GetoptLong::NO_ARGUMENT],
  ['--view-tag',   '-t', GetoptLong::REQUIRED_ARGUMENT],
  ['--set-tag',    '-s', GetoptLong::REQUIRED_ARGUMENT],
  ['--remove',     '-r', GetoptLong::NO_ARGUMENT],
  ['--remove-tag', '-d', GetoptLong::REQUIRED_ARGUMENT],
  ['--generate',   '-g', GetoptLong::NO_ARGUMENT],
  ['--rename',     '-m', GetoptLong::NO_ARGUMENT],
  ['--scheme',     '-n', GetoptLong::REQUIRED_ARGUMENT],
  ['--pretend',    '-p', GetoptLong::NO_ARGUMENT],
  ['--list',       '-l', GetoptLong::NO_ARGUMENT],
  ['--help',       '-h', GetoptLong::NO_ARGUMENT]
).each do |option, argument|
  option = option.delete('-').intern

  case option
  when :scheme
    scheme = argument
  when :help, :list, :pretend
    options << option
  when :viewtag, :settag, :removetag
    actions << [option, argument]
  when :generate, :rename
    actions << [option, scheme]
  else
    actions << [option]
  end
end

help if options.include?(:help)
list if options.include?(:list)
help if actions.empty?

if ARGV.empty?
  puts 'No files specified.'
  exit
end

mime = FileMagic.mime
mime.simplified = true

ARGV.each do |file|
  begin
    raise 'File does not exist.' unless File.exist?(file)

    case mime.file(file)
    when 'audio/x-flac'
      track = FLAC.new(file)
    else
      raise 'File extension not recognised.'
    end

    puts file + ':'

    actions.each do |action|
      catch :next do
        track.send(*action)
      end
    end
  rescue RuntimeError => message
    $stderr.puts file + ':'
    $stderr.puts '  ' + message
  end
end