Finally! We now have mount support in OS X. Ouch.

git-svn-id: https://reductivelabs.com/svn/puppet/trunk@1062 980ebf18-57e1-0310-9a29-db15c13687c0

2 files changed, 85 insertions, 11 deletions

diff --git a/lib/puppet/filetype.rb b/lib/puppet/filetype.rb
index c13c740f0..9054fddbe 100755
--- a/lib/puppet/filetype.rb
+++ b/lib/puppet/filetype.rb
@@ -208,19 +208,32 @@ module Puppet
 
         # Treat netinfo tables as a single file, just for simplicity of certain types
         newfiletype(:netinfo) do
+            class << self
+                attr_accessor :format
+            end
             def read
                 %x{nidump -r /#{@path} /}
             end
 
             # This really only makes sense for cron tabs.
             def remove
-                raise TypeError, "Cannot remove netinfo tables"
+                %x{nireport / /#{@path} name}.split("\n").each do |name|
+                    newname = name.gsub(/\//, '\/').sub(/\s+$/, '')
+                    output = %x{niutil -destroy / '/#{@path}/#{newname}'}
+
+                    unless $? == 0
+                        raise Puppet::Error, "Could not remove %s from %s" %
+                            [name, @path]
+                    end
+                end
             end
 
             # Convert our table to an array of hashes.  This only works for handling one
             # table at a time.
             def to_array(text = nil)
-                text ||= read
+                unless text
+                    text = read
+                end
 
                 hash = nil
 
@@ -260,9 +273,22 @@ module Puppet
             end
 
             def write(text)
-                IO.popen("niload -d #{@path} .", "w") { |p|
+                text.gsub!(/^#.*\n/,'')
+                text.gsub!(/^$/,'')
+                if text == "" or text == "\n"
+                    self.remove
+                    return
+                end
+                unless format = self.class.format
+                    raise Puppe::DevError, "You must define the NetInfo format to inport"
+                end
+                IO.popen("niload -d #{format} . 1>/dev/null 2>/dev/null", "w") { |p|
                     p.print text
                 }
+
+                unless $? == 0
+                    raise ArgumentError, "Failed to write %s" % @path
+                end
             end
         end
     end
diff --git a/lib/puppet/type/parsedtype/mount.rb b/lib/puppet/type/parsedtype/mount.rb
index 08f797f72..5311de1fb 100755
--- a/lib/puppet/type/parsedtype/mount.rb
+++ b/lib/puppet/type/parsedtype/mount.rb
@@ -95,17 +95,36 @@ module Puppet
 
         @instances = []
 
-        case Facter["operatingsystem"].value
+        @platform = Facter["operatingsystem"].value
+        case @platform
         when "Solaris":
             @path = "/etc/vfstab"
             @fields = [:device, :blockdevice, :path, :fstype, :pass, :atboot,
                 :options]
+        when "Darwin":
+            @filetype = Puppet::FileType.filetype(:netinfo)
+            @filetype.format = "fstab"
+            @path = "mounts"
+            @fields = [:device, :path, :fstype, :options, :dump, :pass]
+
+            # How to map the dumped table to what we want
+            @fieldnames = {
+                "name" => :device,
+                "dir" => :path,
+                "dump_freq" => :dump,
+                "passno" => :pass,
+                "vfstype" => :fstype,
+                "opts" => :options
+            }
         else
             @path = "/etc/fstab"
             @fields = [:device, :path, :fstype, :options, :dump, :pass]
         end
 
-        @filetype = Puppet::FileType.filetype(:flat)
+        # Allow Darwin to override the default filetype
+        unless defined? @filetype
+            @filetype = Puppet::FileType.filetype(:flat)
+        end
 
         # Parse a mount tab.
         #
@@ -113,6 +132,11 @@ module Puppet
         # mounts in order, mostly so that comments are retained in the
         # order they were written and in proximity to the same fses.
         def self.parse(text)
+            # provide a single exception for darwin & netinfo
+            if @filetype == Puppet::FileType.filetype(:netinfo)
+                parseninfo(text)
+                return 
+            end
             count = 0
             hash = {}
             text.chomp.split("\n").each { |line|
@@ -138,6 +162,30 @@ module Puppet
             }
         end
 
+        # Parse a netinfo table.
+        def self.parseninfo(text)
+            array = @fileobj.to_array(text)
+
+            hash = {}
+            array.each do |record|
+                @fieldnames.each do |name, field|
+                    if value = record[name]
+                        if field == :options
+                            hash[field] = value.join(",")
+                        else
+                            hash[field] = value[0]
+                        end
+                    else
+                        raise ArgumentError, "Field %s was not provided" % [name]
+                    end
+                end
+
+
+                hash2obj(hash)
+                hash.clear
+            end
+        end
+
         # This only works when the mount point is synced to the fstab.
         def mount
             output = %x{mount #{self[:path]} 2>&1}
@@ -158,9 +206,14 @@ module Puppet
 
         # Is the mount currently mounted?
         def mounted?
+            platform = Facter["operatingsystem"].value
             %x{df}.split("\n").find do |line|
                 fs = line.split(/\s+/)[-1]
-                fs == self[:path]
+                if platform == "Darwin"
+                    fs == "/private/var/automount" + self[:path]
+                else
+                    fs == self[:path]
+                end
             end
         end
 
@@ -170,11 +223,6 @@ module Puppet
                 if value = self.value(field)
                     value
                 else
-                    if @states.include? field
-                        self.warning @states[field].inspect
-                    else
-                        self.warning field.inspect
-                    end
                     raise Puppet::Error,
                         "Could not retrieve value for %s" % field
                 end