File Coverage

blib/lib/Mojolicious/Plugin/AutoSecrets.pm

Criterion	Covered	Total	%
statement	44	46	95.6
branch	4	10	40.0
condition	10	25	40.0
subroutine	8	8	100.0
pod	2	2	100.0
total	68	91	74.7

package Mojolicious::Plugin::AutoSecrets;
# ABSTRACT: Automatic, Rotating Mojolicious Secrets


use Mojo::Base 'Mojolicious::Plugin';
use Mojo::Util qw(sha1_sum);
use Mojo::JSON qw(encode_json decode_json);
use Carp qw(croak);
use Fcntl qw(:DEFAULT :flock);
use autodie;

our $VERSION = '0.01';


sub register {
  my ($self, $app, $config) = @_;
  $config //= {};

  my $path        = $config->{path}        // $app->home->rel_file('.mojo-secrets');
  my $mode        = $config->{mode}        // 0600;
  my $expire_days = $config->{expire_days} // 60;
  my $prune       = $config->{prune}       // 3;
  my $generator   = $config->{generator}   // \&generator;

  my @secrets;

  sysopen my $fh, $path, O_RDWR | O_CREAT, $mode;
  flock $fh, LOCK_EX;

  my $rv;
  my $disk = '';
  while ($rv = $fh->sysread(my $buf, 4096, 0)) { $disk .= $buf }
  croak "Can't read from $path: $!"
    if !defined $rv;

  my $disk_secrets = $disk && decode_json($disk);

  unshift @secrets, @$disk_secrets
    if $disk_secrets;

  if (!@secrets || -z $path || -M _ > $expire_days) {
    unshift @secrets, $generator->();

    @secrets = @secrets[0 .. $prune - 1]
      if $prune && @secrets > $prune;

    $fh->seek(0, 0);
    $fh->syswrite(my $j = encode_json(\@secrets));
  }
  flock $fh, LOCK_UN;
  $fh->close;

  push @secrets, @{$app->{secrets}}
    if $app->{secrets};

  $app->secrets(\@secrets);
}


sub generator {
  sha1_sum(join '', rand(1000) x 2, $$, localtime)
}



1;

__END__