# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
#
# This Source Code Form is "Incompatible With Secondary Licenses", as
# defined by the Mozilla Public License, v. 2.0.

package Bugzilla::Attachment::Database;

use 5.10.1;
use strict;
use warnings;

use Bugzilla::Util qw(trick_taint);

sub new {
    return bless({}, shift);
}

sub store {
    my ($self, $attach_id, $data) = @_;
    my $dbh = Bugzilla->dbh;
    my $sth = $dbh->prepare("INSERT INTO attach_data (id, thedata) VALUES ($attach_id, ?)");
    trick_taint($data);
    $sth->bind_param(1, $data, $dbh->BLOB_TYPE);
    $sth->execute();
}

sub retrieve {
    my ($self, $attach_id) = @_;
    my $dbh = Bugzilla->dbh;
    my ($data) = $dbh->selectrow_array(
        "SELECT thedata FROM attach_data WHERE id = ?",
        undef,
        $attach_id
    );
    return $data;
}

sub remove {
    my ($self, $attach_id) = @_;
    my $dbh = Bugzilla->dbh;
    $dbh->do(
        "DELETE FROM attach_data WHERE id = ?",
        undef,
        $attach_id
    );
}

sub exists {
    my ($self, $attach_id) = @_;
    my $dbh = Bugzilla->dbh;
    my ($exists) = $dbh->selectrow_array(
        "SELECT 1 FROM attach_data WHERE id = ?",
        undef,
        $attach_id
    );
    return !!$exists;
}

1;