Moving musicbrainz gem inside app

2011-11-24 01:45:45 +04:00
parent 3539dbc1d5
commit 83dd697fb9
24 changed files with 783 additions and 8 deletions
@@ -0,0 +1,38 @@
+module MusicBrainz
+  @@last_query_time = 0
+  @@query_interval = 1.5 # Min: 1.0 Safe: 1.5
+  
+  WEB_SERVICE_URL = 'http://musicbrainz.org/ws/2/'
+  USER_AGENT = "gem musicbrainz (https://github.com/magnolia-fan/musicbrainz) @ " + Socket.gethostname
+  
+  def self.query_interval= sec
+    @@query_interval = sec.to_f
+  end
+  
+  def self.load resourse, params = {}
+    url = WEB_SERVICE_URL + resourse.to_s.gsub('_', '-') + '/' + (params[:id].to_s || '')
+    params.delete(:id) unless params[:id].nil?
+    url << '?' + params.map{ |k, v|
+      k.to_s.gsub('_', '-') + '=' + (v.is_a?(Array) ? v.map{ |_| _.to_s.gsub('_', '-') }.join('+') : v.to_s)
+    }.join('&') unless params.empty?
+    self.get_contents url
+  end
+  
+private
+
+  def self.get_contents url
+    time_passed = Time.now.to_f - @@last_query_time
+    sleep @@query_interval - time_passed if time_passed < @@query_interval
+    response = nil
+    5.times do
+      begin
+        response = open(url, "User-Agent" => USER_AGENT)
+        @@last_query_time = Time.now.to_f
+      rescue => e
+        return nil if e.io.status[0].to_i == 404
+      end
+      break unless response.nil?
+    end
+    response
+  end
+end
@@ -0,0 +1,75 @@
+# encoding: UTF-8
+
+module MusicBrainz
+  class Artist < MusicBrainz::Base
+    attr_accessor :id, :type, :name, :country, :date_begin, :date_end, :urls
+    @release_groups
+  
+    def release_groups
+      if @release_groups.nil? and not self.id.nil?
+        @release_groups = []
+        Nokogiri::XML(MusicBrainz.load(:release_group, :artist => self.id)).css('release-group').each do |rg|
+          @release_groups << MusicBrainz::ReleaseGroup.parse_xml(rg)
+        end
+      end
+      @release_groups.sort{ |a, b| a.first_release_date <=> b.first_release_date }
+    end
+  
+    def self.find mbid
+      res = MusicBrainz.load :artist, :id => mbid, :inc => [:url_rels]
+      return nil if res.nil?
+      @artist = self.parse_xml(Nokogiri::XML(res))
+    end
+  
+    def self.parse_xml xml
+      @artist = MusicBrainz::Artist.new
+      @artist.id = self.safe_get_attr(xml, 'artist', 'id')
+      @artist.type = self.safe_get_attr(xml, 'artist', 'type')
+      @artist.name = self.safe_get_value(xml, 'artist > name').gsub(/[`’]/, "'")
+      @artist.country = self.safe_get_value(xml, 'artist > country')
+      @artist.date_begin = self.safe_get_value(xml, 'artist > life-span > begin')
+      @artist.date_end = self.safe_get_value(xml, 'artist > life-span > end')
+      @artist.urls = {}
+      xml.css('relation-list[target-type="url"] > relation').each do |rel|
+        @artist.urls[rel.attr('type').downcase.split(' ').join('_').to_sym] = rel.css('target').text
+      end
+      @artist
+    end
+    
+    def self.discography mbid
+      artist = self.find(mbid)
+      artist.release_groups.each {|rg| rg.releases.each {|r| r.tracks } }
+      artist
+    end
+    
+    def self.find_by_name name
+      matches = self.search name
+      matches.length.zero? ? nil : self.find(matches.first[:mbid])
+    end
+    
+    def self.search name
+      artists = []
+      xml = Nokogiri::XML(MusicBrainz.load(:artist, :query => name.gsub(/\!/, '\!').gsub(/\s/, '+') + '~', :limit => 50))
+      xml.css('artist-list > artist').each do |a|
+        artist = {
+          :name => a.first_element_child.text.gsub(/[`’]/, "'"),
+          :sort_name => self.safe_get_value(a, 'sort-name').gsub(/[`’]/, "'"),
+          :weight => 0,
+          :desc => self.safe_get_value(a, 'disambiguation'),
+          :type => self.safe_get_attr(a, nil, 'type'),
+          :mbid => self.safe_get_attr(a, nil, 'id')
+        }
+        aliases = a.css('alias-list > alias').map{ |item| item.text }
+        if aliases.include? name
+          artist[:weight] += 20
+        elsif aliases.map{ |item| item.downcase }.include? name.downcase
+          artist[:weight] += 10
+        elsif aliases.map{ |item| item.downcase.gsub(/\s/, '') }.include? name.downcase.gsub(/\s/, '')
+          artist[:weight] += 5
+        end
+        artists << artist
+      end
+      artists.sort{ |a, b| b[:weight] <=> a[:weight] }.take(10)
+    end
+  end
+end
@@ -0,0 +1,12 @@
+module MusicBrainz
+  class Base
+    def self.safe_get_attr xml, path, name
+      node = path.nil? ? xml : (xml.css(path).first unless xml.css(path).empty?)
+      node.attr(name) unless node.nil? or node.attr(name).nil?
+    end
+    
+    def self.safe_get_value xml, path
+      xml.css(path).first.text unless xml.css(path).empty?
+    end
+  end
+end
@@ -0,0 +1,41 @@
+module MusicBrainz
+  class Release < MusicBrainz::Base
+    attr_accessor :id, :title, :status, :format, :date, :country
+    @tracks
+  
+    def tracks
+      if @tracks.nil? and not self.id.nil?
+        @tracks = []
+        Nokogiri::XML(MusicBrainz.load(:release, :id => self.id, :inc => [:recordings, :media], :limit => 100)).css('medium-list > medium > track-list > track').each do |r|
+          @tracks << MusicBrainz::Track.parse_xml(r)
+        end
+      end
+      @tracks.sort{ |a, b| a.position <=> b.position }
+    end
+  
+    def self.find mbid
+      xml = Nokogiri::XML(MusicBrainz.load(:release, :id => mbid, :inc => [:media])).css('release').first
+      self.parse_xml(xml) unless xml.nil?
+    end
+  
+    def self.parse_xml xml
+      @release = MusicBrainz::Release.new
+      @release.id = self.safe_get_attr(xml, nil, 'id')
+      @release.title = self.safe_get_value(xml, 'title')
+      @release.status = self.safe_get_value(xml, 'status')
+      @release.format = self.safe_get_value(xml, 'medium-list > medium > format')
+      date = xml.css('date').empty? ? '2030-12-31' : xml.css('date').text
+      if date.length == 0
+        date = '2030-12-31'
+      elsif date.length == 4
+        date += '-12-31'
+      elsif date.length == 7
+        date += '-31'
+      end
+      date = date.split('-')
+      @release.date = Time.utc(date[0], date[1], date[2])
+      @release.country = self.safe_get_value(xml, 'country')
+      @release
+    end
+  end
+end
@@ -0,0 +1,40 @@
+module MusicBrainz
+  class ReleaseGroup < MusicBrainz::Base
+    attr_accessor :id, :type, :title, :disambiguation, :first_release_date
+    @releases
+  
+    def releases
+      if @releases.nil? and not self.id.nil?
+        @releases = []
+        Nokogiri::XML(MusicBrainz.load(:release, :release_group => self.id, :inc => [:media], :limit => 100)).css('release').each do |r|
+          @releases << MusicBrainz::Release.parse_xml(r)
+        end
+      end
+      @releases.sort{ |a, b| a.date <=> b.date }
+    end
+  
+    def self.find mbid
+      xml = Nokogiri::XML(MusicBrainz.load(:release_group, :id => mbid)).css('release-group').first
+      self.parse_xml(xml) unless xml.nil?
+    end
+  
+    def self.parse_xml xml
+      @release_group = MusicBrainz::ReleaseGroup.new
+      @release_group.id = self.safe_get_attr(xml, nil, 'id')
+      @release_group.type = self.safe_get_attr(xml, nil, 'type')
+      @release_group.title = self.safe_get_value(xml, 'title')
+      @release_group.disambiguation = self.safe_get_value(xml, 'disambiguation')
+      date = xml.css('first-release-date').empty? ? '2030-12-31' : xml.css('first-release-date').text
+      if date.length == 0
+        date = '2030-12-31'
+      elsif date.length == 4
+        date += '-12-31'
+      elsif date.length == 7
+        date += '-31'
+      end
+      date = date.split('-')
+      @release_group.first_release_date = Time.utc(date[0], date[1], date[2])
+      @release_group
+    end
+  end
+end
@@ -0,0 +1,20 @@
+module MusicBrainz
+  class Track < MusicBrainz::Base
+    attr_accessor :position, :recording_id, :title, :length
+  
+    def self.find mbid
+      xml = Nokogiri::XML(MusicBrainz.load(:recording, :id => mbid))
+      self.parse_xml(xml) unless xml.nil?
+    end
+  
+    def self.parse_xml xml
+      @track = MusicBrainz::Track.new
+      @track.position = self.safe_get_value(xml, 'position').to_i
+      @track.recording_id = self.safe_get_attr(xml, 'recording', 'id')
+      @track.title = self.safe_get_value(xml, 'recording > title')
+      @track.length = self.safe_get_value(xml, 'length').to_i
+      @track.length = self.safe_get_value(xml, 'recording > length').to_i
+      @track
+    end
+  end
+end
@@ -0,0 +1,11 @@
+require "open-uri"
+require "socket"
+require "nokogiri"
+require "cgi"
+
+require "models/music_brainz"
+require "models/music_brainz/base"
+require "models/music_brainz/artist"
+require "models/music_brainz/release_group"
+require "models/music_brainz/release"
+require "models/music_brainz/track"