lib/Odin.pm, mason/pastebin/dhandler: Apply a size limit on pastes.

[odin-cgi] / bin / pastebin.userv

diff --git a/bin/pastebin.userv b/bin/pastebin.userv
index 4d5d4c01807f46d38f844e3cd2b63a8bf495ff48..ed7ba7a147ff0ac300a9cf555f620e9c2002ce06 100755 (executable)
--- a/bin/pastebin.userv
+++ b/bin/pastebin.userv
@@ -6,7 +6,6 @@ use Odin;
 use DBI;
 use Encode;
 use Encode::Locale;
-use Getopt::Std;
 use POSIX;
 
 my $BAD = 0;
@@ -43,21 +42,26 @@ EOF
   @ARGV == 0 or Odin::fail "usage: list";
   my $db = Odin::open_db;
   for my $r (@{$db->selectall_arrayref
-               ("SELECT " . Odin::sql_timestamp($db, "stamp") .
-                     ", tag, lang, title
+               ("SELECT tag, stamp, lang, title
                  FROM odin_pastebin WHERE owner = ?
                  ORDER BY stamp", undef, $Odin::WHO)}) {
-    my ($stamp, $tag, $lang, $title) = @$r;
+    my ($tag, $stamp, $lang, $title) = @$r;
     my $t = strftime "%Y-%m-%d %H:%M:%S %z", localtime $stamp;
     printf "%-25s  %-12s  %-16s  %s\n",
       $t, $tag, $lang, encode locale => $title;
   }
 } elsif ($op eq "new") {
-  my %o;
-  getopts "l:t:", \%o and @ARGV == 0
-    or Odin::fail "usage: new [-l LANG] [-t TITLE]";
-  my %p = (title => decode(locale => $o{t}), lang => $o{l} // "txt",
-          content => read_content);
+  my $op = Odin::OptParse->new(@ARGV);
+  my $p = (title => undef, lang => "txt");
+  while (my $o = $op->get) {
+    if ($o eq "l") { $p{lang} = $op->arg; }
+    elsif ($o eq "t") { $p{title} = decode locale => $op->arg; }
+    else { $op->unk; }
+  }
+  @ARGV = $op->rest;
+  $op->bad if @ARGV;
+  $op->ok or Odin::fail "usage: new [-l LANG] [-t TITLE]";
+  $p{content} = read_content;
   my $db = Odin::open_db;
   my $c = "";
   while (read STDIN, my $buf, 8192) { $c .= $buf; }
@@ -85,12 +89,19 @@ EOF
   @ARGV or Odin::fail "usage: del TAG ...";
   Odin::delete_pastebin map { $_, undef } @ARGV;
 } elsif ($op eq "update") {
-  my %o;
-  getopts "cl:t:", \%o and @ARGV == 1
-    or Odin::fail "usage: update [-c] [-l LANG] [-t TITLE] TAG";
-  my ($tag) = @ARGV;
-  my %p = (title => decode(locale => $o{t}), lang => $o{l});
-  if ($o{c}) { $p{content} = read_content; }
+  my $op = Odin::OptParse->new(@ARGV);
+  my %p = ();
+  my $contentp = 0;
+  while (my $o = $op->get) {
+    if ($o eq "c") { $contentp = 1; }
+    elsif ($o eq "l") { $p{lang} = $op->arg; }
+    elsif ($o eq "t") { $p{title} = decode locale => $op->arg; }
+    else { $op->unk; }
+  }
+  @ARGV = $op->rest;
+  $op->bad if @ARGV;
+  $op->ok or Odin::fail "usage: new [-l LANG] [-t TITLE]";
+  $p{content} = read_content if $contentp;
   Odin::update_pastebin $tag, undef, %p or Odin::fail "nothing changed";
 } else {
   Odin::fail "unknown operation `$op'";