#!/usr/bin/env bash
# Datadog Agent 用に `locationd/clients.plist` の Authorized キーを追加 / 更新するスクリプト (root 権限が必要)
# 使用方法: bash add_datadaog_agent_to_plist.sh [AGENT_BIN_PATH]
# AGENT_BIN_PATH: 任意 - Agent バイナリ パス (デフォルト: /opt/datadog-agent/bin/agent/agent)

# 設定
PLIST_PATH="/var/db/locationd/clients.plist"
DEFAULT_PATTERN="/opt/datadog-agent/bin/agent/agent"
BACKUP_PATH="${PLIST_PATH}.bak"

# 何か問題が起きた場合にバックアップを復元する関数
restore_backup() {
  echo "[ERROR] バックアップを復元しています..."
  sudo cp "$BACKUP_PATH" "$PLIST_PATH"
  sudo plutil -convert binary1 "$PLIST_PATH"
  echo "[INFO] バックアップを復元しました。終了します。"
  exit 1
}

# エラー ハンドリングを設定
trap restore_backup ERR

# 引数が指定されているか確認
if [ -n "$1" ]; then
  PATTERN="$1"
  echo "[INFO] CLI 引数で指定されたパターンを使用します: $PATTERN"
else
  # 検索するパターンを入力
  read -p "検索するパターンを入力してください [${DEFAULT_PATTERN}]: " PATTERN
  PATTERN=${PATTERN:-$DEFAULT_PATTERN}
fi

# 元のファイルをバックアップ
echo "[INFO] $PLIST_PATH を $BACKUP_PATH にバックアップします"
sudo cp "$PLIST_PATH" "$BACKUP_PATH"

# 解析しやすいように plist を XML に変換
sudo plutil -convert xml1 "$PLIST_PATH"

echo "[INFO] 次を含むエントリを検索します: $PATTERN"

# パターンを含むブロックの最初のキーを見つける (xargs で前後の空白を削除)
KEY_LINE=$(grep "$PATTERN" "$PLIST_PATH" | grep "<key>" | head -n1 | xargs)
if [ -z "$KEY_LINE" ]; then
  echo "[ERROR] 次のパターンを含むエントリが見つかりません: $PATTERN"
  restore_backup
fi

# 行からキーを抽出
KEY=${KEY_LINE#<key>}
KEY=${KEY%</key>}

if [ -z "$KEY" ]; then
  echo "[ERROR] 一致するエントリのキーを特定できませんでした。"
  restore_backup
fi

echo "[INFO] キーを処理します: $KEY"

# <key>$KEY</key> を含む行番号を取得
key_line=$(grep -n "<key>$KEY</key>" "$PLIST_PATH" | cut -d: -f1 | head -n1)
if [ -z "$key_line" ]; then
  echo "[ERROR] キーが見つかりません。"
  restore_backup
fi

# キーの次にある <dict> の行番号を取得
dict_start=$(tail -n +$((key_line + 1)) "$PLIST_PATH" | grep -n "<dict>" | head -n1 | cut -d: -f1)
dict_start=$((key_line + dict_start))

# 対応する </dict> の行番号を取得
dict_end=$(tail -n +$((dict_start + 1)) "$PLIST_PATH" | grep -n "</dict>" | head -n1 | cut -d: -f1)
dict_end=$((dict_start + dict_end))

echo "[INFO] ブロックを検出しました (行 $dict_start 〜 $dict_end)"

# ブロック内に <key>Authorized</key> があるか確認
auth_line=$(sed -n "${dict_start},${dict_end}p" "$PLIST_PATH" | grep -n "<key>Authorized</key>" | cut -d: -f1)

if [ -z "$auth_line" ]; then
  # <key>Authorized</key> が見つからない場合は </dict> の直前に追加
  echo "[INFO] ブロックに <key>Authorized</key><true/> を追加します"
  sed -i "" "${dict_end}i\\
        <key>Authorized</key>\\
        <true/>\\
" "$PLIST_PATH"
else
  # <key>Authorized</key> が見つかった場合は、次の行で値を確認
  auth_line=$((dict_start + auth_line - 1))
  value_line=$((auth_line + 1))

  # 次の行に <false/> が含まれているか確認
  if grep -q "<false/>" <(sed -n "${value_line}p" "$PLIST_PATH"); then
    echo "[INFO] <false/> を <true/> に変更します"
    sed -i "" "${value_line}s/<false\/>/<true\/>/" "$PLIST_PATH"
  else
    echo "[INFO] <key>Authorized</key> は既に正しい値で存在します"
  fi
fi

# システム用に plist をバイナリに戻す
sudo plutil -convert binary1 "$PLIST_PATH"
echo "[INFO] 変更を適用しました。"
echo "[INFO] 変更を反映するには、再起動するか、次を実行してください: sudo killall locationd"
trap - ERR