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 - ";
  35       print DEST "primary key"
  36         if $column eq $primary_key;
  37       print DEST "\n\n";
  38     }
  39     next;
  40
  41   } elsif ( $line=~ /^(\s*)\$self->ut_numbern\('primary_key'\)\s*/ ) {
  42
  43     print DEST "$1\$self->ut_numbern('$primary_key')\n"
  44       if $primary_key;
  45     next;
  46
  47   } elsif (
  48     $line =~ /^(\s*)\|\|\s+\$self->ut_number\('validate_other_fields'\)\s*/
  49   ) {
  50
  51     foreach my $column ( grep { $_ ne $primary_key } $dbdef_table->columns ) {
  52       my $ut = $ut{$dbdef_table->column($column)->type};
  53       $ut .= 'n' if $dbdef_table->column($column)->null;
  54       print DEST "$1|| \$self->ut_$ut('$column')\n";
  55     }
  56     next;
  57
  58   }
  59
  60   print DEST $line;
  61 }
  62
  63 close SRC;
  64 close DEST;
  65
  66 ###
  67 # add FS/t/table.t
  68 ###
  69
  70 open(TEST,">FS/t/$table.t") or die $!;
  71 print TEST <<ENDTEST;
  72 BEGIN { \$| = 1; print "1..1\\n" }
  73 END {print "not ok 1\\n" unless \$loaded;}
  74 use FS::$table;
  75 \$loaded=1;
  76 print "ok 1\\n";
  77 ENDTEST
  78 close TEST;
  79
  80 ###
  81 # add them to MANIFEST
  82 ###
  83
  84 system('cvs edit FS/MANIFEST');
  85
  86 open(MANIFEST,">>FS/MANIFEST") or die $!;
  87 print MANIFEST "FS/$table.pm\n",
  88                "t/$table.t\n";
  89 close MANIFEST;
  90