package_app.sh

#!/bin/bash

# Configuration
# ---------------------------------------------------------
# PACKAGE_NAME: Must be lowercase, no spaces
PACKAGE_NAME="wiretray"

# MENU_NAME: What you see in the App Launcher
MENU_NAME="WireTray"

# Use the first argument as version, default to "1.0" if not provided
VERSION="${1:-1.0}" 
SOURCE_SCRIPT="wiretray.py" 
MAINTAINER="Jack Macqueen"
DESCRIPTION="System tray utility to manage WireGuard connections."

# Detect System Architecture (amd64, arm64, etc.)
ARCH=$(dpkg --print-architecture)
# ---------------------------------------------------------

# Check for PyInstaller
if ! command -v pyinstaller &> /dev/null; then
    echo "Error: PyInstaller is not installed."
    echo "Please run: pip install pyinstaller"
    exit 1
fi

echo "Building standalone binary with PyInstaller..."
# UPDATED: Added --add-data to bundle the SVG icon inside the binary
pyinstaller --noconfirm --onefile --windowed --clean \
            --add-data "icon.svg:." \
            --name "$PACKAGE_NAME" "$SOURCE_SCRIPT"

# Verify build succeeded
if [ ! -f "dist/$PACKAGE_NAME" ]; then
    echo "Error: PyInstaller build failed."
    exit 1
fi

# Create a temporary directory for building .deb
rm -rf build
BUILD_DIR="build/${PACKAGE_NAME}_${VERSION}_${ARCH}"
mkdir -p "$BUILD_DIR/DEBIAN"
mkdir -p "$BUILD_DIR/usr/local/bin"
mkdir -p "$BUILD_DIR/usr/share/applications"
# UPDATED: Create pixmaps directory for the system icon
mkdir -p "$BUILD_DIR/usr/share/pixmaps"

echo "Creating directory structure for $MENU_NAME (v$VERSION) [$ARCH]..."

# 1. Copy the Compiled Binary
cp "dist/$PACKAGE_NAME" "$BUILD_DIR/usr/local/bin/$PACKAGE_NAME"
chmod 755 "$BUILD_DIR/usr/local/bin/$PACKAGE_NAME"

# 2. UPDATED: Copy the Icon for the System Menu
# We rename it to match the PACKAGE_NAME so the .desktop file finds it easily
cp "icon.svg" "$BUILD_DIR/usr/share/pixmaps/$PACKAGE_NAME.svg"

# 3. Create the Control file (Metadata)
cat <<EOF > "$BUILD_DIR/DEBIAN/control"
Package: $PACKAGE_NAME
Version: $VERSION
Section: utils
Priority: optional
Architecture: $ARCH
Depends: wireguard-tools, libxcb-cursor0
Maintainer: $MAINTAINER
Description: $DESCRIPTION
 A system tray application to toggle WireGuard interfaces.
EOF

# 4. Create the Post-Installation Script
cat <<EOF > "$BUILD_DIR/DEBIAN/postinst"
#!/bin/bash
set -e

# Define the sudoers file path
SUDOERS_FILE="/etc/sudoers.d/$PACKAGE_NAME"

# Allow the 'sudo' group to run wg-quick without password
echo "%sudo ALL=(ALL) NOPASSWD: /usr/bin/wg-quick" > "\$SUDOERS_FILE"
chmod 0440 "\$SUDOERS_FILE"

# --- NEW: Fix WireGuard Directory Permissions ---
# This allows the GUI app (running as user) to list the .conf files
if [ -d "/etc/wireguard" ]; then
    echo "Updating /etc/wireguard permissions to allow listing configs..."
    chmod o+rx /etc/wireguard
fi

exit 0
EOF

chmod 755 "$BUILD_DIR/DEBIAN/postinst"

# 5. Create the Post-Removal Script
cat <<EOF > "$BUILD_DIR/DEBIAN/postrm"
#!/bin/bash
set -e
SUDOERS_FILE="/etc/sudoers.d/$PACKAGE_NAME"
if [ "\$1" = "remove" ]; then
    if [ -f "\$SUDOERS_FILE" ]; then
        rm "\$SUDOERS_FILE"
    fi
fi
exit 0
EOF

chmod 755 "$BUILD_DIR/DEBIAN/postrm"

# 6. Create the .desktop file
cat <<EOF > "$BUILD_DIR/usr/share/applications/$PACKAGE_NAME.desktop"
[Desktop Entry]
Name=$MENU_NAME
Comment=$DESCRIPTION
Exec=/usr/local/bin/$PACKAGE_NAME
# UPDATED: Use the package name (which maps to the file in /usr/share/pixmaps)
Icon=$PACKAGE_NAME
Terminal=false
Type=Application
Categories=Network;Utility;
EOF

# 7. Build the .deb package
DEB_FILE="${PACKAGE_NAME}_${VERSION}_${ARCH}.deb"
dpkg-deb --build "$BUILD_DIR" "$DEB_FILE"
rm -rf build dist "$PACKAGE_NAME.spec"

echo "Success! Package created at: $DEB_FILE"
echo "Run these commands to install:"
echo "sudo dpkg -i ./$DEB_FILE"
echo "sudo apt-get install -f"