dockerfile_writer.rb

#!/usr/bin/env ruby
# frozen_string_literal: true

#
# Copyright (C) 2017 - present Instructure, Inc.
#
# This file is part of Canvas.
#
# Canvas is free software: you can redistribute it and/or modify it under
# the terms of the GNU Affero General Public License as published by the Free
# Software Foundation, version 3 of the License.
#
# Canvas is distributed in the hope that it will be useful, but WITHOUT ANY
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
# A PARTICULAR PURPOSE. See the GNU Affero General Public License for more
# details.
#
# You should have received a copy of the GNU Affero General Public License along
# with this program. If not, see <http://www.gnu.org/licenses/>.

require "erubi"
require "json"
require "optparse"
require_relative "./docker_utils"

class DockerfileWriter
  attr_reader :env, :compose_files, :in_file, :out_file, :out_file_suffix

  def initialize(env:, compose_files:, in_file:, out_file:)
    @env = env
    @compose_files = compose_files
    @in_file = in_file
    @out_file = out_file
    @out_file_suffix = ""
  end

  def production?
    env == "production"
  end

  def development?
    env == "development"
  end

  def jenkins?
    env == "jenkins"
  end

  def generation_message
    <<~RUBY
      # GENERATED FILE, DO NOT MODIFY!
      # To update this file please edit the relevant template and run the generation
      # task `build/dockerfile_writer.rb --env #{env} --compose-file #{compose_files.join(",")} --in #{in_file} --out #{out_file}`
    RUBY
  end

  def set_file_suffix(suffix)
    @out_file_suffix = suffix
  end

  class SuffixedStringWriter
    attr_reader :parent, :contents

    def initialize(parent)
      @contents = {}
      @parent = parent
    end

    def <<(obj)
      if @contents[parent.out_file_suffix].nil?
        @contents[parent.out_file_suffix] = +""
      end

      @contents[parent.out_file_suffix] << obj
    end
  end

  def run
    contents = eval(Erubi::Engine.new(File.read(in_file), { bufval: "SuffixedStringWriter.new(self)" }).src + ";_buf.contents") # rubocop:disable Security/Eval

    contents.each do |k, v|
      File.open(k.empty? ? out_file : "#{out_file}.#{k}", "w") do |f|
        f.write "#{v.strip!}\n"
      end
    end
  end

  def docker_compose_volume_paths
    paths = (docker_compose_config["services"]["web"]["volumes"] || []).filter_map do |volume|
      name, path = volume.split(":")
      next unless /\A[a-z]/.match?(name)

      path.sub("/usr/src/app/", "")
    end
    paths.sort_by { |path| [path[0] == "/" ? 1 : 0, path] }
  end

  def docker_compose_config
    DockerUtils.compose_config(*compose_files)
  end

  def yarn_packages
    JSON.parse(File.read("package.json"))["workspaces"]["packages"]
  end
end

options = {}

OptionParser.new do |opts|
  opts.banner = "Usage: dockerfile_writer.rb [options]"

  opts.on("--env [ENVIRONMENT]", String, "Dockerfile Environment") do |v|
    options[:env] = v
  end

  opts.on("--compose-file x,y,z", Array, "List of compose files") do |v|
    options[:compose_files] = v
  end

  opts.on("--in [FILENAME]", String, "Input Template File") do |v|
    options[:in_file] = v
  end

  opts.on("--out [FILENAME]", String, "Output File") do |v|
    options[:out_file] = v
  end
end.parse!

DockerfileWriter.new(**options).run