#!/bin/bash

if [ "$#" -ne 3 ]; then
    echo "Использование: $0 <путь_к_jnlp> <логин> <пароль>"
    exit 1
fi

JNLP_FILE="$1"
USER="$2"
PASS="$3"

if [ ! -f "$JNLP_FILE" ]; then
    echo "Ошибка: файл $JNLP_FILE не найден."
    exit 1
fi

# Удаляем возможные символы возврата каретки (Windows)
sed -i 's/\r$//' "$JNLP_FILE"

# 1. Извлекаем IP-адрес (codebase)
CODEBASE=$(grep -oP 'codebase="[^"]+"' "$JNLP_FILE" | head -1 | cut -d'"' -f2)
IP=$(echo "$CODEBASE" | sed -E 's|https?://||' | cut -d':' -f1 | cut -d'/' -f1)
PORT=$(echo "$CODEBASE" | sed -E 's|https?://[^:/]+:?||' | cut -d'/' -f1)
if [ -z "$PORT" ]; then PORT="80"; fi

# 2. Извлекаем имя основного JAR
JAR_NAME=$(grep -oP '<jar[^>]*>' "$JNLP_FILE" | tr -d '\n' | grep -oP 'href="[^"]+"' | head -1 | cut -d'"' -f2)
if [ -z "$JAR_NAME" ]; then
    JAR_NAME=$(grep -oP 'href="[^"]+\.jar[^"]*"' "$JNLP_FILE" | head -1 | cut -d'"' -f2)
fi

# 3. Извлекаем имя нативной библиотеки для Linux x86_64
NAT_NAME=$(awk '/<resources os="Linux" arch="(x86_64|amd64)">/,/<\/resources>/' "$JNLP_FILE" | grep -oP 'href="[^"]+"' | head -1 | cut -d'"' -f2)
if [ -z "$NAT_NAME" ]; then
    NAT_NAME=$(awk '/<resources os="Linux"/,/<\/resources>/' "$JNLP_FILE" | grep -oP 'href="[^"]+"' | head -1 | cut -d'"' -f2)
fi

# Очищаем имена от возможных суффиксов
JAR_BASE=$(echo "$JAR_NAME" | sed 's/\.pack\.gz$//;s/\.pack$//;s/\.jar$//')
NAT_BASE=$(echo "$NAT_NAME" | sed 's/\.pack\.gz$//;s/\.pack$//;s/\.jar$//')

if [ -z "$JAR_BASE" ] || [ -z "$NAT_BASE" ]; then
    echo "Ошибка: не удалось найти JAR или нативную библиотеку в JNLP."
    exit 1
fi

WORK_DIR="/tmp/ikvm_${IP}"
mkdir -p "$WORK_DIR"
cd "$WORK_DIR" || exit 1

echo "[*] Сервер: $IP:$PORT"
echo "[*] Основной JAR: ${JAR_BASE}.jar"
echo "[*] Нативная библиотека: ${NAT_BASE}.jar"

# 4. Скачивание и распаковка
for BASE in "$JAR_BASE" "$NAT_BASE"; do
    if [ ! -f "${BASE}.jar" ]; then
        echo "[*] Скачивание ${BASE}.jar.pack.gz..."
        wget -q "http://${IP}:${PORT}/${BASE}.jar.pack.gz" -O "${BASE}.jar.pack.gz"
        if [ $? -eq 0 ]; then
            gunzip -f "${BASE}.jar.pack.gz"
            /usr/lib/jvm/java-8-openjdk-amd64/jre/bin/unpack200 "${BASE}.jar.pack" "${BASE}.jar" 2>/dev/null
            rm -f "${BASE}.jar.pack"
        else
            echo "[*] Файл ${BASE}.jar.pack.gz не найден, пробуем ${BASE}.jar..."
            wget -q "http://${IP}:${PORT}/${BASE}.jar"
        fi
    fi
done

# 5. Извлечение нативной библиотеки .so
if [ -f "${NAT_BASE}.jar" ]; then
    unzip -o -q "${NAT_BASE}.jar"
fi

if [ ! -f "${JAR_BASE}.jar" ]; then
    echo "Ошибка: главный JAR файл не найден."
    exit 1
fi

# 6. Запуск консоли
echo "[*] Запуск консоли..."
java -Djava.library.path=. -cp "${JAR_BASE}.jar" tw.com.aten.ikvm.KVMMain "$IP" "$USER" "$PASS" null 5900 623 2 0