summaryrefslogtreecommitdiffstats
path: root/lib/puppet/provider/mcx/mcxcontent.rb
blob: de60de15a684bdf190c91ee4750ab33a7b2c947b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
#--
# 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",
        :computerlist => "ComputerLists",
    }

    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
        raise MCXContentProviderException unless TypeMap.keys.include? ds_type.to_sym

        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