share/media/sevenload.lua

   1 -- libquvi-scripts
   2 -- Copyright (C) 2010-2013  Toni Gundogdu <legatvs@gmail.com>
   3 --
   4 -- This file is part of libquvi-scripts <http://quvi.sourceforge.net/>.
   5 --
   6 -- This program is free software: you can redistribute it and/or
   7 -- modify it under the terms of the GNU Affero General Public
   8 -- License as published by the Free Software Foundation, either
   9 -- version 3 of the License, or (at your option) any later version.
  10 --
  11 -- This program is distributed in the hope that it will be useful,
  12 -- but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 -- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 -- GNU Affero General Public License for more details.
  15 --
  16 -- You should have received a copy of the GNU Affero General
  17 -- Public License along with this program.  If not, see
  18 -- <http://www.gnu.org/licenses/>.
  19 --
  20
  21 local SevenLoad = {} -- Utility functions unique to this script.
  22
  23 -- Identify the media script.
  24 function ident(qargs)
  25   return {
  26     can_parse_url = SevenLoad.can_parse_url(qargs),
  27     domains = table.concat({'sevenload.com'}, ',')
  28   }
  29 end
  30
  31 -- Parse media properties.
  32 function parse(qargs)
  33   local p = quvi.http.fetch(qargs.input_url).data
  34
  35   local E = require 'quvi/entity'
  36   p = E.convert_html(p)
  37
  38   qargs.thumb_url = p:match('"og:image" content="(.-)"') or ''
  39
  40   qargs.title = p:match('"og:title" content="(.-)"') or ''
  41
  42   qargs.id = p:match('videoid":"(.-)"') or ''
  43
  44   qargs.streams = SevenLoad.iter_streams(p)
  45
  46   return qargs
  47 end
  48
  49 --
  50 -- Utility functions
  51 --
  52
  53 function SevenLoad.can_parse_url(qargs)
  54   local U = require 'socket.url'
  55   local t = U.parse(qargs.input_url)
  56   if t and t.scheme and t.scheme:lower():match('^http$')
  57        and t.host   and t.host:lower():match('sevenload%.com$')
  58        and t.path   and t.path:lower():match('^/videos/')
  59   then
  60     return true
  61   else
  62     return false
  63   end
  64 end
  65
  66 function SevenLoad.stream_new(S, t)
  67   local u = t['src'] or error('no match: media stream URL')
  68   local s = S.stream_new(u)
  69   -- 'nostd' is a temporary dictionary and will be ignored by libquvi.
  70   s.nostd = {
  71     quality = u:match('%-(%w+)%.%w+$') or ''
  72   }
  73   s.video.encoding = t['video_codec'] or ''
  74   s.audio.encoding = t['audio_codec'] or ''
  75   s.container = u:match('%.(%w+)$') or ''
  76   s.id = SevenLoad.to_id(s)
  77   return s
  78 end
  79
  80 function SevenLoad.iter_streams(p)
  81   local S = require 'quvi/stream'
  82   local J = require 'json'
  83
  84   local d = p:match('data%-html5="(.-)">') or error('no match: data-html5')
  85   local j = J.decode(d)
  86
  87   local s = j['sources'] or error('"sources" not found')
  88   local r = {}
  89
  90   for _,v in pairs(s) do
  91     table.insert(r, SevenLoad.stream_new(S,v))
  92   end
  93
  94   return r
  95 end
  96
  97 function SevenLoad.to_id(t)
  98   return string.format('%s_%s', t.nostd.quality, t.container)
  99 end
 100
 101 -- vim: set ts=2 sw=2 tw=72 expandtab: