ikiwiki/IkiWiki/UserInfo.pm

96 lines
1.9 KiB
Perl

#!/usr/bin/perl
use warnings;
use strict;
use Storable;
use IkiWiki;
package IkiWiki;
sub userinfo_retrieve () { #{{{
my $userinfo=eval{ Storable::lock_retrieve("$config{wikistatedir}/userdb") };
return $userinfo;
} #}}}
sub userinfo_store ($) { #{{{
my $userinfo=shift;
my $newfile="$config{wikistatedir}/userdb.new";
my $oldmask=umask(077);
my $ret=Storable::lock_store($userinfo, $newfile);
umask($oldmask);
if (defined $ret && $ret) {
if (! rename($newfile, "$config{wikistatedir}/userdb")) {
unlink($newfile);
$ret=undef;
}
}
return $ret;
} #}}}
sub userinfo_get ($$) { #{{{
my $user=shift;
my $field=shift;
my $userinfo=userinfo_retrieve();
if (! defined $userinfo ||
! exists $userinfo->{$user} || ! ref $userinfo->{$user} ||
! exists $userinfo->{$user}->{$field}) {
return "";
}
return $userinfo->{$user}->{$field};
} #}}}
sub userinfo_set ($$$) { #{{{
my $user=shift;
my $field=shift;
my $value=shift;
my $userinfo=userinfo_retrieve();
if (! defined $userinfo ||
! exists $userinfo->{$user} || ! ref $userinfo->{$user}) {
return "";
}
$userinfo->{$user}->{$field}=$value;
return userinfo_store($userinfo);
} #}}}
sub userinfo_setall ($$) { #{{{
my $user=shift;
my $info=shift;
my $userinfo=userinfo_retrieve();
if (! defined $userinfo) {
$userinfo={};
}
$userinfo->{$user}=$info;
return userinfo_store($userinfo);
} #}}}
sub is_admin ($) { #{{{
my $user_name=shift;
return grep { $_ eq $user_name } @{$config{adminuser}};
} #}}}
sub get_banned_users () { #{{{
my @ret;
my $userinfo=userinfo_retrieve();
foreach my $user (keys %{$userinfo}) {
push @ret, $user if $userinfo->{$user}->{banned};
}
return @ret;
} #}}}
sub set_banned_users (@) { #{{{
my %banned=map { $_ => 1 } @_;
my $userinfo=userinfo_retrieve();
foreach my $user (keys %{$userinfo}) {
$userinfo->{$user}->{banned} = $banned{$user};
}
return userinfo_store($userinfo);
} #}}}
1