From bf16077b10d37a317a931c810686db43d8b859eb Mon Sep 17 00:00:00 2001
From: Tom Bloor <t.bloor@shadowcat.co.uk>
Date: Thu, 31 Aug 2017 16:46:25 +0100
Subject: [PATCH] Fix register for new entity work

---
 lib/Pear/LocalLoop/Controller/Api/Register.pm | 50 ++++++++++---------
 1 file changed, 27 insertions(+), 23 deletions(-)

diff --git a/lib/Pear/LocalLoop/Controller/Api/Register.pm b/lib/Pear/LocalLoop/Controller/Api/Register.pm
index dfff302..8ba4f11 100644
--- a/lib/Pear/LocalLoop/Controller/Api/Register.pm
+++ b/lib/Pear/LocalLoop/Controller/Api/Register.pm
@@ -48,7 +48,7 @@ has error_messages => sub {
   };
 };
 
-sub post_register{
+sub post_register {
   my $c = shift;
 
   my $validation = $c->validation;
@@ -87,17 +87,19 @@ sub post_register{
         name => $validation->param('token'),
         used => 0,
       })->update({ used => 1 });
-      # Create customer as a seperate step, so we dont leak data
-      my $customer = $c->schema->resultset('Customer')->create({
-        full_name     => $validation->param('full_name'),
-        display_name  => $validation->param('display_name'),
-        year_of_birth => $validation->param('year_of_birth'),
-        postcode      => $validation->param('postcode'),
-      });
-      $c->schema->resultset('User')->create({
-        customer => $customer,
-        email    => $validation->param('email'),
-        password => $validation->param('password'),
+
+      $c->schema->resultset('Entity')->create({
+        customer => {
+          full_name     => $validation->param('full_name'),
+          display_name  => $validation->param('display_name'),
+          year_of_birth => $validation->param('year_of_birth'),
+          postcode      => $validation->param('postcode'),
+        },
+        user => {
+          email    => $validation->param('email'),
+          password => $validation->param('password'),
+        },
+        type => 'customer',
       });
     });
 
@@ -109,17 +111,19 @@ sub post_register{
         name => $validation->param('token'),
         used => 0,
       })->update({ used => 1 });
-      my $organisation = $c->schema->resultset('Organisation')->create({
-        name        => $validation->param('name'),
-        street_name => $validation->param('street_name'),
-        town        => $validation->param('town'),
-        sector      => $validation->param('sector'),
-        postcode    => $validation->param('postcode'),
-      });
-      $c->schema->resultset('User')->create({
-        organisation => $organisation,
-        email        => $validation->param('email'),
-        password     => $validation->param('password'),
+      $c->schema->resultset('Entity')->create({
+        organisation => {
+          name        => $validation->param('name'),
+          street_name => $validation->param('street_name'),
+          town        => $validation->param('town'),
+          sector      => $validation->param('sector'),
+          postcode    => $validation->param('postcode'),
+        },
+        user => {
+          email        => $validation->param('email'),
+          password     => $validation->param('password'),
+        },
+        type => 'organisation',
       });
     });
   }