#!/bin/bash

LOG_FILE="/volume1/docker/scrutiny/.work/setup.log"

# 既存のログファイルをクリア
> "$LOG_FILE"

echo "$(date): スクリプト実行開始" | tee -a "$LOG_FILE"

LAN_IP=$(ip route get 1 | awk '{print $7}' | head -1)
echo "$(date): 検出されたLAN IP: $LAN_IP" | tee -a "$LOG_FILE"

# 必要なディレクトリを自動作成
mkdir -p /volume1/docker/scrutiny/influxdb /volume1/docker/scrutiny/.work
if [ $? -ne 0 ]; then
    echo "$(date): エラー: 必要なディレクトリの作成に失敗しました。" | tee -a "$LOG_FILE"
    exit 1
fi
echo "$(date): 必要なディレクトリを作成しました。" | tee -a "$LOG_FILE"

# docker-compose.yml と collector.yaml に含めるデバイスを検出
# /dev/sdX (SATA/SCSI), /dev/sataX (DSM独自のSATA), /dev/nvmeXnY (NVMeネームスペース) を対象
DEVICES_LIST=$(find /dev -maxdepth 1 -regextype posix-egrep -regex '/dev/(sd[a-z]+|sata[0-9]+|nvme[0-9]+n[0-9]+)$' | sort)
if [ $? -ne 0 ]; then
    echo "$(date): エラー: デバイスの検出に失敗しました。" | tee -a "$LOG_FILE"
    exit 1
fi
echo "$(date): 検出されたデバイス: $DEVICES_LIST" | tee -a "$LOG_FILE"

# 検出されたデバイスリストをログファイルに保存
echo "$(date): 検出されたデバイスリストを /volume1/docker/scrutiny/.work/dev-list.yaml に保存します。" | tee -a "$LOG_FILE"
echo "$DEVICES_LIST" | sed 's|^|- device: &|' > /volume1/docker/scrutiny/.work/dev-list.yaml
if [ $? -ne 0 ]; then
    echo "$(date): エラー: デバイスリストの保存に失敗しました。" | tee -a "$LOG_FILE"
    exit 1
fi
echo "$(date): デバイスリストを保存しました。" | tee -a "$LOG_FILE"

# docker-compose.yml 用のデバイスリストを生成
DOCKER_DEVICES=""
if [ -n "$DEVICES_LIST" ]; then
    DOCKER_DEVICES=$(echo "$DEVICES_LIST" | awk '{print "      - "$1}' | paste -sd '\n' -)
fi

# 既存の docker-compose.yml を削除して新しいものを生成
rm -f /volume1/docker/scrutiny/docker-compose.yml
if [ $? -ne 0 ]; then
    echo "$(date): エラー: 既存の docker-compose.yml の削除に失敗しました。" | tee -a "$LOG_FILE"
fi
echo "$(date): docker-compose.yml を作成します。" | tee -a "$LOG_FILE"

cat > /volume1/docker/scrutiny/docker-compose.yml << EOF
version: '3.8'
services:
  scrutiny:
    container_name: scrutiny
    image: ghcr.io/analogj/scrutiny:master-omnibus
    cap_add:
      - SYS_RAWIO
      - SYS_ADMIN
    ports:
      - 6090:8080/tcp
      - 8086:8086/tcp
    volumes:
      - /run/udev:/run/udev:ro
      - /volume1/docker/scrutiny:/opt/scrutiny/config
      - /volume1/docker/scrutiny/influxdb:/opt/scrutiny/influxdb
    devices:
$DOCKER_DEVICES
    environment:
      - SCRUTINY_WEB_INFLUXDB_TOKEN=init-token
      - SCRUTINY_WEB_INFLUXDB_INIT_USERNAME=scrutiny
      - SCRUTINY_WEB_INFLUXDB_INIT_PASSWORD=securepass
      - COLLECTOR_CRON_SCHEDULE=0 */6 * * *
      - TZ=Asia/Bangkok
    security_opt:
      - no-new-privileges:true
    restart: unless-stopped
EOF
if [ $? -ne 0 ]; then
    echo "$(date): エラー: docker-compose.yml の書き込みに失敗しました。" | tee -a "$LOG_FILE"
    exit 1
fi
echo "$(date): docker-compose.yml を作成しました。" | tee -a "$LOG_FILE"

# collector.yaml の内容を構築
COLLECTOR_DEVICES=""
if [ -n "$DEVICES_LIST" ]; then
    for DEV in $DEVICES_LIST; do
        DEVICE_NAME=$(basename "$DEV")
        DEVICE_TYPE=""
        if [[ "$DEVICE_NAME" =~ ^sd || "$DEVICE_NAME" =~ ^sata ]]; then
            DEVICE_TYPE="sat"
        elif [[ "$DEVICE_NAME" =~ ^nvme ]]; then
            DEVICE_TYPE="nvme"
        else
            DEVICE_TYPE="unknown"
        fi
        COLLECTOR_DEVICES+="  - device: $DEV\n"
        COLLECTOR_DEVICES+="    type: '$DEVICE_TYPE'\n"
    done
fi

# collector.yaml を作成または上書き
echo "$(date): collector.yaml を作成します。" | tee -a "$LOG_FILE"
cat > /volume1/docker/scrutiny/collector.yaml << EOF
version: 1
host:
  id: ""
devices:
$(echo -e "$COLLECTOR_DEVICES" | sed 's/\\//g')
EOF
if [ $? -ne 0 ]; then
    echo "$(date): エラー: collector.yaml の書き込みに失敗しました。" | tee -a "$LOG_FILE"
    exit 1
fi
echo "$(date): collector.yaml を作成しました。" | tee -a "$LOG_FILE"

# ヘルプファイルを生成
echo "$(date): help.txt を作成します。" | tee -a "$LOG_FILE"
cat > /volume1/docker/scrutiny/help.txt << EOF
■ Scrutiny 初期チェック
DSMのブラウザで http://$LAN_IP:6090 にアクセスしてください。
ドライブが表示されていれば成功です。

■ ディスク検出に問題がある場合 (手動での collector.yaml 編集方法)
1. DSMの「File Station」を開きます。
2. 左パネルで「docker」フォルダ -> 「scrutiny」フォルダへ移動します。
3. 「collector.yaml」ファイルを右クリックし、「テキストエディタで開く」を選択します。
4. 開いたファイルはYAML形式です。`devices:` の下に、以下の書式を参考に、検出されていないディスクの情報を追加または修正します。
  - **SATA HDD/SSD (例: ドライブ1番、2番):**
    - device: /dev/sata1
      type: 'sat'
    - device: /dev/sata2
      type: 'sat'
    # あるいは /dev/sda, /dev/sdb など、お使いのNAS環境で異なる場合があります
    - device: /dev/sda
      type: 'sat'
  - **NVMe SSD (例: M.2 SSD 1番、2番):**
    - device: /dev/nvme0n1
      type: 'nvme'
    - device: /dev/nvme1n1
      type: 'nvme'
  ※ディスクの正確なパスは、Synologyの「ストレージマネージャー」でディスクのモデル名やシリアル番号を確認し、上記の一般的なパスと照らし合わせてみてください。
5. 編集が完了したら、テキストエディタでファイルを「保存」します。
6. DSMの「Container Manager」を開き、左メニューの「コンテナ」をクリックします。
7. リストの中から「scrutiny」コンテナを見つけて選択し、上部の「操作」ボタンから「再起動」を選択します。
8. 再度ScrutinyのURLにアクセスし、ディスクが表示されたか確認します。

■ 完了後
このhelp.txtファイルと、/volume1/docker/scrutiny/.work/setup.log があれば初期化成功です。
この「Scrutiny初期設定」のタスクは一度しか使わないため、タスクスケジューラから削除しても問題ありません。

■ 詳細なログ
スクリプトの実行ログは /volume1/docker/scrutiny/.work/setup.log に記録されています。
検出されたデバイスのリストは /volume1/docker/scrutiny/.work/dev-list.yaml に記録されています。
EOF
if [ $? -ne 0 ]; then
    echo "$(date): エラー: help.txt の作成に失敗しました。" | tee -a "$LOG_FILE"
    exit 1
fi
echo "$(date): help.txt を作成しました。" | tee -a "$LOG_FILE"

# Dockerコンテナを起動（または再起動）
echo "$(date): Dockerコンテナを起動（または再起動）します。" | tee -a "$LOG_FILE"
cd /volume1/docker/scrutiny && docker compose up -d
if [ $? -ne 0 ]; then
    echo "$(date): 重大なエラー: Docker Composeの起動に失敗しました。" | tee -a "$LOG_FILE"
    echo "$(date): 詳細については、DockerのログとContainer Managerを確認してください。" | tee -a "$LOG_FILE"
    exit 1
fi
echo "$(date): Dockerコンテナが正常に起動しました。" | tee -a "$LOG_FILE"

echo "$(date): スクリプト実行完了" | tee -a "$LOG_FILE"