summaryrefslogtreecommitdiffstats
path: root/lib/facter/util/plist/generator.rb
blob: 6c0796b3694be8651ae9d4947b4cab116ca0080a (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
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
#--###########################################################
# Copyright 2006, Ben Bleything <ben@bleything.net> and      #
# Patrick May <patrick@hexane.org>                           #
#                                                            #
# Distributed under the MIT license.                         #
##############################################################
#++
# See Plist::Emit.
module Plist
    # === Create a plist
    # You can dump an object to a plist in one of two ways:
    #
    # * <tt>Plist::Emit.dump(obj)</tt>
    # * <tt>obj.to_plist</tt>
    #   * This requires that you mixin the <tt>Plist::Emit</tt> module, which is already done for +Array+ and +Hash+.
    #
    # The following Ruby classes are converted into native plist types:
    #   Array, Bignum, Date, DateTime, Fixnum, Float, Hash, Integer, String, Symbol, Time, true, false
    # * +Array+ and +Hash+ are both recursive; their elements will be converted into plist nodes inside the <array> and <dict> containers (respectively).
    # * +IO+ (and its descendants) and +StringIO+ objects are read from and their contents placed in a <data> element.
    # * User classes may implement +to_plist_node+ to dictate how they should be serialized; otherwise the object will be passed to <tt>Marshal.dump</tt> and the result placed in a <data> element.
    #
    # For detailed usage instructions, refer to USAGE[link:files/docs/USAGE.html] and the methods documented below.
    module Emit
        # Helper method for injecting into classes.  Calls <tt>Plist::Emit.dump</tt> with +self+.
        def to_plist(envelope = true)
            return Plist::Emit.dump(self, envelope)
        end

        # Helper method for injecting into classes.  Calls <tt>Plist::Emit.save_plist</tt> with +self+.
        def save_plist(filename)
            Plist::Emit.save_plist(self, filename)
        end

        # The following Ruby classes are converted into native plist types:
        #   Array, Bignum, Date, DateTime, Fixnum, Float, Hash, Integer, String, Symbol, Time
        #
        # Write us (via RubyForge) if you think another class can be coerced safely into one of the expected plist classes.
        #
        # +IO+ and +StringIO+ objects are encoded and placed in <data> elements; other objects are <tt>Marshal.dump</tt>'ed unless they implement +to_plist_node+.
        #
        # The +envelope+ parameters dictates whether or not the resultant plist fragment is wrapped in the normal XML/plist header and footer.  Set it to false if you only want the fragment.
        def self.dump(obj, envelope = true)
            output = plist_node(obj)

            output = wrap(output) if envelope

            return output
        end

        # Writes the serialized object's plist to the specified filename.
        def self.save_plist(obj, filename)
            File.open(filename, 'wb') do |f|
                f.write(obj.to_plist)
            end
        end

        private
        def self.plist_node(element)
            output = ''

            if element.respond_to? :to_plist_node
                output << element.to_plist_node
            else
                case element
                when Array
                    if element.empty?
                        output << "<array/>\n"
                    else
                        output << tag('array') {
                            element.collect {|e| plist_node(e)}
                        }
                    end
                when Hash
                    if element.empty?
                        output << "<dict/>\n"
                    else
                        inner_tags = []

                        element.keys.sort.each do |k|
                            v = element[k]
                            inner_tags << tag('key', CGI::escapeHTML(k.to_s))
                            inner_tags << plist_node(v)
                        end

                        output << tag('dict') {
                            inner_tags
                        }
                    end
                when true, false
                    output << "<#{element}/>\n"
                when Time
                    output << tag('date', element.utc.strftime('%Y-%m-%dT%H:%M:%SZ'))
                when Date # also catches DateTime
                    output << tag('date', element.strftime('%Y-%m-%dT%H:%M:%SZ'))
                when String, Symbol, Fixnum, Bignum, Integer, Float
                    output << tag(element_type(element), CGI::escapeHTML(element.to_s))
                when IO, StringIO
                    element.rewind
                    contents = element.read
                    # note that apple plists are wrapped at a different length then
                    # what ruby's base64 wraps by default.
                    # I used #encode64 instead of #b64encode (which allows a length arg)
                    # because b64encode is b0rked and ignores the length arg.
                    data = "\n"
                    Base64::encode64(contents).gsub(/\s+/, '').scan(/.{1,68}/o) { data << $& << "\n" }
                    output << tag('data', data)
                else
                    output << comment( 'The <data> element below contains a Ruby object which has been serialized with Marshal.dump.' )
                    data = "\n"
                    Base64::encode64(Marshal.dump(element)).gsub(/\s+/, '').scan(/.{1,68}/o) { data << $& << "\n" }
                    output << tag('data', data )
                end
            end

            return output
        end

        def self.comment(content)
            return "<!-- #{content} -->\n"
        end

        def self.tag(type, contents = '', &block)
            out = nil

            if block_given?
                out = IndentedString.new
                out << "<#{type}>"
                out.raise_indent

                out << block.call

                out.lower_indent
                out << "</#{type}>"
            else
                out = "<#{type}>#{contents.to_s}</#{type}>\n"
            end

            return out.to_s
        end

        def self.wrap(contents)
            output = ''

            output << '<?xml version="1.0" encoding="UTF-8"?>' + "\n"
            output << '<!DOCTYPE plist PUBLIC "-//Apple Computer//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">' + "\n"
            output << '<plist version="1.0">' + "\n"

            output << contents

            output << '</plist>' + "\n"

            return output
        end

        def self.element_type(item)
            return case item
                   when String, Symbol;          'string'
                   when Fixnum, Bignum, Integer; 'integer'
                   when Float;                   'real'
                   else
                       raise "Don't know about this data type... something must be wrong!"
                   end
        end

        private

        class IndentedString #:nodoc:
            attr_accessor :indent_string

            @@indent_level = 0

            def initialize(str = "\t")
                @indent_string = str
                @contents = ''
            end

            def to_s
                return @contents
            end

            def raise_indent
                @@indent_level += 1
            end

            def lower_indent
                @@indent_level -= 1 if @@indent_level > 0
            end

            def <<(val)
                if val.is_a? Array
                    val.each do |f|
                        self << f
                    end
                else
                    # if it's already indented, don't bother indenting further
                    unless val =~ /\A#{@indent_string}/
                        indent = @indent_string * @@indent_level

                        @contents << val.gsub(/^/, indent)
                    else
                        @contents << val
                    end

                    # it already has a newline, don't add another
                    @contents << "\n" unless val =~ /\n$/
                end
            end
        end
    end
end

# we need to add this so sorting hash keys works properly
class Symbol #:nodoc:
    def <=> (other)
        self.to_s <=> other.to_s
    end
end

class Array #:nodoc:
    include Plist::Emit
end

class Hash #:nodoc:
    include Plist::Emit
end

# $Id: generator.rb 1781 2006-10-16 01:01:35Z luke $