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 Blame History

This file contains invisible Unicode characters!

This file contains invisible Unicode characters that may be processed differently from what appears below. If your use case is intentional and legitimate, you can safely ignore this warning. Use the Escape button to reveal hidden characters.

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;