module Bubble::Messages
  extend ActiveSupport::Concern

  included do
    has_many :messages, -> { chronologically }, dependent: :destroy
    after_save :capture_draft_comment
  end

  def capture(messageable)
    messages.create! messageable: messageable
  end

  def draft_comment
    find_or_build_initial_comment.body_plain_text
  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