vagrant/plugins/commands/init/command.rb

require 'optparse'

require 'vagrant/util/template_renderer'

module VagrantPlugins
  module CommandInit
    class Command < Vagrant.plugin("2", :command)
      def self.synopsis
        "initializes a new Vagrant environment by creating a Vagrantfile"
      end

      def execute
        options = { output: "Vagrantfile" }

        opts = OptionParser.new do |o|
          o.banner = "Usage: vagrant init [box-name] [box-url]"

          o.on("--output FILENAME", String,
               "Output path for the box. '-' for stdout.") do |output|
            options[:output] = output
          end
        end

        # Parse the options
        argv = parse_options(opts)
        return if !argv

        save_path = nil
        if options[:output] != "-"
          save_path = Pathname.new(options[:output]).expand_path(@env.cwd)
          raise Vagrant::Errors::VagrantfileExistsError if save_path.exist?
        end

        template_path = ::Vagrant.source_root.join("templates/commands/init/Vagrantfile")
        contents = Vagrant::Util::TemplateRenderer.render(template_path,
                                                          :box_name => argv[0] || "base",
                                                          :box_url => argv[1])

        if save_path
          # Write out the contents
          save_path.open("w+") do |f|
            f.write(contents)
          end

          @env.ui.info(I18n.t("vagrant.commands.init.success"), prefix: false)
        else
          @env.ui.info(contents, prefix: false)
        end

        # Success, exit status 0
        0
      end
    end
  end
end