adding load report file for resource mate
[migration-tools.git] / mig-bin / mig-link
1 #!/usr/bin/perl -w
2 ###############################################################################
3 =pod
4
5 =head1 NAME
6
7 mig-link 
8
9 Associate the specified file with a parent table within the migration schema.
10
11 =head1 SYNOPSIS
12
13 B<mig-link> <file> <parent table>
14
15 =cut
16
17 ###############################################################################
18
19 use strict;
20 use Switch;
21 use Env qw(
22     HOME PGHOST PGPORT PGUSER PGDATABASE MIGSCHEMA
23     MIGBASEWORKDIR MIGBASEGITDIR MIGGITDIR MIGWORKDIR
24 );
25 use Pod::Usage;
26 use DBI;
27 use Cwd 'abs_path';
28 use FindBin;
29 my $mig_bin = "$FindBin::Bin/";
30 use lib "$FindBin::Bin/";
31 use Mig;
32
33 pod2usage(-verbose => 2) if ! $ARGV[0] || $ARGV[0] eq '--help';
34
35 Mig::die_if_no_env_migschema();
36 Mig::die_if_mig_tracking_table_does_not_exist();
37
38 my $file = abs_path($ARGV[0]);
39 if ($file =~ /^$MIGBASEWORKDIR/) {
40     link_table(@ARGV);
41 } else {
42     print "File falls outside of MIGWORKDIR ($MIGWORKDIR): $file\n";
43 }
44
45 exit 0;
46
47 ###############################################################################
48
49 sub link_table {
50     my $file = abs_path(shift);
51     my $table = shift;
52
53     if (! Mig::check_db_migschema_for_specific_table($table)) {
54         die "table not found in MIGSCHEMA ($MIGSCHEMA): $table\n";
55     }
56
57     my $tracked_file_id = Mig::check_for_tracked_file($file);
58     if ($tracked_file_id) {
59         my $data = Mig::status_this_file($file);
60
61         print "linking file to parent table: $file -> $table\n";
62
63         my $dbh = Mig::db_connect();
64         my $sth = $dbh->prepare("
65             SELECT base_filename
66             FROM $MIGSCHEMA.tracked_file
67             WHERE parent_table = " . $dbh->quote($table) . "
68             AND base_filename <> " . $dbh->quote($file) . ";"
69         );
70         my $rv = $sth->execute()
71             || die "Error checking $MIGSCHEMA.tracked_file: $!";
72         my @cols = $sth->fetchrow_array;
73         $sth->finish;
74         if ($cols[0]) { # found
75             die "table ($table) already linked to a different file: $cols[0]\n";
76         }
77         $rv = $dbh->do("
78             UPDATE $MIGSCHEMA.tracked_file
79             SET parent_table = " . $dbh->quote($table) . "
80             WHERE base_filename = " . $dbh->quote($file) . "
81             ;
82         ") || die "Error updating table $MIGSCHEMA.tracked_file: $!\n";
83         Mig::db_disconnect($dbh);
84     } else {
85         print "File not currently tracked: $file\n";
86     }
87 }