puppetize 0.0.4 → 0.1.0

data/lib/puppetize.rb

@@ -2,6 +2,7 @@ require 'fileutils'
 require 'erb'
 require 'pp'
 require 'open4'
+require 'find'
 require 'puppetize/puppetize'
 require 'puppetize/command_line'
 require 'puppetize/stdlib'

data/lib/puppetize/puppetize.rb

@@ -8,25 +8,7 @@ module Puppetize
       # Check distribution dependency
       puts "\nSorry, Distribution not supported, only redhat based at the moment \n\n" unless File.exist?('/etc/redhat-release') 
 
-      # Create config dir if it doesn't exists:
-      @config_dir = File.join(File.expand_path('~'),'.puppetize')
-      @module_dir = File.join(File.expand_path('~'),'puppetize-module')
-      @rpm_db = File.join(@config_dir,'rpmlist.db')
-      FileUtils.mkdir @config_dir if Dir.glob(@config_dir).empty?
-
-    end
-
-    def init
-
-      rpm_list = cmd "rpm -qa"
-      File.open(@rpm_db, 'w') {|f| f.write(rpm_list) }
-      
-    end
-
-    def build
-
       # Verify necessary commands are installed:
-      #
 
       commands = ['puppet','git']
 
@@ -40,26 +22,83 @@ module Puppetize
 
           if ! test.include? true 
             puts "ERROR - #{command} is not installed or not available on default path"
-            exit -1
+            exit 1
           end
 
         rescue => e
 
           STDERR.puts "ERROR - #{e}"
-          exit -1
+          exit 1
+
+        end
+
+      end
+
+      # Create config dir if it doesn't exists:
+      @config_dir = File.join(File.expand_path('~'),'.puppetize')
+      @module_dir = File.join(File.expand_path('~'),'puppetize-module')
+      @rpm_db = File.join(@config_dir,'rpmlist.db')
+      @trackdirs = [
+        { 
+        :path => '/etc',
+        :ignoredir => [
+          'group',
+          'gshadow',
+          'ld.so.cache',
+          'passwd',
+          'shadow',
+          'group-',
+          'gshadow-',
+          'passwd-',
+          'shadow-',
+          'rc.d' 
+          ]
+        }
+      ]
+
+      FileUtils.mkdir @config_dir if Dir.glob(@config_dir).empty?
+
+    end
+
+    def init
+
+      # Remember what RPMs are installed
+
+      rpm_list = cmd "rpm -qa"
+      File.open(@rpm_db, 'w') {|f| f.write(rpm_list) }
+
+      # Add a git repo to track the filesystem
+
+      @trackdirs.each do |track|
+
+        File.open(File.join(track[:path],".gitignore"),'w') do |f|
+          f.write(track[:ignoredir].join("\n"))
+        end
+
+        Dir.chdir track[:path] do 
+
+          cmd 'git init .'
+          cmd 'git add .'
+          cmd "git commit -am'puppetize: Initial Commit'"
 
         end
 
       end
 
+    end
+
+    def build
+
+      ##################################
       # Generate a standard empty module
 
-      cmd "puppet module generate puppetize-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 
@@ -74,19 +113,105 @@ module Puppetize
       end
 
       # Generate install.pp based on ERB template
+      
       template = File.join(File.dirname(__FILE__), '..', '..', 'tpl', 'install.pp.erb')
-      renderer = ERB.new(File.read(template))
+      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
 
     def reset
 
-      puts "reset"
+      @trackdirs.each do |track|
+        FileUtils.rm_rf File.join(track[:path],'.git') if File.exists? File.join(track[:path],'.git')
+      end
+
+      FileUtils.rm @rpm_db if File.exists? @rpm_db
+      FileUtils.rm_rf 'puppetize-module' if File.exists? 'puppetize-module'
 
     end
 

data/lib/puppetize/version.rb

@@ -1,3 +1,3 @@
 module Puppetize
-  VERSION = "0.0.4"
+  VERSION = "0.1.0"
 end

data/tpl/config.pp.erb

@@ -0,0 +1,17 @@
+
+class module::config {
+
+<% for file in puppet_file_list %>
+  file {'<%= file[:path] %>':
+    ensure  => <%= file[:ensure] %>,
+    owner   => '<%= file[:owner] %>',
+    group   => '<%= file[:group] %>',
+    mode    => '<%= file[:mode] %>',
+<% if file[:ensure] == 'present' %>
+    content => template("${module_name}/<%= file[:name] %>.erb")
+    source  => "puppet:///modules/${module_name}/<%= file[:name] %>"
+<% end %>
+  }
+<% end %>
+
+}

data/tpl/install.pp.erb

@@ -0,0 +1,10 @@
+
+class module::install {
+
+<% for pkg in packagelist %>
+  package {'<%= pkg[0] %>':
+    ensure => '<%= pkg[1] %>-<%= pkg[2] %>',
+  }
+<% end %>
+
+}

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: puppetize
 version: !ruby/object:Gem::Version 
-  hash: 23
+  hash: 27
   prerelease: false
   segments: 
   - 0
+  - 1
   - 0
-  - 4
-  version: 0.0.4
+  version: 0.1.0
 platform: ruby
 authors: 
 - Juan Breinlinger
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2013-11-04 00:00:00 +00:00
+date: 2013-11-05 00:00:00 +00:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -33,6 +33,21 @@ dependencies:
         version: "2.0"
   type: :runtime
   version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: open4
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 15
+        segments: 
+        - 1
+        - 0
+        version: "1.0"
+  type: :runtime
+  version_requirements: *id002
 description: Puppetize is a tool that can generate complete modules from systems
 email: juan.brein@breins.net
 executables: 
@@ -48,6 +63,8 @@ files:
 - lib/puppetize/stdlib.rb
 - lib/puppetize/command_line.rb
 - bin/puppetize
+- tpl/config.pp.erb
+- tpl/install.pp.erb
 - README.md
 - LICENSE
 - CONTRIBUTORS