cleanfb 0.1.5.0 → 0.2.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 6d603676a6ca95f5d14686b165f29cf4b4140a4e
-  data.tar.gz: 91d8e19a731c3c60a4dcc10fbdf24a5a32059735
+  metadata.gz: 89bb9afdedece34a64562247912640a103e15107
+  data.tar.gz: 4875c18855bd4c88997d3a692fe84242a7e289c8
 SHA512:
-  metadata.gz: 05dca4425aea4bdd575f71f3a6536a2048b0629d5c846005e2b66235cc7e5e729cc4fb64085bdcfe087af5913e663cccd13cd2e450375405e12dcae890fad5e9
-  data.tar.gz: 0365ec54817217f08df05e080c41cbfa8fc2faa2076a54e1456a6c4bf2550be0554e7e14d468fadb0c49ab7da8082510bd0d20e3c972a4159eb0b4e8e09344c5
+  metadata.gz: 63950e72bb3cd07dffc49a8554814f35b7f014cfef9da9a995fa29ee0b0b5d11db867312dee83bc7066bdbce2336d444694949ea4467d6fd6699922f383935aa
+  data.tar.gz: 54d33fd1e44b4665dc8de0ca895422c5cf24c6132bd66f625ddf25088f6bd5e5529462cdcbb1e3e578237e4fdbadc91e93fcd26547dab340e6e28776bb0bde2b

data/exe/cleanfb

@@ -8,7 +8,12 @@ end
 
 cleaner = Cleanfb::Clean.new
 
-response = cleaner.remove
+unless ARGV.include? "restore"
+	response = cleaner.remove
+else
+	respone = cleaner.restore
+end
+
 unless response.nil? || response.empty? || (response.include? "Please")
 	puts response.magenta
 else

data/lib/cleanfb/version.rb

@@ -1,3 +1,3 @@
 module Cleanfb
-  VERSION = "0.1.5.0"
+  VERSION = "0.2.0.0"
 end

data/lib/cleanfb.rb

@@ -98,5 +98,52 @@ module Cleanfb
 	  	end
 			return 
 		end
-	end
+
+		def restore()
+
+			# return if no args
+			return "Please supply a file" if ARGV[0].nil? || ARGV[0].empty?
+  			
+			# return the help output if requested
+			if ((ARGV.include? "-h") or (ARGV.include? "--help"))
+				return "Remove backups of an agent from the Puppet master's filebucket.\nBackups are stored at /root/saved_configs/\n\n		Usage:   cleanfb <client> <options>\n\n		options: -h or --help		| help and information\n
+				-y		| defaults all input to yes\n"
+
+			end
+
+			arg = ARGV[0]	
+
+			# set ans to yes automatically if -y option
+			if !ARGV.nil? and (ARGV.include? "-y") and ARGV.length > 1
+  	 		ans = "y"
+				arg = ARGV[1] if (ARGV.index("-y") == 0)
+
+			# else prompt for answer
+			else
+    		print "Restore #{arg} ? y|n: "
+	   		ans = STDIN.gets.chomp
+		  end
+
+			# if yes, retrieve the files and remove them
+  		if ans == "y" || ans == "yes"
+				
+				file = arg
+				sum = Digest::MD5.file file
+				start = (sum.scan /\w/).join("/")
+	    	path = "/opt/puppetlabs/puppet/cache/bucket"
+				
+				path += "/" + start[0..15] + sum + "/"
+
+				if File.exist? "#{path}/contents"
+							
+						puts "Restoring " + path
+						cmd = `mv -f #{file} #{path}contents`
+				else
+	    	  		puts "No file #{arg} found."
+				end
+  	 	else
+	    	  puts "No file #{arg} found."
+		 	end
+			return
+		end	
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: cleanfb
 version: !ruby/object:Gem::Version
-  version: 0.1.5.0
+  version: 0.2.0.0
 platform: ruby
 authors:
 - lhoffl
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2016-08-26 00:00:00.000000000 Z
+date: 2016-09-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler