IkiWiki/UserInfo.pm

   1 #!/usr/bin/perl
   2
   3 package IkiWiki;
   4
   5 use warnings;
   6 use strict;
   7 use Storable;
   8 use IkiWiki;
   9
  10 sub userinfo_retrieve () {
  11         my $userinfo=eval{ Storable::lock_retrieve("$config{wikistatedir}/userdb") };
  12         return $userinfo;
  13 }
  14
  15 sub userinfo_store ($) {
  16         my $userinfo=shift;
  17
  18         my $newfile="$config{wikistatedir}/userdb.new";
  19         my $oldmask=umask(077);
  20         my $ret=Storable::lock_nstore($userinfo, $newfile);
  21         umask($oldmask);
  22         if (defined $ret && $ret) {
  23                 if (! rename($newfile, "$config{wikistatedir}/userdb")) {
  24                         unlink($newfile);
  25                         $ret=undef;
  26                 }
  27         }
  28         return $ret;
  29 }
  30
  31 sub userinfo_get ($$) {
  32         my $user=shift;
  33         my $field=shift;
  34
  35         my $userinfo=userinfo_retrieve();
  36         if (! defined $userinfo ||
  37             ! exists $userinfo->{$user} || ! ref $userinfo->{$user} ||
  38             ! exists $userinfo->{$user}->{$field}) {
  39                 if ($field eq "realname" && defined $ENV{SSL_CLIENT_S_DN_CN}) {
  40                         userinfo_set($user, "realname", $ENV{SSL_CLIENT_S_DN_CN});
  41                         return $ENV{SSL_CLIENT_S_DN_CN};
  42                 }
  43                 return "";
  44         }
  45         if ($field eq "realname" && $userinfo->{$user}->{$field} eq "" &&
  46             defined $ENV{SSL_CLIENT_S_DN_CN}) {
  47                 userinfo_set($user, "realname", $ENV{SSL_CLIENT_S_DN_CN});
  48                 return $ENV{SSL_CLIENT_S_DN_CN};
  49         }
  50         return $userinfo->{$user}->{$field};
  51 }
  52
  53 sub userinfo_set ($$$) {
  54         my $user=shift;
  55         my $field=shift;
  56         my $value=shift;
  57
  58         my $userinfo=userinfo_retrieve();
  59         if (! defined $userinfo ||
  60             ! exists $userinfo->{$user} || ! ref $userinfo->{$user}) {
  61                 return "";
  62         }
  63
  64         $userinfo->{$user}->{$field}=$value;
  65         return userinfo_store($userinfo);
  66 }
  67
  68 sub userinfo_setall ($$) {
  69         my $user=shift;
  70         my $info=shift;
  71
  72         my $userinfo=userinfo_retrieve();
  73         if (! defined $userinfo) {
  74                 $userinfo={};
  75         }
  76         $userinfo->{$user}=$info;
  77         return userinfo_store($userinfo);
  78 }
  79
  80 sub is_admin ($) {
  81         my $user_name=shift;
  82
  83         return grep { $_ eq $user_name } @{$config{adminuser}};
  84 }
  85
  86 1