• Example Search
• Project Search

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• gatling-imap-master
  • src
    • main
      • scala-2.12
        • com
          • linagora
            • gatling
              • imap
                • check
                  • ImapCheckSupport.scala
                  • ImapValidationCheck.scala
                  • package.scala
                  • ImapSimpleCheck.scala
                • ImapTestUtils.scala
                • Imap.scala
                • action
                  • SelectAction.scala
                  • FetchAction.scala
                  • StoreAction.scala
                  • ListAction.scala
                  • ImapResponseHandler.scala
                  • ImapActionContext.scala
                  • LoginAction.scala
                  • UIDFetchAction.scala
                  • AppendAction.scala
                  • ImapActionActor.scala
                  • ImapActionBuilder.scala
                  • ConnectAction.scala
                  • ExpungeAction.scala
                • protocol
                  • Uid.scala
                  • ImapProtocol.scala
                  • ImapResponses.scala
                  • package.scala
                  • UserSession.scala
                  • ImapProtocolBuilder.scala
                  • ImapSessions.scala
                  • command
                    • SelectHandler.scala
                    • AppendHandler.scala
                    • StoreHandler.scala
                    • MessageRanges.scala
                    • ListHandler.scala
                    • ImapSessionExecutor.scala
                    • UIDFetchHandler.scala
                    • FetchHandler.scala
                    • ExpungeHandler.scala
                    • LoginHandler.scala
                    • MessageRange.scala
                • PreDef.scala
                • scenario
                  • ImapSimpleScenario.scala
                  • ImapExpungeScenario.scala
                  • ImapUIDFetchScenario.scala
                  • ImapAuthenticationScenario.scala
    • it
      • scala-2.12
        • com
          • linagora
            • gatling
              • imap
                • CyrusServer.scala
                • Fixture.scala
                • Server.scala
                • protocol
                  • command
                    • ImapSessionsSpec.scala
                    • LoginHandlerSpec.scala
                • scenario
                  • it
                    • ImapAuthenticationScenarioIT.scala
      • resources
        • logback.xml
  • LICENSE
  • project
    • build.properties
    • plugins.sbt
  • LICENSE.md
  • README.adoc
  • .travis.yml
  • build.sbt
  • .gitignore

package com.linagora.gatling.imap.protocol

import java.util.regex.Pattern

import com.sun.mail.imap.protocol.IMAPResponse

import scala.collection.immutable.Seq

case class ImapResponses(responses: Seq[IMAPResponse]) {

  import ImapResponses._

  def mkString(separator: String = ",") = {
    responses.mkString(separator)
  }

  def isBad = responses.lastOption.exists(_.isBAD)

  def isOk = responses.lastOption.exists(_.isOK)

  def isNo = responses.lastOption.exists(_.isNO)

  def countRecent: Option[Int] = {
    responses.map(_.toString).find(_.matches(Recent.regex))
      .map {
        case Recent(actual) => actual.toInt
      }
  }

  def folderList: Seq[String] = {
    responses.map(_.toString).filter(_.matches(List.regex))
      .map {
        case List(name, null) => name
        case List(null, quotedName) => quotedName
      }
  }

  def uidList: Seq[Uid] = {
    responses.map(_.toString).filter(_.matches(UidRegex.regex))
      .map {
        case UidRegex(uid) => Uid(uid.toInt)
      }
  }

  def contains(content: String): Boolean =
    responses.map(_.toString).exists(_.contains(content))
}

object ImapResponses {
  val empty = ImapResponses(Seq.empty)

  private[this] val dotAllFlag = """(?s)"""
  private[this] val startWithStar = """^(?:(?:, )?\*)"""
  private[this] val mailboxName = """(?:"([^"]*)"|([^"\s]*))"""

  private val Recent = (dotAllFlag + startWithStar + """ (\d+) RECENT\s*$""").r
  private val List = ("""^\* LIST .*? """ + mailboxName + """\s*$""").r
  private val UidRegex = (dotAllFlag + startWithStar + """ .*UID (\d+).*$""").r
}