#!/usr/bin/ruby

#

#  Created by Tamara Temple on 2007-03-16.

#  Copyright (c) 2007. All rights reserved.

require 'FileUtils'

include FileUtils::Verbose

LIBRARY_DIRS = 

  ["./Library/Application Support",

    "./Library/Keychains",

    ",/Library/Mail",

    "./Library/Mozilla",

    "./Library/Preferences",

    "./Library/Safari",

    "./Library/StickiesDatabase",

    "./Library/Workflows"]

def getBackDir(timestamp)

  prefix = "/Volumes/BootBackup/Users"

  suffix = timestamp.strftime("%Y%m%d.%H%M%S")

  backupdir = prefix + suffix

  print "** backupdir: ", backupdir, "\n" if $DEBUG

  backupdir

end

def getUserList(parentdir)

  userlist = Dir.entries(parentdir)

  if (!userlist.kind_of?(Array)) then raise "Internal Error: Dir.entries did not return an array." end

  userlist.delete(".") # remove the working directory from the list (this will not affect the directory on the disk)

  userlist.delete("..") # remove the parent directory from the list

  userlist.delete(".localized") # remove an extraneous file from the /Users directory

  print "** userlist: ", userlist.join(", "),"\n" if $DEBUG 

  userlist # return the userlist

end

def getFileList()

  files = Dir.entries('.')

  if (!files.kind_of?(Array)) then raise "Internal Error: Dir.entries did not return an array." end

  files.delete(".") # remove the reference to the directory itself. (note this will not remove the directory -  files is an array not the directory)

  files.delete("..") # remove the parent directory listing 

  files.delete(".Trash") # don't backup the trash

  files.delete("Library") # remove the Library directory as we don't want it all in the backup

  print "** files: ", files.join(", "),"\n" if $DEBUG

  begin

    libdirs = Dir.entries("./Library")

  rescue Errno::ENOENT

    libdirs = []

  ensure

    if (!libdirs.kind_of?(Array)) then raise "Internal Error: Dir.entries did not return an array." end

  end

  # Remove extraneous entries

  libdirs.delete(".")

  libdirs.delete("..")

  libdirs.delete(".localized")

  libdirs.map! {|l| "./Library/" + l} # Need to add in the path to the Library subdirectories

  print "** libdirs: ", libdirs.join(", "),"\n" if $DEBUG

  libdirs = libdirs & LIBRARY_DIRS # get the intersection of library directories with the set we want to keep in backup

  files = files + libdirs

  files.map! {|f| "'#{f}'"} # need to quote file names for find() command in backupUsers() procedure

  print "** files: ", files.join(", "),"\n" if $DEBUG

  files # return the file list

end

def backupUsers(userlist,backupdir)

  for user in userlist

    mkdir("#{backupdir}/#{user}") unless $DEBUG

    print "** mkdir #{backupdir}/#{user}","\n" if $DEBUG

    cd(user) do # change directories in a block, at the end of the block, direcotry will go back to previous working directory

      files = getFileList()

      begin

        `find #{files.join(" ")} -print | cpio -pdv #{backupdir}/#{user}` unless $DEBUG

        print "** find:", files.join(", ")," | cpio: ", "#{backupdir}/#{user}","\n" if $DEBUG

      rescue

        warn "Error with backing up #{user}: #{$!}\nContinuing with next user."

      end

    end

  end

end

backupdir = getBackDir(Time.now)

mkdir(backupdir)

cd("/Users") do # change directories in a block, at the end of the block, directory will go back to the previous working directory

  userlist = getUserList("/Users")

  backupUsers(userlist,backupdir)

end