io.github.fherbreteau.gatling.ftp.protocol.FtpProtocol.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gatling-ftp Show documentation
FTP plugin for Gatling
The newest version!
package io.github.fherbreteau.gatling.ftp.protocol

import com.typesafe.scalalogging.StrictLogging
import io.gatling.commons.model.Credentials
import io.gatling.core.CoreComponents
import io.gatling.core.config.GatlingConfiguration
import io.gatling.core.protocol.{Protocol, ProtocolKey}
import io.github.fherbreteau.gatling.ftp.client.Exchange

import java.nio.file.{Path, Paths}

object FtpProtocol extends StrictLogging {

  val FtpProtocolKey: ProtocolKey[FtpProtocol, FtpComponents] = new ProtocolKey[FtpProtocol, FtpComponents] {

    override def protocolClass: Class[Protocol] = classOf[FtpProtocol].asInstanceOf[Class[Protocol]]

    override def defaultProtocolValue(configuration: GatlingConfiguration): FtpProtocol =
      throw new IllegalArgumentException("Can't provide a default value for ImportProtocol")

    override def newComponents(coreComponents: CoreComponents): FtpProtocol => FtpComponents = {

      ftpProtocol => FtpComponents(ftpProtocol)
    }
  }

  def apply(configuration: GatlingConfiguration): FtpProtocol =
    new FtpProtocol(
      exchange = Exchange(
        server = "localhost",
        port = 22,
        credentials = Credentials("", "")
      ),
      localSourcePath = None,
      localDestinationPath = None,
      remoteSourcePath = None,
      remoteDestinationPath = None
    )
}

final case class FtpProtocol(exchange: Exchange,
                             localSourcePath: Option[Path],
                             localDestinationPath: Option[Path],
                             remoteSourcePath: Option[String],
                             remoteDestinationPath: Option[String]) extends Protocol {
  type Components = FtpComponents

  def localSource(file: String): Path = {
    localSourcePath.getOrElse(Paths.get(".")).resolve(file)
  }

  def localDestination(file: String): Path = {
    localDestinationPath.getOrElse(Paths.get(".")).resolve(file)
  }

  def remoteSource(file: String): String = {
    remoteSourcePath.getOrElse(s"/home/${exchange.credentials.username}").concat("/").concat(file)
  }

  def remoteDestination(file: String): String = {
    remoteDestinationPath.getOrElse(s"/home/${exchange.credentials.username}").concat("/").concat(file)
  }
}