Changeset 7641

Timestamp:

03/08/08 01:32:43 (9 months ago)

Author:

woremacx

Message:

lang/perl/misc/fetch-nico-video.pl: now we have lots of options

Files:

• lang/perl/misc/fetch-nico-video.pl (modified) (2 diffs)

lang/perl/misc/fetch-nico-video.pl

@@ -3 +3 @@
 # copyed from http://coderepos.org/share/browser/lang/perl/plagger/lib/Plagger/Plugin/Filter/FetchNicoVideo.pm
 
+package FetchNicoVideo;
+
 use strict;
-
-use Config::Pit;
+use warnings;
+
+use Carp qw(croak);
+use Cwd ();
 use CGI;
+use Encode;
+use File::Spec;
+use HTTP::Request;
 use LWP::UserAgent;
+use Time::HiRes qw(sleep);
+use URI::Escape;
+
+our $VERSION      = '0.01';
+
+sub new {
+    my $class = shift;
+    my $opt   = ref($_[0]) eq "HASH" ? shift : {@_};
+
+    eval {
+        require Config::Pit;
+        $opt->{pit} = Config::Pit::pit_get("nicovideo.jp");
+    };
+
+    for my $key (qw/mail password/) {
+        $opt->{$key} ||= $opt->{pit}->{$key} if $opt->{pit}->{$key};
+        $opt->{_form}->{$key} = $opt->{$key};
+        croak "conifg '$key' is not set." unless $opt->{$key};
+    }
+    my $self = bless $opt, $class;
+
+    $self->_init_dir;
+    $self->_init_ua;
+
+    $self;
+}
+
+sub _init_dir {
+    my $self = shift;
+
+    unless ($self->{dir}) {
+        $self->{dir} = Cwd::cwd;
+        return;
+    }
+
+    if ($self->{dir} =~ /^[a-zA-Z]/ && $self->conf->{dir} !~ /:/) {
+        $self->{dir} = File::Spec->catfile(Cwd::cwd, $self->conf->{dir});
+    }
+
+    unless (-e $self->{dir} && -d _) {
+        warn $self->{dir} . " does not exist. Creating" unless $self->{quiet};
+        mkpath $self->{dir};
+    }
+}
+
+sub _init_ua {
+    my $self = shift;
+
+    my $ua = LWP::UserAgent->new(keep_alive => 1);
+    $ua->cookie_jar({});
+    $ua->post("https://secure.nicovideo.jp/secure/login?site=niconico" => $self->{_form});
+    $self->{ua} = $ua;
+
+    $self;
+}
 
 sub download {
-    my $ua    = shift;
-    my $entry = shift;
-
-    #get video_id
+    my ($self, $entry) = @_;
+    my $ua = $self->{ua};
+
+    # get video_id
     my ($video_id) = $entry =~ m!www.nicovideo.jp/watch/(.*)!;
-
-    #get flv url
+    croak "invalid entry ($entry)" unless $video_id;
+
+    if (not $self->{id_as_filename} and not exists($self->{title})) {
+        my $resentry = $ua->get($entry);
+        my $content = Encode::decode('utf-8', $resentry->content);
+        $self->{title} = ($content =~ m!<h1><a [^>]+>(.*?)</a></h1>!s)[0];
+    }
+
+    # get flv url
     my $res = $ua->get("http://www.nicovideo.jp/api/getflv?v=$video_id");
-    my $q   = CGI->new( $res->content );
+    my $q   = CGI->new($res->content);
     my $flv_url = $q->param('url');
 
-    unless ($flv_url) {
-        die "Not Found FLV URL : $video_id";
-    }
-    warn "Found FLV URL $flv_url";
+    return warn "Not Found FLV URL : $video_id" unless $flv_url;
+
+    print "Found FLV URL $flv_url" unless $self->{quiet};
+
+    #set local path
+    my $filename;
+    if ($self->{id_as_filename}) {
+        $filename = $video_id;
+
+    } else {
+        $filename = $self->{title};
+        utf8::encode($filename);
+        if ($self->{filename_encode}) {
+            Encode::from_to($filename, "utf-8", $self->{filename_encode});
+        }
+        if ($self->{add_id_to_filename}) {
+            $filename = "$video_id $filename";
+        }
+    }
 
     $flv_url =~ m!^http://[^/]+(?:smilevideo|nicovideo)\.jp/smile\?(\w)=(?:[^.]+)\.\d+(?:low)?!;
@@ -33 +117 @@
     );
     my $ext = exists( $video_type_of{$1} ) ? $video_type_of{$1} : "flv";
-    my $path = "$video_id.$ext";
-
-    unless ( -e $path ) {
-        #access video page
+    my $path = File::Spec->catfile( $self->{dir}, $filename . ".$ext" );
+
+    unless (-e $path) {
+        # access video page
         $ua->get("http://www.nicovideo.jp/watch/$video_id");
 
-        #download flv file
+        # download flv file
         my $req = HTTP::Request->new(GET => $flv_url);
-        warn "Fetching $video_id FLV File from " . $flv_url . "...";
-        my $res = $ua->request($req, $path);
+        print "Fetching $video_id (save as '$filename.$ext') FLV File from " . $flv_url . "..." unless $self->{quiet};;
+        $res = $ua->request($req, $path);
+        print "\n" unless $self->{quiet};
         warn "Fetch FLV Error: $video_id" if $res->is_error;
-    }else{
+    } else {
         warn "Exist FLV File: $video_id";
-        my $sleeping_time = 15;
+        my $sleeping_time = $self->{interval} || 15;
         warn "sleep $sleeping_time.";
-        sleep( $sleeping_time );
-    }
-}
-
-sub ua {
-    my $conf = shift;
-    my $ua = LWP::UserAgent->new( keep_alive => 1 );
-    $ua->cookie_jar( {} );
-    $ua->post( "https://secure.nicovideo.jp/secure/login?site=niconico" => $conf );
-    $ua;
-}
-
-sub conf {
-    my $conf = pit_get("nicovideo.jp", require => {
-        mail     => "your username on example",
-        password => "your password on example",
-    });
-    $conf;
-}
+        sleep($sleeping_time);
+    }
+}
+
+package main;
+
+use strict;
+use warnings;
+
+use Getopt::Long;
+
 sub main {
-    my $conf = conf;
-    my $ua = ua($conf);
+    my %opt = ();
+    GetOptions(\%opt,
+               'mail=s',
+               'password=s',
+               'quiet',
+               'dir=s',
+               'filename_encode=s',
+               'id_as_filename',
+               'add_id_to_filename',
+               'title=s',
+               'interval=i',
+              );
+
+    if (exists($opt{title}) and scalar(@ARGV) > 1) {
+        die "you cannot download multiple urls with --title option.";
+    }
+
+    my $nico = FetchNicoVideo->new(%opt);
     for my $entry (@ARGV) {
-        download($ua, $entry);
+        $nico->download($entry);
     }
 }
 
 main;
+
+__END__
+
+=head1 NAME
+
+fetch-nico-video - command line downloader of NicoVideo
+
+=head1 SYNOPSIS
+
+    fetch-nico-video [options] [urls ...]
+
+=head1 DESCRIPTION
+
+fetch-nico-video is a command line downloader of NicoVideo.
+
+=head1 CONFIG
+
+=over 4
+
+=item ~/.pit/default.yaml (on your pit)
+
+  ---
+  'nicovideo.jp':
+    mail: 'your mail address'
+    password: 'your password'
+
+=item --mail and --password (overrride values of pit)
+
+  --mail 'your mail address'
+  --password 'your password'
+
+=item --dir
+
+Directory to store downloaded enclosures. Optional.
+
+=item --filename_encode
+
+File name encode. Example: euc-jp / shift_jis. Optional. Default is utf-8.
+
+=item --id_as_filename
+
+IF set, set video id as flv file. Default file name is entry title. Optional.
+
+=item --add_id_to_filename
+
+IF set, add video id as prefix of filename. Optional.
+
+=item --title
+
+If set, override entry title. Optional.
+
+=item --interval
+
+Interval between each download. Default is 15.
+
+=back
+
+=head1 SEE ALSO
+
+L<Plagger::Plugin::Filter::FetchNicoVideo>, L<Config::Pit>
+
+=cut