Obsolete
/
dnsmanagerv1
Archived
3
0
Fork 0
This repository has been archived on 2024-06-18. You can view files and clone it, but cannot push or open issues/pull-requests.
dnsmanagerv1/lib/db.pm

270 lines
6.1 KiB
Perl
Raw Normal View History

package db;
use v5.14;
use Moo;
use Modern::Perl;
use autodie;
use DBI;
use getiface ':all';
# db handler
has dbh => ( is => 'rw', builder => '_void');
sub _void { my $x = ''; \$x; }
# reference to the application
has data => qw/is ro required 1/;
sub BUILD {
my $self = shift;
my $db = $$self{data}{database};
my $dsn = "DBI:$$db{sgbd}:database=$$db{name};"
. "host=$$db{host};port=$$db{port}";
$$self{dbh} = DBI->connect($dsn, $$db{user}, $$db{passwd})
|| die "Could not connect to database: $DBI::errstr";
$$self{dbh}->{mysql_enable_utf8} = 1;
$$self{dbh}->do('SET NAMES \'utf8\';') || die;
}
# USER
sub auth {
my ($self, $login, $passwd) = @_;
my $sth;
$sth = $self->dbh->prepare('SELECT * FROM user WHERE login=? and passwd=?');
unless ($sth->execute($login, $passwd)) {
$sth->finish();
die q{Can't authenticate.};
}
# if we can't find the user with this password
unless (my $ref = $sth->fetchrow_arrayref) {
$sth->finish();
die q{The user can't be authenticated.};
}
$sth->finish();
# if this user exists and is auth
$self->get_user($login)
}
sub register_user {
my ($self, $login, $pass) = @_;
my $sth = $self->dbh->prepare('select * from user where login=?');
unless ( $sth->execute($login) ) {
$sth->finish();
die "Impossible to check if the user $login exists.";
}
# if an user already exists
if (my $ref = $sth->fetchrow_arrayref) {
$sth->finish();
die "The user $login already exists.";
}
# if not
$sth = $self->dbh->prepare('insert into user VALUES(?,?,?)');
unless ($sth->execute($login, $pass, 0)) {
$sth->finish();
die "Impossible to register the user $login.";
}
$sth->finish();
}
sub delete_user {
my ($self, $login) = @_;
my $sth;
# TODO : vérifier que ça renvoie la bonne valeur
$sth = $self->dbh->prepare('delete from user where login=?');
unless ( $sth->execute($login) ) {
$sth->finish();
die "Impossible to delete the user $login.";
}
$sth->finish();
$self->delete_domains_from_user($login)
}
sub get_user {
my ($self, $login) = @_;
my ($sth, $user);
$sth = $self->dbh->prepare('SELECT * FROM user WHERE login=?');
unless ( $sth->execute($login)) {
$sth->finish();
die "Impossible to check if the user $login exists.";
}
unless ($user = $sth->fetchrow_hashref) {
$sth->finish();
die "User $login doesn't exist.";
}
$sth->finish();
# the user gets all his domains
$$user{domains} = $self->get_domains($login);
$user
}
sub get_all_users {
my ($self) = @_;
my ($sth, $users);
$sth = $self->dbh->prepare('SELECT * FROM user');
unless ( $sth->execute()) {
$sth->finish();
die q{Impossible to list the users.};
}
while( my $ref = $sth->fetchrow_hashref) {
push @$users, $ref;
}
$sth->finish();
$users
}
sub toggle_admin {
my ($self, $login) = @_;
my $user = $self->get_user($login);
my $val = ($$user{admin}) ? 0 : 1;
my $sth = $self->dbh->prepare('update user set admin=? where login=?');
unless ( $sth->execute( $val, $login ) ) {
$sth->finish();
die "Impossible to toggle admin the user $login.";
}
$sth->finish()
}
sub update_passwd {
my ($self, $login, $new) = @_;
my $sth;
$sth = $self->dbh->prepare('update user set passwd=? where login=?');
unless ( $sth->execute($new, $login) ) {
$sth->finish();
die q{The password can't be updated.};
}
$sth->finish()
}
# DOMAIN
sub get_domains {
my ($self, $login) = @_;
my ($sth);
my $domains = [];
$sth = $self->dbh->prepare('SELECT * FROM domain where login=?');
unless ($sth->execute($login)) {
$sth->finish();
die "Impossible to check if the user $login has domains.";
}
while(my $ref = $sth->fetchrow_hashref) {
push @$domains, $ref;
}
$sth->finish();
$domains
}
sub delete_domain {
my ($self, $domain) = @_;
my $sth;
$sth = $self->dbh->prepare('delete from domain where domain=?');
unless ( $sth->execute($domain) ) {
$sth->finish();
die "Impossible to delete the $domain.";
}
$sth->finish()
}
sub delete_domains_from_user {
my ($self, $login) = @_;
my $sth;
$sth = $self->dbh->prepare('delete from domain where login=?');
unless ( $sth->execute($login) ) {
$sth->finish();
die "Impossible to delete the domains of the user $login.";
}
$sth->finish()
}
# TODO check if the domain is reserved
sub add_domain {
my ($self, $login, $domain) = @_;
my ($sth);
$sth = $self->dbh->prepare('select domain from domain where domain=?');
unless ( $sth->execute($domain) ) {
$sth->finish();
die 'Impossible to search if the domain already exists.';
}
# if the domain already exists
if (my $ref = $sth->fetchrow_arrayref) {
$sth->finish();
die 'The domain already exists.';
}
$sth = $self->dbh->prepare('insert into domain VALUES(?,?,?)');
unless ( $sth->execute($domain, $login, 0) ) {
$sth->finish();
die 'Impossible to add a domain.';
}
$sth->finish();
}
sub get_all_domains {
my ($self) = @_;
my ($sth, $domains);
$sth = $self->dbh->prepare('SELECT * FROM domain');
unless ( $sth->execute()) {
$sth->finish();
die q{Impossible to list the domains.};
}
while( my $ref = $sth->fetchrow_hashref) {
push @$domains, $ref;
}
$sth->finish();
$domains
}
sub disconnect {
my ($self) = @_;
$$self{dbh}->disconnect()
}
sub is_owning_domain {
my ($self, $login, $domain) = @_;
my $sth =
$self->dbh->prepare('SELECT * FROM domain where login=? and domain=?');
unless ($sth->execute($login, $domain)) {
$sth->finish();
die "Impossible to check if the user $login has domains.";
}
unless($sth->fetchrow_hashref) {
$sth->finish();
return 0
}
1
}
1;