git.sesse.net Git - vlc/blob - share/lua/meta/art/01_musicbrainz.lua

   1 --[[
   2  Gets an artwork from amazon
   3
   4  $Id$
   5  Copyright © 2007-2010 the VideoLAN team
   6
   7  This program is free software; you can redistribute it and/or modify
   8  it under the terms of the GNU General Public License as published by
   9  the Free Software Foundation; either version 2 of the License, or
  10  (at your option) any later version.
  11
  12  This program is distributed in the hope that it will be useful,
  13  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  GNU General Public License for more details.
  16
  17  You should have received a copy of the GNU General Public License
  18  along with this program; if not, write to the Free Software
  19  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  20 --]]
  21
  22 function try_query(query)
  23     local l = vlc.object.libvlc()
  24     local t = vlc.var.get( l, "musicbrainz-previousdate" )
  25     if t ~= nil then
  26         if t + 1000000. > vlc.misc.mdate() then
  27             vlc.msg.warn( "We must wait 1 second between requests unless we want to be blacklisted from the musicbrainz server." )
  28             vlc.misc.mwait( t + 1000000. )
  29         end
  30         vlc.var.set( l, "musicbrainz-previousdate", vlc.misc.mdate() )
  31     else
  32         vlc.var.create( l, "musicbrainz-previousdate", vlc.misc.mdate() )
  33     end
  34     l = nil
  35     vlc.msg.dbg( query )
  36     local s = vlc.stream( query )
  37     if not s then return nil end
  38     local page = s:read( 65653 )
  39
  40     -- FIXME: multiple results may be available
  41     _,_,asin = string.find( page, "<asin>(.-)</asin>" )
  42     if asin then
  43         return "http://images.amazon.com/images/P/"..asin..".01._SCLZZZZZZZ_.jpg"
  44     else
  45         return nil
  46     end
  47 end
  48
  49 function fuzzy(query)
  50     -- http://musicbrainz.org/doc/TextSearchSyntax#Fuzzy_searches
  51     -- we could even tweak the fuzziness
  52     return string.gsub(query, "([^%s]+)", "%1~")
  53 end
  54
  55 -- Return the artwork
  56 function fetch_art()
  57     local meta = vlc.input.item():metas()
  58     if not (meta["artist"] and meta["album"]) then
  59         return nil
  60     end
  61
  62     local query1 = "http://musicbrainz.org/ws/1/release/?type=xml&artist="..vlc.strings.encode_uri_component(meta["artist"]).."&title="..vlc.strings.encode_uri_component(meta["album"])
  63     local query2 = "http://musicbrainz.org/ws/1/release/?type=xml&query="..vlc.strings.encode_uri_component(meta["album"].." AND ".."artist:"..meta["artist"])
  64     local query3 = "http://musicbrainz.org/ws/1/release/?type=xml&query="..vlc.strings.encode_uri_component(fuzzy(meta["album"]).." AND ".."artist:"..fuzzy(meta["artist"]))
  65     return try_query(query1) or try_query(query2) or try_query(query3)
  66 end