fixing typo and moving grab file into else clause
[migration-tools.git] / mig-bin / mig-skip-clean
1 #!/usr/bin/perl -w
2 ###############################################################################
3 =pod
4
5 =head1 NAME
6
7 mig-skip-clean 
8
9 Allows you to either use an existing file named <file>.utf8.clean or a
10 named [cleaned file] as if it were the one created by mig-clean
11
12 Note that the clean file, however specified, should contain headers. The
13 remaining tools in the chain will expect this.
14
15 =head1 SYNOPSIS
16
17 B<mig-skip-clean> <file> [cleaned file]
18
19 =cut
20
21 ###############################################################################
22
23 use strict;
24 use Switch;
25 use Env qw(
26     HOME PGHOST PGPORT PGUSER PGDATABASE MIGSCHEMA
27     MIGBASEWORKDIR MIGBASEGITDIR MIGGITDIR MIGWORKDIR
28 );
29 use Pod::Usage;
30 use DBI;
31 use Cwd 'abs_path';
32 use FindBin;
33 my $mig_bin = "$FindBin::Bin/";
34 use lib "$FindBin::Bin/";
35 use Mig;
36
37 pod2usage(-verbose => 2) if ! ($ARGV[0]||$ARGV[1]) || $ARGV[0] eq '--help';
38
39 Mig::die_if_no_env_migschema();
40 Mig::die_if_mig_tracking_table_does_not_exist();
41
42 my $file = abs_path($ARGV[0]);
43 my $clean_file;
44 if ($ARGV[1]) {
45     $clean_file = abs_path($ARGV[1]);
46 } else {
47     $clean_file = $file;
48 }
49 if ($clean_file && ! $clean_file =~ /^$MIGBASEWORKDIR/) {
50     die "File falls outside of MIGWORKDIR ($MIGWORKDIR): $clean_file\n";
51 }
52
53 if ($file =~ /^$MIGBASEWORKDIR/) {
54     skip_clean($file,$clean_file);
55 } else {
56     die "File falls outside of MIGWORKDIR ($MIGWORKDIR): $file\n";
57 }
58
59 exit 0;
60
61 ###############################################################################
62
63 sub skip_clean {
64     my $file = shift;
65     my $clean_file = shift;
66
67     my $tracked_file_id = Mig::check_for_tracked_file($file);
68     if ($tracked_file_id) {
69         my $data = Mig::status_this_file($file);
70
71         if (! $data->{'utf8_filename'}) {
72             die "mig-iconv or mig-skip-iconv needed for UTF8 version of file: $file\n";
73         }
74
75         my $utf8_file = $data->{'utf8_filename'};
76         if (! -e $utf8_file) {
77             die "missing file: $utf8_file\n";
78         }
79
80         print "skipping cleaning of tracked file: $file\n";
81
82         my $dbh = Mig::db_connect();
83         if (! $clean_file) {
84             $clean_file = $utf8_file . '.clean';
85         }
86         if (! -e $clean_file) {
87             die "clean file does not exist: $clean_file\n";
88         }
89
90         my $rv = $dbh->do("
91             UPDATE $MIGSCHEMA.tracked_file
92             SET clean_filename = " . $dbh->quote($clean_file) . "
93             WHERE base_filename = " . $dbh->quote($file) . "
94             ;
95         ") || die "Error inserting into table $MIGSCHEMA.tracked_file: $!\n";
96         Mig::db_disconnect($dbh);
97     } else {
98         die "File not currently tracked: $file\n";
99     }
100 }