Bug 1180572 - create attachment_storage parameter

1 files changed, 59 insertions, 0 deletions

diff --git a/Bugzilla/Attachment/Database.pm b/Bugzilla/Attachment/Database.pm
new file mode 100644
index 000000000..24617cacc
--- /dev/null
+++ b/Bugzilla/Attachment/Database.pm
@@ -0,0 +1,59 @@
+# 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 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;