dnsmanagerv1/lib/db.pm

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;