Rakefile

#
# Treasure Data:
# This script deploy jar packages with a version number that includes commit-hash of git
# instead of -SNAPSHOT so that we can deploy specific commit to the production system. It
# allows us to deploy hotfixes or improvements before stable releases.
#

EXCLUDE_MODULES = %w|
 presto-atop presto-mongodb
 presto-cassandra presto-kafka presto-redis presto-docs
 presto-benchmark presto-benchmark-driver
 presto-example-http presto-base-jdbc
 presto-mysql presto-postgresql presto-hive
 presto-hive-hadoop1 presto-hive-hadoop2
 presto-verifier presto-testing-server-launcher presto-jmx
 presto-hive-cdh4 presto-hive-cdh5 presto-raptor presto-server-rpm
 presto-array presto-record-decoder presto-kafka presto-orc
 presto-local-file presto-ml presto-product-tests
 presto-server-rpm presto-plugin-toolkit|

EXCLUDE_FROM_COMPILE = %w|presto-docs presto-server-rpm|

def presto_modules
  require "rexml/document"
  pom = REXML::Document.new(File.read("pom.xml"))
  modules = []
  REXML::XPath.each(pom, "/project/modules/module"){|m|
    modules << m.text
  }
  modules
end

def active_modules
  presto_modules.keep_if{|m| !EXCLUDE_MODULES.include?(m) }
end

def compile_target_modules
  presto_modules.keep_if{|m| !EXCLUDE_FROM_COMPILE.include?(m) }
end

desc "compile codes"
task "compile" do
  sh "./mvnw -s settings.xml test-compile -pl #{compile_target_modules.join(",")} -DskipTests"
end

desc "run tests"
task "test" do
  sh "./mvnw -s settings.xml -P td -pl #{active_modules.join(",")} test"
end

desc "set a unique version and td-specific settings"
task "update-pom" do
  require "rexml/document"

  # Read the current presto version
  rev = `git rev-parse HEAD`
  pom = REXML::Document.new(File.read("pom.xml"))
  presto_version = REXML::XPath.first(pom, "/project/version")

  # Set (presto-version)-(git revision number:first 7 characters) version to pom.xml files
  version = "#{presto_version.text.gsub("-SNAPSHOT", "")}-#{rev[0...7]}"
  sh "./mvnw versions:set -DgenerateBackupPoms=false -DnewVersion=#{version} -N"

  # Reload pom.xml
  pom = REXML::Document.new(File.read("pom.xml"))

  # delete unncessary modules from pom.xml
  # EXCLUDE_MODULES.each{|m|
  #    pom.delete_element("/project/modules/module[text()='#{m}']")
  # }

  # Inject extension plugin to deploy artifacts to s3
  extension = <<EOF
    <extensions>
      <extension>
        <groupId>org.springframework.build</groupId>
        <artifactId>aws-maven</artifactId>
        <version>5.0.0.RELEASE</version>
      </extension>
    </extensions>
EOF
  REXML::XPath.first(pom, "/project/build").add_element(REXML::Document.new(extension))

  # Inject build profile for TD (disable version check, set distribution management tag)
  profiles = REXML::XPath.first(pom, "/project/profiles")
  unless profiles
    profiles = REXML::XPath.first(pom, "/project").add_element("profiles")
  end
  profiles.add_element(REXML::Document.new(File.read("td-profile.xml")))

  distribution_management = <<EOF
    <distributionManagement>
      <repository>
      	<id>treasuredata</id>
	      <name>treasuredata-releases</name>
	      <url>https://treasuredata.jfrog.io/treasuredata/libs-release-local</url>
      </repository>
      <snapshotRepository>
	      <id>treasuredata</id>
	      <name>treasuredata-snapshots</name>
	      <url>https://treasuredata.jfrog.io/treasuredata/libs-snapshot-local</url>
      </snapshotRepository>
    </distributionManagement>
EOF

  repositories = <<EOF
    <repositories>
      <repository>
	      <id>treasuredata</id>
	      <name>treasuredata-releases</name>
	      <url>https://treasuredata.jfrog.io/treasuredata/libs-release</url>
	      <snapshots>
          <enabled>false</enabled>
      	</snapshots>
      </repository>
      <repository>
	      <id>treasuredata-snapshots</id>
	      <name>treasuredata-snapshots</name>
	      <url>https://treasuredata.jfrog.io/treasuredata/libs-snapshot</url>
	      <releases>
          <enabled>false</enabled>
	      </releases>
      </repository>
    </repositories>
EOF

  REXML::XPath.first(pom, "/project").insert_after("modules", REXML::Document.new(repositories))
  REXML::XPath.first(pom, "/project").insert_before("dependencyManagement", REXML::Document.new(distribution_management))

  # Dump pom.xml
  File.open('pom.xml', 'w'){|f| pom.write(f) }

end

desc "deploy presto"
task "deploy" do
  # Deploy
  # Deploy presto-root
  sh "./mvnw -s settings.xml deploy -P td -N -DskipTests"
  # Deploy presot modules
  sh "./mvnw -s settings.xml deploy -P td -pl #{compile_target_modules.join(",")} -DskipTests"
end