# simple build file to be used locally by Sam
#
require "pty"
require "optparse"

images = {
  base_slim: {
    name: "base",
    tag: "discourse/base:build_slim",
    extra_args: "-f slim.Dockerfile",
  },
  base_slim_arm64: {
    name: "base",
    tag: "discourse/base:build_slim_arm64",
    extra_args: "-f slim.Dockerfile --platform linux/arm64",
  },
  base: {
    name: "base",
    tag: "discourse/base:build",
    extra_args: "-f release.Dockerfile",
  },
  discourse_test_build: {
    name: "discourse_test",
    tag: "discourse/discourse_test:build",
  },
  discourse_dev: {
    name: "discourse_dev",
    tag: "discourse/discourse_dev:build",
  },
}

def run(command)
  lines = []
  PTY.spawn(command) do |stdout, stdin, pid|
    begin
      stdout.each do |line|
        lines << line
        puts line
      end
    rescue Errno::EIO
      # we are done
    end
    Process.wait(pid)
  end

  raise "'#{command}' exited with status #{$?.exitstatus}" if $?.exitstatus != 0

  lines
end

def build(image)
  lines =
    run(
      "cd #{image[:name]} && docker buildx build . --load --no-cache --tag #{image[:tag]} #{image[:extra_args] ? image[:extra_args] : ""}",
    )
  if lines[-1] =~ /successfully built/
    raise "Error building the image for #{image[:name]}: #{lines[-1]}"
  end
end

def dev_deps()
  run(
    "sed -e 's/\(db_name: discourse\)/\1_development/' ../templates/postgres.template.yml > discourse_dev/postgres.template.yml",
  )
  run("cp ../templates/redis.template.yml discourse_dev/redis.template.yml")
end

if ARGV.length != 1
  puts <<~TEXT
    Usage:
    ruby auto_build.rb IMAGE

    Available images:
    #{images.keys.join(", ")}
  TEXT
  exit 1
else
  image = ARGV[0].to_sym

  if !images.include?(image)
    $stderr.puts "Image not found"
    exit 1
  end

  puts "Building #{images[image]}"
  dev_deps() if image == :discourse_dev

  build(images[image])
end