summaryrefslogtreecommitdiffstats
path: root/scripts/fix-attachment-sizes.pl
blob: a50e241ceb9ac91e8d2176aeece1871cf1040956 (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
#!/usr/bin/perl -w

# 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.

use strict;
use feature 'say';

use FindBin qw( $RealBin );
use lib "$RealBin/..";
use lib "$RealBin/../lib";

use Bugzilla;
use Bugzilla::Constants;

Bugzilla->usage_mode(USAGE_MODE_CMDLINE);

my $dbh = Bugzilla->dbh;

$dbh->bz_start_transaction();
my $attachment_sizes = $dbh->selectall_arrayref(q{
        SELECT attachments.attach_id, length(thedata)
        FROM attach_data
        INNER JOIN attachments ON attachments.attach_id = attach_data.id
        WHERE attachments.attach_size != 0
          AND attachments.mimetype = 'image/png'
          AND length(thedata) != attachments.attach_size });
say "Found ", scalar @$attachment_sizes, " attachments to fix";

foreach my $attachment_size (@$attachment_sizes) {
    say "Setting size for $attachment_size->[0] to $attachment_size->[1]";

    $dbh->do("UPDATE attachments SET attach_size = ? WHERE attach_id = ?", undef,
             $attachment_size->[1],
             $attachment_size->[0]);
}

$dbh->bz_commit_transaction();