use DBI;
use Digest::SHA qw(sha256_hex);
use MIME::Base64;
+use POSIX;
###--------------------------------------------------------------------------
### Early utilities.
qr{^https?://}
);
+our $PASTEMAXLEN = 1024*1024;
+
our %COOKIE_DEFAULTS = (
-httponly => undef,
-max_age => 3600
###--------------------------------------------------------------------------
### Miscellaneous utilities.
+our $NOW;
+sub update_now () { $NOW = time; }
+update_now;
+
(our $PROG = $0) =~ s:^.*/::;
sub fail_cmdline ($$%) {
return lc $s;
}
+sub print_columns (@) {
+ my @col = reverse @_;
+ my @fmt = ();
+ my @val = ();
+ while (@col && $col[1] eq "") { splice @col, 0, 2; }
+ my ($wd, $v) = splice @col, 0, 2;
+ push @fmt, "%s"; push @val, $v;
+ while (@col) {
+ my ($wd, $v) = splice @col, 0, 2;
+ push @fmt, "%-${wd}s";
+ push @val, $v;
+ }
+ printf join(" ", reverse @fmt) . "\n", reverse @val;
+}
+
+sub fmt_time ($) {
+ my ($t) = @_;
+ return $t == -1 ? "--" : strftime "%Y-%m-%d %H:%M:%S %z", localtime $t;
+}
+
###--------------------------------------------------------------------------
### Database utilities.
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;
die $exc;
}
-sub sql_timestamp ($$) {
- my ($db, $col) = @_;
- return $db->{private_odin_unixstamp}->($col);
+sub insert_record ($$%) {
+ my ($db, $table, %fields) = @_;
+ my @var = ();
+ my @val = ();
+
+ for my $v (keys %fields) {
+ push @var, $v;
+ push @val, $fields{$v};
+ }
+ $db->do("INSERT INTO $table (" . join(", ", @var) . ")
+ VALUES (" . join(", ", map { "?" } @var) . ")", undef, @val);
}
###--------------------------------------------------------------------------
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);
+ insert_record $db, "odin_shorturl",
+ tag => $tag, stamp => $NOW, owner => $WHO, url => $url;
}
} $db;
return $tag;
merge_hash %$new, %PASTEBIN_DEFAULTS;
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});
+ insert_record $db, "odin_pastebin",
+ tag => $tag, stamp => $NOW, edithash => $hash, owner => $WHO,
+ %$new;
} $db;
return $tag, $editkey;
}
return undef unless defined $content;
$content =~ tr/\r//d;
$content =~ s/([^\n])\z/$1\n/;
+ length $content <= $PASTEMAXLEN or
+ fail "invalid paste content", ".badpaste";
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;