#!/usr/bin/env zsh
#
# Copyright (c) 2015-2016 parazyd
# gtomb is written and maintained by parazyd <parazyd@dyne.org>
#
# This file is part of gtomb
#
# This source code is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This software is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this source code. If not, see <http://www.gnu.org/licenses/>.

ver="0.8"
TOMBPATH="/usr/local/bin/tomb" # Set this to your tomb executable's path
KDFPATH="/usr/local/bin/" # Set this to the path of your KDF binaries (if you're using them)
# The ones below should not need changing
HEXENC="$KDFPATH/tomb-kdb-hexencode"
GENSALT="$KDFPATH/tomb-kdb-pbkdf2-gensalt"
GETITER="$KDFPATH/tomb-kdb-pbkdf2-getiter"
PBKDF="$KDFPATH/tomb-kdb-pbkdf2"
_DD=/bin/dd

# {{{ monmort icon
MONMORT="/tmp/monmort.png"
ICONB64="iVBORw0KGgoAAAANSUhEUgAAACAAAAAgEAQAAACJ4248AAAAAmJLR0T//xSrMc0AAAAJcEhZcwAA\n
AEgAAABIAEbJaz4AAAAJdnBBZwAAACAAAAAgAIf6nJ0AAADrSURBVFjD5ZZBEsMgCEU/TO/l2XMx\n
04VjQ1K1CDYswkwWJnH+E/EL4RP7jluDCACoim/bvfIpFQiKEJcQHCN9xEtLCgDMQM7f33sZrPNG\n
/05loCXujfAtCAVgNgLwIuycjQAra8G9Fm823ADabPRA1QDelfZAVUZktWrNvL8ew5YTnsStx3Am\n
AyOInJVbYF1prZuU+tsR1g9UMDqGuo5oFWhtSEQNEGmeVrqv73Tj0pIZirANMYqRhyw5Bb9MauSW\n
SwA8l9OzG5LnAsiiDQGQRRvaEwInK54J390hndAIYIeQ4k6AAjE/h06ab0SjP08MA1xDAAAAJXRF\n
WHRkYXRlOmNyZWF0ZQAyMDExLTAxLTEyVDA5OjM0OjI0KzAxOjAwo//d7wAAACV0RVh0ZGF0ZTpt\n
b2RpZnkAMjAxMS0wMS0xMlQwOTozNDoyNCswMTowMNKiZVMAAAAASUVORK5CYII="
echo -e "$ICONB64" | base64 --decode > $MONMORT
# }}}

# {{{ sudo functions
function _sudo {
    sudoassword=$(ask_password "Insert sudo password for user $USER")
    echo -e "$sudoassword\n" | sudo -S -v
    _sudowrong
}

function _sudowrong {
    [[ $? = 0 ]] || {
        sudoassword=$(ask_password "Wrong password. Insert sudo password for user $USER")
        echo -e "$sudoassword\n" | sudo -S -v
        _sudowrong
    }
}
# }}}

# {{{ Zenity dialogs
function _zenques {
    zenity \
        --window-icon="$MONMORT" \
        --question \
        --text="$1"
}
function _fsel {
    zenity \
        --window-icon="$MONMORT" \
        --file-selection \
        --title="$1"
}
function _fsave {
    zenity \
        --window-icon="$MONMORT" \
        --file-selection \
        --save \
        --title="$1" \
        --filename="$2"
}
function _zenwarn {
    zenity \
        --window-icon="$MONMORT" \
        --warning \
        --title="$1" \
        --text="$2"
}
function _info {
    which notify-send > /dev/null
    if [[ $? == "0" ]]; then
        _zenotif $*
    else
        _zeninfo $*
    fi
}
function _zenotif {
    zenity \
        --window-icon="$MONMORT" \
        --notification \
        --title="$1" \
        --text="$2"
}
function _zeninfo {
    zenity \
        --window-icon="$MONMORT" \
        --info \
        --title="$1" \
        --text="$2"
}
function _zenerr {
    zenity \
        --window-icon="$MONMORT" \
        --error \
        --title="$1" \
        --text="$2"
}
function _zenprog {
    zenity \
        --window-icon="$MONMORT" \
        --progress \
        --auto-close \
        --pulsate \
        --title="$1" \
        --text="$2"
}
function _zenprognc {
    zenity \
        --window-icon="$MONMORT" \
        --progress \
        --auto-close \
        --no-cancel \
        --pulsate \
        --title="$1" \
        --text="$2"
}
function _zenentry {
    zenity \
        --window-icon="$MONMORT" \
        --entry \
        --title="$1" \
        --text="$2" \
        --entry-text="$3"
}
# }}}

# {{{ Some pinentry code shamelessly stolen from tomb
# Ask user for a password
# Wraps around the pinentry command, from the GnuPG project, as it
# provides better security and conveniently use the right toolkit.
ask_password() {
    local description="$1"
    local title="${2:-Enter tomb password.}"
    local output
    local password
    local gtkrc
    local theme

    # Distributions have broken wrappers for pinentry: they do
    # implement fallback, but they disrupt the output somehow.  We are
    # better off relying on less intermediaries, so we implement our
    # own fallback mechanisms. Pinentry supported: curses, gtk-2, qt4
    # and x11.

    # make sure LANG is set, default to C
    LANG=${LANG:-C}

    _verbose "asking password with tty=$TTY lc-ctype=$LANG"

    if [[ "$DISPLAY" = "" ]]; then

        if _is_found "pinentry-curses"; then
            _verbose "using pinentry-curses"
            output=`cat <<EOF | pinentry-curses
OPTION ttyname=$TTY
OPTION lc-ctype=$LANG
SETTITLE $title
SETDESC $description
SETPROMPT Password:
GETPIN
EOF`
        else
            _failure "Cannot find pinentry-curses and no DISPLAY detected."
        fi

    else # a DISPLAY is found to be active

        # customized gtk2 dialog with a skull (if extras are installed)
        if _is_found "pinentry-gtk-2"; then
            _verbose "using pinentry-gtk2"

            gtkrc=""
            theme=/share/themes/tomb/gtk-2.0-key/gtkrc
            for i in /usr/local /usr; do
                [[ -r $i/$theme ]] && {
                    gtkrc="$i/$theme"
                    break
                }
            done
            [[ "$gtkrc" = "" ]] || {
                gtkrc_old="$GTK2_RC_FILES"
                export GTK2_RC_FILES="$gtkrc"
            }
            output=`cat <<EOF | pinentry-gtk-2
OPTION ttyname=$TTY
OPTION lc-ctype=$LANG
SETTITLE $title
SETDESC $description
SETPROMPT Password:
GETPIN
EOF`
            [[ "$gtkrc" = "" ]] || export GTK2_RC_FILES="$gtkrc_old"

            # TODO QT4 customization of dialog
        elif _is_found "pinentry-qt4"; then
            _verbose "using pinentry-qt4"

            output=`cat <<EOF | pinentry-qt4
OPTION ttyname=$TTY
OPTION lc-ctype=$LANG
SETTITLE $title
SETDESC $description
SETPROMPT Password:
GETPIN
EOF`

            # TODO X11 customization of dialog
        elif _is_found "pinentry-x11"; then
            _verbose "using pinentry-x11"

            output=`cat <<EOF | pinentry-x11
OPTION ttyname=$TTY
OPTION lc-ctype=$LANG
SETTITLE $title
SETDESC $description
SETPROMPT Password:
GETPIN
EOF`

        else

            if _is_found "pinentry-curses"; then
                _verbose "using pinentry-curses"

                _warning "Detected DISPLAY, but only pinentry-curses is found."
                output=`cat <<EOF | pinentry-curses
OPTION ttyname=$TTY
OPTION lc-ctype=$LANG
SETTITLE $title
SETDESC $description
SETPROMPT Password:
GETPIN
EOF`
            else
                _failure "Cannot find any pinentry: impossible to ask for password."
            fi

        fi

    fi # end of DISPLAY block

    # parse the pinentry output
    for i in ${(f)output}; do
        [[ "$i" =~ "^ERR.*" ]] && {
            _warning "Pinentry error: ::1 error::" ${i[(w)3]}
            print "canceled"
            return 1 }

        # here the password is found
        [[ "$i" =~ "^D .*" ]] && password="${i##D }"
    done

    [[ "$password" = "" ]] && {
        _warning "Empty password"
        print "empty"
        return 1 }

    print "$password"
    return 0
}

_is_found() {
    # returns 0 if binary is found in path
    [[ "$1" = "" ]] && return 1
    command -v "$1" 1>/dev/null 2>/dev/null
    return $?
}

function _warning  no() {
    option_is_set -q || _msg warning $@
    return 1
}

function _verbose xxx() {
    option_is_set -D && _msg verbose $@
    return 0
}

function _failure die() {
    typeset -i exitcode=${exitv:-1}
    option_is_set -q || _msg failure $@
    # be sure we forget the secrets we were told
    exit $exitcode
}
# }}}

# {{{ _clean - Clean function, removes sensitive stuff from memory
function _clean {
    unset $?
    local rr="$RANDOM"
    while [[ ${#rr} -lt 500 ]]; do
        rr+="$RANDOM"
    done

    cmnd="$rr";         unset cmnd
    tombname="$rr";     unset tombname
    tombsize="$rr";     unset tombsize
    keyfile="$rr";      unset keyfile
    sudoassword="$rr";  unset sudoassword
    tombtmp="/tmp/tombtmp"
    if [ -f $tombtmp ]; then
        dd if=/dev/urandom of=$tombtmp bs=800 count=1
        rm -f $tombtmp
    fi
    tombtmp="$rr";      unset tombtmp
    newkey="$rr";       unset newkey
    jpegfile="$rr";     unset jpegfile
}
# }}}

# {{{ _main - Main window
function _main {
    _clean
    cmnd=`zenity \
        --window-icon="$MONMORT" \
        --title="gtomb" \
        --width=640 \
        --height=420 \
        --list \
        --hide-header \
        --text="gtomb v$ver\nThe GUI wrapper for Tomb, the crypto undertaker." \
        --separator=" & " \
        --column=Function \
        --column=Description \
        "dig" "Dig a new tomb of chosen size" \
        "forge" "Forge a new key used to lock tombs" \
        "lock" "Lock a non-locked tomb using an existing key" \
        "open" "Open an existing tomb" \
        "index" "Index the contents of all tombs." \
        "search" "Search the content of indexed tombs." \
        "list" "List all open tombs and information on them" \
        "close" "Close a specific tomb (or all)" \
        "slam" "Slam a tomb (or all) killing all programs using it" \
        "resize" "Resize a tomb to a new size (can only grow)" \
        "passwd" "Change the passphrase of a key" \
        "setkey" "Change the key of an existing tomb" \
        "engrave" "Generates a QR code of a key to be saved on paper" \
        "bury" "Hide a key inside a JPEG image" \
        "exhume" "Extract a key from a JPEG image"`
    eval "_$cmnd"
}
# }}}

# {{{ dig - Dig a new tomb
function _dig {
    tombname=`_fsave "Choose where to dig your tomb" "secret.tomb"`
    res=$?

    if [[ -f "$tombname" ]]; then
        _zenerr "Error" "This tomb already exists. I am not digging here."
        exec _main
    elif [[ -z "$tombname" ]]; then
        _info "gtomb" "Cancelled"
        exec _main
    fi

	[[ $res = 0 ]] || exec _main

	tombsize=`_zenentry "Tomb digging" "Enter the size of your tomb in MiB (min. 10 MiB)" "10"`
	res=$?

	re='^[0-9]+$'
	if ! [[ $tombsize =~ $re ]]; then
		_zenerr "Error" "Please choose a valid number."
		exec _main
	elif [[ -z $tombsize ]]; then
		_info "gtomb" "Cancelled"
		exec _main
	fi

	[[ $res = 0 ]] || { _zenwarn "Warning" "Tomb digging canceled." ; exec _main }

	"$TOMBPATH" dig -s "$tombsize" "$tombname" | \
		_zenprog "Digging new tomb" "Please wait while your tomb is being dug..." &

	PID_ZEN=$(ps -C zenity h -o pid,command | awk '/Digging new tomb/ {print $1}')
	while [ -n "$PID_ZEN" ]; do
		PID_ZEN=$(ps h -o pid -p ${PID_ZEN})
		PID_DD=$(ps -C $_DD h -o pid,command | awk -v tombtmp="$tombname" '$0~tombtmp{print $1}')
		sleep 1
	done

	[[ -n "$PID_DD" && -z "$PID_ZEN" ]] && {
		kill -9 $PID_DD
		_zenwarn "Warning" "Tomb digging cancelled."
		rm -f "$tombname"
		exec _main
	}

	_info "Success" "Your tomb has been dug in $tombname"
	exec _main
}
# }}}

# {{{ forge - Forge a new key
function _forge {
    keyfile=`_fsave "Choose where to forge your key" "secret.tomb.key"`
    res=$?

    if [[ -f $keyfile ]]; then
        _zenerr "Error" "This key already exists. I am not overwriting."
        exec _main
    elif [[ -z $keyfile ]]; then
        _info "gtomb" "Canceled"
        exec _main
    fi

    kdf=""
    kdfiter=""
    if [[ -x $HEXENC ]] && [[ -x $GENSALT ]] && [[ -x $GETITER ]] && [[ -x $PBKDF ]]; then
        _zenques "Do you want to use KDF? (Generates passwords armored against dictionary attacks)"
        [[ $? == "0" ]] && {
            kdf="--kdf"
            kdfiter=`_zenentry "Iterations" "Enter the delay (itertime) in seconds for each time \n\
this key is used:" "2"`

            re='^[0-9]+$'
            if ! [[ $kdfiter =~ $re ]]; then
                _zenerr "Error" "Please choose a valid number."
                exec _main
            elif [[ -z $kdfiter ]]; then
                _info "gtomb" "Canceled"
                exec _main
            fi
        }
    else
        _zenotif "gtomb" "KDF binaries not found."
    fi

	[[ $? = 0 ]] || exec _main

	"$TOMBPATH" forge "$keyfile" "$kdf" "$kdfiter" | \
		_zenprog "Forging key" "Please wait while your key is being forged...\n\
You can move your mouse around and use your computer to speed up the process." &

	PID_ZEN=$(ps -C zenity h -o pid,command | awk '/Forging key/ {print $1}')
	while [ -n "$PID_ZEN" ]; do
		PID_ZEN=$(ps h -o pid -p ${PID_ZEN})
		PID_DD=$(ps -C $_DD h -o pid,command | awk '/ if/ {print $1}')
		sleep 1
	done
	[[ -n "$PID_DD"  && -z "$PID_ZEN" ]] && {
		kill -9 $PID_DD
		_zenwarn "Warning" "Forging cancelled."
		rm -f $keyfile
		exec _main
	}

	_info "Success" "Your key is now forged in $keyfile"
	exec _main
}
# }}}

# {{{ lock - Lock a non-locked tomb
function _lock {
    tombname=`_fsel "Select a tomb to lock"`
    [[ -n $tombname ]] || { _zenotif "gtomb" "Cancelled" ; exec _main }
	[[ $? = 0 ]] || exec _main

	keyfile=`_fsel "Choose the key for your tomb"`
	[[ -n $keyfile ]] || { _zenotif "gtomb" "Cancelled" ; exec _main }
	[[ $? = 0 ]] || exec _main

	_sudo
	"$TOMBPATH" lock "$tombname" -k "$keyfile" | \
		_zenprognc "Locking your tomb" "Please wait while your tomb is being locked..."
	_info "Success" "Your tomb is now locked."
	exec _main
}
# }}}

# {{{ open - Open an existing tomb
function _open {
    tombname=`_fsel "Choose a tomb to open"`
	[[ $? = 0 ]] || exec _main

	keyfile=`_fsel "Choose the key for your tomb"`
	[[ $? = 0 ]] || exec _main

	_sudo
	"$TOMBPATH" open "$tombname" -k "$keyfile"
	_info "Success" "Your tomb is now open."
	exec _main
}
# }}}

# {{{ list - list all open tombs, along with their mountpoint
function _list {
    tombtmp="/tmp/tombtmp"
    "$TOMBPATH" list --get-mountpoint > $tombtmp
    tombname=`cat $tombtmp | \
        sed 's/.*\/\([^\/]*\)$/\1\n &/' | \
        zenity \
            --title="Currently open tombs" \
            --window-icon="$MONMORT" \
            --width=640 --height=380 --list \
            --separator=" & " \
            --text="Here are your open tombs" \
            --column=Tomb \
            --column=Path `

    tombname=`echo "$tombname" | cut -c1-16`

	[[ $? = 0 ]] || exec _main

	listchoice=`zenity \
		--title="Choose action" \
		--window-icon="$MONMORT" \
		--width=640 --height=400 --list \
		--separator=" & " \
		--text="What do you want to do with this tomb?" \
		--column=Command \
		--column=Description \
		"disindex" "Disable indexing of this tomb." \
		"enindex" "Enable indexing of this tomb."   \
		"close" "Close the selected tomb." \
		"slam" "Slam the selected tomb." \
		"binds" "Edit current bind-hooks." \
		"posts" "Edit current post-hooks."`

	[[ $? = 0 ]] || exec _main

	case $listchoice in
		close)
			_sudo
			"$TOMBPATH" close "$tombname"
			_zeninfo "Success" "Tomb closed successfully!"
			exec _main
			;;
		slam)
			_sudo
			"$TOMBPATH" slam "$tombname"
			_info "Success" "$tombname slammed successfully!"
			exec _main
			;;
		disindex)
			tombloc=`"$TOMBPATH" list "$tombname" --get-mountpoint`
			touch "$tombloc/.noindex"
			_info "Success" "Indexing disabled for this tomb."
			exec _main
			;;
		enindex)
			tombloc=`"$TOMBPATH" list "$tombname" --get-mountpoint`
			rm -f "$tombloc/.noindex"
			_info "Success" "Indexing of this tomb is now enabled."
			exec _main
			;;
		binds)
			tombloc=`"$TOMBPATH" list "$tombname" --get-mountpoint`
			bindhooks=`zenity \
				--text-info \
				--filename="$tombloc/bind-hooks" \
				--editable`
			[[ $? = 0 ]] && {
				echo "$bindhooks" > "$tombloc/bind-hooks"
				_info "gtomb" "bind-hooks saved"
			}
			exec _main
			;;
		posts)
			tombloc=`"$TOMBPATH" list "$tombname" --get-mountpoint`
			posthooks=`zenity \
				--text-info \
				--filename="$tombloc/post-hooks" \
				--editable`
			[[ $? = 0 ]] && {
				echo "$posthooks" > "$tombloc/post-hooks"
				chmod +x "$tombloc/post-hooks"
				_info "gtomb" "post-hooks saved"
			}
			exec _main
			;;
	esac
}
# }}}

# {{{ close - Close open tomb(s)
function _close {
    tombtmp="/tmp/tombtmp"
    "$TOMBPATH" list --get-mountpoint > $tombtmp
    echo "/all" >> $tombtmp
    tombname=`cat $tombtmp | \
        sed 's/.*\/\([^\/]*\)$/\1\n &/' | \
        zenity \
            --title="Choose a tomb to close" \
            --window-icon="$MONMORT" \
            --width=640 --height=380 --list \
            --separator=" & " \
            --column=Tomb \
            --column=Path `

	[[ $? = 0 ]] || exec _main

	_sudo
	tombname=`echo "$tombname" | cut -c1-16`
	"$TOMBPATH" close "$tombname"
	_info "Success" "Closed successfully!"
	exec _main
}
# }}}

# {{{ slam - Slam open tombs
function _slam {
    tombtmp="/tmp/tombtmp"
    "$TOMBPATH" list --get-mountpoint > $tombtmp
    echo "/all" >> $tombtmp
    tombname=`cat $tombtmp | \
        sed 's/.*\/\([^\/]*\)$/\1\n &/' | \
        zenity \
            --title="Choose a tomb to slam" \
            --window-icon="$MONMORT" \
            --width=640 --height=380 --list \
            --separator=" & " \
            --column=Tomb \
            --column=Path `

	[[ $? = 0 ]] || exec _main

	_sudo
	tombname=`echo "$tombname" | cut -c1-16`
	"$TOMBPATH" slam "$tombname"
	_info "Success" "Slammed successfully!"
	exec _main
}
# }}}

# {{{ resize - Resize an existing *closed* tomb
function _resize {
    tombname=`_fsel "Choose a tomb to resize"`
    res=$?
    _zenques "Is your tomb closed?"

	[[ $? = 0 ]] || { _zenwarn "gtomb" "Please close the tomb before resizing." ; exec _main }

	[[ $res = 0 ]] || exec _main

	tombsize=`_zenentry "New tomb size" "Enter the new size of your tomb in MiB. Must be higher than the current size." "100"`
	[[ -n $tombsize ]] || _zenotif "gtomb" "Cancelled"

	re='^[0-9]+$'
	if ! [[ $tombsize =~ $re ]]; then
		_zenerr "Error" "Please choose a valid number."
		exec _main
	elif [[ -z $tombsize ]]; then
		_info "gtomb" "Cancelled"
		exec _main
	fi

	[[ $? = 0 ]] || exec _main

	keyfile=`_fsel "Choose according keyfile"`
	[[ $? = 0 ]] || exec _main

	_sudo
	"$TOMBPATH" resize "$tombname" -s "$tombsize" -k "$keyfile" | \
		_zenprognc "Resizing tomb." "Please wait while your tomb is being resized..."
	_info "Success" "Tomb resized successfully!"
	exec _main
}
# }}}

# {{{ passwd - Change existing key's passphrase
function _passwd {
    keyfile=`_fsel "Choose a keyfile"`
	[[ $? = 0 ]] || exec _main

	"$TOMBPATH" passwd -k "$keyfile" | \
		_zenprognc "Changing passphrase" "Please wait while your key's passphrase is being changed..."

	_info "Success" "$keyfile passphrase changed successfully!"
	exec _main
}
# }}}

# {{{ setkey - Change a tomb's keyfile
function _setkey {
    tombname=`_fsel "Choose a tomb to change its keyfile"`
	[[ $? = 0 ]] || exec _main

	keyfile=`_fsel "Chosse your tomb's old keyfile"`
	 [[ $? = 0 ]] || exec _main

	newkey=`_fsel "Choose your tomb's new keyfile"`
	[[ $? = 0 ]] || exec _main

	_sudo
    "$TOMBPATH" setkey -k "$newkey" "$keyfile" "$tombname" | \
		_zenprognc "Changing key" "Please wait while your tomb's key is being changed..."

	_info "Success" "$tombname keyfile successfully changed! Now using $newkey"
    exec _main
}
# }}}

# {{{ engrave - generate QR code of a key
function _engrave {
    which qrencode || _zenwarn "Warning" "qrencode is not installed. Install it and try again"
    keyfile=`_fsel "Choose a keyfile to engrave"`
	[[ $? = 0 ]] || exec _main

	jpegfile=`_fsave "Choose where to save your keyfile (PNG format)"`
	[[ $? = 0 ]] || exec _main

	awk '/^-----/ {next}; /^Version/ {next}; {print $0}' $keyfile \
		| qrencode --size 4 --level H --casesensitive -o $jpegfile

	_info "Success" "QR code generated in $jpegfile"
	exec _main
}
# }}}

# {{{ bury - hide a keyfile in a JPEG image
function _bury {
    which steghide || _zenwarn "Warning" "steghide is not installed. Install it and try again"
    keyfile=`_fsel "Choose keyfile"`
	[[ $? = 0 ]] || exec _main

	jpegfile=`_fsel "Choose JPEG file"`
	[[ $? = 0 ]] || exec _main

	"$TOMBPATH" bury -k "$keyfile" "$jpegfile"
	_info "Success" "Your key is now hidden in $jpegfile"
	exec _main
}
# }}}

# {{{ exhume - extract keyfile from JPEG
function _exhume {
    which steghide || _zenwarn "Warning" "steghide is not installed. Install it and try again"
    jpegfile=`_fsel "Choose JPEG file"`
	[[ $? = 0 ]] || exec _main

	keyfile=`_fsave "Choose where to extract your key"`
	[[ $? = 0 ]] || exec _main

	"$TOMBPATH" exhume -k "$keyfile" "$jpegfile"
	_info "Success" "Your keyfile is extracted to $keyfile"
	exec _main
}
# }}}

# {{{ index - index the contents of open tombs
function _index {
    which locate || _zenwarn "Warning" "mlocate is not installed. Install it and try again"
    "$TOMBPATH" index | _zenprognc "Indexing" "Please wait while the open tombs are being indexed..."
    _info "Success" "Tombs indexed!"
    exec _main
}
# }}}

# {{{ search - searches the contents of indexed tombs
function _search {
    strings=""
    _searchstring
    exec _main
}

function _searchstring {
	srchstring=`_zenentry "Search" "Enter an argument to search. Cancel to finish."`
	res=$?
	strings="$strings $srchstring"

	if [[ $res = 1 ]]; then
		tombtmp="/tmp/tombtmp"
		_info "Success" "Search complete!"
		"$TOMBPATH" search "$strings" > $tombtmp
			zenity \
				--text-info \
				--width=800 --height=640 \
				--title="Search results" \
				--filename=$tombtmp
	else
		_searchstring
	fi
}


# }}}

function _ { _clean } # I like cleaning :)

[[ -x $TOMBPATH ]] || {
    _zenwarn "Warning" "Tomb binary is not executable or doesn't exist in the current path. Install it or edit the script to point to the correct path."
	exit 1 }

_main
