* pack usage message to $help = sub {...}

[lab.git] / Dev / twitter / dump_timeline.pl

diff --git a/Dev/twitter/dump_timeline.pl b/Dev/twitter/dump_timeline.pl
index 36fd046..8216287 100755 (executable)
--- a/Dev/twitter/dump_timeline.pl
+++ b/Dev/twitter/dump_timeline.pl
@@ -15,24 +15,38 @@ use YAML::Tiny;
 use Data::Dumper;
 use Encode;
 
-if (@ARGV < 1) {
-    die "usage: $0 screen_name [number_of_pages|all [dump]]\n";
+my $help = sub {
+    die <<EOM;
+usage: $0
+    [user_timeline(default)|retweeted_by_me|mentions
+        [screen_name
+            [number_of_pages|all
+                [dump]
+            ]
+        ]
+    ]
+EOM
+};
+if ($ARGV[0] && ($ARGV[0] eq '--help' || $ARGV[0] eq '-h') ) {
+    &{$help};
 }
-my $screen_name = $ARGV[0];
-my $pages = $ARGV[1] || 1;
+
+my $method = $ARGV[0] || 'user_timeline';
+my $screen_name = $ARGV[1] || '';
+my $pages = $ARGV[2] || 1;
 if ($pages eq 'all') {
     $pages = -1;
 }
-my $dump = $ARGV[2] || 0;
+my $dump = $ARGV[3] || 0;
 
 my $conf = loadconf("$Bin/config.yml");
 if (! defined $conf) {
-    die "$0: cannot parse config file.\n";
+    die "$0: cannot parse config file.";
 }
 
 my $bot = login($conf);
 if (! $bot->authorized) {
-    die "$0: this client is not yet authorized.\n";
+    die "$0: this client is not yet authorized.";
 }
 
 
@@ -40,12 +54,26 @@ eval {
     my $page = 0;
     while ($pages - $page && $page <= 160) {
         $page++;
-        my $res = $bot->user_timeline(
-            {
-                screen_name => $screen_name,
-                page        => $page,
-            }
-        );
+        
+        my $param = ($screen_name)
+            ? { page => $page, screen_name => $screen_name, }
+            : { page => $page, }
+        ;
+    
+        my $res;
+        if ($method eq 'user_timeline') {
+            $res = $bot->user_timeline($param);
+        }
+        elsif ($method eq 'retweeted_by_me') {
+            $res = $bot->retweeted_by_me($param);
+        }
+        elsif ($method eq 'mentions') {
+            $res = $bot->mentions($param);
+        }
+        else {
+            warn "$0: unknown method '$method'";
+            &{$help};
+        }
         
         if ($dump) {
             foreach my $line (split /\n/, Dumper $res) {
@@ -69,7 +97,6 @@ eval {
 if ($@) {
     evalrescue($@);
 }
-print "done\n";
 
 
 sub loadconf {
@@ -82,7 +109,7 @@ sub loadconf {
     my $yaml = YAML::Tiny->read($file);
     
     if ($!) {
-        warn "$0: '$file' $!\n";
+        warn "$0: '$file' $!";
     }
     
     return $yaml->[0];