#!/bin/bash
# by DSR! from https://github.com/xchwarze/wifi-pineapple-cloner

ARCHITECTURE="$1"
FLAVOR="$2"
IMAGEBUILDER_FOLDER="$3"
PROFILE="$4"
declare -a ARCHITECTURE_TYPES=("mips" "mipsel")
declare -a FLAVOR_TYPES=("nano" "tetra" "universal")
if [[ ! -d "$IMAGEBUILDER_FOLDER" || "$PROFILE" == "" ]] || ! grep -q "$ARCHITECTURE" <<< "${ARCHITECTURE_TYPES[*]}" || ! grep -q "$FLAVOR" <<< "${FLAVOR_TYPES[*]}"; then
    echo "Run with \"builder.sh [ARCHITECTURE] [FLAVOR] [IMAGEBUILDER_FOLDER] [PROFILE]\""
    echo "    ARCHITECTURE        -> must be one of these values: mips, mipsel"
    echo "    FLAVOR              -> must be one of these values: nano, tetra, universal"
    echo "    IMAGEBUILDER_FOLDER -> path to openwrt imagebuilder"
    echo "    PROFILE             -> profile for use in imagebuilder build"

    exit 1
fi

# the order in which the dependencies are installed and those that are removed is deliberate!
# no rtl-sdr, no kmod-usb-net-*, no kmod-rtl8192cu, no kmod-usb-acm
PACKAGES_NANO="iw at autossh base-files block-mount ca-certificates chat dnsmasq e2fsprogs ethtool firewall hostapd-utils ip6tables iperf3 iwinfo kmod-crypto-manager kmod-fs-ext4 kmod-fs-nfs kmod-fs-vfat kmod-gpio-button-hotplug kmod-ipt-offload kmod-leds-gpio kmod-ledtrig-default-on kmod-ledtrig-netdev kmod-ledtrig-timer kmod-mt76x2u kmod-nf-nathelper kmod-rt2800-usb kmod-rtl8187 kmod-scsi-generic kmod-usb-ohci kmod-usb-storage-extras kmod-usb-uhci kmod-usb2 libbz2-1.0 libcurl4 libelf1 libffi libgmp10 libiconv-full2 libintl libltdl7 libnet-1.2.x libnl200 libreadline8 libustream-mbedtls20150806 libxml2 logd macchanger mtd nano ncat netcat nginx openssh-client openssh-server openssh-sftp-server openssl-util php7-fpm php7-mod-hash php7-mod-json php7-mod-mbstring php7-mod-session php7-mod-sqlite3 ppp ppp-mod-pppoe procps-ng-pkill procps-ng-ps python-logging python-openssl python-sqlite3 ssmtp tcpdump-mini uci uclibcxx uclient-fetch urandom-seed urngd usb-modeswitch usbreset usbutils wget wireless-tools wpad busybox libatomic1 libstdcpp6 -wpad-basic -dropbear -swconfig -odhcpd-ipv6only -odhcp6c"

# no rtl-sdr, no kmod-usb-net-*, no kmod-usb-serial-*, no kmod-rtl8192cu, no kmod-usb-acm, no kmod-usb-wdm, no kmod-lib-crc-itu-t
PACKAGES_TETRA="iw at autossh base-files bash block-mount ca-certificates chat dnsmasq e2fsprogs ethtool firewall hostapd-utils ip6tables iwinfo kmod-crypto-manager kmod-fs-ext4 kmod-fs-nfs kmod-fs-vfat kmod-gpio-button-hotplug kmod-ipt-offload kmod-leds-gpio kmod-ledtrig-default-on kmod-ledtrig-netdev kmod-ledtrig-timer kmod-mt76x2u kmod-nf-nathelper kmod-rt2800-usb kmod-rtl8187 kmod-scsi-generic kmod-usb-ohci kmod-usb-storage-extras kmod-usb-uhci kmod-usb2 libbz2-1.0 libcurl4 libelf1 libffi libgdbm libgmp10 libiconv-full2 libltdl7 libnet-1.2.x libnl200 libustream-mbedtls20150806 libxml2 logd macchanger mtd nano ncat netcat nginx openssh-client openssh-server openssh-sftp-server openssl-util php7-fpm php7-mod-hash php7-mod-json php7-mod-mbstring php7-mod-session php7-mod-sqlite3 ppp ppp-mod-pppoe procps-ng-pkill procps-ng-ps python-logging python-openssl python-sqlite3 ssmtp tcpdump-mini uci uclibcxx uclient-fetch urandom-seed urngd usb-modeswitch usbreset usbutils wget wireless-tools wpad busybox libatomic1 libstdcpp6 -wpad-basic -dropbear -odhcp6c -odhcpd-ipv6only"

# if you don't install a custom build of busybox you have to install fdisk
# no rtl-sdr, no kmod-usb-net-*, no kmod-usb-serial-*, no kmod-rtl8192cu, no kmod-usb-acm, no kmod-usb-wdm, no kmod-lib-crc-itu-t, no ppp*, no python-*
PACKAGES_UNIVERSAL="iw at autossh base-files bash block-mount ca-certificates chat dnsmasq e2fsprogs ethtool firewall hostapd-utils ip6tables iwinfo kmod-crypto-manager kmod-fs-ext4 kmod-fs-nfs kmod-fs-vfat kmod-gpio-button-hotplug kmod-ipt-offload kmod-leds-gpio kmod-ledtrig-default-on kmod-ledtrig-netdev kmod-ledtrig-timer kmod-mt76x2u kmod-nf-nathelper kmod-rt2800-usb kmod-rtl8187 kmod-scsi-generic kmod-usb-ohci kmod-usb-storage-extras kmod-usb-uhci kmod-usb2 libbz2-1.0 libcurl4 libelf1 libffi libgdbm libgmp10 libiconv-full2 libltdl7 libnet-1.2.x libnl200 libustream-mbedtls20150806 libxml2 logd macchanger mtd nano ncat netcat nginx openssh-client openssh-server openssh-sftp-server openssl-util php7-fpm php7-mod-hash php7-mod-json php7-mod-mbstring php7-mod-session php7-mod-sqlite3 procps-ng-pkill procps-ng-ps ssmtp tcpdump-mini uci uclibcxx uclient-fetch urandom-seed urngd usb-modeswitch usbreset usbutils wget wireless-tools wpad busybox libatomic1 libstdcpp6 -wpad-basic -dropbear -odhcpd-ipv6only -ppp -ppp-mod-pppoe"

# add missing deps and custom busybox build
declare -a FORCE_PACKAGES=("libubus20191227_2019-12-27-041c9d1c-1" "busybox_1.30.1-6")

# fw hash
FW_TETRA_HASH="081278abfc227078b642b80c6a965c26b11bf820"
FW_NANO_HASH="3e087e6e9256431f2022e8e70373569d57f7ec77"
FW_MK7_HASH="32ba9cc3fd7e1b4489a14c1b3c5b4af118dff810"

# fix folders paths
IMAGEBUILDER_FOLDER="$(realpath $IMAGEBUILDER_FOLDER)"
TOOL_FOLDER="$(realpath $(dirname $0)/../tools)"
BUILD_FOLDER="$(realpath $(dirname $0)/../build)"
FW_FOLDER="$(realpath $(dirname $0)/../build/fw-base)"



# steps
check_deps () {
    echo "[*] Checking dependencies"
    echo "******************************"

    if [ ! -f "$FW_FOLDER/basefw.bin" ]; then
        VARIANT=$([[ "$FLAVOR" == "nano" ]] && echo "NANO" || echo "TETRA")

        echo "[!] Attention!"
        echo ""
        echo "File 'build/fw-base/basefw.bin' was not found."
        echo "You must download the original $VARIANT firmware version 2.7.0 and place it with the name basefw.bin in this path:"
        echo "$FW_FOLDER"
        echo ""

        exit 1
    fi

    if [ ! -f "$FW_FOLDER/basefw-mk7.bin" ] && [ "$ARCHITECTURE" == "mipsel" ]; then
        echo "[!] Attention!"
        echo ""
        echo "File 'build/fw-base/basefw-mk7.bin' was not found."
        echo "You must download the original MK7 firmware version 1.1.1 and place it with the name basefw-mk7.bin in this path:"
        echo "$FW_FOLDER"
        echo ""

        exit 1
    fi

    echo "  [+] Check completed"
    echo ""
}

prepare_builder () {
    echo "[*] Prepare builder"
    echo "******************************"

    PACKAGES_ARQ="${ARCHITECTURE}_24kc"
    DOWNLOAD_BASE_URL="https://github.com/xchwarze/wifi-pineapple-community/raw/main/packages/experimental"

    for TARGET in ${FORCE_PACKAGES[@]}; do
        PACKAGE_IPK="${TARGET}_${PACKAGES_ARQ}.ipk"
        PACKAGE_PATH="$IMAGEBUILDER_FOLDER/packages/$PACKAGE_IPK"
        if [ ! -f "$PACKAGE_PATH" ] || [ ! -s "$PACKAGE_PATH" ]; then
            echo "  [+] Install: $TARGET"
            wget -q "$DOWNLOAD_BASE_URL/$PACKAGES_ARQ/$PACKAGE_IPK" -O "$PACKAGE_PATH"
        else
            echo "  [+] Already exist: $TARGET"
        fi
    done

    echo "  [+] Builder setup complete"
    echo ""
}

prepare_build () {
    echo "[*] Prepare build"
    echo "******************************"

    # clean
    rm -rf "$FW_FOLDER/_basefw.*"
    rm -rf "$BUILD_FOLDER/rootfs-base"
    rm -rf "$BUILD_FOLDER/rootfs"
    rm -rf "$BUILD_FOLDER/release"
    mkdir "$BUILD_FOLDER/release"

    # get target firmware
    # this work only with lastest binwalk version!
    if [[ "$FLAVOR" == "tetra" || "$FLAVOR" == "universal" ]]; then
        #echo "  [+] Downloading TETRA firmware..."
        #wget -q https://github.com/xchwarze/wifi-pineapple-community/raw/main/firmwares/2.7.0-tetra.bin -O "$FW_FOLDER/basefw.bin"
        
        FW_HASH=$(sha1sum "$FW_FOLDER/basefw.bin" | awk '{ print $1 }')
        if [ "$FW_HASH" != "$FW_TETRA_HASH" ]; then
            echo "  [!] WARNING file basefw.bin hash does not match"
        fi

        echo "  [+] Unpack firmware for get file system"
        binwalk "$FW_FOLDER/basefw.bin" -e 
        binwalk "$FW_FOLDER/_basefw.bin.extracted/sysupgrade-pineapple-tetra/root" -e --preserve-symlinks
        mv "$FW_FOLDER/_basefw.bin.extracted/sysupgrade-pineapple-tetra/_root.extracted/squashfs-root/" "$BUILD_FOLDER/rootfs-base"
    else
        #echo "  [+] Downloading NANO firmware..."
        #wget -q https://github.com/xchwarze/wifi-pineapple-community/raw/main/firmwares/2.7.0-nano.bin -O "$FW_FOLDER/basefw.bin"

        FW_HASH=$(sha1sum "$FW_FOLDER/basefw.bin" | awk '{ print $1 }')
        if [ "$FW_HASH" != "$FW_NANO_HASH" ]; then
            echo "  [!] WARNING file basefw.bin hash does not match"
        fi

        echo "  [+] Unpack firmware for get file system"
        binwalk "$FW_FOLDER/basefw.bin" -e --preserve-symlinks
        mv "$FW_FOLDER/_basefw.bin.extracted/squashfs-root/" "$BUILD_FOLDER/rootfs-base"
    fi

    echo "  [+] Copying the original files"
    "$TOOL_FOLDER/copier.sh" "$TOOL_FOLDER/../lists/$FLAVOR.filelist" "$BUILD_FOLDER/rootfs-base" "$BUILD_FOLDER/rootfs"
    if [ $? -ne 0 ]; then
        echo "  [!] An error occurred while copying the original files. Check the log for errors."
        exit 1
    fi

    rm -rf "$BUILD_FOLDER/rootfs-base"
    rm -rf "$FW_FOLDER"/_basefw.*

    if [[ "$ARCHITECTURE" == "mipsel" ]]; then
        #echo "  [+] Downloading old MK7 firmware..."
        #wget -q https://github.com/xchwarze/wifi-pineapple-community/raw/main/firmwares/1.1.1-mk7.bin -O "$FW_FOLDER/basefw-mk7.bin"

        FW_HASH=$(sha1sum "$FW_FOLDER/basefw-mk7.bin" | awk '{ print $1 }')
        if [ "$FW_HASH" != "$FW_MK7_HASH" ]; then
            echo "  [!] WARNING file basefw-mk7.bin hash does not match"
        fi

        echo "  [+] Unpack mipsel firmware for get file system"
        binwalk "$FW_FOLDER/basefw-mk7.bin" -e --preserve-symlinks
        mv "$FW_FOLDER/_basefw-mk7.bin.extracted/squashfs-root/" "$BUILD_FOLDER/rootfs-mk7-base"

        echo "  [+] Copying the original mipsel files"
        "$TOOL_FOLDER/copier.sh" "$TOOL_FOLDER/../lists/mipsel-support.filelist" "$BUILD_FOLDER/rootfs-mk7-base" "$BUILD_FOLDER/rootfs" "true"
        if [ $? -ne 0 ]; then
            echo "  [!] An error occurred while copying the original mipsel files. Check the log for errors."
            exit 1
        fi

        rm -rf "$BUILD_FOLDER/rootfs-mk7-base"
        rm -rf "$FW_FOLDER"/_basefw-mk7.*
    fi

    echo "  [+] Patch file system"
    "$TOOL_FOLDER/fs-patcher.sh" "$ARCHITECTURE" "$FLAVOR" "$BUILD_FOLDER/rootfs"
    if [ $? -ne 0 ]; then
        echo "  [!] An error occurred during the execution of the process. Check the log for errors."
        exit 1
    fi

    echo ""
}

build () {
    echo "[*] Build"
    echo "******************************"

    # clean
    echo "  [+] Clean last build data"
    #make clean
    rm -rf "$IMAGEBUILDER_FOLDER/tmp/"
    rm -rf "$IMAGEBUILDER_FOLDER/build_dir"/target-*/root*
    rm -rf "$IMAGEBUILDER_FOLDER/build_dir"/target-*/json_*
    rm -rf "$IMAGEBUILDER_FOLDER/bin/targets/*"

    # set selected packages
    echo "  [+] Executing make"
    selected_packages="$PACKAGES_UNIVERSAL"
    if [[ "$FLAVOR" == "nano" ]];
    then
        selected_packages="$PACKAGES_NANO"
    elif [[ "$FLAVOR" == "tetra" ]];
    then
        selected_packages="$PACKAGES_TETRA"
    fi

    # build
    cd "$IMAGEBUILDER_FOLDER"
    make image PROFILE="$1" PACKAGES="$selected_packages" FILES="$BUILD_FOLDER/rootfs" BIN_DIR="$BUILD_FOLDER/release" > "$BUILD_FOLDER/release/make.log"
    if [ $? -ne 0 ]; then
        echo ""
        echo "  [!] An error occurred in the build process. Check file release/make.log for more information."
        exit 1
    fi

    # add this second check for build process
    checkFwFileExist=$(ls "$BUILD_FOLDER/release"/*-sysupgrade.* 2>/dev/null | wc -l)
    if [ $checkFwFileExist -eq 0 ]; then
        echo ""
        echo "  [!] OpenWRT finished the build process but no firmware was found. Check the release/make.log to see if the process was completed correctly."
        #exit 1
    fi
    echo ""
}



# implement this shitty logic
echo "Wifi Pineapple Cloner - builder"
echo "************************************** by DSR!"
echo ""

check_deps
prepare_builder
prepare_build
build "$PROFILE"

echo "[*] Firmware folder: $BUILD_FOLDER/release"
echo "******************************"
ls -l "$BUILD_FOLDER/release"
echo ""