summaryrefslogtreecommitdiffstats
path: root/Bugzilla/Attachment/S3.pm
blob: 7f47557203a59136480217398056a417a0f2febb (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
62
# 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::S3;

use 5.10.1;
use strict;
use warnings;

use Bugzilla::Error;
use Bugzilla::S3;

sub new {
  my $s3 = Bugzilla::S3->new({
    aws_access_key_id     => Bugzilla->params->{aws_access_key_id},
    aws_secret_access_key => Bugzilla->params->{aws_secret_access_key},
    secure                => 1,
  });
  return
    bless({s3 => $s3, bucket => $s3->bucket(Bugzilla->params->{s3_bucket}),},
    shift);
}

sub store {
  my ($self, $attach_id, $data) = @_;
  unless ($self->{bucket}->add_key($attach_id, $data)) {
    warn "Failed to add attachment $attach_id to S3: "
      . $self->{bucket}->errstr . "\n";
    ThrowCodeError('s3_add_failed',
      {attach_id => $attach_id, reason => $self->{bucket}->errstr});
  }
}

sub retrieve {
  my ($self, $attach_id) = @_;
  my $response = $self->{bucket}->get_key($attach_id);
  if (!$response) {
    warn "Failed to retrieve attachment $attach_id from S3: "
      . $self->{bucket}->errstr . "\n";
    ThrowCodeError('s3_get_failed',
      {attach_id => $attach_id, reason => $self->{bucket}->errstr});
  }
  return $response->{value};
}

sub remove {
  my ($self, $attach_id) = @_;
  $self->{bucket}->delete_key($attach_id)
    or warn "Failed to remove attachment $attach_id from S3: "
    . $self->{bucket}->errstr . "\n";
}

sub exists {
  my ($self, $attach_id) = @_;
  return !!$self->{bucket}->head_key($attach_id);
}

1;