Method: Puppetize::Controler#build

Defined in:
lib/puppetize/puppetize.rb

#build ⇒ Object 



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
 
# File 'lib/puppetize/puppetize.rb', line 90

def build

  ##################################
  # Generate a standard empty module

  cmd "cd /root && puppet module generate puppetize-module"
  FileUtils.mkdir File.join(@module_dir,'files')
  FileUtils.mkdir File.join(@module_dir,'templates')

  #################################
  # Generate the Package resources:

  rpm_list = cmd("rpm -qa").split("\n")
  init_rpm_list = File.read(@rpm_db).split("\n")
  puppet_rpm_list = rpm_list - init_rpm_list 

  packagelist = []

  puppet_rpm_list.each do |rpm|
    name = cmd("rpm --query --qf %{NAME} #{rpm}")
    version = cmd("rpm --query --qf %{VERSION} #{rpm}")
    release = cmd("rpm --query --qf %{RELEASE} #{rpm}")
    packagelist << [name,version,release]
  end

  # Generate install.pp based on ERB template
  
  template = File.join(File.dirname(__FILE__), '..', '..', 'tpl', 'install.pp.erb')
  renderer = ERB.new(File.read(template),nil,'>')
  output = renderer.result(binding)
  File.open(File.join(@module_dir,'manifests','install.pp'),'w') {|f| f.write(output)}


  ######################################
  # Generate the configuration resources

  puppet_files = Array.new

  @trackdirs.each do |track|

    Dir.chdir track[:path] do

      file_list = %x[git status --porcelain].split("\n").map {|f| f[3..-1]}

      flisttmp = []

      # If a file is a directory recurse inside as git does not provide that listing
      file_list.each do |file|
        if File.directory? file 
          Find.find(File.join(track[:path],file)) do |f|
            flisttmp << f.split(File.join(track[:path],''))[1] if File.file? f
          end
        end
      end

      # Merge both lists:
      file_list = (file_list + flisttmp).uniq

      # If the file does not belong to any package or has been modified then include it in the list

      file_list.each do |file|

        filename = File.join(track[:path],file)

        # If file does not belong to any package or it belongs but 
        # it has been modified then puppetize it

        if not system "rpm -qf #{filename} > /dev/null 2>&1" or system "rpm -Vf #{filename}|grep #{filename} > /dev/null 2>&1"
          puppet_files << filename 
        end

      end

    end

  end

  # Copy the files to the files directory:

  puppet_files.each do |file|

    if File.file? file
      FileUtils.cp file, File.join(@module_dir,'files')
      FileUtils.cp file, File.join(@module_dir,'templates',file.split('/').last + ".erb")
    end

  end

  puppet_file_list = Array.new

  puppet_files.each do |file|

    stat = File.stat file
    mode = stat.mode.to_s(8)[-3..-1]
    owner = Etc.getpwuid(stat.uid).name
    group = Etc.getgrgid(stat.gid).name
    type = File.file?(file) ? "present" : "directory"

    puppet_file_list << {
      :path    => file,
      :ensure  => type,
      :owner   => owner,
      :group   => group,
      :mode    => mode,
      :name    => file.split('/').last
    }

  end

  # Generate config.pp based on ERB template
  template = File.join(File.dirname(__FILE__), '..', '..', 'tpl', 'config.pp.erb')
  renderer = ERB.new(File.read(template),nil,'>')
  output = renderer.result(binding)
  File.open(File.join(@module_dir,'manifests','config.pp'),'w') {|f| f.write(output)}

end