Rakefile

# Copyright 2021 Google LLC
#
# Use of this source code is governed by an MIT-style
# license that can be found in the LICENSE file or at
# https://opensource.org/licenses/MIT.

require_relative "config/environment"
require "docker"

def fetch_samples
  Dir.entries(".").select do |entry|
    File.directory?(File.join(".", entry)) \
      && !%w[. ..].include?(entry) \
      && File.exist?(File.join(".", entry, "application.rb"))
  end
end

desc "Lists all available samples."
task :list do
  samples = fetch_samples
  puts "Available samples: "
  samples.sort.each { |dir| puts "  #{dir}" }
  puts ""
  puts "Run a sample with the command `bundle exec rake run\\[<sample-name>\\]`"
end

desc "Runs all samples."
task :all do
  samples = fetch_samples
  ar_version = ENV.fetch "AR_VERSION", "~> 7.1.0"
  less_than_7_1 = ar_version.dup.to_s.sub("~>", "").strip < "7.1.0"
  samples.delete "interleaved-tables" if less_than_7_1
  samples.delete "interleaved-tables-before-7.1" unless less_than_7_1
  samples.delete "bit-reversed-sequence" if less_than_7_1
  samples.delete "auto-generated-primary-key" if less_than_7_1
  samples.delete "query-logs" if less_than_7_1
  samples.each do |sample|
    run_sample sample
  end
end

desc "Runs a simple ActiveRecord tutorial on a Spanner emulator."
task :run, [:sample] do |_t, args|
  sample = args[:sample]
  unless sample
    puts "Missing argument :sample. Running quickstart sample"
    puts ""
    sample = "quickstart"
  end
  run_sample sample
end

def run_sample sample
  puts "Running #{sample}"
  puts "Downloading Spanner emulator image..."
  Docker::Image.create "fromImage" => "gcr.io/cloud-spanner-emulator/emulator:latest"
  puts "Creating Spanner emulator container..."
  container = Docker::Container.create(
    "Image" => "gcr.io/cloud-spanner-emulator/emulator:latest",
    "ExposedPorts" => { "9010/tcp" => {} },
    "HostConfig" => {
      "PortBindings" => {
        "9010/tcp" => [{ "HostPort" => "9010" }]
      }
    }
  )

  begin
    puts "Starting Spanner emulator..."
    container.start!
    Dir.chdir sample do
      sh "ruby ../bin/create_emulator_instance.rb"
      sh "rake db:migrate"
      sh "rake db:seed"
      sh "ruby application.rb"
    end
  ensure
    container.stop!
  end
end