chiark / gitweb /
Store Unix timestamps in the database, rather than SQL ones.
[odin-cgi] / lib / Odin.pm
index e7786f56d6462c1c2ba91336d5bfea3f6845c4cc..551c51f0e83cc7248baa176513a8df201c1af71b 100644 (file)
@@ -50,6 +50,10 @@ our $PASTEBIN = "$BASEURL$PASTEBIN_PATH";
 ###--------------------------------------------------------------------------
 ### Miscellaneous utilities.
 
+our $NOW;
+sub update_now () { $NOW = time; }
+update_now;
+
 (our $PROG = $0) =~ s:^.*/::;
 
 sub fail_cmdline ($$%) {
@@ -95,12 +99,10 @@ sub open_db (@) {
   my $drv = $db->{Driver}{Name};
   if ($drv eq "Pg") {
     $db->{private_odin_retry_p} = sub { $db->state =~ /^40[0P]01$/ };
-    $db->{private_odin_unixstamp} = sub { "extract(epoch from $_[0])" };
   } elsif ($drv eq "SQLite") {
     $db->{private_odin_retry_p} = sub { $db->err == 5 };
-    $db->{private_odin_unixstamp} = sub { "strftime('%s', $_[0])" };
   } else {
-    fail "unsupported database driver `$drv' (patches welcome)", undef;
+    $db->{private_odin_retry_p} = sub { 0 };
   }
 
   return $db;
@@ -127,10 +129,6 @@ sub xact (&$) {
   die $exc;
 }
 
-sub sql_timestamp ($$) {
-  my ($db, $col) = @_;
-  return $db->{private_odin_unixstamp}->($col);
-}
 
 ###--------------------------------------------------------------------------
 ### Sequence numbers and tagging.
@@ -263,8 +261,9 @@ sub new_shorturl ($) {
        undef, $WHOCMP, $url);
     unless (defined $tag) {
       $tag = encode_tag(next_seq($db, "odin_shorturl_seq"));
-      $db->do("INSERT INTO odin_shorturl (tag, owner, url) VALUES (?, ?, ?)",
-             undef, $tag, $WHO, $url);
+      $db->do("INSERT INTO odin_shorturl (tag, stamp, owner, url)
+              VALUES (?, ?, ?, ?)", undef,
+             $tag, $NOW, $WHO, $url);
     }
   } $db;
   return $tag;
@@ -309,7 +308,7 @@ sub delete_shorturl (@) {
 
 our %PASTEBIN_DEFAULTS = (
   title => "(untitled)",
-  lang => "plain-text",
+  lang => "txt",
   content => ""
 );
 our @PASTEBIN_PROPS = keys %PASTEBIN_DEFAULTS;
@@ -327,9 +326,9 @@ sub new_pastebin (\%) {
   xact {
     $tag = encode_tag next_seq $db, "odin_pastebin_seq";
     $db->do("INSERT INTO odin_pastebin
-              (tag, edithash, owner, $PASTEBIN_PROPCOLS)
-            VALUES (?, ?, ?, $PASTEBIN_PROPPLACES)", undef,
-           $tag, $hash, $WHO, @{$new}{@PASTEBIN_PROPS});
+              (tag, stamp, edithash, owner, $PASTEBIN_PROPCOLS)
+            VALUES (?, ?, ?, ?, $PASTEBIN_PROPPLACES)", undef,
+           $tag, $NOW, $hash, $WHO, @{$new}{@PASTEBIN_PROPS});
   } $db;
   return $tag, $editkey;
 }
@@ -431,6 +430,50 @@ sub tidy_pastebin_content ($) {
   return $content;
 }
 
+###--------------------------------------------------------------------------
+### Simple option parser.
+
+package Odin::OptParse;
+
+sub new {
+  my ($cls, @args) = @_;
+  return bless {
+    cur => "",
+    args => \@args,
+    opt => undef,
+    ok => 1
+  }, $cls;
+}
+
+sub get {
+  my ($me) = @_;
+  if (!length $me->{cur}) {
+    my $args = $me->{args};
+    if (!@$args) { return undef; }
+    elsif ($args->[0] =~ /^[^-]|^-$/) { return undef; }
+    elsif ($args->[0] eq "--") { shift @$args; return undef; }
+    $me->{cur} = substr shift @$args, 1;
+  }
+  my $o = $me->{opt} = substr $me->{cur}, 0, 1;
+  $me->{cur} = substr $me->{cur}, 1;
+  return $o;
+}
+
+sub arg {
+  my ($me) = @_;
+  my $a;
+  if (length $me->{cur}) { $a = $me->{cur}; $me->{cur} = ""; }
+  elsif (@{$me->{args}}) { $a = shift @{$me->{args}}; }
+  else { $a = undef; $me->err("option `-$me->{opt}' requires an argument"); }
+  return $a;
+}
+
+sub rest { return @{$_[0]->{args}}; }
+sub ok { return $_[0]->{ok}; }
+sub bad { $_[0]->{ok} = 0; }
+sub err { $_[0]->bad; print STDERR "$PROG: $_[1]\n"; }
+sub unk { $_[0]->err("unknown option `-$_[0]->{opt}'"); }
+
 ###----- That's all, folks --------------------------------------------------
 
 1;