lib/vagabond/vagabondfile.rb in vagabond-0.2.0 vs lib/vagabond/vagabondfile.rb in vagabond-0.2.2

@@ -4,30 +4,34 @@
   class Vagabondfile
 
     attr_reader :path
     attr_reader :config
 
+    DEFAULT_KEYS = %w(boxes mappings test_mappings spec_mappings local_chef_server)
     ALIASES = Mash.new(:boxes => :nodes, :nodes => :boxes)
     
     def initialize(path=nil, *args)
       path = discover_path(Dir.pwd) unless path
       @path = path
       load_configuration!(args.include?(:allow_missing))
     end
 
     def [](k)
+      if(DEFAULT_KEYS.include?(k.to_s))
+        @config[k] ||= Mash.new
+      end
       aliased(k) || @config[k]
     end
 
     def aliased(k)
       if(ALIASES.has_key?(k))
         v = [@config[k], @config[ALIASES[k]]].compact
         if(v.size > 1)
-          case v.first.class
+          case v.first
           when Array
             m = :|
-          when Hash
+          when Hash, Mash
             m = :merge
           else
             m = :+
           end
           v.inject(&m)
@@ -35,25 +39,56 @@
           v.first
         end
       end
     end
     
-    def load_configuration!(no_raise = false)
+    def load_configuration!(*args)
+      unless(args.empty?)
+        no_raise = args.first == true
+        force_store = args.include?(:force_store_path)
+        no_raise ||= force_store
+      end
       if(@path && File.exists?(@path))
         @config = Mash.new(self.instance_eval(IO.read(@path), @path, 1))
-      else
+      end
+      if(!@config || force_store)
         raise 'No Vagabondfile file found!' unless no_raise
-        @config = Mash.new
+        generate_store_path
+        @config = Mash[*DEFAULT_KEYS.map{|k| [k, Mash.new]}.flatten]
       end
     end
 
+    def generate_store_path
+      @path ||= File.expand_path(File.join(Dir.pwd, 'Vagabondfile'))
+      @store_path = File.join('/tmp/vagabond-solos', directory.gsub(%r{[^0-9a-zA-Z]}, '-'), 'Vagabondfile')
+      @store_path = File.expand_path(@store_path.gsub('-', '/'))
+      FileUtils.mkdir_p(File.dirname(@store_path))
+      File.dirname(@store_path)
+    end
+
+    def store_path
+      @store_path || @path
+    end
+
+    def directory
+      File.dirname(@path)
+    end
+
+    def store_directory
+      File.dirname(@store_path || @path)
+    end
+    
     def discover_path(path)
       d_path = Dir.glob(File.join(path, 'Vagabondfile')).first
       unless(d_path)
         cut_path = path.split(File::SEPARATOR)
         cut_path.pop
         d_path = discover_path(cut_path.join(File::SEPARATOR)) unless cut_path.empty?
       end
       d_path
+    end
+
+    def local_chef_server?
+      self[:local_chef_server] && self[:local_chef_server][:enabled]
     end
   end
 end