skinny.mailer.JavaMailOps.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of skinny-mailer_2.10 Show documentation

skinny-mailer

There is a newer version: 2.5.2

package skinny.mailer

import javax.mail._
import javax.mail.internet.MimeMessage
import scala.io.Source
import scala.collection.JavaConverters._
import skinny.logging.Logging

/**
 * Operations about javax.mail APIs.
 */
object JavaMailOps extends Logging {

  def loggingTransport(session: Session): Transport = {
    new Transport(session, null) {
      def sendMessage(msg: Message, addresses: Array[Address]): Unit = {
        val mimeMsg = msg.asInstanceOf[MimeMessage]
        val content = try Source.fromInputStream(mimeMsg.getInputStream).mkString
        catch { case e: MessagingException => "" }
        logger.info {
          s"""
           |
           |##### SkinnyMailer Logging Transport #####
           |
           |${mimeMsg.getAllHeaderLines.asScala.mkString("\n")}
           |
           |${content}
           |""".stripMargin
        }
      }
    }
  }

  /**
   * Returns transport from session.
   */
  def transport(session: Session, auth: Option[SmtpAuthentication], protocol: String): Transport = {
    if (protocol == "logging") {
      loggingTransport(session)
    } else {
      val t = session.getTransport(protocol)
      auth.map(a => t.connect(a.user, a.password)).getOrElse(t.connect())
      t
    }
  }

}