andela/workdey

class TaskManagement < ActiveRecord::Base
  belongs_to :artisan, class_name: "User"
  belongs_to :tasker, class_name: "User"
  belongs_to :task

  has_many :notifications, as: :notifiable

  validates :amount,
            numericality: { greater_than_or_equal_to: 2000 },
            presence: true
  validate :end_time_must_be_later_than_start_time
  validates :task_id,
            :tasker_id,
            :artisan_id,
            :description,
            presence: true
  scope :paid_for, -> { where(paid: true) }

  def self.notifications_count(user_type, id)
    if user_type == "artisan"
      paid_for.where(artisan_id: id).where(artisan_notified: false).count
    else
      where(tasker_id: id).where(tasker_notified: false).
        where.not(status: "unassigned").count
    end
  end

  def self.all_notifications_for(user_type, id)
    if user_type == "artisan"
      where(artisan_id: id, status: "unassigned").
        order(viewed: :asc, created_at: :desc).
        select(:id, :task_id, :tasker_id, :viewed)
    else
      where(tasker_id: id).where.not(status: "unassigned")
    end
  end

  def self.update_all_notifications_as_seen(user)
    query = user.user_type == "tasker" ? "tasker_id" : "artisan_id"
    attribute = query.gsub("id", "notified")
    paid_for.where(query => user.id).where(attribute => false).
      update_all(attribute => true)
  end

  def end_time_must_be_later_than_start_time
    if start_time && end_time
      unless end_time > start_time && end_time > Time.now
        errors[:endtime] = "End_time cannot be in the past"
      end
    else
      errors[:time] = "Task time cannot be nil"
    end
  end
end