#!/usr/bin/env bash
set -euo pipefail

payload_dir="/usr/share/raveos/hyprland-theme/theme-data"

if [[ ! -d "$payload_dir" ]]; then
  echo "Missing payload: $payload_dir" >&2
  exit 1
fi

if [[ ${EUID} -ne 0 ]]; then
  echo "Run as root." >&2
  exit 1
fi

# --- /etc/skel ---

mkdir -p /etc/skel/.config/hypr
if [[ -d "${payload_dir}/hypr" ]]; then
  cp -rf "${payload_dir}/hypr/." /etc/skel/.config/hypr/
  # Remove legacy conf-based config if Lua version exists
  [[ -f /etc/skel/.config/hypr/hyprland.lua ]] && rm -f /etc/skel/.config/hypr/hyprland.conf
fi

mkdir -p /etc/skel/.config/dms
mkdir -p /etc/skel/.config/quickshell/pockets/DMS
mkdir -p /etc/skel/.config/DankMaterialShell
dms_src="${payload_dir}/dms"
[[ -f "${dms_src}/shell.qml" ]] || dms_src="${payload_dir}/DankMaterialShell/quickshell"
if [[ -d "$dms_src" ]]; then
  cp -r "${dms_src}/." /etc/skel/.config/quickshell/pockets/DMS/
  cp -r "${dms_src}/." /etc/skel/.config/dms/
  if [[ -d "${dms_src}/matugen/configs" ]]; then
    mkdir -p /etc/skel/.config/matugen
    cp -r "${dms_src}/matugen/configs/." /etc/skel/.config/matugen/
  fi
fi
if [[ -f "${payload_dir}/DankMaterialShell/settings.json" ]]; then
  install -Dm644 "${payload_dir}/DankMaterialShell/settings.json" /etc/skel/.config/DankMaterialShell/settings.json
fi
if [[ -f "${payload_dir}/DankMaterialShell/.firstlaunch" ]]; then
  install -Dm644 "${payload_dir}/DankMaterialShell/.firstlaunch" /etc/skel/.config/DankMaterialShell/.firstlaunch
fi

mkdir -p /etc/skel/.config/kitty
if [[ -d "${payload_dir}/kitty" ]]; then
  cp -r "${payload_dir}/kitty/." /etc/skel/.config/kitty/
fi

if [[ -f "${payload_dir}/hyprland-pp.png" ]]; then
  install -Dm644 "${payload_dir}/hyprland-pp.png" /etc/skel/.face
fi

mkdir -p /etc/skel/.config/fastfetch
for f in config.jsonc config-kitty.jsonc raveos-logo.png raveos-logo.txt; do
  if [[ -f "${payload_dir}/fastfetch/${f}" ]]; then
    install -Dm644 "${payload_dir}/fastfetch/${f}" "/etc/skel/.config/fastfetch/${f}"
  fi
done

if [[ -f "${payload_dir}/profile.d/raveos-fastfetch.sh" ]]; then
  install -Dm755 "${payload_dir}/profile.d/raveos-fastfetch.sh" /etc/profile.d/raveos-fastfetch.sh
fi

if [[ -d "${payload_dir}/skel" ]]; then
  cp -r --no-preserve=ownership "${payload_dir}/skel/." /etc/skel/
fi

if [[ -f "${payload_dir}/background.jpg" ]]; then
  mkdir -p /usr/share/backgrounds/raveos
  install -Dm644 "${payload_dir}/background.jpg" /etc/skel/.config/background
  install -Dm644 "${payload_dir}/background.jpg" /etc/skel/.config/background.jpg
fi

for d in gtk-3.0 gtk-4.0 nwg-look Thunar xfce4 xsettingsd hyprshell; do
  if [[ -d "${payload_dir}/${d}" ]]; then
    mkdir -p "/etc/skel/.config/${d}"
    cp -rf "${payload_dir}/${d}/." "/etc/skel/.config/${d}/"
  fi
done

if [[ -f "${payload_dir}/sddm/sddm.conf" ]]; then
  install -Dm644 "${payload_dir}/sddm/sddm.conf" /etc/sddm.conf.d/raveos-theme.conf
fi

if [[ -f "${payload_dir}/sddm/hyprland.lua" ]]; then
  install -Dm644 "${payload_dir}/sddm/hyprland.lua" /usr/share/hypr/sddm/hyprland.lua
fi

for candidate in \
  "/usr/share/sddm/themes/sddm-astronaut-theme" \
  "/usr/share/sddm/themes/astronaut" \
  "/usr/share/sddm/themes/sddm-astronaut"; do
  if [[ -d "$candidate" ]]; then
    if [[ -f "${payload_dir}/sddm/new-raveos-main-bg.jpeg" ]]; then
      install -Dm644 "${payload_dir}/sddm/new-raveos-main-bg.jpeg" \
        "${candidate}/Backgrounds/new-raveos-main-bg.jpeg"
    fi
    if [[ -f "${payload_dir}/sddm/astronaut.conf" ]]; then
      install -Dm644 "${payload_dir}/sddm/astronaut.conf" \
        "${candidate}/Themes/astronaut.conf"
    fi
    break
  fi
done

# --- meglévő userek ---

while IFS=: read -r user _ uid gid _ home shell; do
  [[ "$uid" -ge 1000 ]] || continue
  [[ -d "$home" ]] || continue
  [[ "$shell" != "/usr/bin/nologin" && "$shell" != "/bin/false" ]] || continue

  mkdir -p "${home}/.config/hypr" \
           "${home}/.config/quickshell/pockets/DMS" \
           "${home}/.config/dms" \
           "${home}/.config/DankMaterialShell" \
           "${home}/.config/matugen" \
           "${home}/.config/kitty" \
           "${home}/.config/fastfetch"

  pkill -u "$user" hyprpaper 2>/dev/null || true
  if [[ -d "${payload_dir}/hypr" ]]; then
    cp -rf "${payload_dir}/hypr/." "${home}/.config/hypr/"
    [[ -f "${home}/.config/hypr/hyprland.lua" ]] && rm -f "${home}/.config/hypr/hyprland.conf"
  fi
  printf 'preload = %s/.config/background.jpg\nwallpaper = ,%s/.config/background.jpg\nsplash = false\n' \
    "$home" "$home" > "${home}/.config/hypr/hyprpaper.conf"

  dms_src="${payload_dir}/dms"
  [[ -f "${dms_src}/shell.qml" ]] || dms_src="${payload_dir}/DankMaterialShell/quickshell"
  if [[ -d "$dms_src" ]]; then
    cp -r "${dms_src}/." "${home}/.config/quickshell/pockets/DMS/"
    cp -r "${dms_src}/." "${home}/.config/dms/"
    if [[ -d "${dms_src}/matugen/configs" ]]; then
      cp -r "${dms_src}/matugen/configs/." "${home}/.config/matugen/"
    fi
  fi
  if [[ -f "${payload_dir}/DankMaterialShell/settings.json" ]]; then
    install -Dm644 "${payload_dir}/DankMaterialShell/settings.json" "${home}/.config/DankMaterialShell/settings.json"
  fi
  if [[ -f "${payload_dir}/DankMaterialShell/.firstlaunch" ]]; then
    install -Dm644 "${payload_dir}/DankMaterialShell/.firstlaunch" "${home}/.config/DankMaterialShell/.firstlaunch"
  fi

  if [[ -d "${payload_dir}/skel" ]]; then
    cp -r --no-preserve=ownership "${payload_dir}/skel/." "$home/"
  fi

  if [[ -f "${payload_dir}/background.jpg" ]]; then
    install -Dm644 "${payload_dir}/background.jpg" "${home}/.config/background"
    install -Dm644 "${payload_dir}/background.jpg" "${home}/.config/background.jpg"
  fi

  if [[ -d "${payload_dir}/kitty" ]]; then
    cp -r "${payload_dir}/kitty/." "${home}/.config/kitty/"
  fi

  if [[ -f "${payload_dir}/hyprland-pp.png" ]]; then
    install -Dm644 "${payload_dir}/hyprland-pp.png" "${home}/.face"
  fi

  for f in config.jsonc config-kitty.jsonc raveos-logo.png raveos-logo.txt; do
    if [[ -f "${payload_dir}/fastfetch/${f}" ]]; then
      install -Dm644 "${payload_dir}/fastfetch/${f}" "${home}/.config/fastfetch/${f}"
    fi
  done

  if [[ -f "${payload_dir}/hypr/scripts/raveos-monitor-setup.sh" ]]; then
    bash "${payload_dir}/hypr/scripts/raveos-monitor-setup.sh" --hypr-dir "${home}/.config/hypr"
  fi

  for d in gtk-3.0 gtk-4.0 nwg-look Thunar xfce4 xsettingsd hyprshell; do
    if [[ -d "${payload_dir}/${d}" ]]; then
      mkdir -p "${home}/.config/${d}"
      cp -rf "${payload_dir}/${d}/." "${home}/.config/${d}/"
    fi
  done

  runuser -u "$user" -- xdg-user-dirs-update 2>/dev/null || true

  dms_session="${home}/.local/state/DankMaterialShell/session.json"
  mkdir -p "$(dirname "$dms_session")"
  python3 -c "
import json, os
path = '${dms_session}'
d = {}
if os.path.isfile(path):
    try:
        with open(path) as f:
            d = json.load(f)
    except Exception:
        d = {}
if not d.get('wallpaperPath'):
    d['wallpaperPath'] = '/usr/share/raveos/hyprland-theme/theme-data/background.jpg'
    with open(path, 'w') as f:
        json.dump(d, f, indent=2)
" 2>/dev/null || true

  chown -R "${uid}:${gid}" "$home"

  if command -v matugen &>/dev/null && [[ -f "${home}/.config/background" ]]; then
    runuser -u "$user" -- matugen image "${home}/.config/background" 2>/dev/null || true
  fi

  hypr_sig=$(runuser -u "$user" -- bash -c 'ls /run/user/'"$uid"'/hypr/ 2>/dev/null | tail -1')
  if [[ -n "$hypr_sig" ]]; then
    runuser -u "$user" -- bash -c "HYPRLAND_INSTANCE_SIGNATURE='${hypr_sig}' hyprctl dispatch exec 'env LIBGL_ALWAYS_SOFTWARE=1 hyprpaper'" 2>/dev/null || true
  fi
done < /etc/passwd

echo "Automatic apply finished."