summaryrefslogtreecommitdiffstats
path: root/Bugzilla/Attachment/FileSystem.pm
blob: f10a994fd276b21d3095c2aded0441f687b50078 (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
63
# 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::FileSystem;

use 5.10.1;
use strict;
use warnings;

use Bugzilla::Constants qw(bz_locations);

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

sub store {
  my ($self, $attach_id, $data) = @_;
  my $path = _local_path($attach_id);
  mkdir($path, 0770) unless -d $path;
  open(my $fh, '>', _local_file($attach_id));
  binmode($fh);
  print $fh $data;
  close($fh);
}

sub retrieve {
  my ($self, $attach_id) = @_;
  if (open(my $fh, '<', _local_file($attach_id))) {
    local $/;
    binmode($fh);
    my $data = <$fh>;
    close($fh);
    return $data;
  }
  return undef;
}

sub remove {
  my ($self, $attach_id) = @_;
  unlink(_local_file($attach_id));
}

sub exists {
  my ($self, $attach_id) = @_;
  return -e _local_file($attach_id);
}

sub _local_path {
  my ($attach_id) = @_;
  my $hash = sprintf('group.%03d', $attach_id % 1000);
  return bz_locations()->{attachdir} . '/' . $hash;
}

sub _local_file {
  my ($attach_id) = @_;
  return _local_path($attach_id) . '/attachment.' . $attach_id;
}

1;