initial commit from appmail

app/models/concerns/has_message.rb

@@ -0,0 +1,38 @@
+module HasMessage
+
+  def self.included(base)
+    base.extend ClassMethods
+  end
+
+  def message
+    @message ||= self.server.message_db.message(self.message_id)
+  end
+
+  def message=(message)
+    @message = message
+    self.message_id = message&.id
+  end
+
+  module ClassMethods
+    def include_message
+      queued_messages = all.to_a
+      server_ids = queued_messages.map(&:server_id).uniq
+      if server_ids.size == 0
+        return []
+      elsif server_ids.size > 1
+        raise Postal::Error, "'include_message' can only be used on collections of messages from the same server"
+      end
+      message_ids = queued_messages.map(&:message_id).uniq
+      server = queued_messages.first&.server
+      messages = server.message_db.messages(:where => {:id => message_ids}).each_with_object({}) do |message, hash|
+        hash[message.id] = message
+      end
+      queued_messages.each do |queued_message|
+        if m = messages[queued_message.message_id]
+          queued_message.message = m
+        end
+      end
+    end
+  end
+
+end