knife-essentials 0.1

data/lib/chef_fs/file_system.rb

@@ -0,0 +1,69 @@
+require 'chef_fs/path_utils'
+
+module ChefFS
+  module FileSystem
+    # Yields a list of all things under (and including) this entry that match the
+    # given pattern.
+    #
+    # ==== Attributes
+    #
+    # * +entry+ - Entry to start listing under
+    # * +pattern+ - ChefFS::FilePattern to match children under
+    #
+    def self.list(entry, pattern, &block)
+      # Include self in results if it matches
+      if pattern.match?(entry.path)
+        block.call(entry)
+      end
+
+      if entry.dir? && pattern.could_match_children?(entry.path)
+        # If it's possible that our children could match, descend in and add matches.
+        exact_child_name = pattern.exact_child_name_under(entry.path)
+
+        # If we've got an exact name, don't bother listing children; just grab the
+        # child with the given name.
+        if exact_child_name
+          exact_child = entry.child(exact_child_name)
+          if exact_child
+            list(exact_child, pattern, &block)
+          end
+
+        # Otherwise, go through all children and find any matches
+        else
+          entry.children.each do |child|
+            list(child, pattern, &block)
+          end
+        end
+      end
+    end
+
+    # Resolve the given path against the entry, returning
+    # the entry at the end of the path.
+    #
+    # ==== Attributes
+    # 
+    # * +entry+ - the entry to start looking under.  Relative
+    #   paths will be resolved from here.
+    # * +path+ - the path to resolve.  If it starts with +/+,
+    #   the path will be resolved starting from +entry.root+.
+    #
+    # ==== Examples
+    #
+    #     ChefFS::FileSystem.resolve_path(root_path, 'cookbooks/java/recipes/default.rb')
+    #
+    def self.resolve_path(entry, path)
+      return entry if path.length == 0
+      return resolve_path(entry.root, path) if path[0] == "/" && entry.root != entry
+      if path[0] == "/"
+        path = path[1,path.length-1]
+      end
+
+      result = entry
+      ChefFS::PathUtils::split(path).each do |part|
+        result = result.child(part)
+      end
+      result
+    end
+  end
+end
+

data/lib/chef_fs/file_system/base_fs_dir.rb

@@ -0,0 +1,23 @@
+require 'chef_fs/file_system/base_fs_object'
+require 'chef_fs/file_system/nonexistent_fs_object'
+
+module ChefFS
+  module FileSystem
+    class BaseFSDir < BaseFSObject
+      def initialize(name, parent)
+        super
+      end
+
+      def dir?
+        true
+      end
+
+      # Override child(name) to provide a child object by name without the network read
+      def child(name)
+        children.select { |child| child.name == name }.first || NonexistentFSObject.new(name, self)
+      end
+
+      # Abstract: children
+    end
+  end
+end

data/lib/chef_fs/file_system/base_fs_object.rb

@@ -0,0 +1,52 @@
+require 'chef_fs/path_utils'
+
+module ChefFS
+  module FileSystem
+    class BaseFSObject
+      def initialize(name, parent)
+        @parent = parent
+        @name = name
+        if parent
+          @path = ChefFS::PathUtils::join(parent.path, name)
+        else
+          if name != ''
+            raise ArgumentError, "Name of root object must be empty string: was '#{name}' instead"
+          end
+          @path = '/'
+        end
+      end
+
+      attr_reader :name
+      attr_reader :parent
+      attr_reader :path
+
+      def root
+        parent ? parent.root : self
+      end
+
+      def path_for_printing
+        if parent
+          ChefFS::PathUtils::join(parent.path_for_printing, name)
+        else
+          name
+        end
+      end
+
+      def dir?
+        false
+      end
+
+      def exists?
+        true
+      end
+
+      def content_type
+        :text
+      end
+
+      # Important directory attributes: name, parent, path, root
+      # Overridable attributes: dir?, child(name), path_for_printing
+      # Abstract: read, write, delete, children
+    end
+  end
+end

data/lib/chef_fs/file_system/chef_server_root_dir.rb

@@ -0,0 +1,48 @@
+require 'chef_fs/file_system/base_fs_dir'
+require 'chef_fs/file_system/rest_list_dir'
+require 'chef_fs/file_system/data_bags_dir'
+require 'chef_fs/file_system/cookbooks_dir'
+
+module ChefFS
+  module FileSystem
+    class ChefServerRootDir < BaseFSDir
+      def initialize(root_name, config)
+        super("", nil)
+        @chef_server_url = config[:chef_server_url]
+        @chef_username = config[:node_name]
+        @chef_private_key = config[:client_key]
+        @environment = config[:environment]
+        @root_name = root_name
+      end
+
+      attr_reader :chef_server_url
+      attr_reader :chef_username
+      attr_reader :chef_private_key
+      attr_reader :environment
+
+      def rest
+        Chef::REST.new(chef_server_url, chef_username, chef_private_key)
+      end
+
+      def api_path
+        ""
+      end
+
+      def path_for_printing
+        @root_name
+      end
+
+      def children
+        @children ||= [
+          RestListDir.new("clients", self),
+          CookbooksDir.new(self),
+          DataBagsDir.new(self),
+          RestListDir.new("environments", self),
+          RestListDir.new("nodes", self),
+          RestListDir.new("roles", self),
+  #        RestListDir.new("sandboxes", self)
+        ]
+      end
+    end
+  end
+end

data/lib/chef_fs/file_system/cookbook_dir.rb

@@ -0,0 +1,80 @@
+require 'chef_fs/file_system/rest_list_dir'
+require 'chef_fs/file_system/cookbook_subdir'
+require 'chef_fs/file_system/cookbook_file'
+require 'chef/cookbook_version'
+
+module ChefFS
+  module FileSystem
+    class CookbookDir < BaseFSDir
+      def initialize(name, parent, versions = nil)
+        super(name, parent)
+        @versions = versions
+      end
+
+      attr_reader :versions
+
+      def add_child(child)
+        @children << child
+      end
+
+      def api_path
+        "#{parent.api_path}/#{name}/_latest"
+      end
+
+      def child(name)
+        children.select { |child| child.name == name }.first || NonexistentFSObject.new(name, self)
+      end
+
+      def children
+        if @children.nil?
+          @children = []
+          Chef::CookbookVersion::COOKBOOK_SEGMENTS.each do |segment|
+            next unless manifest.has_key?(segment)
+            manifest[segment].each do |segment_file|
+              parts = segment_file['path'].split('/')
+              # Get or create the path to the file
+              container = self
+              parts[0,parts.length-1].each do |part|
+                old_container = container
+                container = old_container.children.select { |child| part == child.name }.first
+                if !container
+                  container = CookbookSubdir.new(part, old_container)
+                  old_container.add_child(container)
+                end
+              end
+              # Create the file itself
+              container.add_child(CookbookFile.new(parts[parts.length-1], container, segment_file))
+            end
+          end
+        end
+        @children
+      end
+
+      def exists?
+        if !@versions
+          child = parent.children.select { |child| child.name == name }.first
+          @versions = child.versions if child
+        end
+        !!@versions
+      end
+
+      def rest
+        parent.rest
+      end
+
+      private
+
+      def manifest
+        begin
+          @manifest ||= rest.get_rest(api_path).manifest
+        rescue Net::HTTPServerException
+          if $!.response.code == "404"
+            raise ChefFS::FileSystem::NotFoundException, $!
+          else
+            raise
+          end
+        end
+      end
+    end
+  end
+end

data/lib/chef_fs/file_system/cookbook_file.rb

@@ -0,0 +1,32 @@
+require 'chef_fs/file_system/base_fs_object'
+
+module ChefFS
+  module FileSystem
+    class CookbookFile < BaseFSObject
+      def initialize(name, parent, file)
+        super(name, parent)
+        @file = file
+      end
+
+      attr_reader :file
+
+      def checksum
+        file[:checksum]
+      end
+
+      def read
+        old_sign_on_redirect = rest.sign_on_redirect
+        rest.sign_on_redirect = false
+        begin
+          rest.get_rest(file['url'])
+        ensure
+          rest.sign_on_redirect = true
+        end
+      end
+
+      def rest
+        parent.rest
+      end
+    end
+  end
+end

data/lib/chef_fs/file_system/cookbook_subdir.rb

@@ -0,0 +1,25 @@
+require 'chef_fs/file_system/base_fs_dir'
+
+# TODO: take environment into account
+
+module ChefFS
+  module FileSystem
+    class CookbookSubdir < BaseFSDir
+      def initialize(name, parent)
+        super(name, parent)
+        @children = []
+      end
+
+      attr_reader :versions
+      attr_reader :children
+
+      def add_child(child)
+        @children << child
+      end
+
+      def rest
+        parent.rest
+      end
+    end
+  end
+end

data/lib/chef_fs/file_system/cookbooks_dir.rb

@@ -0,0 +1,21 @@
+require 'chef_fs/file_system/rest_list_dir'
+require 'chef_fs/file_system/cookbook_dir'
+
+module ChefFS
+  module FileSystem
+    class CookbooksDir < RestListDir
+      def initialize(parent)
+        super("cookbooks", parent)
+      end
+
+      def child(name)
+        result = @children.select { |child| child.name == name }.first if @children
+        result || CookbookDir.new(name, self)
+      end
+
+      def children
+        @children ||= rest.get_rest(api_path).map { |key, value| CookbookDir.new(key, self, value) }
+      end
+    end
+  end
+end

data/lib/chef_fs/file_system/data_bag_dir.rb

@@ -0,0 +1,22 @@
+require 'chef_fs/file_system/base_fs_dir'
+require 'chef_fs/file_system/rest_list_entry'
+
+# TODO: take environment into account
+
+module ChefFS
+  module FileSystem
+    class DataBagDir < RestListDir
+      def initialize(name, parent, exists = nil)
+        super(name, parent)
+        @exists = nil
+      end
+
+      def exists?
+        if @exists.nil?
+          @exists = parent.children.any? { |child| child.name == name }
+        end
+        @exists
+      end
+    end
+  end
+end

data/lib/chef_fs/file_system/data_bags_dir.rb

@@ -0,0 +1,23 @@
+require 'chef_fs/file_system/rest_list_dir'
+require 'chef_fs/file_system/data_bag_dir'
+
+# TODO: take environment into account
+
+module ChefFS
+  module FileSystem
+    class DataBagsDir < RestListDir
+      def initialize(parent)
+        super("data_bags", parent, "data")
+      end
+
+      def child(name)
+        result = @children.select { |child| child.name == name }.first if @children
+        result || DataBagDir.new(name, self)
+      end
+
+      def children
+        @children ||= rest.get_rest(api_path).keys.map { |entry| DataBagDir.new(entry, self, true) }
+      end
+    end
+  end
+end

data/lib/chef_fs/file_system/file_system_entry.rb

@@ -0,0 +1,36 @@
+require 'chef_fs/file_system/base_fs_dir'
+require 'chef_fs/file_system/rest_list_dir'
+require 'chef_fs/path_utils'
+
+module ChefFS
+  module FileSystem
+    class FileSystemEntry < BaseFSDir
+      def initialize(name, parent, file_path = nil)
+        super(name, parent)
+        @file_path = file_path || "#{parent.file_path}/#{name}"
+      end
+
+      attr_reader :file_path
+
+      def path_for_printing
+        ChefFS::PathUtils::relative_to(file_path, File.absolute_path(Dir.pwd))
+      end
+
+      def children
+        @children ||= Dir.entries(file_path).select { |entry| entry[0] != "." }.map { |entry| FileSystemEntry.new(entry, self) }
+      end
+
+      def dir?
+        File.directory?(file_path)
+      end
+
+      def read
+        begin
+          IO.read(file_path)
+        rescue IONotFoundException # TODO real exception
+          raise NotFoundException, $!
+        end
+      end
+    end
+  end
+end