#!/usr/bin/env bash

set -euxo pipefail
source script/lib/blob-store.sh

# Function for displaying help info
help_info() {
  echo "
Usage: ${0##*/} [options]
Build a release .tar.gz for Linux.

Options:
  -h, --help     Display this help and exit.
  --flatpak      Set ZED_BUNDLE_TYPE=flatpak so that this can be included in system info
  "
}

# Parse all arguments manually
while [[ $# -gt 0 ]]; do
    case $1 in
        -h|--help)
            help_info
            exit 0
            ;;
        --flatpak)
            export ZED_BUNDLE_TYPE=flatpak
            shift
            ;;
        --)
            shift
            break
            ;;
        -*)
            echo "Unknown option: $1" >&2
            help_info
            exit 1
            ;;
        *)
            echo "Error: Unexpected argument: $1" >&2
            help_info
            exit 1
            ;;
    esac
done

if [[ -n "${RUSTFLAGS+x}" || -n "${CARGO_ENCODED_RUSTFLAGS+x}" ]]; then
    echo "Unset RUSTFLAGS and CARGO_ENCODED_RUSTFLAGS; put extra flags in target rustflags in .cargo/config.toml so bundle flags are preserved." >&2
    exit 1
fi

export ZED_BUNDLE=true

channel=$(<crates/zed/RELEASE_CHANNEL)
target_dir="${CARGO_TARGET_DIR:-target}"

version="$(script/get-crate-version zed)"
# Set RELEASE_VERSION so it's compiled into GPUI and it knows about the version.
export RELEASE_VERSION="${version}"

commit=$(git rev-parse HEAD | cut -c 1-7)

version_info=$(rustc --version --verbose)
host_line=$(echo "$version_info" | grep host)
target_triple=${host_line#*: }
musl_triple=${target_triple%-gnu}-musl
remote_server_triple=${REMOTE_SERVER_TARGET:-"${musl_triple}"}
rustup_installed=false
if command -v rustup >/dev/null 2>&1; then
    rustup_installed=true
fi

# Generate the licenses first, so they can be baked into the binaries
script/generate-licenses

if "$rustup_installed"; then
    rustup target add "$remote_server_triple"
fi

export CC=${CC:-$(which clang)}

# Build binary in release mode
cargo --config .cargo/bundle-config.toml build --release --target "${target_triple}" --package zed --package cli --features remote/build-remote-server-binary
# Build remote_server in separate invocation to prevent feature unification from other crates
# from influencing dynamic libraries required by it.
if [[ "$remote_server_triple" == "$musl_triple" ]]; then
    musl_cc_var="CC_$(echo "$remote_server_triple" | tr '-' '_')"
    export "$musl_cc_var"=musl-gcc
fi
cargo --config .cargo/bundle-config.toml build --release --target "${remote_server_triple}" --package remote_server

# Upload debug info to sentry.io
if ! command -v sentry-cli >/dev/null 2>&1; then
    echo "sentry-cli not found. skipping sentry upload."
    echo "install with: 'curl -sL https://sentry.io/get-cli | bash'"
else
    if [[ -n "${SENTRY_AUTH_TOKEN:-}" ]]; then
        echo "Uploading zed debug symbols to sentry..."
        # note: this uploads the unstripped binary which is needed because it contains
        # .eh_frame data for stack unwinding. see https://github.com/getsentry/symbolic/issues/783
        for attempt in 1 2 3; do
            echo "Attempting sentry upload (attempt $attempt/3)..."
            if sentry-cli debug-files upload --include-sources --wait -p zed -o zed-dev \
                "${target_dir}/${target_triple}"/release/gwen \
                "${target_dir}/${remote_server_triple}"/release/remote_server; then
                echo "Sentry upload successful on attempt $attempt"
                break
            else
                echo "Sentry upload failed on attempt $attempt"
                if [ $attempt -eq 3 ]; then
                    echo "All sentry upload attempts failed"
                fi
            fi
        done
    else
        echo "missing SENTRY_AUTH_TOKEN. skipping sentry upload."
    fi
fi

# Split debug symbols into local .dbg files, then strip debug info and local
# symbols from the shipped binaries. Sentry receives the full unstripped
# binaries above; the .dbg files keep symbols recoverable when that upload
# is skipped.
# We use llvm-objcopy because GNU objcopy on older distros (e.g. Ubuntu 20.04)
# doesn't understand CREL sections produced by newer LLVM.
llvm-objcopy --only-keep-debug "${target_dir}/${target_triple}/release/gwen" "${target_dir}/${target_triple}/release/gwen.dbg"
llvm-objcopy --only-keep-debug "${target_dir}/${remote_server_triple}/release/remote_server" "${target_dir}/${remote_server_triple}/release/remote_server.dbg"
llvm-objcopy --strip-debug --discard-all "${target_dir}/${target_triple}/release/gwen"
llvm-objcopy --strip-debug --discard-all "${target_dir}/${target_triple}/release/cli"
llvm-objcopy --strip-debug --discard-all "${target_dir}/${remote_server_triple}/release/remote_server"

# Ensure that remote_server does not depend on libssl nor libcrypto, as we got rid of these deps.
if ldd "${target_dir}/${remote_server_triple}/release/remote_server" | grep -q 'libcrypto\|libssl'; then
    if [[ "$remote_server_triple" == *-musl ]]; then
        echo "Error: remote_server still depends on libssl or libcrypto" && exit 1
    else
        echo "Info: Using non-musl remote-server build."
    fi
fi

suffix=""
if [ "$channel" != "stable" ]; then
  suffix="-$channel"
fi

# Move everything that should end up in the final package
# into a temp directory.
temp_dir=$(mktemp -d)
zed_dir="${temp_dir}/gwen$suffix.app"

# Binary
mkdir -p "${zed_dir}/bin" "${zed_dir}/libexec"
cp "${target_dir}/${target_triple}/release/gwen" "${zed_dir}/libexec/gwen"
cp "${target_dir}/${target_triple}/release/cli" "${zed_dir}/bin/gwen"
gzip -c --best "${target_dir}/${remote_server_triple}/release/remote_server" > "${zed_dir}/libexec/gwen-remote-server.gz"

# Libs
# Bundle libstdc++ so older supported systems can run binaries built with our
# toolchain even when their system libstdc++.so.6 lacks required GLIBCXX symbols.
#
# Never bundle the GLib family or its private dependencies: host libraries that
# get dlopen'd are built against the system GLib, and resolving them against an
# older bundled copy fails with missing symbols.
find_libs() {
    ldd ${target_dir}/${target_triple}/release/gwen |\
        cut -d' ' -f3 |\
        grep -v '\<\(libc.so\|libgcc_s.so\|libm.so\|libpthread.so\|libdl.so\|libasound.so\)' |\
        grep -v '\<\(libglib-2.0.so\|libgobject-2.0.so\|libgio-2.0.so\|libgmodule-2.0.so\|libpcre.so\|libpcre2-8.so\|libmount.so\|libblkid.so\|libselinux.so\|libffi.so\|libresolv.so\|libz.so\)'
}

mkdir -p "${zed_dir}/lib"
rm -rf "${zed_dir}/lib/*"
cp $(find_libs) "${zed_dir}/lib"

# PDFium is dlopen'd at runtime by the bundled pdf_viewer for native PDF viewing;
# find_libs cannot see it because the binaries are not linked against it.
case "$(uname -m)" in
    x86_64) pdfium_arch="x64" ;;
    aarch64) pdfium_arch="arm64" ;;
    *) echo "Unsupported architecture for pdfium: $(uname -m)" >&2 && exit 1 ;;
esac
pdfium_tag="chromium/8066"
case "${pdfium_arch}" in
    x64) pdfium_sha256="0b43f405477cf2cfc4dbff06905093c3309756c6bca1fb9da99234a2ca97fed2" ;;
    arm64) pdfium_sha256="0e6f90dccbc6b81fd5d7106abaf164c4222178f024c204d00d526b60fd2ad535" ;;
esac
pdfium_tgz="${target_dir}/pdfium-linux-${pdfium_arch}.tgz"
if ! echo "${pdfium_sha256}  ${pdfium_tgz}" | sha256sum --check --status 2>/dev/null; then
    curl -fsSLo "${pdfium_tgz}" "https://github.com/bblanchon/pdfium-binaries/releases/download/${pdfium_tag}/pdfium-linux-${pdfium_arch}.tgz"
    echo "${pdfium_sha256}  ${pdfium_tgz}" | sha256sum --check -
fi
pdfium_tmp=$(mktemp -d)
tar -xzf "${pdfium_tgz}" -C "${pdfium_tmp}" "lib/libpdfium.so" "LICENSE" "licenses"
cp "${pdfium_tmp}/lib/libpdfium.so" "${zed_dir}/lib/libpdfium.so"
# BSD-3 of the toolset and its bundled third parties requires retaining the
# license notices on redistribution.
mkdir -p "${zed_dir}/share/licenses/pdfium"
cp "${pdfium_tmp}/LICENSE" "${pdfium_tmp}/licenses/"* "${zed_dir}/share/licenses/pdfium/"
rm -rf "${pdfium_tmp}"

# Icons
mkdir -p "${zed_dir}/share/icons/hicolor/512x512/apps"
cp "crates/zed/resources/app-icon-gwen.png" "${zed_dir}/share/icons/hicolor/512x512/apps/gwen.png"
mkdir -p "${zed_dir}/share/icons/hicolor/1024x1024/apps"
cp "crates/zed/resources/app-icon-gwen@2x.png" "${zed_dir}/share/icons/hicolor/1024x1024/apps/gwen.png"

# .desktop
export DO_STARTUP_NOTIFY="true"
export APP_CLI="gwen"
export APP_ICON="gwen"
export APP_ARGS="%U"
if [[ "$channel" == "preview" ]]; then
  export APP_NAME="Gwen Preview"
  APP_ID="com.yrkaierx.Gwen-Preview"
elif [[ "$channel" == "nightly" ]]; then
  export APP_NAME="Gwen Nightly"
  APP_ID="com.yrkaierx.Gwen-Nightly"
elif [[ "$channel" == "dev" ]]; then
  export APP_NAME="Gwen"
  APP_ID="com.yrkaierx.Gwen"
else
  export APP_NAME="Gwen"
  APP_ID="com.yrkaierx.Gwen"
fi

mkdir -p "${zed_dir}/share/applications"
envsubst < "crates/zed/resources/gwen.desktop.in" > "${zed_dir}/share/applications/$APP_ID.desktop"
chmod +x "${zed_dir}/share/applications/$APP_ID.desktop"

# Install/uninstall scripts ship in the archive root so it is self-contained
cp "crates/zed/resources/install.sh" "crates/zed/resources/uninstall.sh" "${temp_dir}/"
chmod 755 "${temp_dir}/install.sh" "${temp_dir}/uninstall.sh"

# Copy generated licenses so they'll end up in archive too
cp "assets/licenses.md" "${zed_dir}/licenses.md"

# Create archive out of everything that's in the temp directory
arch=$(uname -m)
archive="gwen-linux-${arch}.tar.gz"

rm -rf "${archive}"
remove_match="gwen(-[a-zA-Z0-9]+)?-linux-$(uname -m)\.tar\.gz"
ls "${target_dir}/release" | grep -E ${remove_match} | xargs -d "\n" -I {} rm -f "${target_dir}/release/{}" || true
tar -czvf "${target_dir}/release/$archive" -C ${temp_dir} "gwen$suffix.app" install.sh uninstall.sh

gzip -f --stdout --best "${target_dir}/${remote_server_triple}/release/remote_server" > "${target_dir}/gwen-remote-server-linux-${arch}.gz"