#!/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 {
    # shellcheck disable=SC2154
    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" || exit 3648368
        git pull
    else
        git_clone "$TOMB_REPO" "${rootdir}/$INSTALL_DIR/tomb"
    fi

    cd "${rootdir}/$INSTALL_DIR/tomb" || exit 7346384

    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[0]}") == "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" || exit 7684638
        git pull
    else
        git_clone "$TOMB_REPO" "$INSTALL_DIR/tomb"
    fi

    cd "$INSTALL_DIR/tomb" || exit 364863463

    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[0]}"
}

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