openproject/lib/issue_patch.rb

require_dependency 'issue'

module IssuePatch
    def self.included(base) # :nodoc:
        base.extend(ClassMethods)
        base.send(:include, InstanceMethods)

        base.class_eval do
            unloadable

            alias_method_chain :move_to_project_without_transaction, :autolink
            alias_method_chain :update_parent_attributes, :remaining_hours
            before_save   :store_parent
            after_save    :task_follows_story
            after_save    :fix_old_parent
        end
    end

    module ClassMethods
    end

    module InstanceMethods
        def move_to_project_without_transaction_with_autolink(new_project, new_tracker = nil, options = {})
            newissue = move_to_project_without_transaction_without_autolink(new_project, new_tracker, options)

            if self.project_id == newissue.project_id and self.is_story? and newissue.is_story? and self.id != newissue.id
                relation = IssueRelation.new :relation_type => IssueRelation::TYPE_DUPLICATES
                relation.issue_from = self
                relation.issue_to = newissue
                relation.save
            end

            return newissue
        end

        def journalized_update_attributes!(attribs)
            self.init_journal(User.current)
            return self.update_attributes! attribs
        end

        def journalized_update_attributes(attribs)
            self.init_journal(User.current)
            return self.update_attributes attribs
        end

        def journalized_update_attribute(attrib, v)
            self.init_journal(User.current)
            self.update_attribute(attrib, v)
        end

        def is_story?
            return (Story.trackers.include?(self.tracker_id) and self.root?)
        end

        def is_task?
            return (self.parent_id && self.tracker_id == Task.tracker)
        end

        def story
            return Issue.find(:first,
                :conditions => [ "id = ? and tracker_id in (?)", self.root_id, Story.trackers ])
        end

        def blocks
            # return issues that I block that aren't closed
            return [] if closed?
            relations_from.collect {|ir| ir.relation_type == 'blocks' && !ir.issue_to.closed? ? ir.issue_to : nil}.compact
        end

        def blockers
            # return issues that block me
            return [] if closed?
            relations_to.collect {|ir| ir.relation_type == 'blocks' && !ir.issue_from.closed? ? ir.issue_from : nil}.compact
        end

        def velocity_based_estimate
            return nil if !self.is_story? || ! self.story_points || self.story_points <= 0

            v = self.project.velocity
            return nil if ! v or ! v[:velocity] or v[:velocity] <= 0

            return self.story_points * (v[:days] / v[:velocity])
        end

        def update_parent_attributes_with_remaining_hours
            update_parent_attributes_without_remaining_hours

            if parent_id && p = Issue.find_by_id(parent_id) 
                p.remaining_hours = p.leaves.sum(:remaining_hours).to_f
                p.save
            end
        end

        def store_parent
            if self.parent_id_was && !(self.parent_id_was == self.parent_issue_id)
                @old_parent = self.parent_id_was
            else
                @old_parent = nil
            end
        end

        def fix_old_parent
            if @old_parent
                p = Issue.find_by_id(@old_parent)
                c = p.children
                if c.length == 0
                    p.update_attribute(:estimated_hours, nil)
                    p.update_attribute(:remaining_hours, nil)
                else
                    c[0].instance_eval{ update_parent_attributes }
                end
            end
        end

        def task_follows_story
            ## automatically sets the tracker to the task tracker for
            ## any descendant of story, and follow the version_id
            ## Normally one of the _before_save hooks ought to take
            ## care of this, but appearantly neither root_id nor
            ## parent_id are set at that point

            touched_sprint = nil

            if self.is_story?
                # raw sql here because it's efficient and not
                # doing so causes an update loop when Issue calls
                # update_parent
                if not Task.tracker.nil?
                    tasks = self.descendants.collect{|t| connection.quote(t.id)}.join(",")
                    if tasks != ""
                        connection.execute("update issues set tracker_id=#{connection.quote(Task.tracker)}, fixed_version_id=#{connection.quote(self.fixed_version_id)} where id in (#{tasks})")
                    end
                end
                touched_sprint = self.fixed_version

            elsif not Task.tracker.nil?
                begin
                    story = self.story
                    if not story.nil?
                        connection.execute "update issues set tracker_id = #{connection.quote(Task.tracker)}, fixed_version_id = #{connection.quote(story.fixed_version_id)} where id = #{connection.quote(self.id)}"
                        touched_sprint = story.fixed_version
                    end
                end
            end

            if not touched_sprint.nil?
                touched_sprint.touch_burndown
            end
        end
    end
end