vagrant-hivemind 0.1.5 → 0.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 5451a9794ab6ec928dbde47c1a17a16e5a819e10
-  data.tar.gz: e8aca8487f46a8b534a32a392c891e34b4446fa0
+  metadata.gz: 096d7bc5746b1e78bb9df4205c6b6a8bd23d9859
+  data.tar.gz: 5396a552c7076e97a436df6807632aa3f60e605f
 SHA512:
-  metadata.gz: 69bb48ca9a0a5151f347b00ed3522e3c59b687b90771f7df04d55aec9f3fa10c5665cf231097c019a4d8bbed294d424bd25a8e9cacdbdf7f35c0763d654e508f
-  data.tar.gz: a3a7407d0d04b7b7beca80934a3ea323b6a0253c8d04896003c56e9c0fb2bb17144f67c4f85b6a90c33de2cce2bc9fa5543338f4b8a76a21df39252c3adaf926
+  metadata.gz: 1f984063526bde33b71cf8d9f7e14784cd35ebdbd2c2a68ed35f59da838fff9162f327899c4aa73eb21d0111c39758fc681152afa92a850eaae5c7d890de9ab8
+  data.tar.gz: ea4215f9ff49e1497317b27ae4b984c23920bbb12ba1eee489237e6c475c12b1f5dc188f2401fbaa40b71b0a361c1b345fc11beb133203358c66cfae878b91fd

data/lib/vagrant/hivemind/commands/generate.rb

@@ -0,0 +1,71 @@
+require "optparse"
+require "erb"
+
+require "vagrant/hivemind/constants"
+require "vagrant/hivemind/util"
+require "vagrant/hivemind/host"
+
+module Vagrant
+  module Hivemind
+    module Command
+      class Generate < Vagrant.plugin("2", :command)
+          include Vagrant::Hivemind::Constants
+          include Vagrant::Hivemind::Util
+
+        def self.synopsis
+          "Generates a Vagrantfile of the Hive"
+        end
+
+        def execute
+          options = {}
+
+          opts = OptionParser.new do |o|
+            o.banner = "Usage: vagrant hivemind generate [options]"
+            o.separator ""
+            o.separator "Options:"
+            o.separator ""
+
+            o.on("-f", "--force", "Overwrite an existing '#{VAGRANT_FILE}' with a new one") do |f|
+              options[:force] = f
+            end
+
+            o.on("-d", "--directory DIRECTORY", "Specify the directory where '#{HIVE_FILE}' is located (default: current directory)") do |d|
+              options[:directory] = []
+              options[:directory] << d
+            end
+          end.parse!
+
+          root_path = Path.get_root_path_from_options options
+
+          unless HiveFile.exist? root_path
+            @env.ui.error "There is no Hive file in the working directory."
+            return 1
+          end
+
+          hosts = HiveFile.read_from root_path
+
+          if Vagrant::Hivemind::Util::Vagrantfile.exist? root_path
+            if options[:force]
+              write_new_vagrant_file hosts, root_path
+              @env.ui.info "The Vagrantfile has been overwritten."
+              return 0
+            else
+              @env.ui.error "The Vagrantfile already exists!"
+              return 1
+            end
+          end
+
+          write_new_vagrant_file hosts, root_path
+
+          0
+        end
+
+        private
+          def write_new_vagrant_file(hosts, path)
+            Vagrant::Hivemind::Util::Vagrantfile.generate_hivemind_vagrantfile @env, hosts, path, true
+          end
+
+      end
+    end
+  end
+end

data/lib/vagrant/hivemind/commands/root.rb

@@ -55,6 +55,11 @@ module Vagrant
             Halt
           end
 
+          @subcommands.register(:generate) do
+            require_relative "generate"
+            Generate
+          end
+
         end
 
         def execute

data/lib/vagrant/hivemind/constants.rb

@@ -4,6 +4,7 @@ module Vagrant
       HIVE_FILE = "hive.yml"
       ANSIBLE_HOSTS_FILE = "ansible.hosts"
       SYSTEM_HOSTS_FILE = "system.hosts"
+      VAGRANT_FILE = "Vagrantfile"
 
       BOX_TYPES = {
         :server    => { name: "Server",     box_id: "napramirez/ubuntu-14.04-server-amd64",            is_gui: false },

data/lib/vagrant/hivemind/util.rb

@@ -141,7 +141,7 @@ module Vagrant
       end
 
       class Vagrantfile
-        def self.generate_hivemind_vagrantfile(env, hosts, path = Pathname.new(Dir.pwd))
+        def self.generate_hivemind_vagrantfile(env, hosts, path = Pathname.new(Dir.pwd), generate = false)
           box_types = Vagrant::Hivemind::Constants::BOX_TYPES
           box_sizes = Vagrant::Hivemind::Constants::BOX_SIZES
           cache_path = Path.cache_path
@@ -157,19 +157,36 @@ module Vagrant
           template = ERB.new template_string
           template_result = template.result(b)
 
-          tf = Tempfile.new("Hivemind_Vagrantfile", path)
-          tf.write template_result
-          tf.close
+          if generate
+            vagrant_file = File.new(Vagrant::Hivemind::Constants::VAGRANT_FILE, "w+")
+          else
+            vagrant_file = Tempfile.new("Hivemind_Vagrantfile", path)
+          end
+          vagrant_file.write template_result
+          vagrant_file.close
 
           env.define_singleton_method :vagrantfile_name= do |vfn| @vagrantfile_name = vfn end
           env.define_singleton_method :root_path= do |root_path| @root_path = root_path end
           env.define_singleton_method :local_data_path= do |local_data_path| @local_data_path = local_data_path end
-          env.vagrantfile_name = [File.basename(tf)]
+          env.vagrantfile_name = [File.basename(vagrant_file)]
           env.root_path = path
           env.local_data_path = local_data_path
 
           nil
         end
+
+        def self.exist?(path = Pathname.new(Dir.pwd))
+          File.exist? get_vagrant_file_from_path path
+        end
+
+        private
+          def self.get_vagrant_file_from_path(path)
+            if File.directory? path
+              File.join(path, Vagrant::Hivemind::Constants::VAGRANT_FILE)
+            else
+              path
+            end
+          end
       end
 
       class Ansible

data/lib/vagrant/hivemind/version.rb

@@ -1,5 +1,5 @@
 module Vagrant
   module Hivemind
-    VERSION = "0.1.5"
+    VERSION = "0.1.6"
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: vagrant-hivemind
 version: !ruby/object:Gem::Version
-  version: 0.1.5
+  version: 0.1.6
 platform: ruby
 authors:
 - Nap Ramirez
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2015-11-10 00:00:00.000000000 Z
+date: 2016-01-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -74,6 +74,7 @@ files:
 - keys/id_rsa.pub
 - lib/vagrant/hivemind.rb
 - lib/vagrant/hivemind/commands/desc.rb
+- lib/vagrant/hivemind/commands/generate.rb
 - lib/vagrant/hivemind/commands/halt.rb
 - lib/vagrant/hivemind/commands/init.rb
 - lib/vagrant/hivemind/commands/kill.rb