sandro-metamuse 0.1.0

data/.document

@@ -0,0 +1,5 @@
+README.rdoc
+lib/**/*.rb
+bin/*
+features/**/*.feature
+LICENSE

data/.gitignore

@@ -0,0 +1,6 @@
+*.sw?
+.DS_Store
+coverage
+rdoc
+pkg
+.api_keys.yml

data/MIT_LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Sandro Turriate
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.rdoc

@@ -0,0 +1,7 @@
+= metamuse
+
+Search for music metadata
+
+== Copyright
+
+Copyright (c) 2009 Sandro Turriate. See MIT_LICENSE for details.

data/Rakefile

@@ -0,0 +1,48 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "metamuse"
+    gem.summary = %Q{Search for music metadata}
+    gem.email = "sandro.turriate@gmail.com"
+    gem.homepage = "http://github.com/sandro/metamuse"
+    gem.authors = ["Sandro Turriate"]
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+  end
+
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: sudo gem install jeweler"
+end
+
+require 'spec/rake/spectask'
+Spec::Rake::SpecTask.new(:spec) do |spec|
+  spec.libs << 'lib' << 'spec'
+  spec.spec_files = FileList['spec/**/*_spec.rb']
+end
+
+Spec::Rake::SpecTask.new(:rcov) do |spec|
+  spec.libs << 'lib' << 'spec'
+  spec.pattern = 'spec/**/*_spec.rb'
+  spec.rcov = true
+end
+
+
+task :default => :spec
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  if File.exist?('VERSION.yml')
+    config = YAML.load(File.read('VERSION.yml'))
+    version = "#{config[:major]}.#{config[:minor]}.#{config[:patch]}"
+  else
+    version = ""
+  end
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "metamuse #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end
+

data/VERSION

@@ -0,0 +1 @@
+0.1.0

data/lib/array_ext.rb

@@ -0,0 +1,5 @@
+class Array
+  def self.insist(thing)
+    thing.is_a?(Array) ? thing : [thing]
+  end
+end

data/lib/arrayish.rb

@@ -0,0 +1,23 @@
+module Arrayish
+  # List of array methods (that are not in +Object+) that need to be
+  # delegated to +collection+.
+  ARRAY_METHODS = (Array.instance_methods - Object.instance_methods).map { |n| n.to_s }
+
+  # List of additional methods that must be delegated to +collection+.
+  MUST_DEFINE = %w[to_a to_ary inspect]
+
+  (ARRAY_METHODS + MUST_DEFINE).uniq.each do |method|
+    class_eval <<-RUBY, __FILE__, __LINE__ + 1
+      def #{method}(*args, &block)                 # def each(*args, &block)
+        collection.send(:#{method}, *args, &block)  #   collection.send(:each, *args, &block)
+      end                                          # end
+    RUBY
+  end
+
+  # Lie about our class. Borrowed from Rake::FileList
+  # Note: Does not work for case equality (<tt>===</tt>)
+  def is_a?(klass)
+    klass == Array || super(klass)
+  end
+  alias kind_of? is_a?
+end

data/lib/httparty_ext.rb

@@ -0,0 +1,16 @@
+HTTParty::Request.class_eval do
+  class << self
+    attr_accessor :debug
+  end
+
+  def perform_with_debug
+    if self.class.debug
+      puts "HTTParty making #{http_method::METHOD} request to:"
+      puts uri
+    end
+    perform_without_debug
+  end
+
+  alias_method :perform_without_debug, :perform
+  alias_method :perform, :perform_with_debug
+end

data/lib/metamuse.rb

@@ -0,0 +1,71 @@
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+
+begin; require 'rubygems'; rescue LoadError; end
+require 'cgi'
+require 'httparty'
+
+require 'httparty_ext'
+require 'object_ext'
+require 'array_ext'
+require 'arrayish'
+
+require 'metamuse/collection'
+require 'metamuse/association'
+require 'metamuse/services'
+require 'metamuse/services/echonest'
+require 'metamuse/services/freebase'
+require 'metamuse/services/music_brainz'
+require 'metamuse/services/lastfm'
+require 'metamuse/services/lastfm/image'
+
+module Metamuse
+  autoload :Artist, 'metamuse/artist'
+  autoload :Album, 'metamuse/album'
+  autoload :Track, 'metamuse/track'
+
+  class << self
+    attr_reader :echonest_api_key, :lastfm_api_key
+
+    def echonest_api_key=(key)
+      @echonest_api_key = key
+      Services::Echonest.api_key = echonest_api_key
+    end
+
+    def lastfm_api_key=(key)
+      @lastfm_api_key = key
+      Services::Lastfm.api_key = lastfm_api_key
+    end
+
+    def find_artist(name)
+      Services::Echonest.artist(name)
+    end
+
+    private
+
+    def api_keys
+      if File.exists?(name=".api_keys.yml")
+        YAML.load_file name
+      else
+        {}
+      end
+    end
+
+    def register_api_keys
+      api_keys.each do |service, value|
+        send("#{service}_api_key=", value)
+      end
+    end
+  end
+
+  module InstanceInitialize
+    def initialize(attrs={})
+      attrs.each do |k,v|
+        if self.class.has_many_set.values.include?(k)
+          send "#{k}=", v
+        else
+          instance_variable_set "@#{k}", v
+        end
+      end
+    end
+  end
+end

data/lib/metamuse/album.rb

@@ -0,0 +1,19 @@
+module Metamuse
+  class Album
+    include InstanceInitialize
+    extend Association
+    has_many :tracks, Track
+
+    attr_accessor :name, :release_date, :freebase_id, :mbid, :rank, :images
+
+    def <=>(other)
+      if rank
+        rank <=> other.rank
+      else
+        to_s <=> other.to_s
+      end
+    end
+
+  end
+end
+

data/lib/metamuse/artist.rb

@@ -0,0 +1,43 @@
+module Metamuse
+  class Artist
+    include InstanceInitialize
+    extend Association
+    has_many :albums, Album
+
+    attr_accessor :name, :echonest_id, :freebase_guid, :mbid
+
+    def self.build(name)
+      artist = Services::Freebase.artist(name)
+      artist.enhance_albums!
+      artist
+    end
+
+    def tracks
+      @tracks ||= albums.map{|a| a.tracks}.flatten
+    end
+
+    def enhance_albums!
+      return albums if albums.empty?
+      self.albums = albums.map do |album|
+        lastfm_album = lastfm_albums.detect{|a| a.name == album.name}
+        if lastfm_album
+          album.rank = lastfm_album.rank
+          album.mbid = lastfm_album.mbid
+          album.images = lastfm_album.images
+        end
+        album
+      end.sort
+    end
+
+    def fetch_albums_and_tracks!
+      artist = Services::Freebase.artist(name)
+      self.albums = artist.albums
+    end
+
+    private
+
+    def lastfm_albums
+      @lastfm_albums ||= Services::Lastfm.top_albums(name)
+    end
+  end
+end

data/lib/metamuse/association.rb

@@ -0,0 +1,48 @@
+module Metamuse
+  module Association
+    def self.extended(base)
+      class << base
+        attr_accessor :has_many_set, :belongs_to_set
+      end
+      base.belongs_to_set = base.has_many_set = {}
+      base.send :include, InstanceMethods
+    end
+
+    def has_many(accessor, klass)
+      has_many_set.merge! klass => accessor
+
+      define_method(accessor) do
+        ivar = "@#{accessor}"
+        if val = instance_variable_get(ivar)
+          val
+        else
+          instance_variable_set(ivar, Collection.new(self, klass))
+        end
+      end
+
+      define_method("#{accessor}=") do |items|
+        send(accessor).clear
+        send(accessor) << items
+      end
+    end
+
+    def belongs_to(accessor, klass)
+      belongs_to_set.merge! klass => accessor
+
+      define_method(accessor) do
+        instance_variable_get(:"@#{accessor}")
+      end
+
+      define_method(:"#{accessor}=") do |parent|
+        instance_variable_set(:"@#{accessor}", parent)
+      end
+    end
+
+    module InstanceMethods
+      def belongs(owner)
+        accessor = self.class.belongs_to_set[owner.class]
+        send("#{accessor}=", owner) if accessor
+      end
+    end
+  end
+end

data/lib/metamuse/collection.rb

@@ -0,0 +1,25 @@
+module Metamuse
+  class Collection
+    include ::Arrayish
+    attr_reader :collection, :collection_class, :owner
+
+    def initialize(owner, collection_class)
+      @owner = owner
+      @collection_class = collection_class
+      @collection = []
+    end
+
+    alias_method :append, :<<
+    alias_method :push, :<<
+    alias_method :concat, :<<
+
+    def <<(items)
+      Array.insist(items).each do |item|
+        item = collection_class.new(item) unless collection_class === item
+        append item
+        item.belongs owner
+      end
+      self
+    end
+  end
+end

data/lib/metamuse/services.rb

@@ -0,0 +1,16 @@
+module Metamuse
+  module Services
+    module Service
+      def has_api_key()
+        class << self
+          attr_reader :api_key
+
+          def api_key=(key)
+            @api_key = key
+            default_params :api_key => api_key
+          end
+        end
+      end
+    end
+  end
+end

data/lib/metamuse/services/echonest.rb

@@ -0,0 +1,26 @@
+module Metamuse
+  module Services
+    class Echonest
+      extend Service
+      include HTTParty
+      base_uri 'http://developer.echonest.com/api'
+      default_params :version => 3
+      format :xml
+      has_api_key
+
+      def self.artist(name)
+        build_artist get('/search_artists', :query => {:query => name})
+      end
+
+      private
+
+      def self.build_artist(response)
+        if artists = response['response']['artists']
+          artist = artists['artist']
+          artist = artist.first if artist.is_a?(Array)
+          Metamuse::Artist.new :name => artist['name'], :echonest_id => artist['id']
+        end
+      end
+    end
+  end
+end

data/lib/metamuse/services/freebase.rb

@@ -0,0 +1,43 @@
+module Metamuse
+  module Services
+    class Freebase
+      include HTTParty
+      base_uri 'http://www.freebase.com'
+      format :json
+
+      def self.artist(name)
+        envelop = %({"query": #{album_and_track_mql(name)}})
+        response = get("/api/service/mqlread", :query => {:query => envelop})
+        build_artist response
+      end
+
+      def self.artist_search(name)
+        response = get("/api/service/search", :query => album_and_track_search_query(name))
+        build_artist response
+      end
+
+      private
+
+      def self.album_and_track_mql(name)
+        %({"album": [{ "id": null, "name": null, "track": [{ "name": null, "index": null, "sort": "index" }], "release_date": null, "release_type": null, "sort": "-release_date", "/common/topic/image": [{ "id": null, "optional": true, "limit": 3 }] }], "id": null, "name": "#{name}", "type": "/music/artist"})
+      end
+
+      def self.album_and_track_search_query(name)
+        {:query => name, :type => "/music/artist", :limit => 1, :mql_output => album_and_track_mql(name)}
+      end
+
+      def self.build_artist(response)
+        result = response['result']
+        artist = Metamuse::Artist.new :name => CGI.unescapeHTML(result['name']), :freebase_guid => result['guid']
+        albums = Array.insist result['album']
+        albums.each do |album|
+          data = {:name => CGI.unescapeHTML(album['name']), :release_date => album['release_date'], :freebase_id => album['freebase_id']}
+          tracks = album['track']
+          data[:tracks] = tracks.map {|t| t.update 'name' => CGI.unescapeHTML(t['name'])}
+          artist.albums << data
+        end
+        artist
+      end
+    end
+  end
+end