fix youtube lua probing, avoid infinite loops
[vlc.git] / share / luaplaylist / youtube.lua
1 -- $Id$
2
3 -- Helper function to get a parameter's value in a URL
4 function get_url_param( url, name )
5     return string.gsub( url, "^.*[&?]"..name.."=([^&]*).*$", "%1" )
6 end
7
8 function get_arturl( path, video_id )
9     if string.match( vlc.path, "iurl=" ) then
10         return vlc.decode_uri( get_url_param( vlc.path, "iurl" ) )
11     end
12     if not arturl then
13         return "http://img.youtube.com/vi/"..video_id.."/default.jpg"
14     end
15 end
16
17 -- Probe function.
18 function probe()
19     if vlc.access ~= "http" then
20         return false
21     end
22     youtube_site = string.match( string.sub( vlc.path, 1, 8 ), "youtube" )
23     if not youtube_site then
24         -- FIXME we should be using a builtin list of known youtube websites
25         -- like "fr.youtube.com", "uk.youtube.com" etc..
26         youtube_site = string.find( vlc.path, ".youtube.com" )
27         if youtube_site == nil then
28             return false
29         end
30     end
31     return (  string.match( vlc.path, "watch%?v=" ) -- the html page
32             or string.match( vlc.path, "watch_fullscreen%?video_id=" ) -- the fullscreen page
33             or string.match( vlc.path, "p.swf" ) -- the (old?) player url
34             or string.match( vlc.path, "jp.swf" ) -- the (new?) player url (as of 24/08/2007)
35             or string.match( vlc.path, "player2.swf" ) ) -- another player url
36 end
37
38 -- Parse function.
39 function parse()
40     if string.match( vlc.path, "watch%?v=" )
41     then -- This is the HTML page's URL
42         while true do
43             -- Try to find the video's title
44             line = vlc.readline()
45             if not line then break end
46             if string.match( line, "<meta name=\"title\"" ) then
47                 name = string.gsub( line, "^.*content=\"([^\"]*).*$", "%1" )
48             end
49             if string.match( line, "<meta name=\"description\"" ) then
50                 description = string.gsub( line, "^.*content=\"([^\"]*).*$", "%1" )
51             end
52             if string.match( line, "subscribe_to_user=" ) then
53                 artist = string.gsub( line, ".*subscribe_to_user=([^&]*).*", "%1" )
54             end
55             -- var swfArgs = {hl:'en',BASE_YT_URL:'http://youtube.com/',video_id:'XPJ7d8dq0t8',l:'292',t:'OEgsToPDskLFdOYrrlDm3FQPoQBYaCP1',sk:'0gnr-AE6QZJEZmCMd3lq_AC'};
56             if string.match( line, "swfArgs" ) and string.match( line, "video_id" ) then
57                 if string.match( line, "BASE_YT_URL" ) then
58                     base_yt_url = string.gsub( line, ".*BASE_YT_URL:'([^']*)'.*", "%1" )
59                 end
60                 t = string.gsub( line, ".*t:'([^']*)'.*", "%1" )
61                 -- vlc.msg_err( t )
62                 -- video_id = string.gsub( line, ".*&video_id:'([^']*)'.*", "%1" )
63             end
64             if name and description and artist --[[and video_id]] then break end
65         end
66         if not video_id then
67             video_id = get_url_param( vlc.path, "v" )
68         end
69         if not base_yt_url then
70             base_yt_url = "http://youtube.com/"
71         end
72         art_url = get_arturl( vlc.path, video_id )
73         if t then
74             return { { path = base_yt_url .. "get_video?video_id="..video_id.."&t="..t; name = name; description = description; artist = artist; arturl = arturl } }
75         else
76             -- This shouldn't happen ... but keep it as a backup.
77             return { { path = "http://www.youtube.com/v/"..video_id; name = name; description = description; artist = artist; arturl = arturl } }
78         end
79     else -- This is the flash player's URL
80         if string.match( vlc.path, "title=" ) then
81             name = get_url_param( vlc.path, "title" )
82         end
83         video_id = get_url_param( vlc.path, "video_id" )
84         art_url = get_arturl( vlc.path, video_id )
85         if not string.match( vlc.path, "t=" ) then
86             -- This sucks, we're missing "t" which is now mandatory. Let's
87             -- try using another url
88             return { { path = "http://www.youtube.com/v/"..video_id; name = name; arturl = arturl } }
89         end
90         return { { path = "http://www.youtube.com/get_video.php?video_id="..video_id.."&t="..get_url_param( vlc.path, "t" ); name = name; arturl = arturl } }
91     end
92 end