share/lua/website/cbsnews.lua

   1
   2 -- quvi
   3 -- Copyright (C) 2010  quvi project
   4 --
   5 -- This file is part of quvi <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 -- Identify the script.
  24 function ident (self)
  25     package.path = self.script_dir .. '/?.lua'
  26     local C      = require 'quvi/const'
  27     local r      = {}
  28     r.domain     = "cbsnews.com"
  29     r.formats    = "default|best|1100k|598k|386k"
  30     r.categories = C.proto_http
  31     r.handles    =
  32         (self.page_url ~= nil and self.page_url:find(r.domain) ~= nil)
  33     return r
  34 end
  35
  36 -- Parse video URL.
  37 function parse (self)
  38     self.host_id = "cbsnews"
  39     local page   = quvi.fetch(self.page_url)
  40
  41     -- Need "? because some videos have the " and some don't
  42     local _,_,s = page:find('CBSVideo.setVideoId%("?(.-)"?%);')
  43     self.id     = s or error ("no match: video id")
  44
  45     -- Goto cnet and get the url for the xml of the available video formats
  46     s =  "http://api.cnet.com/restApi/v1.0/videoSearch?videoIds="
  47         .. s
  48         .. "&iod=videoMedia"
  49
  50     local xml = quvi.fetch (s, {fetch_type = 'config'})
  51
  52     -- Grab title from the XML.
  53     local _,_,s = xml:find ('<Title>.-CDATA%[(.-)%]')
  54     self.title  = s or error ("no match: video title")
  55
  56     -- Go over the URLs. Figure out 'best' based on bitrate.
  57     local t    = {}
  58     local best = 0
  59
  60     for url in xml:gfind ('<DeliveryUrl>.-CDATA%[(.-)%]') do
  61         local _,_,s = url:find('_(%d+)%.%w+$')
  62         local n     = tonumber (s)
  63         if (best < n) then best = n end
  64         t[n]        = url
  65     end
  66
  67     local url = nil
  68
  69     if (self.requested_format == 'best') then
  70        url = t[best]
  71     else
  72
  73         -- Try to match requested_format to known IDs. Strip non-digits.
  74         local _,_,f = self.requested_format:find ('(%d+)')
  75         table.foreach (t,
  76             function (k,v) if (tonumber (f) == k) then url = v end end)
  77
  78         -- Use whatever is in the table as our 'default' if above failed.
  79         if (url == nil) then
  80             table.foreach (t,
  81                 function (k,v) if (url == nil) then url = v end end)
  82         end
  83     end
  84
  85     self.url = {url}
  86
  87     return self
  88 end
  89
  90 -- vim: set ts=4 sw=4 tw=72 expandtab: