(*
    Prepare secure drive droplet
    Part of 'Simulating encrypted physical disk in OS X' by Constantinos Neophytou
    http://www.cneophytou.com/2008/01/30/simulating-encrypted-physical-disk-in-os-x/
 
    Version 0.1.1
*)
 
property imageType : "UDIF" (* one of "UDIF", "SPARSE" or "SPARSEBUNDLE" - the latter two have faster creation times, but leave empty space on the device *)
property thePass : ""
 
on open dropped_item
    set this_vol_alias to (dropped_item as alias)
    try
        tell application "Finder"
            if the kind of this_vol_alias is "Volume" then
                set freeSpace to free space of this_vol_alias
                set volumeName to quoted form of POSIX path of this_vol_alias
                set imgSize to (round (freeSpace - (1024 * 1024 * 1)) / (1024 * 1024)) (* leave 1mb of free space, just in case *)
 
                display dialog "Enter password for encrypted image:" default answer thePass with hidden answer
                set thePass to text returned of result as string
 
                set createImg to "hdiutil create " & volumeName & ".secure -size " & imgSize & "m -fs HFS+ -type " & imageType & " -encryption -volName Secure -passphrase " & thePass
                set thePass to "" (* clear the password from memory *)
                set hideVol to "chflags hidden " & volumeName
                with timeout of 600 seconds
                    do shell script createImg
                    do shell script hideVol
                    do shell script "hdiutil detach " & quoted form of POSIX path of this_vol_alias
                end timeout
            else
                display alert "This script can only be executed on volumes"
                return
            end if
        end tell
 
    on error error_message number error_number
        display dialog error_message buttons {"OK"} default button 1
    end try
end open

(*
    Automount secure drive folder action
    Part of 'Simulating encrypted physical disk in OS X' by Constantinos Neophytou
    http://www.cneophytou.com/2008/01/30/simulating-encrypted-physical-disk-in-os-x/
 
    Version 0.1.1
*)
 
property secureName : ".secure"
 
on adding folder items to this_folder after receiving these_volumes
    try
        tell application "Finder"
            set the volume_list to every file of this_folder
            (* go through all entries in /Volumes/ *)
            repeat with i from 1 to number of items in volume_list
                set this_item to the original item of item i of volume_list
                if the kind of this_item is "Volume" then
                    set this_disk to (this_item as alias)
 
                    (* is this item the newly mounted disk? *)
                    if this_disk is in these_volumes then
                        (* iterate through all files in the root of disk *)
                        set searchCmd to "ls -d " & quoted form of POSIX path of this_disk & secureName & "*"
                        (* check to see if a hidden secure image  is available *)
                        set searchResult to ""
                        try
                            set searchResult to do shell script searchCmd
                        end try
                        (* mount the secure image *)
                        if (searchResult starts with "/Volumes") then
                            set hiddenFile to POSIX file searchResult as alias
                            open hiddenFile
                        end if
                    end if
                end if
            end repeat
        end tell
    end try
end adding folder items to
 
on removing folder items from this_folder after losing these_items
    tell application "Finder"
        try
            set the volume_list to every file of this_folder
            (* go through all entries in /Volumes/ *)
            repeat with i from 1 to number of items in volume_list
                set this_item to the original item of item i of volume_list
                if the kind of this_item is "Volume" then
                    set this_disk to (this_item as alias)
 
                    set searchCmd to "ls -d " & quoted form of POSIX path of this_disk & secureName & "*"
                    (* check to see if a hidden secure image  is available *)
                    set searchResult to ""
                    try
                        set searchResult to do shell script searchCmd
                    end try
                    if (searchResult starts with "/Volumes") then
                        (* eject the volume *)
                        do shell script "hdiutil detach " & quoted form of POSIX path of this_disk
                        (* eject this_disk *)
                        delay 1
                        display alert "It is now safe to remove the encrypted disk"
                        return
                    end if
                end if
            end repeat
        end try
    end tell
end removing folder items from

(*
    Restore secure drive
    Part of 'Simulating encrypted physical disk in OS X' by Constantinos Neophytou
    http://www.cneophytou.com/2008/01/30/simulating-encrypted-physical-disk-in-os-x/
 
    Version 0.1.1
*)
 
property secureName : ".secure"
tell application "Finder"
    try
        set the volume_list to every file of (POSIX file "/Volumes/" as alias)
        repeat with i from 1 to number of items in volume_list
            set this_item to the original item of item i of volume_list
            if the kind of this_item is "Volume" then
                set this_disk to (this_item as alias)
 
                set searchCmd to "ls -d " & quoted form of POSIX path of this_disk & secureName & "*"
                (* check to see if a hidden secure image  is available *)
                set searchResult to ""
                try
                    set searchResult to do shell script searchCmd
                end try
                if (searchResult starts with "/Volumes") then
                    display dialog "Restore secure disk " & (name of this_disk) & "?" buttons ["No", "Yes"] default button 2
                    set the_button to button returned of result
                    if (the_button is "Yes") then
                        display dialog "Preserve copy of secure data on Desktop?" buttons ["No", "Yes"] default button 2
                        set the_button to button returned of result
                        if (the_button is "Yes") then
                            with timeout of 600 seconds
                                do shell script "mv " & quoted form of searchResult & " ~/Desktop/secure.dmg"
                            end timeout
                        else
                            do shell script "rm -f " & quoted form of searchResult
                        end if
                        set volumeName to quoted form of POSIX path of this_disk
                        do shell script "chflags nohidden " & volumeName
                        do shell script "hdiutil detach " & volumeName
                        delay 1
                        display alert "Device restored. It is now safe to unplug the device."
                        return
                    end if
                end if
            end if
        end repeat
 
    on error error_message number error_number
        display dialog error_message buttons {"OK"} default button 1
    end try
end tell

property restartLog : {"Quicksilver.crash.log"}
property restartID : {"daed"}
 
on adding folder items to this_folder after receiving these_items
    repeat with i from 1 to number of items in these_items
        set fileAlias to item i of these_items as alias
        set fileInfo to info for fileAlias
        set fileName to name of fileInfo as text
        if fileName is in restartLog then
            beep
            set itm to my list_position(fileName, restartLog)
            tell application "Finder"
                open application file id (item itm of restartID)
                delete item fileAlias
            end tell
        end if
    end repeat
end adding folder items to
 
on list_position(this_item, this_list)
    repeat with i from 1 to the count of this_list
        if item i of this_list is this_item then return i
    end repeat
    return 0
end list_position

tell application "Finder" to return creator type of process "Process Name"

property restartName : "Quicksilver"
property restartID : "daed"
 
on adding folder items to this_folder after receiving these_items
    repeat with i from 1 to number of items in these_items
        set fileAlias to item i of these_items as alias
        set fileInfo to info for fileAlias
        set fileName to name of fileInfo as text
        if fileName starts with restartName then
            beep
            tell application "Finder"
                open application file id restartID
            end tell
        end if
    end repeat
end adding folder items to

property backup_script : "rsync.app" (* add/change this string to match the name of your script *)
 
on adding folder items to this_folder after receiving these_volumes
    tell application "Finder"
        try
            set the volume_list to every file of this_folder
 
            (* go through all entries in /Volumes/ *)
            repeat with i from 1 to number of items in volume_list
                set this_item to the original item of item i of volume_list
                if the kind of this_item is "Volume" then
                    set this_disk to (this_item as alias)
 
                    (* is this item the newly mounted disk? *)
                    if this_disk is in these_volumes then
 
                        (* iterate through all files in the root of disk *)
                        set searchCmd to "ls -d " & quoted form of POSIX path of this_disk & backup_script
                        (* check to see if a backup script is available *)
                        set searchResult to ""
                        try
                            set searchResult to do shell script searchCmd
                        end try
                        (* run the backup script *)
                        if (searchResult starts with "/Volumes") then
                            set backupFile to POSIX file searchResult as alias
                            open backupFile
                        end if
 
                    end if
                end if
            end repeat
        on error error_message number error_number
            if the error_number is not -128 then
                display dialog error_message buttons {"OK"} default button 1
            end if
        end try
    end tell
end adding folder items to

property backup_script : "rsync" (* add/change this to match the name of your script *)
 
on open dropped_item
    set this_vol_alias to (dropped_item as alias)
    try
        tell application "Finder"
            if the kind of this_vol_alias is "Volume" then
                set the file_list to every file of this_vol_alias
                repeat with j from 1 to number of items in file_list
                    (* check to see if a backup script is available *)
                    set this_file to name of item j of file_list
                    if this_file is equal to (backup_script & ".app") then
                        open item j of file_list (* run the backup script *)
                    end if
                end repeat
            else
                display alert "This script can only be executed on volumes"
                return
            end if
        end tell
 
        delay 2 (* give some time for the backup script to actually launch *)
        tell application "Finder"
            try
                set bakapp to creator type of process backup_script
                repeat while bakapp is not equal to ""
                    delay 1 (* the backup is still running, we can check again in a second *)
                    set bakapp to creator type of process backup_script
                end repeat
            on error
                if (ejectable of this_vol_alias is true) then
                    (* display alert POSIX path of dropped_item *)
                    eject this_vol_alias
                end if
            end try
        end tell
 
    on error error_message number error_number
        if the error_number is not -128 then
            display dialog error_message buttons {"OK"} default button 1
        end if
    end try
end open

property archive_backup : true
property number_of_archives : 3 (* 0 for unlimited *)
property with_administrator_privileges : false
property one_archive_per_day : true
 
property display_notification : false
property backup_target : "Backups/" (* MUST be a folder location with trailing slash! Always relative to home folder *)
 
property rsync_params : "-aEz --delete-excluded"
(* END OF PREFERENCES *)
 
set WhereImRunningFrom to path to me
tell application "Finder"
    (* can't run from the script editor *)
    set AppCreator to creator type of WhereImRunningFrom
    if AppCreator is "ToyS" then
        activate of me
        beep
        display alert "This script cannot run directly from ScriptEditor"
        return
    end if
 
    (* NEVER run from the hard drive! *)
    set bootVolume to name of disk of home (* safety feature! *)
    set NameOfDisk to name of disk of WhereImRunningFrom
    if NameOfDisk is bootVolume then
        beep
        display alert "Should not run this script from the boot volume!"
        return
    end if
 
    (* setup backup dir *)
    set homeDir to path to home folder from user domain
    set backup_folder to POSIX path of homeDir & backup_target
    try
        (* quick and dirty check to see if folder exists - must be a good way to do this in applescript? *)
        do shell script "cd " & backup_folder
    on error
        try
            do shell script "mkdir " & backup_folder
        on error
            display alert "There was an error creating the backup folder " & backup_folder
            return
        end try
    end try
 
    (* find source and target *)
    set backupBase to NameOfDisk & "-Backup"
    set targetDir to backup_folder & quoted form of backupBase
    set sourceDir to quoted form of ("/Volumes/" & NameOfDisk & "/")
 
    (* set up archive filename *)
    if archive_backup then
        set thedate to current date
        set theday to day of thedate
        set themonth to month of thedate
        set theyear to year of thedate
        set thetime to time of thedate
 
        set thedate to "-" & theyear & "-" & shortMonth(themonth) of me & "-" & theday as string
        if not one_archive_per_day then set thedate to thedate & "-" & thetime as string
 
        set backupFolder to quoted form of (backup_folder & backupBase & thedate  & "/")
    end if
end tell
 
(* do the backup *)
(* try *)
if with_administrator_privileges then
    set excluded to " "
else
    set excluded to " --exclude='.Trash*' --exclude='.Spotlight*'  "
end if
 
if archive_backup then
    set posixDir to POSIX file backup_folder
    set fileList to list folder posixDir
    set archiveList to {}
    tell application "Finder"
        (* Search for existing backups *)
        repeat with i from 1 to number of items in fileList
            set thisFile to item i of fileList
            set filePath to path to resource thisFile in bundle posixDir
            if thisFile starts with backupBase then
                set archiveList to archiveList & POSIX path of filePath
            end if
        end repeat
    end tell
 
    (* List existing backups in order of last modified date *)
    set latestArchive to ""
    set linkDest to ""
    set listString to ""
    repeat with x from 1 to number of items in archiveList
        set listString to listString & " " & quoted form of item x of archiveList
    end repeat
    set newString to do shell script "ls -dt " & listString (* sort files by modified date *)
    set theList to (paragraphs of newString)
 
    (* Clear out outdated backups *)
    if (number of items in theList is greater than number_of_archives) and (number_of_archives is greater than 0) then
        repeat with y from (number_of_archives + 1) to (number of items in theList)
            do shell script "rm -Rf " & quoted form of item y of theList
        end repeat
    end if
    if number of items in theList is greater than 0 then
        set latestArchive to quoted form of item 1 of theList
        if latestArchive is equal to backupFolder then
            (* Preserve same target folder *)
            if number of items in theList is greater than 1 then
                set latestArchive to quoted form of item 2 of theList
            else
                set latestArchive to ""
            end if
        end if
    end if
 
    if latestArchive is not equal to "" then
        set linkDest to " --link-dest=" & latestArchive
    end if
    set theScript to "rsync " & rsync_params & linkDest & excluded & sourceDir & " " & backupFolder & "; touch " & backupFolder
else
    set theScript to "rsync " & rsync_params & excluded & sourceDir & " " & targetDir & "; touch " & targetDir
end if
 
if with_administrator_privileges then
    do shell script theScript with administrator privileges
else
    do shell script theScript
end if
 
activate of me
beep
if display_notification then display alert NameOfDisk & " drive backed up"
(*
on error
    display alert "There was an error backing up " & NameOfDisk
end try
*)
 
 
to shortMonth(themonth)
    set MonthList to {January, February, March, April, May, June, July, August, September, October, November, December}
    set mm to 1
    repeat until item mm of MonthList = themonth
        copy mm + 1 to mm
    end repeat
    if mm < 10 then copy "0" & mm to mm
    return mm
end shortMonth