Ubuntu システムの基本状態チェック自動化

#!/usr/bin/env bash

set -euo pipefail
# set -x  # 必要に応じてデバッグ用に有効化

########################################
# 基本設定およびチェック
########################################

DATE=$(date +"%Y-%m-%d")
WORK_DIR="/var/log/system_checking_by_chatgpt"
RAW_REPORT="$WORK_DIR/raw_report_$DATE.txt"
CHUNKS_DIR="$WORK_DIR/chunks_$DATE"
SUMMARY_DIR="$WORK_DIR/summary_$DATE"
FINAL_REPORT="$WORK_DIR/final_report_$DATE.md"  # Markdown形式で出力
DEBUG_LOG="$WORK_DIR/debug_$DATE.log"

mkdir -p "$WORK_DIR"
mkdir -p "$CHUNKS_DIR"
mkdir -p "$SUMMARY_DIR"

# 必須コマンドの存在チェック
REQUIRED_COMMANDS=("curl" "jq" "smartctl" "systemctl" "docker" "sensors")
for cmd in "${REQUIRED_COMMANDS[@]}"; do
    if ! command -v "$cmd" &>/dev/null; then
        echo "[ERROR] 必須コマンド $cmd がインストールされていません。" | tee -a "$DEBUG_LOG"
        exit 1
    fi
done

# ディスク空き容量のチェック（100MB以上必要）
AVAILABLE_SPACE=$(df "$WORK_DIR" | tail -1 | awk '{print $4}')
if [ "$AVAILABLE_SPACE" -lt 102400 ]; then
    echo "[ERROR] $WORK_DIR のディスク空き容量が不足しています。" | tee -a "$DEBUG_LOG"
    exit 1
fi

# ログ出力関数（標準エラー出力へ出力し、関数戻り値を妨げない）
log() {
    local LEVEL=$1
    shift
    local MESSAGE="$@"
    {
        echo "[$LEVEL] $MESSAGE" | tee -a "$DEBUG_LOG" >&2
        logger -t "system_checking_by_chatgpt" "[$LEVEL] $MESSAGE"
    }
}

log "INFO" "スクリプト実行開始..."

########################################
# APIキー読み込み
########################################

OPENAI_KEY_FILE="/home/your_user_name/.openai_api_key"
if [ ! -f "$OPENAI_KEY_FILE" ]; then
    log "ERROR" "$OPENAI_KEY_FILE が存在しません。"
    exit 1
fi

OPENAI_API_KEY=$(cat "$OPENAI_KEY_FILE")
if [ -z "$OPENAI_API_KEY" ]; then
    log "ERROR" "OPENAI_API_KEY が空です。"
    exit 1
fi
export OPENAI_API_KEY

log "INFO" "APIキーを正常に読み込みました。"

API_URL="https://api.openai.com/v1/chat/completions"
MODEL="gpt-4o"

########################################
# 関数定義
########################################

collect_system_info() {
    {
        echo "=== システム稼働時間 (uptime) ==="
        uptime
        echo ""

        echo "=== 日時情報 ==="
        date
        echo ""

        echo "=== CPU・メモリ使用状況 ==="
        top -b -n1 | head -n 20
        echo ""

        echo "=== メモリ使用状況 (free) ==="
        free -h
        echo ""

        echo "=== ディスク使用量 (df) ==="
        df -h
        echo ""

        echo "=== 異常なサービス一覧 (Failed Services) ==="
        systemctl list-units --state=failed
        echo ""

        echo "=== 温度センサー情報 ==="
        sensors 2>/dev/null || echo "[WARNING] 温度情報を取得できません。"
        echo ""

        echo "=== NVMe / SMART 状態 ==="
        sudo smartctl -a /dev/nvme0n1 2>/dev/null || echo "[WARNING] /dev/nvme0n1 をチェックできません。"
        echo ""
        sudo smartctl -a /dev/nvme1n1 2>/dev/null || echo "[WARNING] /dev/nvme1n1 をチェックできません。"
        echo ""

        echo "=== dmesg (直近100行) ==="
        dmesg | tail -n 100
        echo ""

        echo "=== システムジャーナル (エラーのみ直近300行) ==="
        journalctl -p err -n 300
        echo ""

        echo "=== ネットワーク状況 (ss -tulpn) ==="
        ss -tulpn 2>/dev/null || echo "[WARNING] ネットワーク状況を取得できません。"
        echo ""

        echo "=== ネットワークインターフェイス統計 (ip -s link) ==="
        ip -s link
        echo ""

        echo "=== Dockerコンテナ一覧 (ある場合) ==="
        docker ps -a 2>/dev/null || echo "[INFO] Dockerは起動していないか、インストールされていません。"
        echo ""

    } > "$RAW_REPORT"
}

generate_chunk_summaries() {
    # RAW_REPORTをチャンク分割
    CHUNK_SIZE=200
    split -l $CHUNK_SIZE "$RAW_REPORT" "$CHUNKS_DIR/chunk_"

    CHUNKS=("$CHUNKS_DIR"/chunk_*)

    if [ ${#CHUNKS[@]} -eq 0 ]; then
        log "ERROR" "チャンクが生成されませんでした。"
        exit 1
    fi

    # チャンクごとの要約用プロンプト
    CHUNK_SYSTEM_PROMPT="あなたはプロのシステム管理コンサルタントです。以下はシステム情報の一部です。この情報から以下を抽出して要約してください： (1) 重要なシステム状態情報（CPU/メモリ使用率、ディスク使用率、ネットワーク状況、サービス稼働状況、エラーメッセージなど）、(2) 異常やエラー、注意が必要な事象。可能な限り詳細に、かつ整理された形で記述してください。"

    summaries=()

    for chunk_file in "${CHUNKS[@]}"; do
        CHUNK_CONTENT=$(cat "$chunk_file")

        CHUNK_API_PAYLOAD=$(jq -n \
            --arg system_prompt "$CHUNK_SYSTEM_PROMPT" \
            --arg user_prompt "$CHUNK_CONTENT" \
            --arg model "$MODEL" \
            '{
                "model": $model,
                "messages": [
                    {"role": "system", "content": $system_prompt},
                    {"role": "user", "content": $user_prompt}
                ]
            }')

        log "INFO" "$chunk_file の要約リクエストを実行します..."
        START_TIME=$(date +%s)
        RESPONSE=$(curl -sS -X POST "$API_URL" \
          -H "Authorization: Bearer $OPENAI_API_KEY" \
          -H "Content-Type: application/json" \
          -d "$CHUNK_API_PAYLOAD")

        if [ $? -ne 0 ]; then
            log "ERROR" "ChatGPT APIとの通信に失敗 (要約段階): $chunk_file"
            exit 1
        fi
        END_TIME=$(date +%s)
        log "INFO" "APIリクエスト所要時間: $((END_TIME - START_TIME)) 秒。"

        SUMMARY=$(echo "$RESPONSE" | jq -r '.choices[0].message.content' 2>>"$DEBUG_LOG" || echo "")

        if [ -z "$SUMMARY" ] || [ "$SUMMARY" = "null" ]; then
            log "ERROR" "$chunk_file の要約取得に失敗しました。"
            echo "Raw Response: $RESPONSE" >> "$DEBUG_LOG"
            exit 1
        fi

        SUMMARY_FILE="$SUMMARY_DIR/summary_$(basename "$chunk_file").txt"
        echo "$SUMMARY" > "$SUMMARY_FILE"
        summaries+=("$SUMMARY_FILE")
    done

    # 要約ファイルリストを標準出力へ返す
    for s in "${summaries[@]}"; do
        echo "$s"
    done
}

combine_summaries() {
    local summaries=("$@")
    COMBINED_SUMMARY="$WORK_DIR/combined_summary_$DATE.txt"
    rm -f "$COMBINED_SUMMARY"
    touch "$COMBINED_SUMMARY"

    log "INFO" "全チャンク要約を統合します..."
    {
        echo "以下は複数チャンクの要約を統合した結果です："
        echo "------------------------------------"
        for sfile in "${summaries[@]}"; do
            echo "=== チャンク要約 ==="
            cat "$sfile"
            echo ""
        done
    } > "$COMBINED_SUMMARY"

    echo "$COMBINED_SUMMARY"
}

generate_final_report() {
    local COMBINED_SUMMARY="$1"

    # HEREDOCを使ってプロンプトを記述
    FINAL_SYSTEM_PROMPT=$(cat <<EOF
あなたはプロのシステム管理コンサルタントです。以下は複数のチャンクからまとめたシステム情報要約です。これに基づいて、**非常に詳細な最終報告書**を **Markdown形式** で作成してください。報告書には、明確なMarkdown見出しを付けた以下3つの主要セクションを含めてください。

**1. システム現況の記述**：
CPU、メモリ、ディスクなどのリソース使用状況、サービスの稼働状況、ネットワーク状態、重要なログ情報、エラーメッセージ、異常な事象を列挙・記述してください。

**2. システム現況の分析**：
上記情報を分析し、潜在的な問題点、異常原因の推測、パフォーマンスボトルネック、特異な状況やリスクを指摘してください。

**3. 改善策の提案**：
分析結果に基づき、実行可能で具体的な改善策、診断手順、パフォーマンス最適化、エラー修正、システム安定性を強化するための提案を示してください。

論理的で分かりやすく、十分な情報量を確保した詳細な報告書にしてください。
EOF
)

    FINAL_INPUT=$(cat "$COMBINED_SUMMARY")

    FINAL_API_PAYLOAD=$(jq -n \
        --arg system_prompt "$FINAL_SYSTEM_PROMPT" \
        --arg user_prompt "$FINAL_INPUT" \
        --arg model "$MODEL" \
        '{
            "model": $model,
            "messages": [
                {"role": "system", "content": $system_prompt},
                {"role": "user", "content": $user_prompt}
            ]
        }')

    log "INFO" "最終分析を開始します..."
    log "DEBUG" "最終分析リクエストペイロード: $FINAL_API_PAYLOAD"

    START_TIME=$(date +%s)
    RESPONSE=$(curl -sS -X POST "$API_URL" \
      -H "Authorization: Bearer $OPENAI_API_KEY" \
      -H "Content-Type: application/json" \
      -d "$FINAL_API_PAYLOAD")

    if [ $? -ne 0 ]; then
        log "ERROR" "ChatGPT APIとの通信に失敗 (最終分析段階)"
        exit 1
    fi
    END_TIME=$(date +%s)
    log "INFO" "APIリクエスト所要時間: $((END_TIME - START_TIME)) 秒。"

    log "DEBUG" "最終分析レスポンス: $RESPONSE"

    FINAL_ANALYSIS=$(echo "$RESPONSE" | jq -r '.choices[0].message.content' 2>>"$DEBUG_LOG" || echo "")

    if [ -z "$FINAL_ANALYSIS" ] || [ "$FINAL_ANALYSIS" = "null" ]; then
        log "ERROR" "最終分析結果の取得に失敗しました。"
        echo "Raw Response: $RESPONSE" >> "$DEBUG_LOG"
        exit 1
    fi

    {
        echo "# 日次システム点検報告書 - $DATE"
        echo ""
        echo "$FINAL_ANALYSIS"
    } > "$FINAL_REPORT"

    log "INFO" "最終報告書を生成しました: $FINAL_REPORT"
}

########################################
# メイン処理フロー
########################################

log "INFO" "システム情報を収集します..."
collect_system_info
log "INFO" "システム情報を収集しました: $RAW_REPORT"

# generate_chunk_summariesの出力を配列summariesとして取得
mapfile -t summaries < <(generate_chunk_summaries)
COMBINED_SUMMARY_FILE=$(combine_summaries "${summaries[@]}")
generate_final_report "$COMBINED_SUMMARY_FILE"

log "INFO" "スクリプト実行完了。"