#!/usr/bin/env bash

# 堅牢なシェルスクリプトの基本設定


# -e: コマンドが失敗したら即座に終了


# -u: 未定義変数を使用したらエラー


# -o pipefail: パイプライン中のコマンドが失敗した場合、パイプライン全体を失敗とする

set -euo pipefail

# --- グローバル変数と設定 ---

API_URL="https://api.example.com/data/metrics"
TMP_DIR="/tmp/json_parser_$$"
export AWK_SCRIPT="awk_metrics.awk"

# --- トラップとクリーンアップ ---

# EXITシグナル（スクリプト終了時）に実行されるクリーンアップ関数

cleanup() {
    local exit_status=$?
    if [ -d "${TMP_DIR}" ]; then
        rm -rf "${TMP_DIR}"
        echo "[INFO] Temporary directory cleaned up: ${TMP_DIR}" 1>&2
    fi
    trap - EXIT # トラップ解除
    return $exit_status
}
trap cleanup EXIT HUP INT TERM

# --- AWKスクリプト定義 (SAX風パーサー) ---

# AWKスクリプトを一時ファイルに書き出す


# FSはタブ(\t)に設定し、JSON Path ($1) と値 ($2) を分離する。

cat << 'EOF_AWK' > "${TMP_DIR}/${AWK_SCRIPT}"
BEGIN {
    FS="\t";
    sum = 0;
    count = 0;
}

# パスが 'metrics.[任意の数値].value' にマッチする場合、値を集計する


# 注意: AWKでは正規表現の\dは非標準。POSIX互換性のために[0-9]を使用。

$1 ~ /^metrics\.[0-9]+\.value$/ {

    # $2 (値)を数値として処理

    value = $2 + 0;
    sum += value;
    count++;
}

END {
    if (count > 0) {
        average = sum / count;
        print "Total Records Processed:", count;
        print "Total Value Sum:", sum;
        print "Average Metric Value:", average;
    } else {

        # 集計対象のデータが見つからなかった場合

        print "ERROR: No target metrics found." | "cat 1>&2";
        exit 1;
    }
}
EOF_AWK

# --- メイン処理関数 ---

fetch_and_process_metrics() {
    local output_file=$1
    echo "[INFO] Fetching JSON stream from ${API_URL}..." 1>&2

    # 1. APIからのデータ取得とストリーム処理


    # -s: サイレントモード (-S: エラー表示を抑制しない)


    # -L: リダイレクトを追跡


    # -f: エラー発生時にゼロ以外の終了コードを返す


    # --retry 5: 5回リトライ

    curl -sSLf --retry 5 "${API_URL}" \
    | jq -r '

        # 2. jqによるストリーム整形（パスと値のTSV出力）


        # paths: 全てのパスを配列として生成


        # select: オブジェクトや配列全体ではなく、リーフノードの値のみを選択

        paths as $p |
        select( (.[$p | unique] | type) != "object" and (.[$p | unique] | type) != "array" ) |

        # パスをドット区切りに変換し、タブと値を連結

        ($p | map(tostring) | join(".")) + "\t" + (.[$p | unique] | tostring)
    ' \
    | awk -f "${TMP_DIR}/${AWK_SCRIPT}" > "${output_file}"

    local pipe_status=("${PIPESTATUS[@]}")

    # JQ (2番目) と AWK (3番目) のステータスをチェック

    if [[ ${pipe_status[1]} -ne 0 ]]; then
        echo "[ERROR] JQ filtering failed with status: ${pipe_status[1]}" 1>&2
        return 1
    fi
    if [[ ${pipe_status[2]} -ne 0 ]]; then
        echo "[ERROR] AWK processing failed with status: ${pipe_status[2]}" 1>&2
        return 1
    fi

    echo "[INFO] Processing completed successfully. Output saved to ${output_file}" 1>&2
}

# --- 実行ブロック ---

mkdir -p "${TMP_DIR}"

# ダミーAPI応答ファイルの作成 (テスト用)

cat << EOF_JSON > "${TMP_DIR}/response.json"
{
  "timestamp": "2024-01-01T10:00:00Z",
  "status": "OK",
  "metrics": [
    { "id": "A1", "value": 150.5, "unit": "ms" },
    { "id": "B2", "value": 120.0, "unit": "ms" },
    { "id": "C3", "value": 180.5, "unit": "ms" }
  ],
  "metadata": { "region": "us-east-1" }
}
EOF_JSON

# 実際に外部APIを叩く代わりに、ダミーファイルから処理する例


# 実際の運用では curl ... | jq ... とする。

echo "[INFO] Starting JSON Stream Parse Test..."
cat "${TMP_DIR}/response.json" \
| jq -r '
    paths as $p |
    select( (.[$p | unique] | type) != "object" and (.[$p | unique] | type) != "array" ) |
    ($p | map(tostring) | join(".")) + "\t" + (.[$p | unique] | tostring)
' \
| awk -f "${TMP_DIR}/${AWK_SCRIPT}"

# fetch_and_process_metrics "processed_data.txt" # 外部API呼び出し時

# スクリプトの正常終了を示す

exit 0

# ユニット名が 'metrics-processor.service' の場合

journalctl -u metrics-processor.service --since "1 hour ago" -e