bin/generate-table-module

   1 #!/usr/bin/perl
   2
   3 use FS::Schema qw( dbdef_dist );
   4
   5 my $table = shift;
   6
   7 ###
   8 # add a new FS/FS/table.pm
   9 ###
  10
  11 my %ut = ( #just guesses
  12   'int'     => 'number',
  13   'number'  => 'float',
  14   'varchar' => 'text',
  15   'text'    => 'text',
  16   'serial'  => 'number',
  17 );
  18
  19 my $dbdef_table = dbdef_dist->table($table)
  20   or die "define table in Schema.pm first";
  21 my $primary_key = $dbdef_table->primary_key;
  22
  23 open(SRC,"<eg/table_template.pm") or die $!;
  24 -e "FS/FS/$table.pm" and die "FS/FS/$table.pm already exists!";
  25 open(DEST,">FS/FS/$table.pm") or die $!;
  26
  27 while (my $line = <SRC>) {
  28
  29   $line =~ s/table_name/$table/g;
  30
  31   if ( $line =~ /^=item\s+field\s+-\s+description\s*$/ ) {
  32
  33     foreach my $column ( $dbdef_table->columns ) {
  34       print DEST "=item $column\n\n";
  35       if ( $column eq $primary_key ) {
  36         print DEST "primary key\n\n";
  37       } else {
  38         print DEST "$column\n\n";
  39       }
  40     }
  41     next;
  42
  43   } elsif ( $line=~ /^(\s*)\$self->ut_numbern\('primary_key'\)\s*/ ) {
  44
  45     print DEST "$1\$self->ut_numbern('$primary_key')\n"
  46       if $primary_key;
  47     next;
  48
  49   } elsif (
  50     $line =~ /^(\s*)\|\|\s+\$self->ut_number\('validate_other_fields'\)\s*/
  51   ) {
  52
  53     foreach my $column ( grep { $_ ne $primary_key } $dbdef_table->columns ) {
  54       my $ut = $ut{$dbdef_table->column($column)->type};
  55       $ut .= 'n' if $dbdef_table->column($column)->null;
  56       print DEST "$1|| \$self->ut_$ut('$column')\n";
  57     }
  58     next;
  59
  60   }
  61
  62   print DEST $line;
  63 }
  64
  65 close SRC;
  66 close DEST;
  67
  68 ###
  69 # add FS/t/table.t
  70 ###
  71
  72 open(TEST,">FS/t/$table.t") or die $!;
  73 print TEST <<ENDTEST;
  74 BEGIN { \$| = 1; print "1..1\\n" }
  75 END {print "not ok 1\\n" unless \$loaded;}
  76 use FS::$table;
  77 \$loaded=1;
  78 print "ok 1\\n";
  79 ENDTEST
  80 close TEST;
  81
  82 ###
  83 # add them to MANIFEST
  84 ###
  85
  86 system('cvs edit FS/MANIFEST');
  87
  88 open(MANIFEST,">>FS/MANIFEST") or die $!;
  89 print MANIFEST "FS/$table.pm\n",
  90                "t/$table.t\n";
  91 close MANIFEST;
  92