share/media/funnyordie.lua

   1 -- libquvi-scripts
   2 -- Copyright (C) 2011,2013  Toni Gundogdu <legatvs@gmail.com>
   3 -- Copyright (C) 2010 quvi project
   4 --
   5 -- This file is part of libquvi-scripts <http://quvi.sourceforge.net/>.
   6 --
   7 -- This program is free software: you can redistribute it and/or
   8 -- modify it under the terms of the GNU Affero General Public
   9 -- License as published by the Free Software Foundation, either
  10 -- version 3 of the License, or (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 Affero General Public License for more details.
  16 --
  17 -- You should have received a copy of the GNU Affero General
  18 -- Public License along with this program.  If not, see
  19 -- <http://www.gnu.org/licenses/>.
  20 --
  21
  22 local FunnyOrDie = {} -- Utility functions unique to this script
  23
  24 -- Identify the media script.
  25 function ident(qargs)
  26   return {
  27     can_parse_url = FunnyOrDie.can_parse_url(qargs),
  28     domains = table.concat({'funnyordie.com'}, ',')
  29   }
  30 end
  31
  32 -- Parse media properties.
  33 function parse(qargs)
  34   local p = quvi.http.fetch(qargs.input_url).data
  35
  36   qargs.thumb_url = p:match('"og:image" content="(.-)"') or ''
  37   qargs.title = p:match('"og:title" content="(.-)">') or ''
  38   qargs.id = p:match('key:%s+"(.-)"') or ''
  39
  40   qargs.streams = FunnyOrDie.iter_streams(p)
  41
  42   return qargs
  43 end
  44
  45 --
  46 -- Utility functions
  47 --
  48
  49 function FunnyOrDie.can_parse_url(qargs)
  50   local U = require 'socket.url'
  51   local t = U.parse(qargs.input_url)
  52   if t and t.scheme and t.scheme:lower():match('^http$')
  53        and t.host   and t.host:lower():match('funnyordie%.com$')
  54        and t.path   and t.path:lower():match('^/videos/%w+')
  55   then
  56     return true
  57   else
  58     return false
  59   end
  60 end
  61
  62 function FunnyOrDie.iter_streams(p)
  63   local t = {}
  64
  65   -- 41a7516647: added the pattern in the next line, and removed...
  66   for u in p:gmatch('type: "video/mp4", src: "(.-)"') do table.insert(t,u) end
  67   for u in p:gmatch('source src="(.-)"') do table.insert(t,u) end -- ... This.
  68   -- Keep both of them.
  69
  70   if #t ==0 then
  71     error('no match: media stream URL')
  72   end
  73
  74   local S = require 'quvi/stream'
  75   local r = {}
  76
  77   -- nostd is a dictionary used by this script only. libquvi ignores it.
  78   for _,u in pairs(t) do
  79     local q,c = u:match('/(%w+)%.(%w+)$')
  80     if q and c then
  81       local s = S.stream_new(u)
  82       s.nostd = {quality=q}
  83       s.container = c
  84       s.id = FunnyOrDie.to_id(s)
  85       table.insert(r,s)
  86     end
  87   end
  88
  89   if #r >1 then
  90     FunnyOrDie.ch_best(r)
  91   end
  92
  93   return r
  94 end
  95
  96 function FunnyOrDie.ch_best(t)
  97   t[1].flags.best = true
  98 end
  99
 100 function FunnyOrDie.to_id(t)
 101   return string.format("%s_%s", t.nostd.quality, t.container)
 102 end
 103
 104 -- vim: set ts=2 sw=2 tw=72 expandtab: