From: Tilman Sauerbeck <tilman@code-monkey.de>
Date: Wed, 3 Mar 2010 07:21:55 +0000 (+0100)
Subject: Initial commit.
X-Git-Url: http://git.code-monkey.de/?p=rb-rip.git;a=commitdiff_plain;h=5805d23e939c5a1c20c80da3207bb6c82c080853

Initial commit.
---

5805d23e939c5a1c20c80da3207bb6c82c080853
diff --git a/rb-rip b/rb-rip
new file mode 100755
index 0000000..164d581
--- /dev/null
+++ b/rb-rip
@@ -0,0 +1,337 @@
+#!/usr/bin/ruby -w
+
+require 'fileutils'
+require "musicbrainz"
+require "yaml"
+require "ostruct"
+#require "glib2"
+require "tempfile"
+
+class Track
+	attr_reader :id, :name
+
+	def initialize(mb)
+		@name = mb.result(MusicBrainz::Query::TrackGetTrackName)
+
+		tmp = mb.result(MusicBrainz::Query::TrackGetTrackId)
+		@id = mb.id_from_url(tmp)
+	end
+end
+
+class Artist
+	attr_reader :id, :name
+
+	def initialize(mb)
+		tmp = mb.result(MusicBrainz::Query::AlbumGetAlbumArtistId)
+		@id = mb.id_from_url(tmp)
+
+		@name = mb.result(MusicBrainz::Query::AlbumGetArtistName, 1)
+	end
+end
+
+class Album
+	attr_reader :id, :name, :artist, :tracks
+	attr_accessor :genre, :year
+
+	def initialize(mb)
+		@genre = nil
+		@year = nil
+
+		@artist = Artist.new(mb)
+
+		tmp = mb.result(MusicBrainz::Query::AlbumGetAlbumId)
+		@id = mb.id_from_url(tmp)
+
+		@name = mb.result(MusicBrainz::Query::AlbumGetAlbumName)
+
+		num_tracks = mb.result(MusicBrainz::Query::AlbumGetNumTracks).to_i
+		@tracks = Array.new(num_tracks)
+
+		1.upto(num_tracks) do |j|
+			mb.select(MusicBrainz::Query::SelectTrack, j)
+
+			@tracks[j - 1] = Track.new(mb)
+
+			mb.select(MusicBrainz::Query::Back)
+		end
+	end
+end
+
+class Encoder
+	def initialize(config)
+		@cfg = config
+	end
+
+	def encode(dest, metadata)
+	end
+
+	def apply_tags(dest, metadata)
+	end
+
+	def compute_replaygain
+	end
+
+	def suffix
+	end
+end
+
+class VorbisEncoder < Encoder
+	def encode(dest, m)
+		<<EOF
+		oggenc -q #{@cfg.vorbis_quality} -Q -o "#{dest}" -
+EOF
+	end
+
+	def apply_tags(dest, m)
+		Tempfile.open("rbrip.vorbiscomments") do |tf|
+			tf.puts("artist=" + m[:artist])
+			tf.puts("title=" + m[:track_name])
+			tf.puts("album=" + m[:disc])
+			tf.puts("tracknumber=" + m[:track_no])
+			tf.puts("genre=" + m[:genre])
+			tf.puts("date=" + m[:year].to_s)
+			tf.puts("musicbrainz_albumid=" + m[:disc_id])
+			tf.puts("musicbrainz_artistid=" + m[:artist_id])
+			tf.puts("musicbrainz_trackid=" + m[:track_id])
+			tf.flush
+
+			FileUtils.mv(dest, dest + ".untagged")
+			`vorbiscomment -w "#{dest}.untagged" "#{dest}" -c #{tf.path}`
+			FileUtils.rm(dest + ".untagged")
+		end
+	end
+
+	def compute_replaygain(files)
+		"vorbisgain -a " + files.map { |f| f + suffix }.join(" ")
+	end
+
+	def suffix
+		".ogg"
+	end
+end
+
+class FlacEncoder < Encoder
+	def encode(dest, m)
+		s =<<EOF
+flac --no-ogg
+  --sign=signed --endian=little --channels=2 --bps=16 --sample-rate=44100
+  -o "#{dest}" -
+EOF
+
+		s.gsub!(/\n/, " ")
+		s
+	end
+
+	def apply_tags(dest, m)
+		Tempfile.open("rbrip.vorbiscomments") do |tf|
+			tf.puts("artist=" + m[:artist])
+			tf.puts("title=" + m[:track_name])
+			tf.puts("album=" + m[:disc])
+			tf.puts("tracknumber=" + m[:track_no])
+			tf.puts("genre=" + m[:genre])
+			tf.puts("date=" + m[:year].to_s)
+			tf.puts("musicbrainz_albumid=" + m[:disc_id])
+			tf.puts("musicbrainz_artistid=" + m[:artist_id])
+			tf.puts("musicbrainz_trackid=" + m[:track_id])
+			tf.flush
+
+			`metaflac --import-tags-from=#{tf.path} #{dest}`
+		end
+	end
+
+	def compute_replaygain(files)
+		"metaflac --add-replay-gain " + files.map { |f| f + suffix }.join(" ")
+	end
+
+	def suffix
+		".flac"
+	end
+end
+
+class String
+	def sanitize
+		# cut leading and trailing "..."
+		if self[0..2] == "..."
+			tmp = self[3..-1]
+		else
+			tmp = self.dup
+		end
+
+		if tmp[-3..-1] == "..."
+			tmp = tmp[0..-4]
+		end
+
+		tmp.gsub(/\s+/, "_").gsub(/&/, "and").delete(",'()[]/\\!?\":").
+		tr("-", "_").
+		squeeze(" ").squeeze(".").squeeze("_")
+	end
+
+	def sanitize!
+		replace(sanitize)
+	end
+
+	def rpad(total)
+		n = [total - length, 0].max
+		self + (" " * n)
+	end
+end
+
+cfg_file = ARGV.first || File.expand_path("~/.rbrip.yaml")
+
+begin
+	config = YAML.load(File.open(cfg_file))
+rescue
+	config = OpenStruct.new(
+		:device => "/dev/hdd",
+		:destdir => ".",
+		:dirmask => "%a/%y-%l",
+		:filemask => "%n-%a-%t",
+		:sanitize_filenames => true,
+		:vorbis_quality => 5
+	)
+
+	File.open(cfg_file, "w") { |f| YAML.dump(config, f) }
+end
+
+mb = MusicBrainz::Client.new
+mb.device = config.device
+
+unless mb.query(MusicBrainz::Query::GetCDInfo)
+	STDERR.puts "Error: " + mb.error
+	exit 1
+end
+
+num_albums = mb.result(MusicBrainz::Query::GetNumAlbums).to_i
+
+if num_albums < 1
+	STDERR.puts "Album not found. Guess you need to submit it:"
+	puts mb.url
+	exit 1
+elsif num_albums > 1
+	STDERR.puts "Multiple albums found. What now?"
+	exit 1
+end
+
+mb.select(MusicBrainz::Query::Rewind)
+mb.select(MusicBrainz::Query::SelectAlbum, 1)
+
+album = Album.new(mb)
+
+if album.id == ""
+	STDERR.puts "Album not found. Guess you need to submit it:"
+	puts mb.url
+	exit 1
+end
+
+puts "Disc info:\n
+ Artist: #{album.artist.name.rpad(20)} (#{album.artist.id})
+ Name:   #{album.name.rpad(20)} (#{album.id})
+
+ Tracks:\n"
+
+album.tracks.each_with_index do |t, i|
+	#tmp = GLib.locale_from_utf8(t.name)
+	tmp = t.name
+	puts "%3s." % (i + 1) + " #{tmp}"
+end
+
+print "\nContinue [Y/n]? "
+STDOUT.flush
+exit if STDIN.getc == ?n
+
+print "\nEnter year of release: "
+STDOUT.flush
+
+album.year = STDIN.gets.strip.to_i
+
+print "Enter genre: "
+STDOUT.flush
+
+album.genre = STDIN.gets.strip
+
+# where to put the encoded files?
+destdir = config.dirmask.dup
+
+dir_map = {
+	"%A" => album.artist.name.sanitize,
+	"%L" => album.name.sanitize,
+	"%Y" => album.year.to_s
+}
+
+if config.sanitize_filenames
+	dir_map.each_value { |v| v.sanitize! }
+end
+
+dir_map.each do |k, v|
+	destdir.gsub!(/#{k}/, v)
+	destdir.gsub!(/#{k.downcase}/, v.downcase)
+end
+
+destdir = File.join(config.destdir, destdir)
+FileUtils.makedirs(destdir) unless File.directory?(destdir)
+
+puts "\nRipping to #{destdir}\n\n"
+
+files = []
+enc = VorbisEncoder.new(config)
+#enc = FlacEncoder.new(config)
+
+metadata = {
+	:artist => album.artist.name,
+	:artist_id => album.artist.id,
+	:disc => album.name,
+	:disc_id => album.id,
+	:genre => album.genre,
+	:year => album.year
+}
+file_map = dir_map.dup
+
+album.tracks.each_with_index do |t, i|
+	t = album.tracks[i]
+
+	track_no = i + 1
+	file = config.filemask.dup
+
+	file_map["%N"] = "%02i" % track_no
+
+	if config.sanitize_filenames
+		file_map["%T"] = t.name.sanitize
+	else
+		file_map["%T"] = t.name
+	end
+
+	metadata[:track_no] = track_no.to_s
+	metadata[:track_name] = t.name
+	metadata[:track_id] = t.id
+
+	file_map.each do |k, v|
+		file.gsub!(/#{k}/, v)
+		file.gsub!(/#{k.downcase}/, v.downcase)
+	end
+
+	if config.sanitize_filenames
+		if file[-1] == ?.
+			file = file[0..-2]
+		end
+	end
+
+	file = File.join(destdir, file)
+	files << file
+
+	rip_cmd = "cdparanoia -q -d \"#{config.device}\" -w -- #{track_no} -"
+	encode_cmd = enc.encode(file + enc.suffix, metadata)
+
+	puts "Ripping: #{track_no}. #{t.name} => #{file + enc.suffix}"
+	`#{rip_cmd} | #{encode_cmd}`
+
+	apltags = enc.apply_tags(file + enc.suffix, metadata)
+	puts "Applying tags..."
+	`#{apltags} > /dev/null`
+end
+
+rplgain_cmd = enc.compute_replaygain(files)
+
+unless rplgain_cmd.nil?
+	puts "Computing replaygain..."
+	`#{rplgain_cmd} > /dev/null`
+end