#!/bin/bash

set -e

# From: https://people.debian.org/~mpitt/systemd.conf-2016-graphical-session.pdf

if command -v systemctl >/dev/null; then
    # robustness: if the previous graphical session left some failed units,
    # reset them so that they don't break this startup
    for unit in $(systemctl --user --no-legend --state=failed --plain list-units | cut -f1 -d' '); do
        partof="$(systemctl --user show -p PartOf --value "$unit")"
        for target in cosmic-session.target graphical-session.target; do
            if [ "$partof" = "$target" ]; then
                systemctl --user reset-failed "$unit"
                break
            fi
        done
    done
fi

# use the user's preferred shell to acquire environment variables
# see: https://github.com/pop-os/cosmic-session/issues/23
if [ -n "${SHELL}" ]; then
    # --in-login-shell: our flag to indicate that we don't need to recurse any further
    if [ "${1}" != "--in-login-shell" ]; then
        # `exec -l`: like `login`, prefixes $SHELL with a hyphen to start a login shell
        exec bash -c "exec -l '${SHELL}' -c '${0} --in-login-shell'"
    fi
fi

export XDG_CURRENT_DESKTOP="${XDG_CURRENT_DESKTOP:=COSMIC}"
export XDG_SESSION_TYPE="${XDG_SESSION_TYPE:=wayland}"
export _JAVA_AWT_WM_NONREPARENTING=1
export GDK_BACKEND=wayland,x11
export MOZ_ENABLE_WAYLAND=1
export QT_QPA_PLATFORM="wayland;xcb"
export QT_AUTO_SCREEN_SCALE_FACTOR=1
export QT_ENABLE_HIGHDPI_SCALING=1
export DCONF_PROFILE=/usr/share/dconf/profile/cosmic

# Set the QT platform theme to CuteCosmic. Fallback to qt6ct if CuteCosmic is not installed.
export QT_QPA_PLATFORMTHEME=cosmic
for QT_PLUGIN_PATH in /usr/lib{*,/*}/qt6/plugins; do
    if [ -f "${QT_PLUGIN_PATH}/platformthemes/libcutecosmictheme.so" ]; then
        # CuteCosmic found, no need for a fallback.
        export QT_QPA_PLATFORMTHEME=cosmic
        break
    elif [ -f "${QT_PLUGIN_PATH}/platformthemes/libqt6ct.so" ]; then
        # Fallback to qt6ct, but keep looking for CuteCosmic.
        export QT_QPA_PLATFORMTHEME=qt6ct
    fi
done

# Start gnome keyring components if the daemon is active
# -> check if /run/user/$UID/keyring exists
if [ -d "/run/user/$(id -u)/keyring" ]; then

    # Use PATH lookup instead of hardcoding /usr/bin
    if command -v gnome-keyring-daemon >/dev/null 2>&1; then
        eval "$(gnome-keyring-daemon --start --components=pkcs11,secrets,ssh > /dev/null 2>&1)"
    else
        echo "gnome-keyring-daemon not found in PATH" >&2
    fi

    # Only set SSH_AUTH_SOCK if the socket actually exists. Either
    # set the correct one, or don't set one at all. Don't set the
    # wrong value.
    if [ -S "/run/user/$(id -u)/gcr/ssh" ]; then
        export SSH_AUTH_SOCK="/run/user/$(id -u)/gcr/ssh"
    elif [ -S "/run/user/$(id -u)/keyring/ssh" ]; then
        export SSH_AUTH_SOCK="/run/user/$(id -u)/keyring/ssh"
    fi
fi

if command -v systemctl >/dev/null; then
    # Import some variables that we explicitly want to have available
    # in the user session.
    systemctl --user import-environment XDG_SESSION_TYPE XDG_CURRENT_DESKTOP DCONF_PROFILE SSH_AUTH_SOCK

    # For environment variables already imported into the user's
    # session, if the value imported differs from the value in this
    # environment, update it.
    mapfile -t existing_env_vars < <(systemctl --user show-environment)
    for env_var in "${existing_env_vars[@]}"; do
        env_var_name="$(echo "${env_var}" | awk -F '=' '{print $1}')"
        env_var_val_str_to_compare="${env_var_name}=${!env_var_name:-}"

        if [[ "${env_var}" != "${env_var_val_str_to_compare}" ]]; then
            # Update only if the value in current environment is non-empty
            env_var_unassigned_str="${env_var_name}="
            if [[ "${env_var_val_str_to_compare}" != "${env_var_unassigned_str}" ]]; then
                systemctl --user import-environment "${env_var_name}" ||:
            fi
        fi
    done
fi

# Run cosmic-session
if [[ -z "${DBUS_SESSION_BUS_ADDRESS}" ]]; then
    exec /usr/bin/dbus-run-session -- /usr/bin/cosmic-session
else
    exec /usr/bin/cosmic-session
fi
