summaryrefslogtreecommitdiffstats
path: root/lib/puppet/util/rdoc/code_objects.rb
blob: 312fe0c9b260d86ee3d68adfc9a6bf37ba03b4ef (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
require 'rdoc/code_objects'

module RDoc

    # This modules contains various class that are used to hold information
    # about the various Puppet language structures we found while parsing.
    #
    # Those will be mapped to their html counterparts which are defined in
    # PuppetGenerator.

    # PuppetTopLevel is a top level (usually a .pp/.rb file)
    class PuppetTopLevel < TopLevel
        attr_accessor :module_name, :global

        # will contain all plugins
        @@all_plugins = {}

        # contains all cutoms facts
        @@all_facts = {}

        def initialize(toplevel)
            super(toplevel.file_relative_name)
        end

        def self.all_plugins
            @@all_plugins.values
        end

        def self.all_facts
            @@all_facts.values
        end
    end

    # PuppetModule holds a Puppet Module
    # This is mapped to an HTMLPuppetModule
    # it leverage the RDoc (ruby) module infrastructure
    class PuppetModule < NormalModule
        attr_accessor :facts, :plugins

        def initialize(name,superclass=nil)
            @facts = []
            @plugins = []
            super(name,superclass)
        end

        def initialize_classes_and_modules
            super
            @nodes = {}
        end

        def add_plugin(plugin)
            add_to(@plugins, plugin)
        end

        def add_fact(fact)
            add_to(@facts, fact)
        end

        def add_node(name,superclass)
            cls = @nodes[name]
            unless cls
                cls = PuppetNode.new(name, superclass)
                @nodes[name] = cls if !@done_documenting
                cls.parent = self
                cls.section = @current_section
            end
            cls
        end

        def each_fact
            @facts.each {|c| yield c}
        end

        def each_plugin
            @plugins.each {|c| yield c}
        end

        def each_node
            @nodes.each {|c| yield c}
        end

        def nodes
            @nodes.values
        end
    end

    # PuppetClass holds a puppet class
    # It is mapped to a HTMLPuppetClass for display
    # It leverages RDoc (ruby) Class
    class PuppetClass < ClassModule
        attr_accessor :resource_list

        def initialize(name, superclass)
            super(name,superclass)
            @resource_list = []
        end

        def add_resource(resource)
            add_to(@resource_list, resource)
        end

        def is_module?
            false
        end
    end

    # PuppetNode holds a puppet node
    # It is mapped to a HTMLPuppetNode for display
    # A node is just a variation of a class
    class PuppetNode < PuppetClass
        def initialize(name, superclass)
            super(name,superclass)
        end

        def is_module?
            false
        end
    end

    # Plugin holds a native puppet plugin (function,type...)
    # It is mapped to a HTMLPuppetPlugin for display
    class Plugin < Context
        attr_accessor :name, :type

        def initialize(name, type)
            super()
            @name = name
            @type = type
            @comment = ""
        end

        def <=>(other)
            @name <=> other.name
        end

        def full_name
            @name
        end

        def http_url(prefix)
            path = full_name.split("::")
            File.join(prefix, *path) + ".html"
        end

        def is_fact?
            false
        end

        def to_s
            res = self.class.name + ": " + @name + " (" + @type + ")\n"
            res << @comment.to_s
            res
        end
    end

    # Fact holds a custom fact
    # It is mapped to a HTMLPuppetPlugin for display
    class Fact < Context
        attr_accessor :name, :confine

        def initialize(name, confine)
            super()
            @name = name
            @confine = confine
            @comment = ""
        end

        def <=>(other)
            @name <=> other.name
        end

        def is_fact?
            true
        end

        def full_name
            @name
        end

        def to_s
            res = self.class.name + ": " + @name + "\n"
            res << @comment.to_s
            res
        end
    end

    # PuppetResource holds a puppet resource
    # It is mapped to a HTMLPuppetResource for display
    # A resource is defined by its "normal" form Type[title]
    class PuppetResource < CodeObject
        attr_accessor :type, :title, :params

        def initialize(type, title, comment, params)
            super()
            @type = type
            @title = title
            @comment = comment
            @params = params
        end

        def <=>(other)
            full_name <=> other.full_name
        end

        def full_name
            @type + "[" + @title + "]"
        end

        def name
            full_name
        end

        def to_s
            res = @type + "[" + @title + "]\n"
            res << @comment.to_s
            res
        end
    end
end