- Check for existance of hash key before using
[videosite.git] / videosite / YouTubeGrabber.pm
1 # (c) 2007 by Ralf Ertzinger <ralf@camperquake.de>
2 # licensed under GNU GPL v2
3 #
4 # Grabber for youtube.com/de/...
5 #
6 # download strategy revised using
7 # http://www.kde-apps.org/content/show.php?content=41456
8
9 package YouTubeGrabber;
10
11 use GrabberBase;
12 @ISA = qw(GrabberBase);
13
14 use LWP::UserAgent;
15 use HTTP::Cookies;
16 use HTML::TokeParser;
17 use Data::Dumper;
18
19 use strict;
20
21 sub new {
22     my $class = shift;
23     my $self = $class->SUPER::new();
24
25     $self->{'NAME'} = 'youtube';
26     $self->{'PATTERNS'} = ['(http://(?:[-a-zA-Z0-9_.]+\.)*youtube.(?:com|de|co.uk)/watch\?(?:.+=.+&)*v=([-a-zA-Z0-9_]+))',
27                            '(http://(?:[-a-zA-Z0-9_.]+\.)*youtube.(?:com|de|co.uk)/v/([-a-zA-Z0-9_]+))'];
28     $self->{'_PARAMS'} = {'QUALITY' => ['normal', 'Quality of the video to download. normal = standard resolution flash video, high = higher resolution flash video, h264 = high resolution MPEG4 video'], 'USERNAME' => ['', 'Username to use for YouTube login'], 'PASSWORD' => ['', 'Password to use for YouTube login']};
29
30     bless($self, $class);
31     $self->_prepare_parameters();
32
33     return $self;
34 }
35
36 sub _parse {
37     my $self = shift;
38     my $url = shift;
39     my $pattern = shift;
40     my $content;
41     my $metadata = {};
42     my $p;
43     my $e;
44     my $tag;
45     my $jar = HTTP::Cookies->new;
46     my $ua = LWP::UserAgent->new('agent' => 'Mozilla/5.0');
47     my $r;
48     my $videourl;
49     my $quality = $self->_getval('QUALITY');
50     my $append = '';
51
52     $url =~ m|$pattern|;
53     $url = $1;
54
55     $metadata->{'URL'} = $url;
56     $metadata->{'ID'} = $2;
57     $metadata->{'TYPE'} = 'video';
58     $metadata->{'SOURCE'} = $self->{'NAME'};
59     $metadata->{'TITLE'} = undef;
60     $metadata->{'DLURL'} = undef;
61
62     if ($quality eq 'high') {
63         $append = '&fmt=6';
64     } elsif ($quality eq 'h264') {
65         $append = '&fmt=18';
66     }
67
68     $videourl = sprintf('http://www.youtube.com/watch?v=%s%s', $2, $append);
69
70     unless(defined($r = $ua->get($videourl))) {
71         $self->error('Could not download %s', $url);
72         return undef;
73     }
74
75     if ($r->base->as_string() =~ m,/verify_age,) {
76         $self->debug('Video requires age verification');
77         $ua->cookie_jar($jar);
78         $r = $self->__login($videourl, $ua);
79         unless(defined($r)) {
80             $self->error('Could not log into YouTube');
81             return undef;
82         }
83     }
84     $content = $r->content();
85
86     $p = HTML::TokeParser->new(\$content);
87
88     while ($tag = $p->get_tag('div', 'meta', 'script')) {
89         if ('meta' eq $tag->[0]) {
90             if ('title' eq $tag->[1]->{'name'}) {
91                 $metadata->{'TITLE'} = $tag->[1]->{'content'};
92                 $self->debug('Title found: %s', $metadata->{'TITLE'});
93             }
94         } elsif ('script' eq $tag->[0]) {
95             $e = $p->get_text();
96             if ($e =~ m|/watch_fullscreen\?(.+)\x27|) {
97                 my %args = map { split(/=/, $_, 2); } split(/&/, $1);
98                 $metadata->{'DLURL'} = sprintf('http://www.youtube.com/get_video.php?video_id=%s&t=%s%s',
99                         $metadata->{'ID'}, $args{'t'}, $append);
100                 $self->debug('URL found: %s', $metadata->{'DLURL'});
101             }
102         } elsif ('div' eq $tag->[0]) {
103             if (exists($tag->[1]->{'class'}) and ('errorBox' eq $tag->[1]->{'class'})) {
104                 $self->error("Could not get video data for youtube %s: %s",
105                         $metadata->{'ID'}, $p->get_trimmed_text());
106                 return undef;
107             }
108         }
109     }
110
111     unless(defined($metadata->{'DLURL'}) && defined($metadata->{'TITLE'})) {
112         $self->error('Could not determine download URL');
113         return undef;
114     }
115
116     return $metadata;
117 }
118
119 sub __login {
120     my $self = shift;
121     my $videourl = shift;
122     my $ua = shift;
123     my $user = $self->_getval('USERNAME');
124     my $pass = $self->_getval('PASSWORD');
125     my $r;
126     my $p;
127     my $c;
128     my $token;
129
130     sub check_cookie {
131
132         my $jar = shift;
133         my $key = shift;
134         my $found = undef;
135
136         $jar->scan(sub { $found = 1 if ( $key eq $_[1]) });
137
138         return $found;
139     }
140
141     if (($user eq '') or ($pass eq '')) {
142         $self->error('No username or password defined for YouTube');
143         return undef;
144     }
145
146     $self->debug('Logging in');
147     $r = $ua->post('http://www.youtube.com/signup', { 'next' => '/', 'current_form' => 'login', 'action_login' => '1', 'username' => $user, 'password' => $pass });
148
149     unless(check_cookie($ua->cookie_jar, 'LOGIN_INFO')) {
150         $self->error('Could not log into YouTube');
151         return undef;
152     }
153
154     $r = $ua->get($videourl);
155     if ($r->base->as_string() =~ m,/verify_age,) {
156         $self->debug("Looking for session token...");
157         $c = $r->decoded_content();
158         $p = HTML::TokeParser->new(\$c);
159         while (my $tag = $p->get_tag('script')) {
160             if ($p->get_text() =~ /gXSRF_token = '(.+)'/) {
161                 $token = $1;
162                 last;
163             }
164         }
165
166         unless(defined($token)) {
167             $self->error("Could not find session token");
168             return undef;
169         }
170
171         $self->debug('Authenticating session...');
172         $r = $ua->post($r->base->as_string, { 'next_url' => $r->base->path, 'action_confirm' => 'Confirm Birth Date', 'session_token' => $token });
173     }
174
175     unless(check_cookie($ua->cookie_jar, 'is_adult')) {
176         $self->error('Could not authenticate session');
177         return undef;
178     }
179
180     return $ua->get($videourl);
181 }
182
183 1;