share/lua/website/imdb.lua

   1 --
   2 -- libquvi-scripts
   3 -- Copyright (C) 2011  quvi project
   4 --
   5 -- This file is part of libquvi-scripts <http://quvi.sourceforge.net/>.
   6 --
   7 -- This library is free software; you can redistribute it and/or
   8 -- modify it under the terms of the GNU Lesser General Public
   9 -- License as published by the Free Software Foundation; either
  10 -- version 2.1 of the License, or (at your option) any later version.
  11 --
  12 -- This library 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 GNU
  15 -- Lesser General Public License for more details.
  16 --
  17 -- You should have received a copy of the GNU Lesser General Public
  18 -- License along with this library; if not, write to the Free Software
  19 -- Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  20 -- 02110-1301  USA
  21 --
  22
  23 local IMDB = {} -- Utility functions unique to this script
  24
  25 -- Identify the script.
  26 function ident(self)
  27     package.path = self.script_dir .. '/?.lua'
  28     local C      = require 'quvi/const'
  29     local r      = {}
  30     r.domain     = 'imdb%.com'
  31     r.formats    = 'default|best'
  32     local B      = require 'quvi/bit'
  33     r.categories = B.bit_or(C.proto_http, C.proto_rtmp)
  34     local U      = require 'quvi/util'
  35     r.handles    = U.handles(self.page_url,
  36             {r.domain},
  37             {"/video/.+"})
  38     return r
  39 end
  40
  41 -- Query available formats.
  42 function query_formats(self)
  43     local page    = quvi.fetch(self.page_url)
  44     local formats = IMDB.iter_formats(page)
  45
  46     local t = {}
  47     for _,v in pairs(formats) do
  48         table.insert(t, IMDB.to_s(v))
  49     end
  50
  51     table.sort(t)
  52     self.formats = table.concat(t, "|")
  53
  54     return self
  55 end
  56
  57 -- Parse media URL.
  58 function parse(self)
  59     self.host_id = 'imdb'
  60
  61     self.id = self.page_url:match('/video/%w+/(vi%d-)/')
  62
  63     local page = quvi.fetch(self.page_url)
  64     self.title = page:match('<title>(.-) %- IMDb</title>')
  65
  66     --
  67     -- Format codes (for most videos):
  68     -- uff=1 - 480p RTMP
  69     -- uff=2 - 480p HTTP
  70     -- uff=3 - 720p HTTP
  71     --
  72     local formats = IMDB.iter_formats(page)
  73     local U       = require 'quvi/util'
  74     local format  = U.choose_format(self, formats,
  75                                     IMDB.choose_best,
  76                                     IMDB.choose_default,
  77                                     IMDB.to_s)
  78                         or error("unable to choose format")
  79     if not format.path then
  80         error("no match: media url")
  81     end
  82
  83     local url = 'http://www.imdb.com' .. format.path
  84     local iframe = quvi.fetch(url, {fetch_type = 'config'})
  85     local file = iframe:match('so%.addVariable%("file", "(.-)"%);')
  86     file = U.unescape(file)
  87
  88     if file:match('^http.-') then
  89         self.url = {file}
  90     else
  91         local path  = iframe:match('so%.addVariable%("id", "(.-)"%);')
  92         path = U.unescape(path)
  93         self.url = {file .. path}
  94     end
  95
  96     -- Optional: we can live without these
  97
  98     local thumb = iframe:match('so%.addVariable%("image", "(.-)"%);')
  99     self.thumbnail_url = thumb or ''
 100
 101     return self
 102 end
 103
 104 --
 105 -- Utility functions
 106 --
 107
 108 function IMDB.iter_formats(page)
 109     local p = "case '(.-)' :.-url = '(.-)';"
 110     local t = {}
 111     for c,u in page:gmatch(p) do
 112         table.insert(t, {path=u, container=c})
 113         --print(u,c)
 114     end
 115
 116     -- First item is ad, so remove it
 117     table.remove(t, 1)
 118
 119     return t
 120 end
 121
 122 function IMDB.choose_best(t) -- Expect the last to be the 'best'
 123   return t[#t]
 124 end
 125
 126 function IMDB.choose_default(t) -- Expect the second to be the 'default'
 127   return t[2]
 128 end
 129
 130 function IMDB.to_s(t)
 131   return t.container
 132 end
 133
 134 -- vim: set ts=4 sw=4 tw=72 expandtab: