summaryrefslogtreecommitdiffstats
path: root/Bugzilla/Attachment/Database.pm
blob: 42cdd2d6f2a72993fad9742c90fa0387b84524e7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
# 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;