type/mcx.rb Feature #1026 - MCX Type

Added new MCX type and base test.

This type manages MCX settings on DirectoryService nodes.  These
settings take the form of plist XML documents attached to Users,
Groups, and Computers in DirectoryService.

1 files changed, 199 insertions, 0 deletions

diff --git a/lib/puppet/provider/mcx/mcxcontent.rb b/lib/puppet/provider/mcx/mcxcontent.rb
new file mode 100644
index 000000000..fdcc8cc5d
--- /dev/null
+++ b/lib/puppet/provider/mcx/mcxcontent.rb
@@ -0,0 +1,199 @@
+#--
+# Copyright (C) 2008 Jeffrey J McCune.
+
+# This program and entire repository is free software; you can
+# redistribute it and/or modify it under the terms of the GNU 
+# General Public License as published by the Free Software 
+# Foundation; either version 2 of the License, or any later version.
+
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+
+# Author: Jeff McCune <mccune.jeff@gmail.com>
+
+require 'tempfile'
+
+Puppet::Type.type(:mcx).provide :mcxcontent, :parent => Puppet::Provider do
+
+    desc "MCX Settings management using DirectoryService on OS X.
+
+This provider manages the entire MCXSettings attribute available
+to some directory services nodes.  This management is 'all or nothing'
+in that discrete application domain key value pairs are not managed
+by this provider.
+
+It is recommended to use WorkGroup Manager to configure Users, Groups,
+Computers, or ComputerLists, then use 'ralsh mcx' to generate a puppet
+manifest from the resulting configuration.
+
+Original Author: Jeff McCune (mccune.jeff@gmail.com)"
+
+    # This provides a mapping of puppet types to DirectoryService
+    # type strings.
+    TypeMap = {
+        :user => "Users",
+        :group => "Groups",
+        :computer => "Computers",
+        :computergroup => "ComputerGroups",
+    }
+
+    class MCXContentProviderException < Exception
+
+    end
+
+    commands :dscl => "/usr/bin/dscl"
+    confine :operatingsystem => :darwin
+    defaultfor :operatingsystem => :darwin
+
+    # self.instances is all important.
+    # This is the only class method, it returns
+    # an array of instances of this class.
+    def self.instances
+        mcx_list = []
+        for ds_type in TypeMap.keys
+            ds_path = "/Local/Default/#{TypeMap[ds_type]}"
+            output = dscl 'localhost', '-list', ds_path
+            member_list = output.split
+            for ds_name in member_list
+                content = mcxexport(ds_type, ds_name)
+                if content.empty?
+                    Puppet.debug "/#{TypeMap[ds_type]}/#{ds_name} has no MCX data."
+                else 
+                    # This node has MCX data.
+                    rsrc = self.new(:name => "/#{TypeMap[ds_type]}/#{ds_name}",
+                                 :ds_type => ds_type,
+                                 :ds_name => ds_name,
+                                 :content => content)
+                    mcx_list << rsrc
+                end
+            end
+        end
+        return mcx_list
+    end
+
+    private
+
+    # mcxexport is used by instances, and therefore
+    # a class method.
+    def self.mcxexport(ds_type, ds_name)
+        ds_t = TypeMap[ds_type]
+        ds_n = ds_name.to_s
+        ds_path = "/Local/Default/#{ds_t}/#{ds_n}"
+
+        dscl 'localhost', '-mcxexport', ds_path
+    end
+
+    def mcximport(ds_type, ds_name, val)
+        ds_t = TypeMap[ds_type]
+        ds_n = ds_name.to_s
+        ds_path = "/Local/Default/#{ds_t}/#{ds_name}"
+
+        tmp = Tempfile.new('puppet_mcx')
+        begin
+            tmp << val
+            tmp.flush
+            dscl 'localhost', '-mcximport', ds_path, tmp.path
+        ensure
+            tmp.close
+            tmp.unlink
+        end
+    end
+
+    # Given the resource name string, parse ds_type out.
+    def parse_type(name)
+        tmp = name.split('/')[1]
+        if ! tmp.is_a? String
+            raise MCXContentProviderException,
+            "Coult not parse ds_type from resource name '#{name}'.  Specify with ds_type parameter."
+        end
+        # De-pluralize and downcase.
+        tmp = tmp.chop.downcase.to_sym
+        if not TypeMap.keys.member? tmp
+            raise MCXContentProviderException,
+            "Coult not parse ds_type from resource name '#{name}'.  Specify with ds_type parameter."
+        end
+        return tmp
+    end
+
+    # Given the resource name string, parse ds_name out.
+    def parse_name(name)
+        ds_name = name.split('/')[2]
+        if ! ds_name.is_a? String
+            raise MCXContentProviderException,
+            "Could not parse ds_name from resource name '#{name}'.  Specify with ds_name parameter."
+        end
+        return ds_name
+    end
+
+    # Gather ds_type and ds_name from resource or
+    # parse it out of the name.
+    # This is a private instance method, not a class method.
+    def get_dsparams
+        ds_type = resource[:ds_type]
+        if ds_type.nil?
+            ds_type = parse_type(resource[:name])
+        end
+
+        ds_name = resource[:ds_name]
+        if ds_name.nil?
+            ds_name = parse_name(resource[:name])
+        end
+
+        rval = {
+            :ds_type => ds_type.to_sym,
+            :ds_name => ds_name,
+        }
+
+        return rval
+
+    end
+
+    public
+
+    def create
+        self.content=(resource[:content])
+    end
+
+    def destroy
+        ds_parms = get_dsparams
+        ds_t = TypeMap[ds_parms[:ds_type]]
+        ds_n = ds_parms[:ds_name].to_s
+        ds_path = "/Local/Default/#{ds_t}/#{ds_n}"
+
+        dscl 'localhost', '-mcxdelete', ds_path
+    end
+
+    def exists?
+        # JJM Just re-use the content method and see if it's empty.
+        begin
+            mcx = content
+        rescue Puppet::ExecutionFailure => e
+            return false
+        end
+        has_mcx = ! mcx.empty?
+        return has_mcx
+    end
+
+    def content
+        ds_parms = get_dsparams
+        mcx = self.class.mcxexport(ds_parms[:ds_type],
+                                   ds_parms[:ds_name])
+        return mcx
+    end
+
+    def content=(value)
+        # dscl localhost -mcximport
+        ds_parms = get_dsparams
+        mcx = mcximport(ds_parms[:ds_type],
+                        ds_parms[:ds_name],
+                        resource[:content])
+        return mcx
+    end
+
+end