From 29a9af3cf0001ee0df55c510adb41d6032f97e0f Mon Sep 17 00:00:00 2001
From: Ilion Beyst <ilion.beyst@gmail.com>
Date: Tue, 8 Sep 2015 12:11:48 +0200
Subject: [PATCH] Added associations

---
 app/models/client.rb      | 3 +++
 app/models/transaction.rb | 8 ++++++--
 app/models/user.rb        | 4 ++++
 3 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/app/models/client.rb b/app/models/client.rb
index 84199ac..3a825c3 100644
--- a/app/models/client.rb
+++ b/app/models/client.rb
@@ -10,4 +10,7 @@
 #
 
 class Client < ActiveRecord::Base
+  def transactions
+    Transaction.where(origin: name)
+  end
 end
diff --git a/app/models/transaction.rb b/app/models/transaction.rb
index d3c868d..1e48af6 100644
--- a/app/models/transaction.rb
+++ b/app/models/transaction.rb
@@ -13,6 +13,10 @@
 #
 
 class Transaction < ActiveRecord::Base
-  belongs_to :debtor
-  belongs_to :creditor
+  belongs_to :debtor, class_name: 'User'
+  belongs_to :creditor, class_name: 'User'
+
+  def client
+    Client.find_by name: origin
+  end
 end
diff --git a/app/models/user.rb b/app/models/user.rb
index 1a49da2..2b5a25d 100644
--- a/app/models/user.rb
+++ b/app/models/user.rb
@@ -12,4 +12,8 @@
 
 class User < ActiveRecord::Base
   devise :timeoutable, :omniauthable, :omniauth_providers => [:zeuswpi]
+  has_many :incoming_transactions,
+    class_name: 'Transaction', foreign_key: 'creditor_id'
+  has_many :outgoing_transactions,
+    class_name: 'Transaction', foreign_key: 'debtor_id'
 end