Create clientbucket.rb

clientbucket.rb

@@ -0,0 +1,128 @@
+#!/usr/bin/ruby
+
+require 'rubygems'
+require 'highline/import'
+require 'find'
+require 'pp'
+
+if ARGV.length < 1
+  puts "Usage: clientbucket.rb /path/to/file"
+  exit 1
+end
+
+target_path = ARGV.shift
+clientbucket_path = "/var/lib/puppet/clientbucket"
+
+available_files = Array.new
+
+Find.find(clientbucket_path) do |file_path|
+
+  # Skip directories and "contents" files
+  next if FileTest.directory?(file_path)
+  next unless File.basename(file_path) =~ /paths$/
+
+  # See if this file has a path the user was looking for
+  path = File.open(file_path).first
+  path.chomp!
+  if target_path == path
+
+    # Get the md5 string the file is referred to by
+    file_path =~ /([^\/]+)\/paths$/;
+    file_hash = $1
+
+    # Get the "contents" file containing the old file's contents
+    contents_path = File.dirname(file_path) + "/contents"
+
+    # Get the creation time
+    ctime = File.stat(file_path).ctime
+
+    details = {:hash => file_hash, :ctime => ctime, :contents_path => contents_path}
+    available_files.push(details)
+
+  end
+
+end
+
+# See if we found any files for the user
+if available_files.length == 0
+  puts "No files with path #{target_path} exist in the clientbucket"
+  exit 2
+end
+
+
+while true
+
+  # Sort by creation time
+  available_files.sort! { |x, y| x[:ctime] <=> y[:ctime] }
+  n = 0
+  available_files.each {|file|
+    puts "[#{n}]: #{file[:hash]} #{file[:ctime]}"
+    # Stash the option number for this file
+    available_files[n][:option_number] = n
+    n+=1
+  }
+
+  puts "------------------------"
+
+  print "Pick a file, or x to exit: "
+  number = gets.strip
+  number.chomp!
+
+  if number !~ /^(x|\d+)$/
+    puts "Invalid input"
+    next
+  end
+
+  if number == "x"
+    puts "Exiting"
+    exit 0
+  end
+
+  number = number.to_i
+
+  if number > n
+    puts "Invalid file number"
+    next
+  end
+
+  while true
+
+    print "Restore (r), view (v), diff (d), unified diff (u), or x to go back: "
+    choice = gets.strip
+    choice.chomp!
+
+    if choice !~ /^[rvdux]$/
+      puts "Invalid choice"
+      next
+    end
+
+    case choice
+    when "x"
+      break
+    when "v"
+      system("vim #{available_files[number][:contents_path]}")
+    when "d"
+      system("diff #{available_files[number][:contents_path]} #{target_path}")
+    when "u"
+      system("diff -u #{available_files[number][:contents_path]} #{target_path}")
+    when "r"
+      print "Restore to (default is to restore to #{target_path}: "
+      choice = gets.strip
+      choice.chomp!
+
+      # If not specified, use the default
+      restore_path = target_path
+      if choice != ''
+        restore_path = choice
+      end
+
+      puts "Restoring to #{restore_path}"
+      system("cp -p #{available_files[number][:contents_path]} #{restore_path}")
+      puts "Done"
+      exit 0
+
+    end
+
+  end
+
+end