Only allow creation of log if you're not already working

dmastylo · Apr 12, 2014 · 4680e78 · 4680e78
1 parent e09d090
commit 4680e78
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/app/controllers/time_spents_controller.rb b/app/controllers/time_spents_controller.rb
@@ -5,6 +5,11 @@ class TimeSpentsController < ApplicationController
   before_filter :user_owns_time_spent, only: [:edit, :update, :destroy]
 
   def create
+    if current_user.currently_working?
+      flash[:error] = "You are already working!"
+      redirect_to root_path
+    end
+
     @time_spent = current_user.time_spents.create(time_spent_params)
     @time_spent.set_amount_paid
 

diff --git a/app/models/user.rb b/app/models/user.rb
@@ -55,6 +55,10 @@ def projects_sorted_by_recent_work
 
   # Time Spents
   # ========================================================
+  def currently_working?
+    !time_spents.where("finished_at IS NULL").blank?
+  end
+
   def working_time_spent(existing_task)
     existing_task ? time_spents.last : time_spents.build
   end