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"

OPENAI_KEY_FILE="/home/your_user_name/.openai_api_key"

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

# 檢查磁碟空間
AVAILABLE_SPACE=$(df "$WORK_DIR" | tail -1 | awk '{print $4}')
if [ "$AVAILABLE_SPACE" -lt 102400 ]; then # 少於 100MB
    echo "[ERROR] $WORK_DIR 磁碟空間不足。" | tee -a "$DEBUG_LOG"
    exit 1
fi

# 日誌函數：將訊息輸出到stderr，以免干擾函數回傳值
log() {
    local LEVEL=$1
    shift
    local MESSAGE="$@"
    # 將訊息寫入DEBUG_LOG並透過stderr顯示
    {
        echo "[$LEVEL] $MESSAGE" | tee -a "$DEBUG_LOG" >&2
        logger -t "system_checking_by_chatgpt" "[$LEVEL] $MESSAGE"
    }
}

log "INFO" "開始執行程式..."

########################################
# 讀取 API 金鑰
########################################

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() {
    # 收集系統資訊到 RAW_REPORT
    {
        echo "=== System Uptime ==="
        uptime
        echo ""

        echo "=== Date & Time ==="
        date
        echo ""

        echo "=== CPU & Memory Usage ==="
        top -b -n1 | head -n 20
        echo ""

        echo "=== Memory Usage (free) ==="
        free -h
        echo ""

        echo "=== Disk Usage (df) ==="
        df -h
        echo ""

        echo "=== Failed Services ==="
        systemctl list-units --state=failed
        echo ""

        echo "=== Temperature Sensors ==="
        sensors 2>/dev/null || echo "[WARNING] 無法取得溫度資訊"
        echo ""

        echo "=== NVMe / SMART Status ==="
        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 "=== Recent dmesg Entries (Last 100 lines) ==="
        dmesg | tail -n 100
        echo ""

        echo "=== System Journal (Last 300 lines, errors only) ==="
        journalctl -p err -n 300
        echo ""

        echo "=== Network Status (ss -tulpn) ==="
        ss -tulpn 2>/dev/null || echo "[WARNING] 無法檢查網路狀態"
        echo ""

        echo "=== Network Interface Statistics (ip -s link) ==="
        ip -s link
        echo ""

        echo "=== Docker Containers (if any) ==="
        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" "沒有產生任何 chunk！"
        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": "user", "content": $user_prompt},
                    {"role": "system", "content": $system_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

    # 僅輸出 summary 檔案清單到stdout（不輸出其他日誌），確保呼叫者可正確取得檔名列表
    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"

    FINAL_SYSTEM_PROMPT="你是一位專業的系統管理顧問。以上是由多個分段摘要組成的合併內容，該系統可能遭到惡意攻擊，請針對整體系統狀況產出「詳細的最終報告」。並包含以下三個主要章節：

**1. 系統現況描述**：
列出目前系統的狀況、服務運行情形、網路狀態、重要日誌資訊、錯誤訊息及任何異常事件的細節。

**2. 系統現況分析**：
針對上述資訊進行分析，指出可能的問題點、異常原因推測、系統效能瓶頸，以及任何不尋常的狀況或風險。

**3. 建議解決方式**：
根據分析結果，提出可行且明確的改善方案、診斷指令或檢查流程建議，包括優化效能、修正錯誤、強化系統穩定性的建議。

在分析過程中，你必須針對每個問題點補上對應的檢查指令和流程建議，必要時可以繪製詳細表格進行記錄。"

    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": "user", "content": $user_prompt},
                {"role": "system", "content": $system_prompt}
            ]
        }')

    log "INFO" "開始最終分析..."

    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"

# 使用 readarray 讀取 generate_chunk_summaries 的輸出至陣列
mapfile -t summaries < <(generate_chunk_summaries)
COMBINED_SUMMARY_FILE=$(combine_summaries "${summaries[@]}")
generate_final_report "$COMBINED_SUMMARY_FILE"

log "INFO" "程式執行完成。"