From 93de8fdf9e31da70759f230e9761f4e40e19aaf2 Mon Sep 17 00:00:00 2001
From: Finn <toyou1995@gmail.com>
Date: Mon, 15 Jan 2018 16:57:19 +0000
Subject: [PATCH] functional category loading and category in upload

---
 lib/Pear/LocalLoop/Controller/Api/Upload.pm | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/lib/Pear/LocalLoop/Controller/Api/Upload.pm b/lib/Pear/LocalLoop/Controller/Api/Upload.pm
index 96a0b03..72c9b3e 100644
--- a/lib/Pear/LocalLoop/Controller/Api/Upload.pm
+++ b/lib/Pear/LocalLoop/Controller/Api/Upload.pm
@@ -104,6 +104,7 @@ sub post_upload {
 
   #Check a proper purchase time was submitted
   $validation->optional('purchase_time')->is_full_iso_datetime;
+  $validation->optional('category');
 
   # First pass of required items
   return $c->api_validation_error if $validation->has_error;
@@ -179,6 +180,7 @@ sub post_upload {
   my $purchase_time = $c->parse_iso_datetime($validation->param('purchase_time') || '');
   $purchase_time ||= DateTime->now();
   my $file = defined $upload ? $c->store_file_from_upload( $upload ) : undef;
+  my $category = defined $validation->param('category') ? $validation->param('category') : undef;
   my $distance = $c->get_distance_from_coords( $user->entity->type_object, $organisation );
 
   my $new_transaction = $organisation->entity->create_related(
@@ -203,6 +205,13 @@ sub post_upload {
     );
   }
 
+  if ( defined $category ) {
+    my $c->schema->resultset('TransactionCategory')->create({
+      category_id => $category,
+      transaction_id => $new_transaction->id,
+    });
+  }
+
   return $c->render( json => {
     success => Mojo::JSON->true,
     message => 'Upload Successful',
@@ -213,9 +222,14 @@ sub post_category {
   my $c = shift;
   my $self = $c;
 
+  my $categories = { ids => [], names => [] };
+
   my $category_rs = $c->schema->resultset('Category');
-  $category_rs->result_class('DBIx::Class::ResultClass::HashRefInflator');
-  my $categories => $category_rs->all;
+
+  for ( $category_rs->all ) {
+    push @{ $categories->{ ids } }, $_->get_column('id');
+    push @{ $categories->{ names } }, $_->get_column('name');
+  }
 
   return $self->render( json => {
     success => Mojo::JSON->true,