From bbe3d277675494c8e11be2d480b642c609287211 Mon Sep 17 00:00:00 2001
From: Andrew Gregory <andrew.gregory.8@gmail.com>
Date: Sat, 27 Jun 2015 11:32:01 -0400
Subject: [PATCH] Schema/Field: add foreign_key_reference builder

foreign_key_reference was only being set from the is_foreign_key
builder.  If a user manually set is_foreign_key, as DBIx::Class does,
foreign_key_reference would never be set, breaking the HTML producer.
---
 lib/SQL/Translator/Schema/Field.pm | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/lib/SQL/Translator/Schema/Field.pm b/lib/SQL/Translator/Schema/Field.pm
index a88263233..817bad9ae 100644
--- a/lib/SQL/Translator/Schema/Field.pm
+++ b/lib/SQL/Translator/Schema/Field.pm
@@ -179,8 +179,10 @@ Get or set the field's foreign key reference;
 has foreign_key_reference => (
     is => 'rw',
     predicate => '_has_foreign_key_reference',
-    isa => schema_obj('Constraint'),
+    isa => sub { not defined $_[0] or schema_obj('Constraint')->(@_) },
+    builder => 1,
     weak_ref => 1,
+    lazy => 1,
 );
 
 around foreign_key_reference => sub {
@@ -197,6 +199,22 @@ around foreign_key_reference => sub {
     $self->$orig;
 };
 
+sub _build_foreign_key_reference {
+    my ( $self ) = @_;
+
+    if ( my $table = $self->table ) {
+        for my $c ( $table->get_constraints ) {
+            if ( $c->type eq FOREIGN_KEY ) {
+                my %fields = map { $_, 1 } $c->fields;
+                if ( $fields{ $self->name } ) {
+                    return $c;
+                }
+            }
+        }
+    }
+    return undef;
+}
+
 =head2 is_auto_increment
 
 Get or set the field's C<is_auto_increment> attribute.