#!/bin/bash
#
# .---.                  .              .
# |                      |              |
# |--- .--. .-.  .-.  .-.|  .-. .--.--. |.-.  .-. .--.  .-.
# |    |   (.-' (.-' (   | (   )|  |  | |   )(   )|  | (.-'
# '    '     --'  --'  -' -  -' '  '   -' -'   -' '   -  --'
#
#                    Freedom in the Cloud
#
# Filesystem encryption, etc
#
# License
# =======
#
# Copyright (C) 2014-2018 Bob Mottram <bob@freedombone.net>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program 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 Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

TOMB_REPO="https://github.com/dyne/Tomb"
TOMB_COMMIT='c80ebd6d6ed77980eb5b559757e03ea13a29bdd1'

function mesh_install_tomb {
    chroot ${rootdir} apt-get -yq install cryptsetup zsh pinentry-curses

    if [ ! -d ${rootdir}/$INSTALL_DIR ]; then
        mkdir -p ${rootdir}/$INSTALL_DIR
    fi

    if [ -d /repos/tomb ]; then
        mkdir ${rootdir}/$INSTALL_DIR/tomb
        cp -r -p /repos/tomb/. ${rootdir}/$INSTALL_DIR/tomb
        cd ${rootdir}/$INSTALL_DIR/tomb
        git pull
    else
        git_clone $TOMB_REPO ${rootdir}/$INSTALL_DIR/tomb
    fi

    cd ${rootdir}/$INSTALL_DIR/tomb

    git checkout $TOMB_COMMIT -b $TOMB_COMMIT

    chroot ${rootdir} /bin/bash -x <<EOF
cd $INSTALL_DIR/tomb
make install
EOF
    if [ ! -f ${rootdir}/usr/local/bin/tomb ]; then
        exit 93462
    fi
}

function install_tomb {
    if [ $INSTALLING_MESH ]; then
        mesh_install_tomb
        return
    fi

    function_check set_repo_commit
    set_repo_commit $INSTALL_DIR/tomb "tomb commit" "$TOMB_COMMIT" $TOMB_REPO

    if [ -f $COMPLETION_FILE ]; then
        if [[ $(is_completed $FUNCNAME) == "1" ]]; then
            return
        fi
    fi

    apt-get -yq install cryptsetup zsh pinentry-curses

    if [ ! -d $INSTALL_DIR ]; then
        mkdir -p $INSTALL_DIR
    fi

    if [ -d /repos/tomb ]; then
        mkdir $INSTALL_DIR/tomb
        cp -r -p /repos/tomb/. $INSTALL_DIR/tomb
        cd $INSTALL_DIR/tomb
        git pull
    else
        git_clone $TOMB_REPO $INSTALL_DIR/tomb
    fi

    cd $INSTALL_DIR/tomb

    git checkout $TOMB_COMMIT -b $TOMB_COMMIT
    set_completion_param "tomb commit" "$TOMB_COMMIT"

    make install
    if [ ! -f /usr/local/bin/tomb ]; then
        exit 93462
    fi

    mark_completed $FUNCNAME
}

function defrag_filesystem {
    fs_type=$(stat -f /)

    if [[ "$fs_type" != *"btrfs"* && "$fs_type" != *"ext4"* ]]; then
        return
    fi

    echo $'Defragmenting root directory'

    if [[ "$fs_type" == *"btrfs"* ]]; then
        btrfs filesystem defragment -r -clzo /
    fi

    if [[ "$fs_type" == *"ext4"* ]]; then
        e4defrag /
    fi

    echo $'Defragmentation completed'
}

function optimise_filesystem {
    if grep -q "btrfs" /etc/fstab; then
        if ! grep -q "btrfs defaults,subvol=@,compress=lzo,ssd" /etc/fstab; then
            sed -i 's|btrfs subvol=@|btrfs defaults,subvol=@,compress=lzo,ssd|g' /etc/fstab
        fi
    fi
}

# NOTE: deliberately no exit 0