From 7ff428d5c2b79fe7ba50fe921c7534ddb444b0a0 Mon Sep 17 00:00:00 2001
From: Mitchell Hashimoto <mitchell.hashimoto@gmail.com>
Date: Thu, 11 Feb 2010 00:51:21 -0800
Subject: [PATCH] `vagrant init` command. Used for initializing directories
 with vagrant.

---
 .gitignore                    |  3 ++-
 lib/vagrant/commands.rb       | 14 +++++++++++++-
 templates/Vagrantfile         |  5 +++++
 test/vagrant/commands_test.rb | 20 ++++++++++++++++++++
 4 files changed, 40 insertions(+), 2 deletions(-)
 create mode 100644 templates/Vagrantfile

diff --git a/.gitignore b/.gitignore
index c1b26a4ed..47f22ca4c 100644
--- a/.gitignore
+++ b/.gitignore
@@ -4,4 +4,5 @@ Vagrantfile
 .bundle
 *.lock
 cookbooks/*
-_site/*
\ No newline at end of file
+_site/*
+!templates/*
\ No newline at end of file
diff --git a/lib/vagrant/commands.rb b/lib/vagrant/commands.rb
index 99cac954e..27afdd279 100644
--- a/lib/vagrant/commands.rb
+++ b/lib/vagrant/commands.rb
@@ -7,9 +7,21 @@ module Vagrant
     extend Vagrant::Util
 
     class <<self
-      # TODO: Coming soon to a theatre near you.
+      # Initializes a directory for use with vagrant. This command copies an
+      # initial `Vagrantfile` into the current working directory so you can
+      # begin using vagrant. The configuration file contains some documentation
+      # to get you started.
       def init
+        rootfile_path = File.join(Dir.pwd, Env::ROOTFILE_NAME)
+        if File.exist?(rootfile_path)
+          error_and_exit(<<-error)
+It looks like this directory is already setup for vagrant! (A #{Env::ROOTFILE_NAME}
+already exists.)
+error
+        end
 
+        # Copy over the rootfile template into this directory
+        File.copy(File.join(PROJECT_ROOT, "templates", Env::ROOTFILE_NAME), rootfile_path)
       end
 
       # Bring up a vagrant instance. This handles everything from importing
diff --git a/templates/Vagrantfile b/templates/Vagrantfile
new file mode 100644
index 000000000..39741849f
--- /dev/null
+++ b/templates/Vagrantfile
@@ -0,0 +1,5 @@
+Vagrant::Config.run do |config|
+  # All Vagrant configuration is done here. For a detailed explanation
+  # and listing of configuration options, please check the documentation
+  # online.
+end
diff --git a/test/vagrant/commands_test.rb b/test/vagrant/commands_test.rb
index 3b39f3740..ecfc0b1b5 100644
--- a/test/vagrant/commands_test.rb
+++ b/test/vagrant/commands_test.rb
@@ -8,6 +8,26 @@ class CommandsTest < Test::Unit::TestCase
     Vagrant::Env.stubs(:persisted_vm).returns(@persisted_vm)
   end
 
+  context "init" do
+    setup do
+      File.stubs(:copy)
+      @rootfile_path = File.join(Dir.pwd, Vagrant::Env::ROOTFILE_NAME)
+      @template_path = File.join(PROJECT_ROOT, "templates", Vagrant::Env::ROOTFILE_NAME)
+    end
+
+    should "error and exit if a rootfile already exists" do
+      File.expects(:exist?).with(@rootfile_path).returns(true)
+      Vagrant::Commands.expects(:error_and_exit).once
+      Vagrant::Commands.init
+    end
+
+    should "copy the templated rootfile to the current path" do
+      File.expects(:exist?).with(@rootfile_path).returns(false)
+      File.expects(:copy).with(@template_path, @rootfile_path).once
+      Vagrant::Commands.init
+    end
+  end
+
   context "up" do
     setup do
       Vagrant::Env.stubs(:persisted_vm).returns(nil)