hiera-template 1.0.3 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 306f24d5d884749ca90a11cf54dbb8e2c5fbbe30
-  data.tar.gz: f6f3be67ede23520fb60ad214ce9a645c1997d71
+  metadata.gz: ba9af3b56b0fd2b81b46cb4e08214125bfd985ed
+  data.tar.gz: bd55eb318d26f6fd05a0b57fad7024a89292ca52
 SHA512:
-  metadata.gz: b5b99dff21fa867c1b180efb8b1860eedbe4bcf7e24323a079083c0f4d159825b4dd32e0f6d760bb1ea6d66213861b4e01285fcf9a661c2c9569f8878f32e5bd
-  data.tar.gz: f39f15a8fcf86793ffcb139d9f74ddd3ac4ba4c2fa767d48d05334aa3e0a6fe8178718e867ee3bf2db2352bd5cecfbc97036f442da7a977d919229ac9220808f
+  metadata.gz: 98514fc593241f4225c9b4073ec51e99cb27bac21e8b8d5803b49e3a3f2cc5c531b3c974d6f449309865b4400c6c22e8c1b3f1557020db0c83341507badd0467
+  data.tar.gz: 10367edaf9817e96446ed847ff1e9fe92d5bc2127c7daebd4aed4997722acc02604397763e7d249fda6dea963fc52df5c04ea97df61880a1cdb7a805aff1329e

data/lib/template.rb

@@ -23,9 +23,12 @@ module Template
 			# Init some class vars
 			@new_template_path 	= files[1] 
 			@profile_path 		= files[0]
-#			@hiera_data_path 	= Extras.get_hiera_data_path
 			@keys 			= Hash.new 
-			
+			@data_groups		= Hash.new 
+			@data_groups["unknown"] = current_group = []
+			@cwd = File.expand_path File.dirname(__FILE__)
+	
+
 			# Kick back to stdout 
 			debug("In init definition")
 			info("Creating new template for #{@profile_path}")
@@ -33,6 +36,7 @@ module Template
 
 			# Run it 
 			parse()
+			debug(@data_groups)
 			write()
 		end
 
@@ -56,28 +60,59 @@ module Template
 		def parse
 			info("Parsing #{@profile_path}")
 			profile = File.open(@profile_path)
-			profile.each_line do |line|
-				if line.match(/hiera\(/)
-					debug("Data item found: #{line}")
-					data = line.split("(").last.chomp
-					debug("Data item split to: #{data}")
-					data.delete! (")")
-					data.delete! ("\"")
-					data.delete! ("'")
-					data.delete! (",")
-					debug("Data item after deletes: #{data}")
-					@keys[data] = nil 
-					info("Adding #{data}")
+
+			# For each data grouop, parse the file
+			profile.each_line do |line| 
+				catch :unknown_group do 
+					# If the line matches the data group, init a new array for the keys in that data group
+					if line.match(/\#\[/)
+						debug("Data group found on line: #{line}")
+						data_group = line.split('[').last.delete(']').chomp
+						info("Creating data group: #{data_group}") 
+						@data_groups[data_group] = @current_group = []
+						debug("Data groups now include: #{@data_groups}")
+
+					# For all other lines, parse it for keys
+					elsif line.match(/hiera\(/)
+						debug("Current data group being added to: #{@current_group}")
+						debug("Data item found: #{line}")
+						data = line.split("(").last.chomp
+						data.delete! (")")
+						data.delete! ("\"")
+						data.delete! ("'")
+						data.delete! (",")
+						
+						@current_group << data
+						
+						info("Adding #{data}")
+						debug("Adding #{data} to #{@current_group} hash")
+					end
 				end
 			end
 		end
 
 		def write
-			info("Writing keys to new template #{@new_template_path}")
-			template = File.open(@new_template_path, 'w')
-			@keys.to_hash
-			template.write(@keys.to_yaml)
-			template.close
+			# Create dotdir if needed:
+			unless Dir.exists?("#{Dir.home}/.hiera-template")
+				Dir.mkdir(File.join(Dir.home, '.hiera-template'), 0755)
+			end
+			unless Dir.exists?("#{Dir.home}/.hiera-template/templates")
+				Dir.mkdir("#{Dir.home}/.hiera-template/templates", 0755)
+			end
+
+			temp_path = "#{Dir.home}/.hiera-template/templates"
+			profile_name = @profile_path.split("/").last.split(".").first
+
+			@data_groups.each do |k,v|
+				write_hash = Hash.new
+				v.each do |key|
+					write_hash[key] = nil
+				end
+				info("Writing keys to new template #{temp_path}/#{profile_name}-#{k}-template.yaml")
+				template = File.open("#{temp_path}/#{profile_name}-#{k}-template.yaml", "w")
+				template.write(write_hash.to_yaml)
+				template.close
+			end
 		end
 
 		def verify(path)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: hiera-template
 version: !ruby/object:Gem::Version
-  version: 1.0.3
+  version: 1.1.0
 platform: ruby
 authors:
 - Jeff Malnick
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-11-01 00:00:00.000000000 Z
+date: 2014-11-02 00:00:00.000000000 Z
 dependencies: []
 description: 
 email: