restore-studio-margin.sh

#!/bin/bash
set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
INJECTOR="$SCRIPT_DIR/injector.mjs"
STATE_ROOT="$HOME/Library/Application Support/StudioMarginCodex"
STATE_PATH="$STATE_ROOT/state.json"
PLIST_PATH="$HOME/Library/LaunchAgents/com.studiomargin.apply.plist"
LAUNCH_LABEL="com.studiomargin.apply"
PORT=""

while [[ $# -gt 0 ]]; do
  case "$1" in
    --port) PORT="$2"; shift 2 ;;
    *) echo "Unknown argument: $1" >&2; exit 2 ;;
  esac
done

launchctl bootout "gui/$(id -u)/$LAUNCH_LABEL" >/dev/null 2>&1 || true

safe_stop_injector() {
  local pid="$1" command
  [[ "$pid" =~ ^[0-9]+$ ]] || return 0
  command="$(ps -p "$pid" -o command= 2>/dev/null || true)"
  if [[ "$command" == *"$INJECTOR"* && "$command" == *"--watch"* ]]; then
    kill "$pid" 2>/dev/null || true
  fi
}

if [[ -f "$STATE_PATH" ]]; then
  STATE_VALUES="$(node -e 'try { const s=JSON.parse(require("fs").readFileSync(process.argv[1],"utf8")); console.log(`${s.port || ""}\t${s.injectorPid || ""}`) } catch {}' "$STATE_PATH")"
  STATE_PORT="${STATE_VALUES%%$'\t'*}"
  STATE_PID="${STATE_VALUES#*$'\t'}"
  [[ -z "$PORT" ]] && PORT="$STATE_PORT"
  [[ -n "$STATE_PID" ]] && safe_stop_injector "$STATE_PID"
fi
PORT="${PORT:-9335}"
sleep 0.25
node "$INJECTOR" --remove --port "$PORT" --timeout-ms 3000 >/dev/null 2>&1 || true
rm -f "$PLIST_PATH"
echo "The live Studio Margin theme was removed."