RT# 71419 Updated Advanced wireless broadband report with new option to display packa...
[freeside.git] / bin / fix-sequences
1 #!/usr/bin/perl -Tw
2
3 # run dbdef-create first!
4
5 use strict;
6 use DBI;
7 use DBIx::DBSchema 0.26;
8 use DBIx::DBSchema::Table;
9 use DBIx::DBSchema::Column;
10 use DBIx::DBSchema::ColGroup::Unique;
11 use DBIx::DBSchema::ColGroup::Index;
12 use FS::UID qw(adminsuidsetup driver_name);
13 use FS::Record qw(dbdef);
14
15 my $user = shift or die &usage;
16 my $dbh = adminsuidsetup $user;
17
18 my $schema = dbdef();
19
20 #false laziness w/fs-setup
21 my @tables = scalar(@ARGV)
22                ? @ARGV
23                : grep { ! /^h_/ } $schema->tables;
24 foreach my $table ( @tables ) {
25   my $tableobj = $schema->table($table)
26     or die "unknown table $table (did you run dbdef-create?)\n";
27
28   my $primary_key = $tableobj->primary_key;
29   next unless $primary_key;
30
31   my $col = $tableobj->column($primary_key);
32
33
34   next unless uc($col->type) eq 'SERIAL'
35               || ( driver_name eq 'Pg'
36                      && defined($col->default)
37                      && $col->default =~ /^nextval\(/i
38                  )
39               || ( driver_name eq 'mysql'
40                      && defined($col->local)
41                      && $col->local =~ /AUTO_INCREMENT/i
42                  );
43
44   my $seq = "${table}_${primary_key}_seq";
45   if ( driver_name eq 'Pg'
46        && defined($col->default) 
47        && $col->default =~ /^nextval\('"(public\.)?(\w+_seq)"'::text\)$/
48      ) {
49     $seq = $2;
50   }
51
52   warn "fixing sequence for $table\n";
53
54
55   my $sql = "SELECT setval( '$seq',
56                             ( SELECT max($primary_key) FROM $table ) );";
57
58   #warn $col->default. " $seq\n$sql\n";
59   $dbh->do( $sql ) or die $dbh->errstr;
60
61 }
62
63 $dbh->commit or die $dbh->errstr;
64 $dbh->disconnect or die $dbh->errstr;
65
66 sub usage {
67   die "Usage:\n  fix-sequences user [ table table ... ] \n";
68 }
69