image/build.rb

   1 # simple build file to be used locally by Sam
   2 #
   3 require 'pty'
   4 require 'optparse'
   5
   6 def run(command)
   7   lines = []
   8   PTY.spawn(command) do |stdin, stdout, pid|
   9     begin
  10       stdin.each do |line|
  11         lines << line
  12         puts line
  13       end
  14     rescue Errno::EIO
  15       # we are done
  16     end
  17   end
  18
  19   lines
  20 end
  21
  22 def ensure_docker_squash
  23   docker_squash = "https://github.com/goldmann/docker-squash/archive/master.zip"
  24   run ("pip install --user #{$docker_squash} --upgrade")
  25 end
  26
  27
  28 def build(image)
  29   lines = run("cd #{image[:name]} && docker build .")
  30   img = lines[-1]["successfully built ".length..-1].strip
  31
  32   if image[:squash]
  33
  34     if image[:layers_to_keep] == nil
  35       run("docker-squash -t #{image[:tag]} --cleanup --verbose #{img}")
  36     else
  37       layers_to_squash = run("docker history #{img} | wc -l").first.to_i - (1 + image[:layers_to_keep])
  38       run("docker-squash -t #{image[:tag]} --cleanup --verbose -f #{layers_to_squash} #{img}")
  39     end
  40
  41     run("docker rmi #{img}")
  42
  43   else
  44     run("docker tag #{img} #{image[:tag]}")
  45   end
  46 end
  47
  48 def bump(image, image_version)
  49   run("echo #{image_version} > base/VERSION") if image == 'base'
  50   run("sed -i '' -e 's/^\(# NAME:\).*$$/\1     discourse\/#{image}/' #{image}/Dockerfile")
  51   run("sed -i '' -e 's/^\(# VERSION:\).*$$/\1  #{image_version}/' #{image}/Dockerfile")
  52   run("sed -i '' -e 's/^\(FROM discourse\/[^:]*:\).*/\1#{image_version}/' #{image}/Dockerfile")
  53 end
  54
  55 def dev_deps()
  56   run("sed -e 's/\(db_name: discourse\)/\1_development/' ../templates/postgres.template.yml > discourse_dev/postgres.template.yml")
  57   run("cp ../templates/redis.template.yml discourse_dev/redis.template.yml")
  58 end
  59
  60 options = {}
  61 OptionParser.new do |parser|
  62   parser.on("-i", "--image image",
  63             "Build the image. No parameter means [base discourse discourse_test].") do |i|
  64     options[:image] = [i.to_sym]
  65   end
  66   parser.on("-b", "--bump version",
  67             "Bumps the version in the Dockerfiles specified by --image") do |v|
  68     options[:version] = [v]
  69   end
  70 end.parse!
  71
  72 DEFAULT_IMAGES = %i[base discourse discourse_test discourse_dev discourse_bench]
  73
  74 todo = options[:image] || DEFAULT_IMAGES
  75 version = options[:version] || '1.3.7'
  76
  77 if ENV["USER"] != "root"
  78   STDERR.puts "Build script must be ran as root due to docker-squash"
  79   exit 1
  80 end
  81
  82 ensure_docker_squash
  83
  84 images = {
  85   base: { name: 'base', tag: "discourse/base:#{version}", squash: true },
  86   discourse: { name: 'discourse', tag: "discourse/discourse:#{version}", squash: true, layers_to_keep: 1 },
  87   discourse_test: { name: 'discourse_test', tag: "discourse/discourse_test:#{version}", squash: true, layers_to_keep: 2 },
  88   discourse_dev: { name: 'discourse_dev', tag: "discourse/discourse_dev:#{version}", squash: false },
  89   discourse_bench: { name: 'discourse_bench', tag: "discourse/discourse_bench:#{version}", squash: false }
  90 }
  91
  92 todo.each do |image|
  93   puts images[image]
  94   bump(images[image][:name], options[:version]) if options[:version]
  95
  96   dev_deps() if image == 'discourse_dev'
  97   run "(cd base && ./download_phantomjs)" if image == 'base'
  98
  99   build(images[image])
 100 end