removing migschema defs from resource mate xml
[migration-tools.git] / mig-bin / mig-init
1 #!/usr/bin/perl -w
2 ###############################################################################
3 =pod
4
5 =head1 NAME
6
7 mig-init - This will add or recreate tracking tables for the B<mig> toolset to
8 the migration schema specified by the MIGSCHEMA environment variable, in the
9 PostgreSQL database specified by various PG environment variables.
10
11 In practice, you should invoke 'mig env use schema_name' prior to calling
12 B<init>
13
14 =head1 SYNOPSIS
15
16 B<mig-init>
17
18 B<mig-init> <help>
19
20 =cut
21
22 ###############################################################################
23
24 use strict;
25 use Switch;
26 use Env qw(
27     HOME PGHOST PGPORT PGUSER PGDATABASE MIGSCHEMA
28     MIGBASEWORKDIR MIGBASEGITDIR MIGGITDIR MIGWORKDIR
29 );
30 use Pod::Usage;
31 use DBI;
32 use FindBin;
33 my $mig_bin = "$FindBin::Bin/";
34 my $mig_sql = $mig_bin . "../mig-sql/init/";
35 use lib "$FindBin::Bin/";
36 use Mig;
37
38 pod2usage(-verbose => 2) if $ARGV[0];
39
40 Mig::die_if_no_env_migschema();
41
42 if (! Mig::check_for_db_migschema()) {
43     try_to_create_schema();
44 }
45
46 if (! Mig::check_db_migschema_for_migration_tables()) {
47     try_to_init_schema_with_migration_tools();
48 }
49 Mig::die_if_mig_tracking_table_exists();
50 Mig::die_if_mig_column_tracking_table_exists();
51 loop_through_mig_sql_templates();
52
53 exit 0;
54
55 ###############################################################################
56
57 sub try_to_create_schema {
58     if ($MIGSCHEMA =~ /[^\w_]/) {
59         die "$MIGSCHEMA is not suitable for a schema name in PostgreSQL\n";
60     }
61     my $dbh = Mig::db_connect();
62     my $rv = $dbh->do("CREATE SCHEMA $MIGSCHEMA;")
63         || die "Error creating migration schema ($MIGSCHEMA): $!\n";
64     print "Created schema $MIGSCHEMA\n";
65     Mig::db_disconnect($dbh);
66 }
67
68 sub try_to_init_schema_with_migration_tools {
69     Mig::die_if_no_migration_tools();
70     print "Calling migration_tools.init() and .build()\n";
71     my $dbh = Mig::db_connect();
72     my $rv = $dbh->do("SELECT migration_tools.init(" . $dbh->quote($MIGSCHEMA) . ");")
73         || die "Error running migration_tools.init($MIGSCHEMA): $!\n";
74     print "migration_tools.init() finished\n";
75     my $rv2 = $dbh->do("SELECT migration_tools.build(" . $dbh->quote($MIGSCHEMA) . ");")
76         || die "Error running migration_tools.build($MIGSCHEMA): $!\n";
77     print "migration_tools.build() finished\n";
78     Mig::db_disconnect($dbh);
79 }
80
81 sub loop_through_mig_sql_templates {
82     print "Looping through mig-sql/init/ templates\n";
83     opendir my $dir, $mig_sql or die "Cannot open directory: $!";
84     my @files = sort readdir $dir;
85     closedir $dir;
86     foreach my $file (@files) {
87         if ($file =~ /.sql$/) {
88             print "executing $file:\n";
89             system( $mig_bin . "mig-sql", ('-f',$mig_sql . $file) )
90         }
91     }
92 }
93