Start of GS5.

1 files changed, 25 insertions, 0 deletions

diff --git a/app/models/tenant_membership.rb b/app/models/tenant_membership.rb
new file mode 100644
index 0000000..122f702
--- /dev/null
+++ b/app/models/tenant_membership.rb
@@ -0,0 +1,25 @@
+class TenantMembership < ActiveRecord::Base
+  belongs_to :tenant
+  belongs_to :user
+    
+  validates_presence_of :tenant
+  validates_presence_of :user
+  
+  after_create :set_current_tenant_if_necessary
+
+  # State Machine stuff
+  default_scope where(:state => 'active')
+  state_machine :initial => :active do
+  end
+
+  private
+  # The first TenantMembership becomes the current_tenant by default.
+  #
+  def set_current_tenant_if_necessary
+    if !self.user.current_tenant
+      self.user.current_tenant = self.tenant
+      self.user.save
+    end
+  end
+
+end