shroom 0.0.1

data/lib/sh_song.rb

@@ -0,0 +1,164 @@
+require 'rubygems'
+require 'mp3info'
+require 'flacinfo'
+require 'ogginfo'
+require 'earworm'
+require 'rbrainz'
+include MusicBrainz
+
+module Sh
+  class Song
+    attr_reader :path, :mime, :matches
+    attr_accessor :title, :artist, :album, :image, :lyrics, :track_num, :year
+
+    def initialize path
+      @path = File.expand_path path
+      ext = File.extname path
+      @mime = {
+        '.mp3' => 'audio/mpeg',
+        '.flac' => 'audio/flac',
+        '.ogg' => 'audio/ogg'
+      }[ext]
+    end
+
+    def duration
+      if not @duration
+        case
+        when 'audio/mpeg'
+          Mp3Info.open(@path) {|mp3| @duration = mp3.length}
+        when 'audio/flac'
+        when 'audio/ogg'
+        end
+      end
+
+      return @duration
+    end
+
+
+    def lookup!
+      track = lookup_multiple.first
+      if track
+        # Don't distinguish between bands with the same name by adding to the name
+        track.artist.disambiguation = false
+        # Pull data from response
+        self.title = track.title
+        self.artist = track.artist.to_s
+        artistid = track.artist.id.to_mbid.uuid
+        rel = track.releases.to_a.first
+        self.album = rel.title
+        releaseid = rel.id.to_mbid.uuid
+        # Determine track number
+        query = Webservice::Query.new
+        filter = Webservice::TrackFilter.new(:artistid => artistid, :releaseid => releaseid)
+        tracks = query.get_tracks(filter).entities
+        tracks.to_a.each_with_index do |t, i|
+          self.track_num = i + 1 if t.title == track.title and t.duration == track.duration
+        end
+      end
+    end
+
+    private
+    def lookup_multiple
+      # Generate fingerprint and send to MusicDNS
+      ew = Earworm::Client.new(Sh::KEYS[:music_dns])
+      begin
+        info = ew.identify(:file => path)
+      rescue
+        printf "Couldn't generate fingerprint for %s\n> %s\n\n", path, $!
+        return
+      end
+      puids = info.puid_list
+      self.title = info.title
+      self.artist = info.artist_name
+      # Return if no matches are found
+      return if not title or not artist
+      # Get more information from MusicBrainz
+      query = Webservice::Query.new
+      filter = Webservice::TrackFilter.new(:artist => artist, :title => title, :puid => puids.first)
+      tracks = query.get_tracks(filter).entities
+      @matches = tracks || []
+    end
+
+    public
+    def read_tags!
+      case @mime
+      when 'audio/mpeg'
+        read_tags_mp3!
+      when 'audio/flac'
+        read_tags_flac!
+      when 'audio/ogg'
+        read_tags_ogg!
+      end
+    end
+
+    private
+    def read_tags_mp3!
+      Mp3Info.open path do |mp3|
+        t = mp3.tag
+        @title = t.title unless t.title and t.title.is_binary_data?
+        @artist = t.artist unless t.artist and t.artist.is_binary_data?
+        @album = t.album unless t.album and t.album.is_binary_data?
+        @year = t.year
+        @track_num = t.tracknum
+        @duration = mp3.length
+      end
+    end
+
+    def read_tags_ogg!
+      OggInfo.open path do |ogg|
+        t = ogg.tag
+        @title = t.title unless t.title and t.title.is_binary_data?
+        @artist = t.artist unless t.artist and t.artist.is_binary_data?
+        @album = t.album unless t.album and t.album.is_binary_data?
+        #TODO: handle alternative date formats, like yyyy/mm/dd
+        @year = t.date.split('-').first.to_i
+        @track_num = t.tracknumber.to_i
+        @duration = ogg.length
+      end rescue Exception
+      self
+    end
+
+    def read_tags_flac!
+      flac = FlacInfo.new(path)
+      comments = flac.comment
+      t = {}
+      comments.each do |comment|
+        split = comment.split('=')
+        key = split.first.downcase
+        value = (split[1..-1]).join
+        value = nil if value == '' or value.is_binary_data?
+        case key
+        when 'title'
+          @title = value
+        when 'artist'
+          @artist = value
+        when 'album'
+          @album = value
+        when 'year'
+          #TODO: handle alternative date formats, like yyyy/mm/dd
+          @year = value.split('-').first.to_i
+        when 'tracknumber'
+          @track_num = value.to_i
+        end
+      end
+    end
+
+    public
+    def to_s
+      sprintf("%s, %s, %.2d - %s", artist, album, track_num, title)
+    end
+
+    def to_html
+      t = title
+      t = "Unknown" if not t or t.is_binary_data?
+      t.gsub!("<", "&lt;")
+      ar = artist
+      ar = "Unknown" if not ar or ar.is_binary_data?
+      ar.gsub!("<", "&lt;")
+      al = album
+      al = "Unknown" if not al or al.is_binary_data?
+      al.gsub!("<", "&lt;")
+      return "<b>#{t}</b> by <i>#{ar}</i> from <i>#{al}</i>".gsub("&", "&amp;")
+    end
+  end
+end

data/lib/sh_util.rb

@@ -0,0 +1,21 @@
+class Object
+  def try_require lib
+    name = lib
+    name << '.rb' unless File.extname(name) == '.rb'
+    found = false
+    $:.each do |d|
+      found = !Dir["#{d}/#{name}"].empty?
+      break if found
+    end
+    require lib if found
+    found
+  end
+
+  def find_in_load_path(file)
+    $:.each do |path|
+      abs_file = "#{path}/#{file}"
+      return abs_file if File.exists? abs_file
+    end
+    return nil
+  end
+end

data/lib/sh_view.rb

@@ -0,0 +1,430 @@
+require 'sh_browse'
+require 'sh_lyrics'
+require 'sh_cover_art'
+require 'libglade2'
+require 'gtk2'
+include Gtk
+
+module Sh
+  class View
+    TAB_QUEUE, TAB_BROWSE, TAB_LYRICS, NUM_TABS = *(0..3).to_a
+    TAB_NAMES = {
+      TAB_QUEUE => 'Queue',
+      TAB_BROWSE => 'Browse',
+      TAB_LYRICS => 'Lyrics'
+    }
+
+    def initialize
+      Gtk.init
+
+      @rnotify = try_require 'RNotify'
+
+      Notify.init 'Shroom' if @rnotify
+
+      @glade= GladeXML.new(Global.locate('shroom.glade'))
+
+      cancelled = false
+      dialog = Dialog.new("Adding songs to database",
+        nil,
+        Dialog::MODAL,
+        [Stock::CANCEL, Dialog::RESPONSE_NONE])
+      dialog.signal_connect('response') { cancelled = true; dialog.destroy }
+      progress = ProgressBar.new
+      dialog.vbox.pack_start Label.new('Adding songs to database'), false, false, 8
+      lbl_path = Label.new
+      dialog.vbox.pack_start lbl_path, false, false, 0
+      dialog.vbox.pack_start progress, false, false, 8
+      dialog.width_request = 300
+      
+      new_songs = []
+      Dir[$prefs[:library_dir]+'/**/*'].each do |path|
+        ext = File.extname path
+        if [".mp3", ".ogg", ".flac"].include? ext
+          new_songs << Song.new(path) unless $db.contains? path
+        end
+      end
+      if new_songs.length > 0
+        dialog.show_all
+        while (Gtk.events_pending?)
+          Gtk.main_iteration
+        end
+      end
+      inc = 1 / new_songs.length.to_f
+      new_songs.each do |song|
+        lbl_path.text = "..." + song.path[-30..-1]
+        puts "Adding: #{song.path}"
+        song.read_tags!
+        $db.save_song song
+        progress.fraction += inc
+        while (Gtk.events_pending?)
+          Gtk.main_iteration
+        end
+        break if cancelled
+      end
+      progress.pulse unless cancelled
+      $db.clean
+      dialog.destroy unless cancelled
+
+      # Clean up
+      new_songs = nil
+      dialog = nil
+      progress = nil
+      lbl_path = nil
+
+      icon = Gdk::Pixbuf.new Global.locate('icon_16x16.png')
+
+      @window = Window.new "Shroom"
+      @window.icon = icon
+      @window.resize 800, 600
+      @window.signal_connect('destroy') do
+        quit
+      end
+
+      @status_icon = StatusIcon.new
+      @status_icon.pixbuf = icon
+      @status_icon.visible = true
+      @status_icon.signal_connect('activate') do |widget|
+        @window.visible = !@window.visible?
+      end
+
+      content_pane = VBox.new(false, 0)
+      @window.add content_pane
+
+      content_pane.pack_start(create_menu, false, true, 0)
+
+      vpaned = VBox.new
+      content_pane.add vpaned
+
+      # Controls
+      vbox = VBox.new false, 0
+      vpaned.pack_start vbox, false, true, 8
+      hbox = HBox.new
+      lbl_song = Label.new
+      hbox.add lbl_song
+      lbl_time = Label.new(time_to_s(0) + "/" + time_to_s(0))
+      hbox.pack_start lbl_time, false, false, 8
+      vbox.pack_start hbox, false, false, 8
+      box_controls = HBox.new
+      vbox.add box_controls
+      @btn_play = ToggleButton.new
+      @btn_play.add Image.new Stock::MEDIA_PLAY, IconSize::SMALL_TOOLBAR
+      box_controls.pack_start @btn_play, false, true, 0
+      btn_prev = Button.new
+      btn_prev.add Image.new Stock::MEDIA_PREVIOUS, IconSize::SMALL_TOOLBAR
+      box_controls.pack_start btn_prev, false, true, 0
+      seeker = HScale.new 0, 1, 0.01
+      box_controls.add seeker
+      seeker.draw_value = false
+      btn_next = Button.new
+      btn_next.add Image.new Stock::MEDIA_NEXT, IconSize::SMALL_TOOLBAR
+      box_controls.pack_start btn_next, false, true, 0
+
+      seeker.signal_connect('change_value') do |range, scroll, value|
+        if @player
+          seeker.value = value if value < 0.999
+          pos = seeker.value * @player.duration
+          @player.seek pos
+        else
+          seeker.value = 0
+        end
+      end
+      GLib::Timeout.add(100) do
+        if @player
+          pos = @player.position
+          seeker.value = pos / @player.duration
+          lbl_time.text = time_to_s(pos) + "/" + time_to_s(@player.duration)
+          lbl_song.set_markup @player.song.to_html
+        else
+          seeker.value = 0
+          lbl_time.text = time_to_s(0) + "/" + time_to_s(0)
+          lbl_song.set_markup "<b>Not playing</b>"
+        end
+        true
+      end
+      @btn_play.signal_connect('clicked') do |toggle|
+        if toggle.active?
+          @player.play
+        else
+          @player.pause
+        end if @player
+      end
+      btn_next.signal_connect('clicked') do |btn|
+        if @queue
+          playing = @player.playing?
+          @player.stop
+          @queue_pos += 1
+          if @queue_pos < @queue.size
+            prepare_song
+            play if playing
+          else
+            @queue_pos = 0
+            prepare_song
+            @btn_play.active = false
+          end
+        end
+      end
+      btn_prev.signal_connect('clicked') do |btn|
+        if @queue
+          playing = @player.playing?
+          @player.stop
+          @queue_pos -= 1
+          if @queue_pos >= 0
+            prepare_song
+            play if playing
+          else
+            @queue_pos = 0
+            prepare_song
+            @btn_play.active = false
+          end
+        end
+      end
+
+      # Horizontally split area
+      hpaned = HPaned.new
+      vpaned.add hpaned
+
+      # Content
+      frm_content = Frame.new
+      hpaned.add2 frm_content
+
+      # Sidebar
+      sw = ScrolledWindow.new(nil, nil)
+      hpaned.add1 sw
+      sw.set_policy(POLICY_AUTOMATIC, POLICY_NEVER)
+      sw.width_request = 150
+      sidebar = VBox.new(false, 0)
+      sw.add_with_viewport sidebar
+      sto_tabs = ListStore.new(String)
+      NUM_TABS.times do |tab|
+        iter = sto_tabs.append
+        iter[0] = TAB_NAMES[tab]
+      end
+      lst_tabs = TreeView.new sto_tabs
+      lst_tabs.headers_visible = false
+      renderer = CellRendererText.new
+      column = TreeViewColumn.new('Invisible header',
+        renderer,
+        'text' => 0)
+      lst_tabs.append_column column
+      sidebar.pack_start lst_tabs, true, true, 0
+      # Cover image widget
+      @img_cover = Image.new
+      @img_cover.height_request = 150
+      sidebar.pack_start @img_cover, false, false, 0
+      # Lyrics text view
+      @txt_lyrics = TextView.new
+      @txt_lyrics.editable = false
+      scr_lyrics = ScrolledWindow.new(nil, nil)
+      scr_lyrics.set_policy(POLICY_AUTOMATIC, POLICY_AUTOMATIC)
+      scr_lyrics.add @txt_lyrics
+      # Browse area
+      browse = Sh::Browse.new self
+      lst_tabs.selection.signal_connect('changed') do |selection|
+        frm_content.remove frm_content.child if frm_content.child
+        case selection.selected.path.indices[0]
+        when TAB_QUEUE
+          frm_content.child = Label.new 'Queue'
+        when TAB_BROWSE
+          frm_content.child = browse.widget
+        when TAB_LYRICS
+          frm_content.child = scr_lyrics
+        end
+        frm_content.show_all
+      end
+      # Select 'Browse' tab
+      i = 0
+      lst_tabs.model.each do |row|
+        lst_tabs.selection.select_path row[1] if i == TAB_BROWSE
+        i += 1
+      end
+
+      # Show everything
+      @window.show_all
+    end
+
+    private
+    def create_menu
+      menu_items = [
+        ['/_File'],
+        ['/File/sep1', '<Separator>', nil, nil, lambda {}],
+        ['/File/Quit', '<StockItem>', '<control>Q', Stock::QUIT, lambda {Gtk.main_quit}],
+        ['/_Edit'],
+        ['/Edit/Preferences...', '<StockItem>', nil, Stock::PREFERENCES, lambda {show_preferences}],
+        ['/_Help'],
+        ['/Help/About', '<StockItem>', nil, Stock::ABOUT, lambda {show_about}]
+      ]
+      accel_group = AccelGroup.new
+      item_factory = ItemFactory.new(ItemFactory::TYPE_MENU_BAR, '<main>', accel_group)
+      @window.add_accel_group(accel_group)
+      item_factory.create_items(menu_items)
+      item_factory.get_widget('<main>')
+    end
+
+    def show_preferences
+      #FIXME: disable close button on dialog
+      dlg_prefs = @glade.get_widget("dlg_preferences")
+      dlg_prefs.signal_connect('delete-event') do
+        dlg_prefs.hide
+      end
+      fbtn_library = @glade.get_widget("fbtn_library")
+      fbtn_library.current_folder = $prefs[:library_dir]
+      chk_lastfm = @glade.get_widget("chk_lastfm")
+      chk_lastfm.active = $prefs[:lastfm]
+      txt_lastfm_user = @glade.get_widget("txt_lastfm_user")
+      txt_lastfm_user.text = $prefs[:lastfm_user]
+      txt_lastfm_pass = @glade.get_widget("txt_lastfm_pass")
+      txt_lastfm_pass.text = $prefs[:lastfm_password]
+      btn_ok = @glade.get_widget("btn_ok")
+      ok_handle = btn_ok.signal_connect('clicked') do
+        $prefs[:library_dir] = fbtn_library.filename
+        $prefs[:lastfm] = chk_lastfm.active?
+        $prefs[:lastfm_user] = txt_lastfm_user.text
+        $prefs[:lastfm_password] = txt_lastfm_pass.text
+        Sh::Global.save_prefs
+        dlg_prefs.hide
+      end
+      btn_close = @glade.get_widget("btn_close")
+      close_handle = btn_close.signal_connect('clicked') do
+        dlg_prefs.hide
+      end
+      dlg_prefs.show
+      while dlg_prefs.visible?
+        Gtk.main_iteration
+      end
+      btn_ok.signal_handler_disconnect ok_handle
+      btn_close.signal_handler_disconnect close_handle
+    end
+
+    def show_about
+      version = '0.0.0'
+      Gem::SourceIndex.from_installed_gems.each do |n, spec|
+        version = spec.version.to_s if spec.name == 'shroom'
+      end
+      dlg_about = Gnome::About.new('Shroom', version,
+        "Copyright (C) 2009 Aiden Nibali",
+        "Shroom - A music player and organizer in Ruby",
+        ["Aiden Nibali"], ["Aiden Nibali"], nil)
+      dlg_about.logo = Gdk::Pixbuf.new(Global.locate('icon_16x16.png'))
+      dlg_about.show
+    end
+
+    def prepare_song
+      @player = Sh::Player.new @queue[@queue_pos]
+      @player.on_finished do |player|
+        if player == @player
+          @player.stop
+          @queue_pos += 1
+          if @queue_pos < @queue.size
+            prepare_song
+            play
+          else
+            @queue_pos = 0
+            prepare_song
+            @btn_play.active = false
+          end
+        end
+      end
+      on_song_changed
+    end
+
+    # Format time (in seconds) as mins:secs
+    def time_to_s(time)
+      mins_float = time / 60
+      mins = mins_float.floor
+      secs = ((mins_float - mins) * 60).round
+      if secs == 60
+        mins += 1
+        secs = 0
+      end
+      "#{mins}:#{sprintf('%.2d', secs)}"
+    end
+
+    def on_song_changed
+      if @player
+        song = @player.song
+        @note.close if @note
+        @note = nil
+        if @rnotify
+          # Prepare notification
+          msg = "by <i>#{song.artist}</i> from <i>#{song.album}</i>"
+          @note = Notify::Notification.new(song.title || 'Unknown track', msg, nil, @status_icon)
+        end
+        # Cover art
+        if $prefs[:cover_art]
+          @pixbuf = nil
+          (@pixbuf = Gdk::Pixbuf.new(song.image)) rescue Exception
+          if @pixbuf
+            @note.pixbuf_icon = @pixbuf.scale(48, 48) if @rnotify
+            @img_cover.pixbuf = @pixbuf.scale(132, 132)
+          else
+            @img_cover.pixbuf = nil
+            Thread.new do
+              song.image = Sh::CoverArt.get_cover(song)
+              $db.save_song song
+              # Show cover unless requests have been shuffled
+              if song == @player.song
+                @pixbuf = nil
+                (@pixbuf = Gdk::Pixbuf.new(song.image)) rescue Exception
+                if @pixbuf
+                  if @rnotify
+                    @note.close
+                    @note.pixbuf_icon = @pixbuf.scale(48, 48)
+                    @note.show
+                  end
+                  @img_cover.pixbuf = @pixbuf.scale(132, 132)
+                end
+              end
+            end
+          end
+        end
+        # Lyrics
+        if $prefs[:lyrics]
+          @txt_lyrics.buffer.text = "Loading..."
+          Thread.new do
+            if not song.lyrics
+              song.lyrics = Sh::Lyrics.get_lyrics(song)
+              $db.save_song song
+            end
+            # Show lyrics unless requests have been shuffled
+            @txt_lyrics.buffer.text = song.lyrics if @player.song == song
+          end
+        end
+        if @rnotify
+          #Show notification
+          @note.timeout = 5000
+          @note.show
+        end
+        # Tooltip
+        @status_icon.tooltip = @player.song.to_s
+      end
+    end
+
+    public
+    def queue=(queue)
+      @queue_pos = 0
+      @queue = queue
+      prepare_song
+    end
+
+    def play
+      stop
+      if @player
+        @btn_play.active = true
+        @player.play
+      end
+    end
+
+    def stop
+      @btn_play.active = false
+      @player.stop if @player
+    end
+
+    def show
+      Gtk.main
+    end
+
+    def quit
+      Gtk.main_quit
+    end
+  end
+end