build.sbt

organization := "com.amazonaws"
organizationName := "Amazon Web Services"
organizationHomepage := Some(url("https://aws.amazon.com"))
name := "sagemaker-spark"
description := "SageMaker Spark provides a Spark interface to Amazon SageMaker, allowing " +
  "customers to train using the Spark Estimator API, host their model on Amazon SageMaker, and " +
  "make predictions with their model using the Spark Transformer API."
homepage := Some(url("https://github.com/aws/sagemaker-spark"))
scmInfo := Some(
  ScmInfo(
    url("https://github.com/aws/sagemaker-spark"),
    "https://github.com/aws/sagemaker-spark.git"
  )
)
licenses := Seq("Apache License, Version 2.0" -> url("https://aws.amazon.com/apache2.0"))

scalaVersion := "2.12.16"

// to change the version of spark add -DSPARK_VERSION=2.x.x when running sbt
// for example: "sbt -DSPARK_VERSION=2.1.1 clean compile test doc package"
val sparkVersion = System.getProperty("SPARK_VERSION", "3.3.0")

lazy val SageMakerSpark = (project in file("."))

version := {
    val base = baseDirectory.in(SageMakerSpark).value
    val packageVersion = IO.read(base / ".." / "VERSION").trim
    s"spark_$sparkVersion-$packageVersion"
}

libraryDependencies ++= Seq(
  "org.apache.hadoop" % "hadoop-aws" % "3.3.1",
  "com.amazonaws" % "aws-java-sdk-s3" % "1.12.262",
  "com.amazonaws" % "aws-java-sdk-sts" % "1.12.262",
  "com.amazonaws" % "aws-java-sdk-sagemaker" % "1.12.262",
  "com.amazonaws" % "aws-java-sdk-sagemakerruntime" % "1.12.262",
  "org.apache.spark" %% "spark-core" % sparkVersion % "provided",
  "org.apache.spark" %% "spark-mllib" % sparkVersion % "provided",
  "org.apache.spark" %% "spark-sql" % sparkVersion % "provided",
  "org.scalatest" %% "scalatest" % "3.0.9" % "test",
  "org.mockito" % "mockito-all" % "2.0.2-beta" % "test"
)

// add a task to print the classpath. Also use the packaged JAR instead
// of the .class files.
exportJars := true
lazy val printClasspath = taskKey[Unit]("Dump classpath")
printClasspath := (fullClasspath in Runtime value) foreach { e => println(e.data) }

// set coverage threshold
coverageFailOnMinimum := true
coverageMinimumStmtTotal := 90
coverageMinimumBranchTotal := 90
coverageMinimumStmtPerPackage := 83
coverageMinimumBranchPerPackage := 75

// make scalastyle gate the build
(compile in Compile) := {
  scalastyle.in(Compile).toTask("").value
  (compile in Compile).value
}

(test in Test) := {
  scalastyle.in(Test).toTask("").value
  (test in Test).value
}

parallelExecution in Test := false

// https://github.com/sbt/sbt/issues/3570
updateOptions := updateOptions.value.withGigahorse(false)

// publishing configuration
publishMavenStyle := true
pomIncludeRepository := { _ => false }
publishArtifact in Test := false
val nexusUriHost = "aws.oss.sonatype.org"
val nexusUriHostWithScheme = "https://" + nexusUriHost + "/"
val snapshotUrl = nexusUriHostWithScheme + "content/repositories/snapshots"
val localReleaseUrl = nexusUriHostWithScheme + "service/local"
val releaseUrl = localReleaseUrl + "/staging/deploy/maven2"
publishTo := {
  if (isSnapshot.value)
    Some("snapshots" at snapshotUrl)
  else
    Some("releases" at releaseUrl)
}
Sonatype.SonatypeKeys.sonatypeRepository := localReleaseUrl
Sonatype.SonatypeKeys.sonatypeCredentialHost := nexusUriHost

credentials += Credentials(
  "Sonatype Nexus Repository Manager",
  nexusUriHost,
  sys.env.getOrElse("SONATYPE_USERNAME", "NOT_A_PUBLISH_BUILD"),
  sys.env.getOrElse("SONATYPE_PASSWORD", "NOT_A_PUBLISH_BUILD")
)
pomExtra := (
  <developers>
    <developer>
      <id>amazonwebservices</id>
      <organization>Amazon Web Services</organization>
      <organizationUrl>https://aws.amazon.com</organizationUrl>
      <roles>
        <role>developer</role>
      </roles>
    </developer>
  </developers>
  )