#!/bin/bash
#  _____               _           _
# |   __|___ ___ ___ _| |___ _____| |_ ___ ___ ___
# |   __|  _| -_| -_| . | . |     | . | . |   | -_|
# |__|  |_| |___|___|___|___|_|_|_|___|___|_|_|___|
#
#                              Freedom in the Cloud
#
# Riot Web user interface for Matrix
#
# License
# =======
#
# Copyright (C) 2017-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/>.

VARIANTS='full full-vim chat'

IN_DEFAULT_INSTALL=0
SHOW_ON_ABOUT=1
NOT_ON_ONION=1

RIOT_VERSION='0.15.3'
RIOT_FILENAME="riot-v${RIOT_VERSION}"
RIOT_HASH='0aecaa1c0d1e387c1730fea33cdb01b1a296e6146b7aef6a819fa90d9efc026e'
RIOT_DOWNLOAD_URL="https://github.com/vector-im/riot-web/releases/download/v${RIOT_VERSION}"
RIOT_ONION_PORT=8115
RIOT_ONION_HOSTNAME=
RIOT_DOMAIN_NAME=
RIOT_CODE=

riot_variables=(MY_USERNAME
                RIOT_DOMAIN_NAME
                MATRIX_DOMAIN_NAME
                SYSTEM_TYPE
                ONION_ONLY
                DDNS_PROVIDER)

function logging_on_riot {
    echo -n ''
}

function logging_off_riot {
    echo -n ''
}

function remove_user_riot {
    echo -n ''
}

function add_user_riot {
    echo '0'
}

function riot_remove_bad_links {
    sed -i '/riot.im/d' /var/www/$RIOT_DOMAIN_NAME/htdocs/home.html
}

function install_interactive_riot {
    if [[ $ONION_ONLY != "no" ]]; then
        RIOT_DOMAIN_NAME='riot.local'
    else
        RIOT_DETAILS_COMPLETE=
        while [ ! $RIOT_DETAILS_COMPLETE ]
        do
            data=$(mktemp 2>/dev/null)
            if [[ "$DDNS_PROVIDER" == *"freedns"* ]]; then
                dialog --backtitle $"Freedombone Configuration" \
                       --title $"Riot Web user interface for Matrix" \
                       --form $"\\nPlease enter your details.\\n\\nIMPORTANT: This should be a domain name which is supported by Let's Encrypt:" 13 65 3 \
                       $"Domain:" 1 1 "$(grep 'RIOT_DOMAIN_NAME' temp.cfg | awk -F '=' '{print $2}')" 1 15 40 40 \
                       $"Code:" 2 1 "$(grep 'RIOT_CODE' temp.cfg | awk -F '=' '{print $2}')" 2 15 40 255 \
                       2> "$data"
            else
                dialog --backtitle $"Freedombone Configuration" \
                       --title $"Riot Web user interface for Matrix" \
                       --form $"\\nPlease enter your details. The background image URL can be left blank.\\n\\nIMPORTANT: This should be a domain name which is supported by Let's Encrypt:" 13 65 2 \
                       $"Domain:" 1 1 "$(grep 'RIOT_DOMAIN_NAME' temp.cfg | awk -F '=' '{print $2}')" 1 15 40 40 \
                       2> "$data"
            fi
            sel=$?
            case $sel in
                1) rm -f "$data"
                   exit 1;;
                255) rm -f "$data"
                     exit 1;;
            esac
            RIOT_DOMAIN_NAME=$(sed -n 1p < "$data")
            if [ "$RIOT_DOMAIN_NAME" ]; then
                TEST_DOMAIN_NAME=$RIOT_DOMAIN_NAME
                validate_domain_name
                if [[ "$TEST_DOMAIN_NAME" != "$RIOT_DOMAIN_NAME" ]]; then
                    RIOT_DOMAIN_NAME=
                    dialog --title $"Domain name validation" --msgbox "$TEST_DOMAIN_NAME" 15 50
                else
                    if [[ "$DDNS_PROVIDER" == *"freedns"* ]]; then
                        RIOT_CODE=$(sed -n 2p < "$data")
                        validate_freedns_code "$RIOT_CODE"
                        if [ ! "$VALID_CODE" ]; then
                            RIOT_DOMAIN_NAME=
                        fi
                    fi
                fi
            fi
            if [ $RIOT_DOMAIN_NAME ]; then
                RIOT_DETAILS_COMPLETE="yes"
            fi
            rm -f "$data"
        done

        # save the results in the config file
        write_config_param "RIOT_CODE" "$RIOT_CODE"
    fi
    write_config_param "RIOT_DOMAIN_NAME" "$RIOT_DOMAIN_NAME"
    APP_INSTALLED=1
}

function reconfigure_riot {
    echo -n ''
}

function riot_download {
    # download
    if [ ! -f "$INSTALL_DIR/${RIOT_FILENAME}.tar.gz" ]; then
        wget "${RIOT_DOWNLOAD_URL}/${RIOT_FILENAME}.tar.gz" -O "$INSTALL_DIR/${RIOT_FILENAME}.tar.gz"
    fi
    if [ ! -f "$INSTALL_DIR/${RIOT_FILENAME}.tar.gz" ]; then
        echo $'Unable to download Riot Web from releases'
        exit 62836
    fi
    cd "$INSTALL_DIR" || exit 2468724628

    # check the hash
    curr_hash=$(sha256sum ${RIOT_FILENAME}.tar.gz | awk -F ' ' '{print $1}')
    if [[ "$curr_hash" != "$RIOT_HASH" ]]; then
        echo $'Riot download hash does not match'
        exit 78352
    fi

    tar -xzvf ${RIOT_FILENAME}.tar.gz
    if [ ! -d "$INSTALL_DIR/${RIOT_FILENAME}" ]; then
        echo $'Unable to extract Riot Web tarball'
        exit 542826
    fi
    cp -r "$INSTALL_DIR/${RIOT_FILENAME}/"* "/var/www/$RIOT_DOMAIN_NAME/htdocs"

    chown -R www-data:www-data "/var/www/$RIOT_DOMAIN_NAME/htdocs"
}

function upgrade_riot {
    if ! grep -q 'riot version:' "$COMPLETION_FILE"; then
        return
    fi

    CURR_RIOT_VERSION=$(get_completion_param "riot version")
    echo "riot current version: ${CURR_RIOT_VERSION}"
    echo "riot app version: ${RIOT_VERSION}"
    if [[ "${CURR_RIOT_VERSION}" == "${RIOT_VERSION}" ]]; then
        return
    fi

    riot_download
    sed -i "s|riot version.*|riot version:$RIOT_VERSION|g" "${COMPLETION_FILE}"
    riot_remove_bad_links

    systemctl restart nginx
}

function backup_local_riot {
    echo -n ''
}

function restore_local_riot {
    echo -n ''
}

function backup_remote_riot {
    echo -n ''
}

function restore_remote_riot {
    echo -n ''
}

function remove_riot {
    function_check remove_onion_service
    remove_onion_service riot ${RIOT_ONION_PORT}

    nginx_dissite $RIOT_DOMAIN_NAME
    if [ -f /etc/nginx/sites-available/$RIOT_DOMAIN_NAME ]; then
        rm /etc/nginx/sites-available/$RIOT_DOMAIN_NAME
    fi

    if [ -d /var/www/$RIOT_DOMAIN_NAME ]; then
        rm -rf /var/www/$RIOT_DOMAIN_NAME
    fi

    remove_completion_param install_riot
    sed -i '/riot /d' "$COMPLETION_FILE"
}

function install_riot {
    if [[ $ONION_ONLY != 'no' ]]; then
        return
    fi

    # check that matrix has been installed
    if [ ! "$MATRIX_DOMAIN_NAME" ]; then
        exit 687292
    fi
    if [[ "$MATRIX_DOMAIN_NAME" != *'.'* ]]; then
        exit 256288
    fi
    if [ ! -d /var/lib/matrix ]; then
        exit 827334
    fi

    function_check get_completion_param
    MATRIX_ONION_DOMAIN_NAME=$(get_completion_param "matrix onion domain")

    apt-get -yq install wget

    if [ ! -d /var/www/$RIOT_DOMAIN_NAME/htdocs ]; then
        mkdir -p /var/www/$RIOT_DOMAIN_NAME/htdocs
    fi

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

    riot_download

    cd "/var/www/$RIOT_DOMAIN_NAME/htdocs" || exit 4628462876

    if [[ "$ONION_ONLY" == 'no' ]]; then
        riot_config_file="config.${RIOT_DOMAIN_NAME}.json"
        cp config.sample.json $riot_config_file
        sed -i "s|\"default_hs_url\":.*|\"default_hs_url\": \"https://${MATRIX_DOMAIN_NAME}\",|g" "$riot_config_file"
        sed -i "s|\"default_is_url\":.*|\"default_is_url\": \"https://${MATRIX_DOMAIN_NAME}\",|g" "$riot_config_file"
        sed -i "s|\"bug_report_endpoint_url\":.*|\"bug_report_endpoint_url\": \"https://${MATRIX_DOMAIN_NAME}/bugs\",|g" "$riot_config_file"
        sed -i "/\"servers\":/a \"${MATRIX_DOMAIN_NAME}\"," "$riot_config_file"
    else
        riot_config_file="config.${MATRIX_ONION_DOMAIN_NAME}.json"
        cp config.sample.json "$riot_config_file"
        sed -i "s|\"default_hs_url\":.*|\"default_hs_url\": \"http://${MATRIX_ONION_DOMAIN_NAME}\",|g" "$riot_config_file"
        sed -i "s|\"default_is_url\":.*|\"default_is_url\": \"http://${MATRIX_ONION_DOMAIN_NAME}\",|g" "$riot_config_file"
        sed -i "s|\"bug_report_endpoint_url\":.*|\"bug_report_endpoint_url\": \"http://${MATRIX_ONION_DOMAIN_NAME}/bugs\",|g" "$riot_config_file"
        sed -i "/\"servers\":/a \"${MATRIX_ONION_DOMAIN_NAME}\"," "$riot_config_file"
    fi
    sed -i "s|\"integrations_ui_url\":.*|\"integrations_ui_url\": \"\",|g" "$riot_config_file"
    sed -i "s|\"integrations_rest_url\":.*|\"integrations_rest_url\": \"\",|g" "$riot_config_file"
    sed -i 's|https://piwik.riot.im/||g' "$riot_config_file"

    RIOT_ONION_HOSTNAME=$(add_onion_service riot 80 ${RIOT_ONION_PORT})

    riot_nginx_site=/etc/nginx/sites-available/$RIOT_DOMAIN_NAME
    if [[ $ONION_ONLY == "no" ]]; then
        function_check nginx_http_redirect
        nginx_http_redirect $RIOT_DOMAIN_NAME
        { echo 'server {';
          echo '  listen 443 ssl;';
          echo '  #listen [::]:443 ssl;';
          echo "  server_name $RIOT_DOMAIN_NAME;";
          echo '';
          echo '  # Security'; } >> $riot_nginx_site
        function_check nginx_ssl
        nginx_ssl $RIOT_DOMAIN_NAME

        function_check nginx_security_options
        nginx_security_options $RIOT_DOMAIN_NAME

        { echo '  add_header Strict-Transport-Security max-age=15768000;';
          echo '';
          echo '  # Logs';
          echo '  access_log /dev/null;';
          echo '  error_log /dev/null;';
          echo '';
          echo '  # Root';
          echo "  root /var/www/$RIOT_DOMAIN_NAME/htdocs;";
          echo '';
          echo '  index index.html;';
          echo '';
          echo '  location / {'; } >> $riot_nginx_site
        function_check nginx_limits
        nginx_limits $RIOT_DOMAIN_NAME '15m'
        { echo '  }';
          echo '}';
          echo ''; } >> $riot_nginx_site
    else
        echo -n '' > $riot_nginx_site
    fi
    { echo 'server {';
      echo "    listen 127.0.0.1:$RIOT_ONION_PORT default_server;";
      echo "    server_name $RIOT_ONION_HOSTNAME;";
      echo ''; } >> $riot_nginx_site
    function_check nginx_security_options
    nginx_security_options $RIOT_DOMAIN_NAME
    { echo '';
      echo '  # Logs';
      echo '  access_log /dev/null;';
      echo '  error_log /dev/null;';
      echo '';
      echo '  # Root';
      echo "  root /var/www/$RIOT_DOMAIN_NAME/htdocs;";
      echo '';
      echo '  index index.html;';
      echo '';
      echo '  location / {'; } >> $riot_nginx_site
    function_check nginx_limits
    nginx_limits $RIOT_DOMAIN_NAME '15m'
    echo '  }' >> "$riot_nginx_site"
    echo '}' >> "$riot_nginx_site"

    sed '/Content-Security-Policy/d' "$riot_nginx_site"
    sed -i 's| DENY;| SAMEORIGIN;|g' "$riot_nginx_site"
    sed -i 's|limit_conn conn_limit_per_ip.*|limit_conn conn_limit_per_ip 50;|g' "$riot_nginx_site"
    sed -i 's|limit_req zone.*|limit_req zone=req_limit_per_ip burst=20 nodelay;|g' "$riot_nginx_site"

    function_check create_site_certificate
    if [ ! -f "/etc/ssl/certs/${RIOT_DOMAIN_NAME}.pem" ]; then
        create_site_certificate "$RIOT_DOMAIN_NAME" 'yes'
    fi

    function_check nginx_ensite
    nginx_ensite $RIOT_DOMAIN_NAME

    function_check add_ddns_domain
    add_ddns_domain $RIOT_DOMAIN_NAME

    riot_remove_bad_links
    chown -R www-data:www-data /var/www/$RIOT_DOMAIN_NAME/htdocs

    systemctl restart nginx

    set_completion_param "riot domain" "$RIOT_DOMAIN_NAME"
    if ! grep -q "riot version:" "${COMPLETION_FILE}"; then
        echo "riot version:${RIOT_VERSION}" >> "${COMPLETION_FILE}"
    else
        sed -i "s|riot version.*|riot version:${RIOT_VERSION}|g" "${COMPLETION_FILE}"
    fi
    APP_INSTALLED=1
}

# NOTE: deliberately no exit 0