puppet 0.9.2

data/lib/puppet/type/nameservice.rb

@@ -0,0 +1,3 @@
+require 'puppet/type/nameservice/posix'
+require 'puppet/type/nameservice/netinfo'
+require 'puppet/type/nameservice/objectadd'

data/lib/puppet/type/nameservice/netinfo.rb

@@ -0,0 +1,173 @@
+# Manage NetInfo POSIX objects.  Probably only used on OS X, but I suppose
+# it could be used elsewhere.
+
+require 'puppet'
+require 'puppet/type/nameservice/posix'
+
+module Puppet
+    class Type
+        # Return the NetInfo directory in which a given object type is stored.
+        # Defaults to the type's name if @netinfodir is unset.
+    end
+
+    module NameService
+        module NetInfo
+            def self.test
+                system("which niutil > /dev/null 2>&1")
+
+                if $? == 0
+                    return true
+                else
+                    Puppet.err "Could not find niutil"
+                    return false
+                end
+
+                system("which nireport > /dev/null 2>&1")
+
+                if $? == 0
+                    return true
+                else
+                    Puppet.err "Could not find nireport"
+                    return false
+                end
+            end
+
+            def self.exists?(obj)
+                cmd = "nidump -r /%s/%s /" %
+                    [obj.class.netinfodir, obj.name]
+
+                output = %x{#{cmd} 2>/dev/null}
+                if output == ""
+                    return false
+                else
+                    #Puppet.debug "%s exists: %s" % [obj.name, output]
+                    return true
+                end
+            end
+
+            def self.flush
+                output = %x{lookupd -flushcache 2>&1}
+
+                if $? != 0
+                    Puppet.err "Could not flush lookupd cache: %s" % output
+                end
+            end
+
+            class NetInfoState < POSIX::POSIXState
+                def self.allatonce?
+                    false
+                end
+
+                def self.netinfokey
+                    if defined? @netinfokey and @netinfokey
+                        return @netinfokey
+                    else
+                        return self.name
+                    end
+                end
+
+                def retrieve
+                    NetInfo.flush
+                    dir = @parent.class.netinfodir
+                    cmd = ["nireport", "/", "/%s" % dir, "name"]
+
+                    if key = self.class.netinfokey
+                        cmd << key.to_s
+                    else
+                        raise Puppet::DevError,
+                            "Could not find netinfokey for state %s" %
+                            self.class.name
+                    end
+                    self.debug "Executing %s" % cmd.join(" ").inspect
+
+                    output = %x{#{cmd.join(" ")} 2>&1}.split("\n").each { |line|
+                        if line =~ /^(\w+)\s+(.+)$/
+                            name = $1
+                            value = $2.sub(/\s+$/, '')
+
+                            if name == @parent.name
+                                if value =~ /^[-0-9]+$/
+                                    @is = Integer(value)
+                                else
+                                    @is = value
+                                end
+                            end
+                        else
+                            raise Puppet::DevError, "Could not match %s" % line
+                        end
+                    }
+
+                    unless defined? @is
+                        @is = :notfound
+                    end
+                end
+
+                def addcmd
+                    creatorcmd("-create")
+                end
+
+                def creatorcmd(arg)
+                    cmd = ["niutil"]
+                    cmd << arg
+
+                    cmd << "/" << "/%s/%s" %
+                        [@parent.class.netinfodir, @parent.name]
+
+                    #if arg == "-create"
+                    #    return [cmd.join(" "), self.modifycmd].join(";")
+                    #else
+                        return cmd.join(" ")
+                    #end
+                end
+
+                def deletecmd
+                    creatorcmd("-destroy")
+                end
+
+                def modifycmd
+                    cmd = ["niutil"]
+
+                    cmd << "-createprop" << "/" << "/%s/%s" %
+                        [@parent.class.netinfodir, @parent.name]
+
+                    if key = self.class.netinfokey
+                        cmd << key << "'%s'" % self.should
+                    else
+                        raise Puppet::DevError,
+                            "Could not find netinfokey for state %s" %
+                            self.class.name
+                    end
+                    cmd.join(" ")
+                end
+            end
+
+            class GroupGID < NetInfoState
+            end
+
+            class UserUID       < NetInfoState
+            end
+
+            class UserGID       < NetInfoState
+            end
+
+            class UserComment   < NetInfoState
+                @netinfokey = "realname"
+            end
+
+            class UserHome      < NetInfoState
+            end
+
+            class UserShell     < NetInfoState
+            end
+
+            class UserLocked    < NetInfoState
+            end
+
+            class UserExpire    < NetInfoState
+            end
+
+            class UserInactive  < NetInfoState
+            end
+        end
+    end
+end

data/lib/puppet/type/nameservice/objectadd.rb

@@ -0,0 +1,146 @@
+require 'puppet'
+
+module Puppet
+    class State
+        def self.objectaddflag
+            if defined? @objectaddflag and @objectaddflag
+                return @objectaddflag
+            else
+                return @name
+            end
+        end
+    end
+
+    module NameService
+        module ObjectAdd
+            # Verify that we've got the commands necessary to manage flat files.
+            def self.test
+                system("which groupadd > /dev/null 2>&1")
+
+                if $? == 0
+                    return true
+                else
+                    Puppet.err "Could not find groupadd"
+                    return false
+                end
+            end
+
+            def self.exists?(obj)
+                if obj.getinfo
+                    return true
+                else
+                    return false
+                end
+            end
+
+            class ObjectAddGroup < POSIX::POSIXState
+                def self.allatonce?
+                    true
+                end
+
+                def addcmd
+                    cmd = ["groupadd"]
+                    if gid = @parent.should(:gid)
+                        unless gid == :auto
+                            cmd << "-g" << gid 
+                        end
+                    end
+                    cmd << @parent.name
+
+                    return cmd.join(" ")
+                end
+
+                def deletecmd
+                    "groupdel %s" % @parent.name
+                end
+
+                def modifycmd
+                    [
+                        "groupmod",
+                        self.class.objectaddflag,
+                        "'%s'" % self.should,
+                        @parent.name
+                    ].join(" ")
+                end
+            end
+
+            class GroupGID       < ObjectAddGroup
+                @objectaddflag = "-g"
+                @autogen = true
+            end
+
+            class ObjectAddUser < POSIX::POSIXState
+                def self.allatonce?
+                    true
+                end
+
+                def addcmd
+                    cmd = ["useradd"]
+                    @parent.eachstate { |state|
+                        # the value needs to be quoted, mostly because -c might
+                        # have spaces in it
+                        cmd << state.class.objectaddflag << "'%s'" % state.should
+                    }
+                    # stupid fedora
+                    case Facter["distro"].value
+                    when "Fedora", "RedHat":
+                        cmd << "-M"
+                    else
+                    end
+                    cmd << @parent.name
+
+                    cmd.join(" ")
+                end
+                
+                def deletecmd
+                    ["userdel", @parent.name].join(" ")
+                end
+
+                def modifycmd
+                cmd = [
+                    "usermod",
+                    self.class.objectaddflag,
+                    "'%s'" % self.should,
+                    @parent.name
+                ].join(" ")
+                end
+            end
+            class UserUID       < ObjectAddUser
+                @objectaddflag = "-u"
+                @autogen = true
+            end
+
+            class UserGID       < ObjectAddUser
+                @objectaddflag = "-g"
+                @autogen = true
+            end
+
+            class UserComment   < ObjectAddUser
+                @objectaddflag = "-c"
+            end
+
+            class UserHome      < ObjectAddUser
+                @objectaddflag = "-d"
+                @autogen = true
+            end
+
+            class UserShell     < ObjectAddUser
+                @objectaddflag = "-s"
+                @autogen = true
+            end
+
+            class UserLocked    < ObjectAddUser
+            end
+
+            class UserExpire    < ObjectAddUser
+                @objectaddflag = "-e"
+                @autogen = true
+            end
+
+            class UserInactive  < ObjectAddUser
+                @objectaddflag = "-f"
+                @autogen = true
+            end
+        end
+    end
+end

data/lib/puppet/type/nameservice/posix.rb

@@ -0,0 +1,200 @@
+require 'puppet'
+
+module Puppet
+    class Type
+        def self.posixmethod
+            if defined? @posixmethod and @posixmethod
+                return @posixmethod
+            else
+                return @name
+            end
+        end
+    end
+
+    module NameService
+        # This is the base module for basically all of the NSS stuff.  It
+        # should be able to retrieve the info for almost any system, but
+        # it can't create any information on its own.  You need to define
+        # a subclass of these classes to actually modify the system.
+        module POSIX
+            class POSIXState < Puppet::State
+                class << self
+                    attr_accessor :extender
+
+                    def autogen?
+                        if defined? @autogen
+                            return @autogen
+                        else
+                            return false
+                        end
+                    end
+
+                    def optional?
+                        if defined? @optional
+                            return @optional
+                        else
+                            return false
+                        end
+                    end
+                end
+
+                def self.doc
+                    if defined? @extender
+                        @extender.doc
+                    else
+                        nil
+                    end
+                end
+
+                def self.complete
+                    mod = "Puppet::State::%s" %
+                        self.to_s.sub(/.+::/,'')
+                    begin
+                        modklass = eval(mod)
+                    rescue NameError
+                        raise Puppet::Error,
+                            "Could not find extender module %s for %s" %
+                                [mod, self.to_s]
+                    end
+                    include modklass
+
+                    self.extender = modklass
+                end
+
+                def self.posixmethod
+                    if defined? @extender
+                        if @extender.respond_to?(:posixmethod)
+                            return @extender.posixmethod
+                        else
+                            return @extender.name
+                        end
+                    else
+                        raise Puppet::DevError,
+                            "%s could not retrieve posixmethod" % self
+                    end
+                end
+
+                def self.name
+                    @extender.name
+                end
+
+                # we use the POSIX interfaces to retrieve all information,
+                # so we don't have to worry about abstracting that across
+                # the system
+                def retrieve
+                    if obj = @parent.getinfo(true)
+
+                        if method = self.class.posixmethod || self.class.name
+                            @is = obj.send(method)
+                        else
+                            raise Puppet::DevError,
+                                "%s has no posixmethod" % self.class
+                        end
+                    else
+                        @is = :notfound
+                    end
+                end
+
+                def sync
+                    event = nil
+                    # they're in sync some other way
+                    if self.insync?
+                        return nil
+                    end
+                    if @is == :notfound
+                        self.retrieve
+                        if self.insync?
+                            return nil
+                        end
+                    end
+                    # if the object needs to be created or deleted,
+                    # depend on another method to do it all at once
+                    if @is == :notfound or self.should == :notfound
+                        event = syncname()
+
+                        return event
+                        # if the whole object is created at once, just return
+                        # an event saying so
+                        #if self.class.allatonce?
+                        #    return event
+                        #end
+                    end
+
+                    unless @parent.exists?
+                        raise Puppet::DevError,
+                            "%s %s does not exist; cannot set %s" %
+                            [@parent.class.name, @parent.name, self.class.name]
+                    end
+
+                    # this needs to be set either by the individual state
+                    # or its parent class
+                    cmd = self.modifycmd
+
+                    self.debug "Executing %s" % cmd.inspect
+
+                    output = %x{#{cmd} 2>&1}
+
+
+                    unless $? == 0
+                        raise Puppet::Error, "Could not modify %s on %s %s: %s" %
+                            [self.class.name, @parent.class.name,
+                                @parent.name, output]
+                    end
+
+                    if event
+                        return event
+                    else
+                        return "#{@parent.class.name}_modified".intern
+                    end
+                end
+
+                private
+                def syncname
+                    cmd = nil
+                    event = nil
+                    if self.should == :notfound
+                        # we need to remove the object...
+                        unless @parent.exists?
+                            # the group already doesn't exist
+                            return nil
+                        end
+
+                        # again, needs to be set by the ind. state or its
+                        # parent
+                        cmd = self.deletecmd
+                        type = "delete"
+                    else
+                        if @parent.exists?
+                            return nil
+                        end
+
+                        # blah blah, define elsewhere, blah blah
+                        cmd = self.addcmd
+                        type = "create"
+                    end
+                    self.debug "Executing %s" % cmd.inspect
+
+                    output = %x{#{cmd} 2>&1}
+
+                    unless $? == 0
+                        raise Puppet::Error, "Could not %s %s %s: %s" %
+                            [type, @parent.class.name, @parent.name, output]
+                    end
+
+                    # we want object creation to show up as one event, 
+                    # not many
+                    unless self.class.allatonce?
+                        if type == "create"
+                            @parent.eachstate { |state|
+                                state.sync
+                                state.retrieve
+                            }
+                        end
+                    end
+
+                    return "#{@parent.class.name}_#{type}d".intern
+                end
+            end
+        end
+    end
+end