#!/usr/bin/env bash

# Copyright (C) 2025-2026 Dmitry Messerman
# SPDX-License-Identifier: GPL-3.0

# Avoid interactive prompts during apt install
export DEBIAN_FRONTEND=noninteractive

# ---------------------------------------------------------------------------
# 1. System packages
# ---------------------------------------------------------------------------
apt-get update && apt-get install -y \
        g++ \
        gcc \
        git \
        glpk-utils \
        jq \
        libgfortran5 \
        libgmp-dev \
        libgomp1 \
        libmpfr6 \
        locales \
        python3 \
        python-is-python3 \
        python3-dev \
        python3-pip \
        python3-pytest-mock \
        python3-setuptools \
        python3-tk \
        python3-venv \
        python3-z3 \
        tcsh \
        tzdata \
        vim \
        wget \
        x11-apps \
        x11-xserver-utils \
        xvfb \
        z3 

if [[ $? -ne 0 ]]; then
    echo -e "\nERROR: system packages installation failed\n"
    exit 1
fi    

# ---------------------------------------------------------------------------
# 2. python3.11 and python3.13
# ---------------------------------------------------------------------------

apt-get install -y --no-install-recommends \
        gnupg \
        curl \
        ca-certificates && \
        curl -fsSL "https://keyserver.ubuntu.com/pks/lookup?op=get&search=0xF23C5A6CF475977595C89F51BA6932366A755776" \
            | gpg --dearmor -o /etc/apt/trusted.gpg.d/deadsnakes.gpg && \
        echo "deb https://ppa.launchpadcontent.net/deadsnakes/ppa/ubuntu noble main" \
            > /etc/apt/sources.list.d/deadsnakes.list && \
    apt-get update && \
    apt-get install -y --no-install-recommends \
        python3.11 \
        python3.11-venv \
        python3.11-dev \
        python3.11-tk \
        python3.13 \
        python3.13-venv \
        python3.13-dev \
        python3.13-tk && \
    apt-get clean && rm -rf /var/lib/apt/lists/*

if [[ $? -ne 0 ]]; then
    echo -e "\nERROR: python3.11 and/or python3.13 installation failed\n"
    exit 1
fi    

# ---------------------------------------------------------------------------
# 3. Conda 
# ---------------------------------------------------------------------------
wget --tries=5 --timeout=30 --waitretry=2 https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh && \
    bash ./Miniconda3-latest-Linux-x86_64.sh -b -p /opt/miniconda3 && \
    /opt/miniconda3/bin/conda tos accept --override-channels --channel https://repo.anaconda.com/pkgs/main && \
    /opt/miniconda3/bin/conda tos accept --override-channels --channel https://repo.anaconda.com/pkgs/r
    
if [[ $? -ne 0 ]]; then
    echo -e "\nERROR: conda installation failed\n"
    exit 1
fi    

# ---------------------------------------------------------------------------
# 4. scip and ipopt
# ---------------------------------------------------------------------------
/opt/miniconda3/bin/conda install -y -c conda-forge ipopt scip && \
    ln -sf /opt/miniconda3/bin/scip /usr/local/bin && \
    ln -sf /opt/miniconda3/bin/ipopt /usr/local/bin && \
    rm ./Miniconda3-latest-Linux-x86_64.sh

if [[ $? -ne 0 ]]; then
    echo -e "\nERROR: SCIP and/or IPOPT installation failed\n"
    exit 1
fi    

# ---------------------------------------------------------------------------
# 5. Mathsat
# ---------------------------------------------------------------------------
current_dir=$PWD
MATHSAT=mathsat-5.6.15
ARCH=$(uname -m)
MATHSAT_LINUX=${MATHSAT}-linux-${ARCH}
MATHSAT_LINUX_ARCHIVE=${MATHSAT}-linux-${ARCH}.tar.gz
wget --tries=5 --timeout=30 --waitretry=2 https://mathsat.fbk.eu/release/${MATHSAT_LINUX_ARCHIVE}
if [[ $? -ne 0 ]]; then
    echo -e "\nERROR: $MATHSAT_LINUX_ARCHIVE download failed\n"
    exit 1
fi
for pyver in 3.11 3.12 3.13; do
    cpython_tag="cpython-$(echo $pyver | tr -d '.')"
    dist_dir=/usr/local/lib/python${pyver}/dist-packages
    tar -xvf ${MATHSAT_LINUX_ARCHIVE} && \
        cd ${MATHSAT_LINUX}/python && \
        python${pyver} setup.py build && \
        mkdir -p ${dist_dir} && \
        mv -f mathsat.py ${dist_dir} && \
        mv -f build/lib.linux-${ARCH}-${cpython_tag}/_mathsat.${cpython_tag}-${ARCH}-linux-gnu.so ${dist_dir}/_mathsat.so && \
        cd $current_dir && \
        rm -rf ${MATHSAT_LINUX}
        if [[ $? -ne 0 ]]; then
            echo -e "\nERROR: Mathsat python${pyver} installation failed\n"
            exit 1
        fi    
done
rm -f $MATHSAT_LINUX_ARCHIVE


# ---------------------------------------------------------------------------
# 6. UTF-8 fonts
# ---------------------------------------------------------------------------
locale-gen en_US.UTF-8
update-locale LANG=en_US.UTF-8 LC_ALL=en_US.UTF-8 LANGUAGE=en_US:en

if [[ $? -ne 0 ]]; then
    echo -e "\nERROR: UTF-8 fonts installation failed\n"
    exit 1
fi