share/lua/website/vimeo.lua

   1
   2 -- libquvi-scripts
   3 -- Copyright (C) 2010-2012  Toni Gundogdu <legatvs@gmail.com>
   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 --
  24 -- NOTE: Vimeo is picky about the user-agent string.
  25 --
  26
  27 local Vimeo = {} -- Utility functions unique to this script.
  28
  29 -- Identify the script.
  30 function ident(self)
  31     package.path = self.script_dir .. '/?.lua'
  32     local C      = require 'quvi/const'
  33     local r      = {}
  34     r.domain     = "vimeo%.com"
  35     r.formats    = "default|best"
  36     r.categories = C.proto_http
  37     local U      = require 'quvi/util'
  38     r.handles    = U.handles(self.page_url, {r.domain}, {"/%d+$"})
  39     return r
  40 end
  41
  42 -- Query available formats.
  43 function query_formats(self)
  44     local config  = Vimeo.get_config(self)
  45     local formats = Vimeo.iter_formats(self, config)
  46
  47     local t = {}
  48     for _,v in pairs(formats) do
  49         table.insert(t, Vimeo.to_s(v))
  50     end
  51
  52     table.sort(t)
  53     self.formats = table.concat(t, "|")
  54
  55     return self
  56 end
  57
  58 -- Parse media URL.
  59 function parse(self)
  60     self.host_id  = "vimeo"
  61
  62     local c = Vimeo.get_config(self)
  63
  64     local s = c:match('"title":(.-),') or error("no match: media title")
  65     local U = require 'quvi/util'
  66     self.title = U.slash_unescape(s):gsub('^"',''):gsub('"$','')
  67
  68     self.duration = (tonumber(c:match('"duration":(%d+)')) or 0) * 1000
  69
  70     local s = c:match('"thumbnail":"(.-)"') or ''
  71     if #s >0 then
  72       self.thumbnail_url = U.slash_unescape(s)
  73     end
  74
  75     local formats = Vimeo.iter_formats(self, c)
  76     local format  = U.choose_format(self, formats,
  77                                      Vimeo.choose_best,
  78                                      Vimeo.choose_default,
  79                                      Vimeo.to_s)
  80                         or error("unable to choose format")
  81     self.url      = {format.url or error("no match: media stream URL")}
  82     return self
  83 end
  84
  85 --
  86 -- Utility functions
  87 --
  88
  89 function Vimeo.normalize(url)
  90     url = url:gsub("player.", "") -- player.vimeo.com
  91     url = url:gsub("/video/", "/") -- player.vimeo.com
  92     return url
  93 end
  94
  95 function Vimeo.get_config(self)
  96     self.page_url = Vimeo.normalize(self.page_url)
  97
  98     self.id = self.page_url:match('vimeo.com/(%d+)')
  99                 or error("no match: media ID")
 100
 101     local c_url = "http://vimeo.com/" .. self.id
 102     local c = quvi.fetch(c_url, {fetch_type='config'})
 103
 104     if c:match('<error>') then
 105         local s = c:match('<message>(.-)[\n<]')
 106         error( (not s) and "no match: error message" or s )
 107     end
 108
 109     return c
 110 end
 111
 112 function Vimeo.iter_formats(self, config)
 113     local t = {}
 114     local qualities = config:match('"qualities":%[(.-)%]')
 115                         or error('no match: qualities')
 116     for q in qualities:gmatch('"(.-)"') do
 117         Vimeo.add_format(self, config, t, q)
 118     end
 119     return t
 120 end
 121
 122 function Vimeo.add_format(self, config, t, quality)
 123     table.insert(t, {quality=quality,
 124                      url=Vimeo.to_url(self, config, quality)})
 125 end
 126
 127 function Vimeo.choose_best(t) -- First 'hd', then 'sd' and 'mobile' last.
 128     for _,v in pairs(t) do
 129         local f = Vimeo.to_s(v)
 130         for _,q in pairs({'hd','sd','mobile'}) do
 131             if f == q then return v end
 132         end
 133     end
 134     return Vimeo.choose_default(t)
 135 end
 136
 137 function Vimeo.choose_default(t)
 138   for _,v in pairs(t) do
 139       if Vimeo.to_s(v) == 'sd' then return v end -- Default to 'sd'.
 140   end
 141   return t[1] -- Or whatever is the first.
 142 end
 143
 144 function Vimeo.to_url(self, config, quality)
 145     local sign = config:match('"signature":"(.-)"')
 146                   or error("no match: request signature")
 147
 148     local exp = config:match('"timestamp":(%d+)')
 149                   or error("no match: request timestamp")
 150
 151     local s = "http://player.vimeo.com/play_redirect?clip_id=%s"
 152               .. "&sig=%s&time=%s&quality=%s&type=moogaloop_local"
 153
 154     return string.format(s, self.id, sign, exp, quality)
 155 end
 156
 157 function Vimeo.to_s(t)
 158     return string.format("%s", t.quality)
 159 end
 160
 161 -- vim: set ts=4 sw=4 tw=72 expandtab: