[dbsrgits/SQL-Translator.git] / lib / SQL / Translator / Schema / Graph.pm

package SQL::Translator::Schema::Graph;

use strict;
use warnings;

use base 'Class::Base';

use Data::Dumper;
local $Data::Dumper::Maxdepth = 3;

use SQL::Translator::Schema::Graph::Node;
use SQL::Translator::Schema::Graph::Edge;
use SQL::Translator::Schema::Graph::Port;
use SQL::Translator::Schema::Graph::CompoundEdge;
use SQL::Translator::Schema::Graph::HyperEdge;

use constant Node => 'SQL::Translator::Schema::Graph::Node';
use constant Edge => 'SQL::Translator::Schema::Graph::Edge';
use constant Port => 'SQL::Translator::Schema::Graph::Port';
use constant CompoundEdge => 'SQL::Translator::Schema::Graph::CompoundEdge';
use constant HyperEdge => 'SQL::Translator::Schema::Graph::HyperEdge';

use Class::MakeMethods::Template::Hash (
  'new --and_then_init' => 'new',
  object => [
   'translator' => {class => 'SQL::Translator'},
  ],
  'hash' => [ qw( node ) ],
  'number --counter' => [ qw( order ) ],
);

use vars qw/$DEBUG/;
$DEBUG = 0 unless defined $DEBUG;

sub init {
  my $self = shift;

  #
  # build package objects
  #
  foreach my $table ($self->translator->schema->get_tables){
   die __PACKAGE__." table ".$table->name." doesn't have a primary key!" unless $table->primary_key;
   die __PACKAGE__." table ".$table->name." can't have a composite primary key!" if ($table->primary_key->fields)[1];

   my $node = Node->new();

   $self->node_push($table->name => $node);

   if ($table->is_trivial_link) { $node->is_trivial_link(1); }
   else { $node->is_trivial_link(0); }

   $node->order($self->order_incr());
   $node->name( $self->translator->format_package_name($table->name) );
   $node->table( $table );
   $node->primary_key( ($table->primary_key->fields)[0] );

   # Primary key may have a differenct accessor method name
   $node->primary_key_accessor(
                        defined($self->translator->format_pk_name)
                        ? $self->translator->format_pk_name->( $node->name, $node->primary_key )
                        : undef
                        );
  }

  foreach my $node ($self->node_values){
   foreach my $field ($node->table->get_fields){
     if (!$field->is_foreign_key && !$field->is_primary_key) { $node->data_fields->{$field->name} = 1; }
     elsif($field->is_foreign_key) {
     my $that = $self->node($field->foreign_key_reference->reference_table);

     #this means we have an incomplete schema
     next unless $that;

     my $edge = Edge->new(
                     type => 'import',
                     thisnode => $node,
                     thisfield => $field,
                     thatnode => $that,
                     #can you believe this sh*t just to get a field obj?
                     thatfield => $self->translator->schema->get_table($field->foreign_key_reference->reference_table)->get_field(($field->foreign_key_reference->reference_fields)[0])
                    );

     $node->edgecount($that->name, $node->edgecount($that->name)+1);

     $node->has($that->name, $node->has($that->name)+1);
     $that->many($node->name, $that->many($node->name)+1);

     $that->edgecount($node->name, $that->edgecount($node->name)+1);

          #warn "\t" . $node->name . "\t" . $node->edgecount($that->name);
     $node->push_edges( $edge );
     $that->push_edges( $edge->flip );
      }
   }

    #warn Dumper($node->edgecount());
    #warn "*****";
  }

  #
  # type MM relationships
  #
  #foreach linknode
  foreach my $lnode (sort $self->node_values){
   next if $lnode->table->is_data;
   foreach my $inode1 (sort $self->node_values){
     #linknode can't link to itself
     next if $inode1 eq $lnode;

     my @inode1_imports = grep { $_->type eq 'import' and $_->thatnode eq $inode1 } $lnode->edges;
     next unless @inode1_imports;

     foreach my $inode2 (sort $self->node_values){
      #linknode can't link to itself
      next if $inode2 eq $lnode;

      #identify tables that import keys to linknode
      my %i = map {$_->thatnode->name => 1} grep { $_->type eq 'import'} $lnode->edges;

      if(scalar(keys %i) == 1) {
      } else {
        last if $inode1 eq $inode2;
      }

      my @inode2_imports =  grep { $_->type eq 'import' and $_->thatnode eq $inode2 } $lnode->edges;
      next unless @inode2_imports;

      my $cedge = CompoundEdge->new();
      $cedge->via($lnode);

      #warn join ' ', map {$_->thisfield->name} map {$_->flip} $lnode->edges;
      #warn join ' ', map {$_->thisfield->name} $lnode->edges;
      #warn join ' ', map {$_->thisfield->name} map {$_->flip} grep {$_->type eq 'import'} $lnode->edges;
      #warn join ' ', map {$_->thatfield->name} map {$_->flip} grep {$_->type eq 'import'} $lnode->edges;
      $cedge->push_edges(
                     map {$_->flip}
                     grep {$_->type eq 'import'
                           and
                         ($_->thatnode eq $inode1 or $_->thatnode eq $inode2)
                         } $lnode->edges
                    );

      if(scalar(@inode1_imports) == 1 and scalar(@inode2_imports) == 1){
        $cedge->type('one2one');

        $inode1->via($inode2->name,$inode1->via($inode2->name)+1);
        $inode2->via($inode1->name,$inode2->via($inode1->name)+1);
      }
      elsif(scalar(@inode1_imports)  > 1 and scalar(@inode2_imports) == 1){
        $cedge->type('many2one');

        $inode1->via($inode2->name,$inode1->via($inode2->name)+1);
        $inode2->via($inode1->name,$inode2->via($inode1->name)+1);
      }
      elsif(scalar(@inode1_imports) == 1 and scalar(@inode2_imports)  > 1){
        #handled above
      }
      elsif(scalar(@inode1_imports)  > 1 and scalar(@inode2_imports)  > 1){
        $cedge->type('many2many');

        $inode1->via($inode2->name,$inode1->via($inode2->name)+1);
        $inode2->via($inode1->name,$inode2->via($inode1->name)+1);
      }
#warn Dumper($cedge);

      $inode1->push_compoundedges($cedge);
      $inode2->push_compoundedges($cedge) unless $inode1 eq $inode2;
#        if($inode1->name ne $inode2->name){
#          my $flipped_cedge = $cedge;
#          foreach my $flipped_cedge_edge ($flipped_cedge->edges){
#            warn Dumper $flipped_cedge_edge;
#            warn "\t". Dumper $flipped_cedge_edge->flip;
#          }
#        }
     }
   }
  }

  my $graph = $self; #hack

  #
  # create methods
  #
  # this code needs to move to Graph.pm
  foreach my $node_from ($graph->node_values) {

    next unless $node_from->table->is_data or !$node_from->table->is_trivial_link;

    foreach my $cedge ( $node_from->compoundedges ) {

      my $hyperedge = SQL::Translator::Schema::Graph::HyperEdge->new();

      my $node_to;
      foreach my $edge ($cedge->edges) {
        if ($edge->thisnode->name eq $node_from->name) {
          $hyperedge->vianode($edge->thatnode);

          if ($edge->thatnode->name ne $cedge->via->name) {
            $node_to ||= $graph->node($edge->thatnode->table->name);
          }

          $hyperedge->push_thisnode($edge->thisnode);
          $hyperedge->push_thisfield($edge->thisfield);
          $hyperedge->push_thisviafield($edge->thatfield);

        } else {
          if ($edge->thisnode->name ne $cedge->via->name) {
            $node_to ||= $graph->node($edge->thisnode->table->name);
          }
          $hyperedge->push_thatnode($edge->thisnode);
          $hyperedge->push_thatfield($edge->thisfield);
          $hyperedge->push_thatviafield($edge->thatfield);
        }
        $self->debug($edge->thisfield->name);
        $self->debug($edge->thatfield->name);
      }

      if ($hyperedge->count_thisnode == 1 and $hyperedge->count_thatnode == 1) {
        $hyperedge->type('one2one');
      } elsif ($hyperedge->count_thisnode  > 1 and $hyperedge->count_thatnode == 1) {
        $hyperedge->type('many2one');
      } elsif ($hyperedge->count_thisnode == 1 and $hyperedge->count_thatnode  > 1) {
        $hyperedge->type('one2many');
      } elsif ($hyperedge->count_thisnode  > 1 and $hyperedge->count_thatnode  > 1) {
        $hyperedge->type('many2many');
      }

      $self->debug($_) foreach sort keys %::SQL::Translator::Schema::Graph::HyperEdge::;

      #node_to won't always be defined b/c of multiple edges to a single other node
      if (defined($node_to)) {
        $self->debug($node_from->name);
        $self->debug($node_to->name);

        if (scalar($hyperedge->thisnode) > 1) {
          $self->debug($hyperedge->type ." via ". $hyperedge->vianode->name);
          my $i = 0;
          foreach my $thisnode ( $hyperedge->thisnode ) {
            $self->debug($thisnode->name .' '.
                        $hyperedge->thisfield_index(0)->name .' -> '.
                        $hyperedge->thisviafield_index($i)->name .' '.
                        $hyperedge->vianode->name .' '.
                        $hyperedge->thatviafield_index(0)->name .' <- '.
                        $hyperedge->thatfield_index(0)->name .' '.
                        $hyperedge->thatnode_index(0)->name ."\n"
                       );
            $i++;
          }
        }
        #warn Dumper($hyperedge) if $hyperedge->type eq 'many2many';
        $node_from->push_hyperedges($hyperedge);
      }
    }
  }

}

1;
Commit	Line	Data
0caaf4c3	1	package SQL::Translator::Schema::Graph;
	2
	3	use strict;
f27f9229	4	use warnings;
0caaf4c3	5
4b5c50b9	6	use base 'Class::Base';
4b5c50b9	7
ba506e52	8	use Data::Dumper;
11ad2df9	9	local $Data::Dumper::Maxdepth = 3;
11ad2df9	10
0caaf4c3	11	use SQL::Translator::Schema::Graph::Node;
	12	use SQL::Translator::Schema::Graph::Edge;
	13	use SQL::Translator::Schema::Graph::Port;
	14	use SQL::Translator::Schema::Graph::CompoundEdge;
	15	use SQL::Translator::Schema::Graph::HyperEdge;
	16
11ad2df9	17	use constant Node => 'SQL::Translator::Schema::Graph::Node';
	18	use constant Edge => 'SQL::Translator::Schema::Graph::Edge';
	19	use constant Port => 'SQL::Translator::Schema::Graph::Port';
	20	use constant CompoundEdge => 'SQL::Translator::Schema::Graph::CompoundEdge';
	21	use constant HyperEdge => 'SQL::Translator::Schema::Graph::HyperEdge';
0caaf4c3	22
0caaf4c3	23	use Class::MakeMethods::Template::Hash (
11ad2df9	24	'new --and_then_init' => 'new',
	25	object => [
	26	'translator' => {class => 'SQL::Translator'},
	27	],
	28	'hash' => [ qw( node ) ],
	29	'number --counter' => [ qw( order ) ],
0caaf4c3	30	);
0caaf4c3	31
4b5c50b9	32	use vars qw/$DEBUG/;
	33	$DEBUG = 0 unless defined $DEBUG;
	34
0caaf4c3	35	sub init {
11ad2df9	36	my $self = shift;
	37
	38	#
	39	# build package objects
	40	#
	41	foreach my $table ($self->translator->schema->get_tables){
ea93df61	42	die __PACKAGE__." table ".$table->name." doesn't have a primary key!" unless $table->primary_key;
ea93df61	43	die __PACKAGE__." table ".$table->name." can't have a composite primary key!" if ($table->primary_key->fields)[1];
11ad2df9	44
ea93df61	45	my $node = Node->new();
11ad2df9	46
ea93df61	47	$self->node_push($table->name => $node);
11ad2df9	48
ea93df61	49	if ($table->is_trivial_link) { $node->is_trivial_link(1); }
ea93df61	50	else { $node->is_trivial_link(0); }
11ad2df9	51
ea93df61	52	$node->order($self->order_incr());
	53	$node->name( $self->translator->format_package_name($table->name) );
	54	$node->table( $table );
	55	$node->primary_key( ($table->primary_key->fields)[0] );
11ad2df9	56
ea93df61	57	# Primary key may have a differenct accessor method name
	58	$node->primary_key_accessor(
	59	defined($self->translator->format_pk_name)
	60	? $self->translator->format_pk_name->( $node->name, $node->primary_key )
	61	: undef
	62	);
11ad2df9	63	}
	64
	65	foreach my $node ($self->node_values){
ea93df61	66	foreach my $field ($node->table->get_fields){
	67	if (!$field->is_foreign_key && !$field->is_primary_key) { $node->data_fields->{$field->name} = 1; }
	68	elsif($field->is_foreign_key) {
	69	my $that = $self->node($field->foreign_key_reference->reference_table);
11ad2df9	70
ea93df61	71	#this means we have an incomplete schema
ea93df61	72	next unless $that;
11ad2df9	73
ea93df61	74	my $edge = Edge->new(
	75	type => 'import',
	76	thisnode => $node,
	77	thisfield => $field,
	78	thatnode => $that,
	79	#can you believe this sh*t just to get a field obj?
	80	thatfield => $self->translator->schema->get_table($field->foreign_key_reference->reference_table)->get_field(($field->foreign_key_reference->reference_fields)[0])
	81	);
11ad2df9	82
ea93df61	83	$node->edgecount($that->name, $node->edgecount($that->name)+1);
11ad2df9	84
ea93df61	85	$node->has($that->name, $node->has($that->name)+1);
ea93df61	86	$that->many($node->name, $that->many($node->name)+1);
11ad2df9	87
ea93df61	88	$that->edgecount($node->name, $that->edgecount($node->name)+1);
11ad2df9	89
11ad2df9	90	#warn "\t" . $node->name . "\t" . $node->edgecount($that->name);
ea93df61	91	$node->push_edges( $edge );
ea93df61	92	$that->push_edges( $edge->flip );
11ad2df9	93	}
ea93df61	94	}
11ad2df9	95
	96	#warn Dumper($node->edgecount());
	97	#warn "*****";
	98	}
	99
	100	#
	101	# type MM relationships
	102	#
	103	#foreach linknode
	104	foreach my $lnode (sort $self->node_values){
ea93df61	105	next if $lnode->table->is_data;
	106	foreach my $inode1 (sort $self->node_values){
	107	#linknode can't link to itself
	108	next if $inode1 eq $lnode;
	109
	110	my @inode1_imports = grep { $_->type eq 'import' and $_->thatnode eq $inode1 } $lnode->edges;
	111	next unless @inode1_imports;
	112
	113	foreach my $inode2 (sort $self->node_values){
	114	#linknode can't link to itself
	115	next if $inode2 eq $lnode;
	116
	117	#identify tables that import keys to linknode
	118	my %i = map {$_->thatnode->name => 1} grep { $_->type eq 'import'} $lnode->edges;
	119
	120	if(scalar(keys %i) == 1) {
	121	} else {
	122	last if $inode1 eq $inode2;
	123	}
	124
	125	my @inode2_imports = grep { $_->type eq 'import' and $_->thatnode eq $inode2 } $lnode->edges;
	126	next unless @inode2_imports;
	127
	128	my $cedge = CompoundEdge->new();
	129	$cedge->via($lnode);
	130
	131	#warn join ' ', map {$_->thisfield->name} map {$_->flip} $lnode->edges;
	132	#warn join ' ', map {$_->thisfield->name} $lnode->edges;
	133	#warn join ' ', map {$_->thisfield->name} map {$_->flip} grep {$_->type eq 'import'} $lnode->edges;
	134	#warn join ' ', map {$_->thatfield->name} map {$_->flip} grep {$_->type eq 'import'} $lnode->edges;
	135	$cedge->push_edges(
	136	map {$_->flip}
	137	grep {$_->type eq 'import'
	138	and
	139	($_->thatnode eq $inode1 or $_->thatnode eq $inode2)
	140	} $lnode->edges
	141	);
	142
	143	if(scalar(@inode1_imports) == 1 and scalar(@inode2_imports) == 1){
	144	$cedge->type('one2one');
	145
	146	$inode1->via($inode2->name,$inode1->via($inode2->name)+1);
	147	$inode2->via($inode1->name,$inode2->via($inode1->name)+1);
	148	}
	149	elsif(scalar(@inode1_imports) > 1 and scalar(@inode2_imports) == 1){
	150	$cedge->type('many2one');
	151
	152	$inode1->via($inode2->name,$inode1->via($inode2->name)+1);
	153	$inode2->via($inode1->name,$inode2->via($inode1->name)+1);
	154	}
	155	elsif(scalar(@inode1_imports) == 1 and scalar(@inode2_imports) > 1){
	156	#handled above
	157	}
	158	elsif(scalar(@inode1_imports) > 1 and scalar(@inode2_imports) > 1){
	159	$cedge->type('many2many');
	160
	161	$inode1->via($inode2->name,$inode1->via($inode2->name)+1);
	162	$inode2->via($inode1->name,$inode2->via($inode1->name)+1);
	163	}
11ad2df9	164	#warn Dumper($cedge);
11ad2df9	165
ea93df61	166	$inode1->push_compoundedges($cedge);
ea93df61	167	$inode2->push_compoundedges($cedge) unless $inode1 eq $inode2;
11ad2df9	168	# if($inode1->name ne $inode2->name){
	169	# my $flipped_cedge = $cedge;
	170	# foreach my $flipped_cedge_edge ($flipped_cedge->edges){
	171	# warn Dumper $flipped_cedge_edge;
	172	# warn "\t". Dumper $flipped_cedge_edge->flip;
	173	# }
	174	# }
ea93df61	175	}
ea93df61	176	}
11ad2df9	177	}
	178
	179	my $graph = $self; #hack
	180
	181	#
	182	# create methods
	183	#
	184	# this code needs to move to Graph.pm
	185	foreach my $node_from ($graph->node_values) {
	186
	187	next unless $node_from->table->is_data or !$node_from->table->is_trivial_link;
	188
	189	foreach my $cedge ( $node_from->compoundedges ) {
	190
	191	my $hyperedge = SQL::Translator::Schema::Graph::HyperEdge->new();
	192
	193	my $node_to;
	194	foreach my $edge ($cedge->edges) {
	195	if ($edge->thisnode->name eq $node_from->name) {
	196	$hyperedge->vianode($edge->thatnode);
	197
	198	if ($edge->thatnode->name ne $cedge->via->name) {
	199	$node_to \|\|= $graph->node($edge->thatnode->table->name);
	200	}
	201
	202	$hyperedge->push_thisnode($edge->thisnode);
	203	$hyperedge->push_thisfield($edge->thisfield);
	204	$hyperedge->push_thisviafield($edge->thatfield);
	205
	206	} else {
	207	if ($edge->thisnode->name ne $cedge->via->name) {
	208	$node_to \|\|= $graph->node($edge->thisnode->table->name);
	209	}
	210	$hyperedge->push_thatnode($edge->thisnode);
	211	$hyperedge->push_thatfield($edge->thisfield);
	212	$hyperedge->push_thatviafield($edge->thatfield);
b046b0b9	213	}
11ad2df9	214	$self->debug($edge->thisfield->name);
	215	$self->debug($edge->thatfield->name);
	216	}
	217
	218	if ($hyperedge->count_thisnode == 1 and $hyperedge->count_thatnode == 1) {
	219	$hyperedge->type('one2one');
	220	} elsif ($hyperedge->count_thisnode > 1 and $hyperedge->count_thatnode == 1) {
	221	$hyperedge->type('many2one');
	222	} elsif ($hyperedge->count_thisnode == 1 and $hyperedge->count_thatnode > 1) {
	223	$hyperedge->type('one2many');
	224	} elsif ($hyperedge->count_thisnode > 1 and $hyperedge->count_thatnode > 1) {
	225	$hyperedge->type('many2many');
	226	}
	227
	228	$self->debug($_) foreach sort keys %::SQL::Translator::Schema::Graph::HyperEdge::;
	229
	230	#node_to won't always be defined b/c of multiple edges to a single other node
	231	if (defined($node_to)) {
	232	$self->debug($node_from->name);
	233	$self->debug($node_to->name);
	234
	235	if (scalar($hyperedge->thisnode) > 1) {
	236	$self->debug($hyperedge->type ." via ". $hyperedge->vianode->name);
	237	my $i = 0;
	238	foreach my $thisnode ( $hyperedge->thisnode ) {
	239	$self->debug($thisnode->name .' '.
	240	$hyperedge->thisfield_index(0)->name .' -> '.
	241	$hyperedge->thisviafield_index($i)->name .' '.
	242	$hyperedge->vianode->name .' '.
	243	$hyperedge->thatviafield_index(0)->name .' <- '.
	244	$hyperedge->thatfield_index(0)->name .' '.
	245	$hyperedge->thatnode_index(0)->name ."\n"
	246	);
	247	$i++;
	248	}
b046b0b9	249	}
11ad2df9	250	#warn Dumper($hyperedge) if $hyperedge->type eq 'many2many';
	251	$node_from->push_hyperedges($hyperedge);
	252	}
b046b0b9	253	}
11ad2df9	254	}
b046b0b9	255
0caaf4c3	256	}
	257
	258	1;