#!/usr/bin/env bash

# Migration: Migrate individual state files to unified state.json
# Version: 20260404

OLD_TOGGLES_DIR="$HOME/.local/state/nomarchy/toggles"
IDLE_STATE_FILE="$HOME/.config/home-manager/idle-state.json"
NIGHTLIGHT_STATE_FILE="$HOME/.config/home-manager/hyprsunset-state.json"
HYPRLAND_STATE_FILE="$HOME/.config/home-manager/hyprland-state.json"
THEME_STATE_FILE="$HOME/.config/home-manager/theme-state.nix"
WALLPAPER_STATE_FILE="$HOME/.config/home-manager/wallpaper-state.nix"
FONT_STATE_FILE="$HOME/.config/home-manager/font-state.nix"
NEW_STATE_FILE="$HOME/.config/home-manager/state.json"

mkdir -p "$(dirname "$NEW_STATE_FILE")"
[[ ! -f $NEW_STATE_FILE ]] && echo "{}" > "$NEW_STATE_FILE"

# 1. Migrate .local/state/nomarchy/toggles (these were empty 'off' flag files)
if [[ -d $OLD_TOGGLES_DIR ]]; then
    for file in "$OLD_TOGGLES_DIR"/*; do
        [[ -e "$file" ]] || continue
        filename=$(basename "$file")
        case "$filename" in
            suspend-off)
                echo "Migrating suspend-off -> suspend: false"
                jq '.suspend = false' "$NEW_STATE_FILE" > "$NEW_STATE_FILE.tmp" && mv "$NEW_STATE_FILE.tmp" "$NEW_STATE_FILE"
                ;;
            screensaver-off)
                echo "Migrating screensaver-off -> screensaver: false"
                jq '.screensaver = false' "$NEW_STATE_FILE" > "$NEW_STATE_FILE.tmp" && mv "$NEW_STATE_FILE.tmp" "$NEW_STATE_FILE"
                ;;
            skip-vscode-theme-changes)
                echo "Migrating skip-vscode-theme-changes -> skipVsCodeTheme: true"
                jq '.skipVsCodeTheme = true' "$NEW_STATE_FILE" > "$NEW_STATE_FILE.tmp" && mv "$NEW_STATE_FILE.tmp" "$NEW_STATE_FILE"
                ;;
        esac
        rm "$file"
    done
    rmdir "$OLD_TOGGLES_DIR" 2>/dev/null || true
fi

# 2. Migrate existing JSON state files
if [[ -f $IDLE_STATE_FILE ]]; then
    echo "Migrating idle-state.json -> state.json"
    ENABLED=$(jq '.enabled' "$IDLE_STATE_FILE")
    jq ".idle = $ENABLED" "$NEW_STATE_FILE" > "$NEW_STATE_FILE.tmp" && mv "$NEW_STATE_FILE.tmp" "$NEW_STATE_FILE"
    rm "$IDLE_STATE_FILE"
fi

if [[ -f $NIGHTLIGHT_STATE_FILE ]]; then
    echo "Migrating hyprsunset-state.json -> state.json"
    ENABLED=$(jq '.enabled' "$NIGHTLIGHT_STATE_FILE")
    TEMP=$(jq '.temperature' "$NIGHTLIGHT_STATE_FILE")
    jq ".nightlight = $ENABLED | .nightlightTemperature = $TEMP" "$NEW_STATE_FILE" > "$NEW_STATE_FILE.tmp" && mv "$NEW_STATE_FILE.tmp" "$NEW_STATE_FILE"
    rm "$NIGHTLIGHT_STATE_FILE"
fi

if [[ -f $HYPRLAND_STATE_FILE ]]; then
    echo "Migrating hyprland-state.json -> state.json"
    GAPS_OUT=$(jq '.gaps_out' "$HYPRLAND_STATE_FILE")
    GAPS_IN=$(jq '.gaps_in' "$HYPRLAND_STATE_FILE")
    BORDER_SIZE=$(jq '.border_size' "$HYPRLAND_STATE_FILE")
    jq ".hyprland = {\"gaps_out\": $GAPS_OUT, \"gaps_in\": $GAPS_IN, \"border_size\": $BORDER_SIZE}" "$NEW_STATE_FILE" > "$NEW_STATE_FILE.tmp" && mv "$NEW_STATE_FILE.tmp" "$NEW_STATE_FILE"
    rm "$HYPRLAND_STATE_FILE"
fi

# 3. Migrate plaintext string state files
if [[ -f $THEME_STATE_FILE ]]; then
    echo "Migrating theme-state.nix -> state.json"
    THEME=$(cat "$THEME_STATE_FILE" | tr -d '\n')
    jq ".theme = \"$THEME\"" "$NEW_STATE_FILE" > "$NEW_STATE_FILE.tmp" && mv "$NEW_STATE_FILE.tmp" "$NEW_STATE_FILE"
    rm "$THEME_STATE_FILE"
fi

if [[ -f $WALLPAPER_STATE_FILE ]]; then
    echo "Migrating wallpaper-state.nix -> state.json"
    WALLPAPER=$(cat "$WALLPAPER_STATE_FILE" | tr -d '\n')
    jq ".wallpaper = \"$WALLPAPER\"" "$NEW_STATE_FILE" > "$NEW_STATE_FILE.tmp" && mv "$NEW_STATE_FILE.tmp" "$NEW_STATE_FILE"
    rm "$WALLPAPER_STATE_FILE"
fi

if [[ -f $FONT_STATE_FILE ]]; then
    echo "Migrating font-state.nix -> state.json"
    FONT=$(cat "$FONT_STATE_FILE" | tr -d '\n')
    jq ".font = \"$FONT\"" "$NEW_STATE_FILE" > "$NEW_STATE_FILE.tmp" && mv "$NEW_STATE_FILE.tmp" "$NEW_STATE_FILE"
    rm "$FONT_STATE_FILE"
fi

echo "State migration complete."