hearken 0.0.1 → 0.0.2

data/lib/hearken/indexing/audio_traverser.rb

@@ -0,0 +1,19 @@
+require 'pathname'
+require 'hearken/indexing/file'
+
+class Hearken::Indexing::AudioTraverser
+  attr_reader :current
+  EXTS = %w{m4a mp3 ogg wma}.map {|e| '.'+e }
+
+  def initialize path
+    @path = Pathname.new path
+  end
+
+  def each
+    @path.find { |child| yield Hearken::Indexing::File.new child, @path if is_audio? child }
+  end
+private
+  def is_audio? current
+    current.file? and EXTS.include? current.extname.downcase
+  end
+end

data/lib/hearken/indexing/executor.rb

@@ -0,0 +1,33 @@
+require 'hearken/tagged'
+
+class String
+  def escape char
+    gsub(char, "\\#{char}")
+  end
+
+  def escape2 char
+    split(char).join("\\#{char}")
+  end
+end
+
+module Hearken::Indexing::Executor
+  include Hearken::Tagged
+
+  def extract_file_attributes path
+    @path = path.to_s
+    @timestamp = path.timestamp
+  end
+
+  def clean_path path
+    path.escape(" ").escape2("'").escape("!").escape2("`").escape("(").escape(")").escape2("&").escape2(";")
+  end
+
+  def execute command
+    debug command
+    `#{command} 2>&1`
+  end
+
+  def debug message
+    puts message if ENV['DEBUG']
+  end
+end

data/lib/hearken/indexing/ffmpeg_file.rb

@@ -0,0 +1,51 @@
+require 'hearken/indexing/executor'
+
+class Hearken::Indexing::FfmpegFile
+  include Hearken::Indexing::Executor
+
+  def initialize path
+    extract_file_attributes path
+    content = execute "ffmpeg -i #{clean_path @path}"
+    state = :draining
+    @meta = {}
+    content.each_line do |line|
+      l = line.chomp
+      case l
+        when "  Metadata:"
+          state = :metadata
+        else
+          if state == :metadata
+            begin
+              m = / *: */.match l
+              @meta[m.pre_match.strip] = m.post_match.strip if m
+            rescue ArgumentError => e
+            end
+          end
+      end
+    end
+
+    @title = @meta['TIT2'] || @meta['title']
+    @album = @meta['TALB'] || @meta ['album']
+    @artist = @meta['TPE1'] || @meta['TPE2'] || @meta['artist']
+    @albumartist = @meta['TSO2'] || @meta['album_artist']
+    @time = to_duration @meta['Duration']
+    @date = @meta['TDRC'] || @meta['TYER'] || @meta['date']
+    @track = @meta['TRCK'] || @meta['track']
+    @puid = @meta['MusicIP PUID']
+    @mbartistid = @meta['MusicBrainz Artist Id']
+    @mbalbumid = @meta['MusicBrainz Album Id']
+    @mbalbumartistid = @meta['MusicBrainz Album Artist Id']
+    @asin = @meta['ASIN']
+  end
+
+  def method_missing method
+    @meta[method.to_s]
+  end
+private
+  def to_duration s
+    return nil unless s
+    first, *rest = s.split ','
+    hours, minutes, seconds = first.split ':'
+    seconds.to_i + (minutes.to_i * 60) + (hours.to_i * 60 * 60)
+  end
+end

data/lib/hearken/indexing/file.rb

@@ -0,0 +1,13 @@
+class Hearken::Indexing::File
+  def initialize path, root
+    @path, @root = path, root
+  end
+
+  def timestamp
+    @path.mtime.to_i.to_s
+  end
+
+  def to_s
+    @path.to_s
+  end
+end

data/lib/hearken/indexing/indexer.rb

@@ -0,0 +1,28 @@
+require 'splat'
+require 'hearken/indexing/persistant_traverser'
+
+class Hearken::Indexing::Indexer
+  def initialize path
+    @path = path
+  end
+
+  def execute
+    start = Time.now
+    count = 0
+    traverser = Hearken::Indexing::PersistantTraverser.new @path, Hearken::Indexing::PATH
+
+    traverser.each do |audio_file|
+      count += 1
+      show_progress start, count if count % 1000 == 0
+    end
+
+    show_progress start, count
+
+    (File.dirname(__FILE__)+'/../../../media/applause.mp3').to_player
+  end
+private
+  def show_progress start, count
+    elapsed = Time.now-start
+    $stderr.puts "Processed #{count} audio files in #{elapsed} seconds (#{elapsed/count} per file)"
+  end
+end

data/lib/hearken/indexing/parser.rb

@@ -0,0 +1,7 @@
+require 'hearken/indexing/ffmpeg_file'
+
+module Hearken::Indexing::Parser
+  def self.parse path
+    Hearken::Indexing::FfmpegFile.new path
+  end
+end

data/lib/hearken/indexing/persistant_traverser.rb

@@ -0,0 +1,32 @@
+require 'hearken/indexing/persisted_traverser'
+require 'hearken/indexing/audio_traverser'
+require 'hearken/indexing/parser'
+
+class Hearken::Indexing::PersistantTraverser
+  def initialize audio_path, store_path
+    @audio_path, @store_path = audio_path, store_path
+  end
+
+  def each
+    with_existing_entries do |existing_entries, persisted|
+      Hearken::Indexing::AudioTraverser.new(@audio_path).each do |path|
+        existing = existing_entries[path.to_s]
+        existing = nil if existing and existing.timestamp != path.timestamp
+        existing = nil if existing and existing.no_tag_fields?
+        track = existing || Hearken::Indexing::Parser.parse(path)
+        persisted.append track
+        yield track
+      end
+    end
+  end
+private
+  def with_existing_entries
+    persisted = Hearken::Indexing::PersistedTraverser.new @store_path
+    entries = {}
+    persisted.each do |entry|
+      entries[entry.path] = entry
+    end
+    persisted.clear
+    yield entries, persisted
+  end
+end

data/lib/hearken/indexing/persisted_traverser.rb

@@ -0,0 +1,30 @@
+require 'hearken/track'
+
+class Hearken::Indexing::PersistedTraverser
+  include Enumerable
+
+  def initialize path
+    @path = path
+  end
+
+  def each
+    File.open @path  do |file|
+      while line = file.gets
+        row = line.chomp.split '<->'
+        track = Hearken::Track.new
+        Hearken::Tagged::FIELDS.each {|field| track.send "#{field}=", row.shift }
+        yield track
+      end
+    end if File.exist? @path
+  end
+
+  def clear
+    File.open @path, 'w'
+  end
+
+  def append track
+    File.open(@path, 'a') do |file|
+      file.puts track.to_a.join('<->')
+    end
+  end
+end

data/lib/hearken/library.rb

@@ -0,0 +1,46 @@
+require 'hearken/track'
+
+class Hearken::Library
+  FILE_FIELDS = %w{path timestamp}
+  TAG_FIELDS = %w{album track title artist time date albumartist puid mbartistid mbalbumid mbalbumartistid asin}
+  FIELDS = FILE_FIELDS + TAG_FIELDS
+
+  include Hearken::Debug
+  attr_reader :tracks
+
+  def initialize preferences
+  end
+
+  def count
+    @tracks.count
+  end
+
+  def row id
+    @tracks[id]
+  end
+
+  def path row
+    row.path
+  end
+
+  def with_track id
+    yield @tracks[id]
+  end
+
+  def reload
+    s = Time.now.to_i
+    path = File.expand_path('~')+'/.music'
+    @tracks = []
+    File.open path do |file|
+      id = 0
+      while line = file.gets
+        row = line.chomp.split '<->'
+        track = Hearken::Track.new id
+        FIELDS.each {|field| track.send "#{field}=", row.shift }
+        @tracks << track
+        id += 1
+      end
+    end if File.exist? path
+    puts "Reloaded db with #{@tracks.size} tracks in #{Time.now.to_i-s} seconds"
+  end
+end

data/lib/hearken/player.rb

@@ -0,0 +1,91 @@
+require 'splat'
+
+require 'hearken/queue'
+require 'hearken/scrobbler'
+require 'hearken/library'
+
+module Hearken
+  class Player
+    include Queue
+    attr_reader :library, :scrobbler
+    attr_accessor :scrobbling, :matches
+
+    def initialize preferences
+      @scrobbler = Scrobbler.new preferences
+      @scrobbling = true
+      @library = Library.new preferences
+      @library.reload
+    end
+
+    def c text,colour
+      text.to_s.foreground colour
+    end
+
+    def status
+      if @pid
+        track = self.current
+        puts "Since #{c Time.at(track.started), :cyan}\n\t#{track}"
+        played = Time.now.to_i-track.started
+        puts "#{c played, :yellow} seconds (#{c track.time.to_i-played, :yellow} remaining)" if track.time
+      else
+        puts 'not playing'.foreground(:yellow)
+      end
+    end
+
+    def current
+      (@pid and File.exist?('current_song')) ? YAML.load_file('current_song') : nil
+    end
+
+    def register track
+      track.started = Time.now.to_i
+      File.open('current_song', 'w') {|f| f.print track.to_yaml }
+    end
+
+    def start
+      if @pid
+        puts "Already started (pid #{@pid})"
+        return
+      end
+      @pid = fork do
+        player_pid = nil
+        Signal.trap('TERM') do
+          Process.kill 'TERM', player_pid if player_pid
+          exit
+        end
+        total_tracks = @library.count
+        loop do
+          id = dequeue || (rand * total_tracks).to_i
+          row = @library.row id
+          unless row
+            puts "track with id #{id} did not exist"
+            next
+          end
+          path = @library.path row
+          unless path and File.exist? path
+            puts "track with id #{id} did not refer to a file"
+            next
+          end
+          @library.with_track(id) do |track|
+            @scrobbler.update track if @scrobbling
+            register track
+            player_pid = path.to_player
+            Process.wait player_pid
+            @scrobbler.scrobble track if @scrobbling
+          end
+        end
+      end
+      puts "Started (pid #{@pid})"
+    end
+
+    def stop
+      return unless @pid
+      Process.kill 'TERM', @pid 
+      @pid = nil
+    end
+
+    def restart
+      stop
+      start
+    end
+  end
+end

data/lib/hearken/preferences.rb

@@ -0,0 +1,29 @@
+module Hearken
+  class Preferences
+    def initialize
+      @preference_path = home_path '.hearken'
+      if File.exists? @preference_path
+        @preferences = YAML.load File.read(@preference_path)
+      else
+        @preferences = {}
+      end
+    end
+
+    def home_path *paths
+      File.join File.expand_path('~'), *paths
+    end
+
+    def [] key
+      @preferences[key]
+    end
+
+    def []= key, value
+      @preferences[key] = value
+      persist
+    end
+
+    def persist
+      File.open(@preference_path, 'w') {|f| f.puts @preferences.to_yaml}
+    end
+  end
+end

data/lib/hearken/queue.rb

@@ -0,0 +1,26 @@
+module Hearken
+  module Queue
+    def enqueue id
+      @sequence ||= 0
+      @library.with_track id do |track|
+        File.open("#{Time.now.to_i}-#{@sequence.to_s.rjust(8,'0')}.song", 'w') {|f| f.print track.to_yaml }
+        @sequence += 1
+      end
+    end
+
+    def each
+      Dir.glob('*.song').sort.each do |file|
+        yield YAML.load File.read(file)
+      end
+    end
+
+    def dequeue
+      file = Dir.glob('*.song').sort.first
+      return nil unless file
+      hash = YAML.load(File.read(file))
+      id = hash[:id] if hash
+      FileUtils.rm file
+      id
+    end
+  end
+end

data/lib/hearken/range_expander.rb

@@ -0,0 +1,30 @@
+module Hearken
+  class RangeExpander
+    def expand text
+      text.split(/[^0-9a-z-]/).inject([]) {|acc, term| acc + expand_term(term) }
+    end
+
+    def expand_to_ids text
+      expand(text).map {|number| from_number number }
+    end
+  private
+    def expand_term term
+      words = term.split '-'
+      words.empty? ? [] : range(words.first, words.last)
+    end
+
+    def range from, to
+      f, t = to_number(from), to_number(to)
+      t = to_number(from.slice(0...from.size-to.size)+to) if t < f
+      (f..t).to_a
+    end
+
+    def from_number term
+      term.to_s 36
+    end
+
+    def to_number term
+      term.to_i 36
+    end
+  end
+end