module Card::DraftCommenting
  extend ActiveSupport::Concern

  included do
    after_save :capture_draft_comment
  end

  def draft_comment
    find_or_build_initial_comment.body.content
  end

  def draft_comment=(body)
    if body.present?
      @draft_comment = body
    else
      messages.comments.destroy_all
    end
  end

  private
    def find_or_build_initial_comment
      message = messages.comments.first || messages.new(messageable: Comment.new)
      message.comment
    end

    def capture_draft_comment
      if @draft_comment.present?
        find_or_build_initial_comment.update! body: @draft_comment, creator: creator
      end

      @draft_comment = nil
    end
end