gitlab-org--gitlab-foss/app/models/timelog.rb

51 lines
1.2 KiB
Ruby

# frozen_string_literal: true
class Timelog < ApplicationRecord
include Importable
before_save :set_project
validates :time_spent, :user, presence: true
validate :issuable_id_is_present, unless: :importing?
belongs_to :issue, touch: true
belongs_to :merge_request, touch: true
belongs_to :project
belongs_to :user
belongs_to :note
scope :for_issues_in_group, -> (group) do
joins(:issue).where(
'EXISTS (?)',
Project.select(1).where(namespace: group.self_and_descendants)
.where('issues.project_id = projects.id')
)
end
scope :between_times, -> (start_time, end_time) do
where('spent_at BETWEEN ? AND ?', start_time, end_time)
end
def issuable
issue || merge_request
end
private
def issuable_id_is_present
if issue_id && merge_request_id
errors.add(:base, _('Only Issue ID or merge request ID is required'))
elsif issuable.nil?
errors.add(:base, _('Issue or merge request ID is required'))
end
end
def set_project
self.project_id = issuable.project_id
end
# Rails5 defaults to :touch_later, overwrite for normal touch
def belongs_to_touch_method
:touch
end
end