#!/usr/bin/env bash

# Memcached Installer
# Min. Requirement  : GNU/Linux Ubuntu 14.04 & 16.04
# Last Build        : 17/07/2019
# Author            : ESLabs.ID (eslabs.id@gmail.com)
# Since Version     : 1.0.0

# Include helper functions.
if [ "$(type -t run)" != "function" ]; then
    BASEDIR=$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )
    # shellchechk source=scripts/helper.sh
    # shellcheck disable=SC1090
    . "${BASEDIR}/helper.sh"
fi

# Make sure only root can run this installer script.
requires_root

function enable_memcache {
    # PHP version.
    local PHPv="${1}"
    if [ -z "${PHPv}" ]; then
        PHPv=${PHP_VERSION:-"7.3"}
    fi

    if "${DRYRUN}"; then
        echo "Optimizing PHP Memcache module in dryrun mode."
    else
        # Custom Memcache setting.
        local RAM_SIZE && \
        RAM_SIZE=$(get_ram_size)
        if [[ ${RAM_SIZE} -le 1024 ]]; then
            # If machine RAM less than / equal 1GiB, set Memcached to 1/8 of RAM size.
            local MEMCACHED_SIZE=$((RAM_SIZE / 8))
        elif [[ ${RAM_SIZE} -gt 2048 && ${RAM_SIZE} -le 8192 ]]; then
            # If machine RAM less than / equal 8GiB and greater than 2GiB, set Memcached to 1/4 of RAM size.
            local MEMCACHED_SIZE=$((RAM_SIZE / 4))
        else
            # Otherwise, set Memcached to max of 2048GiB.
            local MEMCACHED_SIZE=2048
        fi
        sed -i "s/-m 64/-m ${MEMCACHED_SIZE}/g" /etc/memcached.conf

        if [ -d "/etc/php/${PHPv}/mods-available/" ]; then
            cat >> "/etc/php/${PHPv}/mods-available/memcache.ini" <<EOL

; Optimized for LEMPer stack
memcache.dbpath="/var/lib/memcache"
memcache.maxreclevel=0
memcache.maxfiles=0
memcache.archivememlim=0
memcache.maxfilesize=0
memcache.maxratio=0
; custom setting for WordPress + W3TC
session.bak_handler = memcache
session.bak_path = "tcp://127.0.0.1:11211"
EOL

            # Reload PHP-FPM service
            if [[ $(pgrep -c "php-fpm${PHPv}") -gt 0 ]]; then
                run service "php${PHPv}-fpm" reload
                status "PHP${PHPv}-FPM restarted successfully."
            elif [[ -n $(command -v "php${PHPv}") ]]; then
                run service "php${PHPv}-fpm" start

                if [[ $(pgrep -c "php-fpm${PHPv}") -gt 0 ]]; then
                    status "PHP${PHPv}-FPM started successfully."
                else
                    warning "Something wrong with PHP & FPM ${PHPv} installation."
                fi
            fi

        else
            warning "It seems that PHP ${PHPv} not yet installed. Please install it before!"
        fi
    fi
}

function init_memcache_install() {
    if "${AUTO_INSTALL}"; then
        DO_INSTALL_MEMCACHED="y"
    else
        while [[ "${DO_INSTALL_MEMCACHED}" != "y" && "${DO_INSTALL_MEMCACHED}" != "n" ]]; do
            read -rp "Do you want to install Memcached server? [y/n]: " -i y -e DO_INSTALL_MEMCACHED
        done
    fi

    if [[ "${DO_INSTALL_MEMCACHED}" == y* && ${INSTALL_MEMCACHED} == true ]]; then
        echo "Installing Memcache and PHP memcached module..."

        # Install memcached
        run apt-get install -y libmemcached11 memcached php-igbinary \
            php-memcache php-memcached php-msgpack

        # Enable PHP module
        echo "Enabling PHP memcached module..."

        # Set PHP version to install.
        if [ -z "${PHP_VERSION}" ]; then PHP_VERSION="7.3"; fi

        if [ "${PHP_VERSION}" != "all" ]; then
            enable_memcache "${PHP_VERSION}"

            # Default PHP Required for LEMPer
            if [ "${PHP_VERSION}" != "7.3" ]; then
                enable_memcache "7.3"
            fi
        else
            enable_memcache "5.6"
            enable_memcache "7.0"
            enable_memcache "7.1"
            enable_memcache "7.2"
            enable_memcache "7.3"
        fi

        # Installation status.
        if "${DRYRUN}"; then
            warning -e "\nMemcache server installed in dryrun mode."
        else
            if [[ $(pgrep -c "php-fpm${PHPv}") -gt 0 ]]; then
                service memcached restart
                status -e "\nMemcache server installed successfully."
            else
                warning -e "\nSomething wrong with Memcache installation."
            fi
        fi 
    fi
}

echo "[Welcome to Memcached Installer]"
echo ""

# Start running things from a call at the end so if this script is executed
# after a partial download it doesn't do anything.
if [[ -n $(command -v memcached) ]]; then
    warning "Memcached server already exists. Installation skipped..."
else
    init_memcache_install "$@"
fi