got_mp3 0.1.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ba5c71c2a3dd150c8be5e6f1d370c6437145e8b6dd95f139647f67ddba469a1f
-  data.tar.gz: c267589160c6a9ba438b997ccec0365c6c069766947fdf5bc802a9f8f1687e9d
+  metadata.gz: 2ec37a296a75bda0bea9d2ce9b7b40534fa6425e7cddfc29d1002cd067f5cddb
+  data.tar.gz: de3b11e29c23e10f1a920f96e82e40c2073de1ff1d25b32761f2e92cdc2dafd3
 SHA512:
-  metadata.gz: 988db3e77e2d5695efaf0a47f27f84f5c34c774dd49d81ee63ff095221b1cf641e9f827f79b22c84e88c8c2734c02ef8add3314a2d0da87a25f6ebf989e76763
-  data.tar.gz: eb1e2afa7793029dc227c55476138bc6aa02664ddf92a57afabe2b3f6d24d219a55e0609905a374d2b762fa8469578f44a9c3a13ba11d50339139d6280721102
+  metadata.gz: 6cc885998b65f09146163ef5d68ed3c261a3046ecea177b4b61d2ae64827784a9bf7924bd0d14f6209ede86631e693a31d23aaebcd938b2010bff68f5bc14901
+  data.tar.gz: b92b911b2d1f4696a21df1800b4c9047e8ca6e03f8cbac8df6a066c9f399b3ba8a4ffe87870b151d2d09018d3cb72bc625a86bfae75b5464bf3974f0d1da2150

data/lib/got_mp3.rb

@@ -1,6 +1,8 @@
 #!/usr/bin/env ruby
+
 # file: got_mp3.rb
 
+require 'ostruct'
 require "mp3info"
 
 
@@ -11,32 +13,155 @@ class GotMP3
   end
 
 
+  # Adds album art to each MP3 file
+  #
   # Example usage:
   #   gotmp3 = GotMP3.new(dir: '/tmp/tree/Da Fustra - Over The Waves To Shetland')
   #   gotmp3.add_jpg
-  # 
+  #
   def add_jpg()
 
-    a = Dir[File.join(@dir, "**", "*.jpg")]
-    puts 'a: ' + a.inspect if @debug
-    directories = a.inject({}) {|r, x| r[File.dirname(x)] = File.basename(x); r }
+    find_by_ext('.jpg').each do |directory, img_filename|
 
-    directories.each do |directory, img_filename|
+      puts 'add_jpg to directory: ' + directory.inspect if @debug
       add_image directory, img_filename
     end
 
   end
 
+  # Adds a track title to each MP3 file
+  #
+  # Example usage:
+  #   gotmp3 = GotMP3.new(dir: '/tmp/tree/Da Fustra - Over The Waves To Shetland')
+  #   gotmp3.add_titles
+  #
+  def add_titles()
+
+    find_by_ext('.txt').each do |directory, txt_filename|
+      add_tracktitles directory, txt_filename
+    end
+
+  end
+
+  def each_mp3_file(directory='.', &blk)
+
+    puts 'each_mp3 - directory: ' + directory.inspect if @debug
+    found = Dir[File.join(directory, "*.mp3")].sort_by { |x| File.mtime(x) }
+    puts 'each_mp3 - found: ' + found.inspect if @debug
+
+    found.reverse.each.with_index do |mp3_filepath, i|
+
+      puts 'each_mp3 - mp3_filepath: ' + mp3_filepath.inspect if @debug
+
+      next unless File.exists? mp3_filepath
+
+      blk.call(mp3_filepath, i )
+
+    end
+
+  end
+
+  # Adds the album art, track title, renames the MP3 file, and adds a playlist
+  #
+  def go()
+
+    find_by_ext('.mp3').each do |directory, _ |
+
+      # find the image file
+      img_filename = Dir[File.join(directory, '*.jpg')].first
+      puts 'img_filename: ' + img_filename.inspect if @debug
+
+      # find the text file
+      txt_filename = Dir[File.join(directory, '*.txt')].first
+      next unless txt_filename
+
+      add_image_and_titles(directory, img_filename, txt_filename)
+
+    end
+
+  end
+
+  # rename 1 or more mp3 files within 1 or more file directories
+  #
+  # example usage:
+  #   rename() {|mp3file| mp3files.sub(/Disc \d - /,'')}
+  #   rename() {|mp3file| mp3file.sub(/Disc \d - (\d+) - /,'\1. ')}
+  #
+  def rename()
+
+    each_mp3_file do |mp3_filepath|
+
+      mp3_directory = File.dirname(mp3_filepath)
+      mp3_filename = File.basename(mp3_filepath)
+
+      newname = yield(mp3_filename)
+      File.rename(mp3_filepath, File.join(mp3_directory, newname))
+
+    end
+
+  end
+
+  def write_titles()
+
+    puts 'inside write_titles()' if @debug
+
+    find_by_ext('.mp3').each do |directory, _ |
+
+      puts 'write_titles() - directory: ' + directory.inspect if @debug
+      txt_filename = Dir[File.join(directory, '*.txt')].first
+
+      next if txt_filename
+
+      tracks = []
+
+      each_mp3_track(directory) do |mp3, trackno, mp3_filepath|
+
+        tracks << OpenStruct.new({
+          title: mp3.tag.title,
+          artist: mp3.tag.artist,
+          album: mp3.tag.album,
+          album_artist: mp3.tag2['TPE2'],
+          disc: mp3.tag2['TPOS'],
+          tracknum: mp3.tag.tracknum,
+          filename: File.basename(mp3_filepath)
+        })
+
+      end
+
+      heading = tracks[0].album_artist + ' - ' + tracks[0].album
+      s = "# %s\n\n" % [heading]
+      h = tracks.group_by(&:disc)
+
+      body = if h.length == 1 then
+
+        list(tracks)
+
+      else
+
+        "\n" + h.map do |disc, tracks2|
+          ("## Disc %d\n\n" % disc) + list(tracks2) + "\n\n"
+        end.join("\n")
+
+      end
+
+      File.write File.join(directory, heading + '.txt'), s + body
+
+    end
+
+  end
+
   private
 
   # adds album art to each mp3 file in a file directory
   #
   def add_image(directory, img_filename)
 
-    image_file = File.new(File.join(directory, img_filename),'rb') 
+    puts 'inside add_image - directory: ' + directory.inspect if @debug
+    puts 'img_filename: ' + img_filename.inspect if @debug
+    image_file = File.new(File.join(directory, img_filename),'rb')
     img = image_file.read
 
-    each_mp3(directory) do |mp3|
+    each_mp3_track(directory) do |mp3, _, _|
 
       mp3.tag2.remove_pictures
       mp3.tag2.add_picture img
@@ -45,16 +170,125 @@ class GotMP3
 
   end
 
-  def each_mp3(directory, &blk)
+  def add_tracktitles(directory, txt_filename)
+
+    txt_file = File.join(directory, txt_filename)
+    track_titles = File.read(txt_file).lines[1..-1].map(&:strip)
+
+    each_mp3_track(directory) do |mp3, trackno, _|
+
+      mp3.tag.title = track_titles[trackno-1]
+
+    end
+  end
+
+  def add_image_and_titles(directory, img_file, txt_file)
+
+    if img_file then
+      img = File.new(img_file,'rb').read
+    end
+
+    track_titles = File.read(txt_file).lines[1..-1].map(&:strip)
+
+    titles_mp3 = track_titles.map do |title|
+      title[/^[^\/]+/].gsub(/:/,'_').rstrip + '.mp3'
+    end
+
+    found = Dir[File.join(directory, "*.mp3")].sort_by { |x| File.mtime(x) }
+    found.each.with_index do |mp3_filepath, i|
 
-    Dir[File.join(directory, "*.mp3")].each do |mp3_filepath|
+      Mp3Info.open(mp3_filepath) do |mp3|
 
-      Mp3Info.open mp3_filepath do |mp3|
-        blk.call mp3
+        if img_file then
+          mp3.tag2.remove_pictures
+          mp3.tag2.add_picture img
+        end
+
+        mp3.tag.title = track_titles[i]
       end
 
+      File.rename(mp3_filepath, File.join(directory, titles_mp3[i]))
+
     end
 
+    File.write File.join(directory, 'playlist.m3u'), titles_mp3.join("\n")
+
+  end
+
+  def each_mp3_track(directory, &blk)
+
+    each_mp3_file(directory) do |mp3_filepath, i|
+
+      Mp3Info.open(mp3_filepath) {|mp3|  blk.call(mp3, i+1, mp3_filepath) }
+
+    end
+
+  end
+
+  def find_by_ext(extension)
+
+    puts 'find_by_ext() - @dir' + @dir.inspect if @debug
+    a = Dir[File.join(@dir, "**", "*" + extension)]
+    puts 'a: ' + a.inspect if @debug
+
+    a.inject({}) do |r, x|
+
+      r[File.dirname(x)] = File.basename(x)
+      r
+
+    end
+  end
+
+  def list(tracks)
+
+    a = if tracks.map(&:artist).uniq.length < 2 then
+      tracks.map {|x| "%02d. %s" % [x.tracknum, x.title] }
+    else
+      tracks.map {|x| "%02d. %s - %s" % [x.tracknum, x.title, x.artist] }
+    end
+
+    a.join("\n")
+
+  end
+
+end
+
+class Titles
+
+  def initialize(filename, target_directory: 'titles')
+    @titles = File.read filename
+    @target_directory = target_directory
+  end
+
+  def titleize()
+    @titles.gsub(/[\w']+/) {|x| x[0].upcase + x[1..-1]}
+  end
+
+  def titleize!()
+    @titles.gsub!(/[\w']+/) {|x| x[0].upcase + x[1..-1]}
+  end
+
+  def split(target_directory: @target_directory)
+
+    FileUtils.mkdir_p @target_directory
+    a = @titles.strip.split(/(?=^#)/)
+
+    a.each do |x|
+
+      filename = x.lines.first.chomp[/(?<=# cd ).*/i].strip + '.txt'
+      puts 'processing file ' + filename.inspect
+      heading = x.lstrip.lines.first
+
+      tracks = x.strip.lines[1..-1].map.with_index do |line, i|
+        "%02d. %s" % [i+1, line]
+      end
+
+      File.write File.join(target_directory, filename), heading + tracks.join
+
+    end
+
+    puts 'split done'
+
   end
 
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: got_mp3
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.1
 platform: ruby
 authors:
 - James Robertson
@@ -35,7 +35,7 @@ cert_chain:
   RFrZsrDO1Cwk4v03qCUvuZ8H/nCojYYJGMrJE1JhBBJNJXCAULivh/zlOYcDhG69
   uPtZOD7YKrG45OftX8aNxlNv
   -----END CERTIFICATE-----
-date: 2021-11-24 00:00:00.000000000 Z
+date: 2021-11-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby-mp3info
@@ -87,6 +87,6 @@ rubyforge_project:
 rubygems_version: 2.7.10
 signing_key: 
 specification_version: 4
-summary: A ruby-mp3info wrapper to make it convenient to update the metadata of multiple
-  MP3 files at once.
+summary: A ruby-mp3info wrapper to make it convenient to update the album art and
+  track titles of multiple MP3 files in a batch process.
 test_files: []