#!/bin/bash
#  _____               _           _
# |   __|___ ___ ___ _| |___ _____| |_ ___ ___ ___
# |   __|  _| -_| -_| . | . |     | . | . |   | -_|
# |__|  |_| |___|___|___|___|_|_|_|___|___|_|_|___|
#
#                              Freedom in the Cloud
#
# Removes an ignore rule for either an email address or text in the
# subject line or from xmpp MUC filtered words
#
# License
# =======
#
# Copyright (C) 2015-2019 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/>.

PROJECT_NAME='freedombone'

export TEXTDOMAIN=${PROJECT_NAME}-unignore
export TEXTDOMAINDIR="/usr/share/locale"

XMPP_CONFIG=/etc/prosody/prosody.cfg.lua
PLEROMA_CONFIG=/etc/pleroma/config/dev.secret.exs
MYUSERNAME=$USER
EMAIL_ADDRESS=
SUBJECT_TEXT=
ALL_TEXT=

function show_help {
    echo ''
    echo $"${PROJECT_NAME}-unignore -u [username] -e [mail address] -t [text in subject line] -a [text anywhere]"
    echo ''
    exit 0
}

while [ $# -gt 1 ]
do
key="$1"

case $key in
    -h|--help)
    show_help
    ;;
    -u|--user)
    shift
    MYUSERNAME="$1"
    ;;
    -e|--email)
    shift
    EMAIL_ADDRESS="$1"
    ;;
    -t|--text)
    shift
    SUBJECT_TEXT="$1"
    ;;
    -a|--anywhere|--all)
    shift
    ALL_TEXT="$1"
    ;;
    *)
    # unknown option
    ;;
esac
shift
done

if ! [[ $MYUSERNAME && $EMAIL_ADDRESS ]]; then
    if ! [[ $MYUSERNAME && $SUBJECT_TEXT ]]; then
        show_help
    fi
fi

MUTTRC="/home/$MYUSERNAME/.muttrc"
PM=/home/$MYUSERNAME/.procmailrc

# unignore if subject line contains text
if [ "$SUBJECT_TEXT" ]; then
    if grep -q "Ignore rule for $SUBJECT_TEXT" "$PM"; then
        sed -i "/# Ignore rule for $SUBJECT_TEXT/,/# End of ignore rule/d" "$PM"
    fi
fi

# unignore if email contains text
if [ "$ALL_TEXT" ]; then
    if grep -q "Ignore rule for words '$ALL_TEXT'" "$PM"; then
        sed -i "/# Ignore rule for words '$ALL_TEXT'/,/# End of ignore words rule/d" "$PM"
    fi

    if [ -f "$PLEROMA_CONFIG" ]; then
        if grep -q ':mrf_keyword' "$PLEROMA_CONFIG"; then
            lines=$(grep -Pzo '.*mrf_keyword(.*\n)*' "$PLEROMA_CONFIG")
            if [[ "$lines" == *' reject: '* ]]; then
                if [[ "$lines" != *' reject: []'* ]]; then
                    curr_filter_words=$(printf "%s" "$lines" | grep ' reject: ')
                    if [ "$curr_filter_words" ]; then
                        blocked_word="${ALL_TEXT}"
                        recompile=1
                        if [[ "$curr_filter_words" == *",\"${blocked_word}\""* ]]; then
                            sed -i "/mrf_keyword/{N;s|,\"${blocked_word}\"||}" "$PLEROMA_CONFIG"
                            sed -i "/mrf_keyword/{N;N;s|,\"${blocked_word}\"||}" "$PLEROMA_CONFIG"
                            sed -i "/mrf_keyword/{N;N;N;s|,\"${blocked_word}\"||}" "$PLEROMA_CONFIG"
                        else
                            if [[ "$curr_filter_words" == *"\"${blocked_word}\","* ]]; then
                                sed -i "/mrf_keyword/{N;s|\"${blocked_word}\",||}" "$PLEROMA_CONFIG"
                                sed -i "/mrf_keyword/{N;N;s|\"${blocked_word}\",||}" "$PLEROMA_CONFIG"
                                sed -i "/mrf_keyword/{N;N;N;s|\"${blocked_word}\",||}" "$PLEROMA_CONFIG"
                            else
                                if [[ "$curr_filter_words" == *" \"${blocked_word}\" "* ]]; then
                                    sed -i "/mrf_keyword/{N;s| \"${blocked_word}\" ||}" "$PLEROMA_CONFIG"
                                    sed -i "/mrf_keyword/{N;N;s| \"${blocked_word}\" ||}" "$PLEROMA_CONFIG"
                                    sed -i "/mrf_keyword/{N;N;N;s| \"${blocked_word}\" ||}" "$PLEROMA_CONFIG"
                                else
                                    recompile=
                                fi
                            fi
                        fi
                        if [ $recompile ]; then
                            systemctl stop pleroma
                            cd /etc/pleroma || exit 0
                            chown -R pleroma:pleroma /etc/pleroma
                            sudo -u pleroma mix clean
                            sudo -u pleroma mix deps.compile
                            sudo -u pleroma mix compile
                            systemctl restart pleroma
                        fi
                    fi
                fi
            fi
        fi
    fi

    if [ -f "$XMPP_CONFIG" ]; then
        if grep -q 'filter_words =' "$XMPP_CONFIG"; then
            if ! grep -q 'filter_words = {}' "$XMPP_CONFIG"; then
                curr_filter_words=$(grep 'filter_words =' "$XMPP_CONFIG")
                if [ "$curr_filter_words" ]; then
                    if [[ "$curr_filter_words" == *",\"${ALL_TEXT}\""* ]]; then
                        sed -i "s|,\"${ALL_TEXT}\"||g" "$XMPP_CONFIG"
                    else
                        if [[ "$curr_filter_words" == *"\"${ALL_TEXT}\","* ]]; then
                            sed -i "s|\"${ALL_TEXT}\",||g" "$XMPP_CONFIG"
                        else
                            if [[ "$curr_filter_words" == *" \"${ALL_TEXT}\" "* ]]; then
                                sed -i "s| \"${ALL_TEXT}\" ||g" "$XMPP_CONFIG"
                            fi
                        fi
                    fi
                fi
            fi
        fi
    fi

fi

# unignore an email address
if [ "$EMAIL_ADDRESS" ]; then
    if grep -q "Ignore rule for $EMAIL_ADDRESS" "$PM"; then
        sed -i "/# Ignore rule for $EMAIL_ADDRESS/,/# End of ignore rule/d" "$PM"
    fi
fi

exit 0