#!/usr/bin/env bash

#


# kubectl-get-node-info: Kubernetesノード情報を取得するkubectlプラグイン

#


# 入力: なし (またはオプションでノード名)


# 出力: ノード名、CPU/メモリ使用率などの情報


# 前提: kubectl, jq, curlがインストールされ、kubectl proxyが実行中であること


# 計算量: Kubernetes APIへの1回のHTTPリクエスト、jqによるJSON処理


# メモリ: 少ない (一時ファイルは一時ディレクトリに作成され、スクリプト終了時に削除)

set -euo pipefail

# 一時ディレクトリの作成と終了時のクリーンアップ


# mktemp -dは安全な一時ディレクトリを作成する。


# trapコマンドはEXITシグナル（スクリプト終了時）に指定されたコマンドを実行し、


# 一時ディレクトリとその内容を確実に削除する。これは冪等性にも貢献する。

TMP_DIR=$(mktemp -d -t k8s-plugin-XXXXXXXXXX)
trap 'rm -rf "$TMP_DIR"' EXIT

# kubectl proxyが動いていることを想定し、localhost経由でAPIにアクセス


# プロダクション環境ではサービスアカウントトークンなど別の認証方法を検討する。

KUBE_API_SERVER="http://localhost:8001" # kubectl proxyのデフォルトポート

# APIパス

NODES_API_PATH="/api/v1/nodes"

# オプション引数としてノード名を受け取る

TARGET_NODE=""
if [[ "$#" -ge 1 ]]; then
    TARGET_NODE="$1"
fi

echo "--- Fetching Kubernetes Node Information ---"

# curlでKubernetes APIを呼び出す例


# --retry: 最大5回のリトライ


# --retry-delay: 最初のリトライまでの遅延秒数 (1秒)


# --retry-max-time: リトライ試行の最大時間 (60秒)


# -s: サイレントモード (プログレスメーター非表示)


# -S: エラー表示


# --tlsv1.2: TLSv1.2の使用を強制 (セキュリティ強化)


# --fail: HTTPエラー (4xx, 5xx) の場合、エラーとして終了


# --show-error: エラー発生時に詳細を表示

RESPONSE=$(curl -sS --fail --show-error --tlsv1.2 \
                --retry 5 --retry-delay 1 --retry-max-time 60 \
                "${KUBE_API_SERVER}${NODES_API_PATH}" \
                -H "Accept: application/json")

# jqでJSONレスポンスを処理し、ノード情報を抽出


# jqは強力なJSON処理ツール。ここではノードリストから必要な情報を整形して出力する。


# -r: raw出力 (引用符をつけない)

if [[ -z "$TARGET_NODE" ]]; then
    echo "$RESPONSE" | jq -r '
        .items[] | {
            name: .metadata.name,
            os_image: .status.nodeInfo.osImage,
            kernel_version: .status.nodeInfo.kernelVersion,
            cri_version: .status.nodeInfo.containerRuntimeVersion,
            kubelet_version: .status.nodeInfo.kubeletVersion,
            allocatable_cpu: .status.allocatable.cpu,
            allocatable_memory: .status.allocatable.memory,
            capacity_cpu: .status.capacity.cpu,
            capacity_memory: .status.capacity.memory
        } | to_entries | map("\(.key): \(.value)") | .[]
    ' | sed 's/^/  /' # インデントを追加
else
    echo "$RESPONSE" | jq -r --arg node_name "$TARGET_NODE" '
        .items[] | select(.metadata.name == $node_name) | {
            name: .metadata.name,
            os_image: .status.nodeInfo.osImage,
            kernel_version: .status.nodeInfo.kernelVersion,
            cri_version: .status.nodeInfo.containerRuntimeVersion,
            kubelet_version: .status.nodeInfo.kubeletVersion,
            allocatable_cpu: .status.allocatable.cpu,
            allocatable_memory: .status.allocatable.memory,
            capacity_cpu: .status.capacity.cpu,
            capacity_memory: .status.capacity.memory
        } | to_entries | map("\(.key): \(.value)") | .[]
    ' | sed 's/^/  /' # インデントを追加
fi

echo "--- Node Information Retrieval Complete ---"

# /etc/systemd/system/k8s-node-info-collector.service

#


# Description: Kubernetesノード情報を定期的に収集するサービス

#


# [Unit]セクション: サービスの説明と、他のユニットとの依存関係を定義。


#   After: network-online.target は、ネットワークが利用可能になってからサービスを起動することを示す。

[Unit]
Description=Kubernetes Node Information Collector Plugin
After=network-online.target

# [Service]セクション: サービス自体の実行方法を定義。


#   Type=oneshot: サービスが単一のコマンドを実行し、完了したら終了することを示す。


#   User: サービスを実行するユーザーを指定。root権限を避けるため、専用の非特権ユーザー（例: k8s-collector）を作成することを推奨。


#         ここでは例として "your_user" を使用しているが、実際の運用ではより制限されたユーザーが良い。


#   Group: サービスを実行するグループを指定。


#   ExecStart: 実行するコマンド。フルパスで指定する。


#   WorkingDirectory: コマンドが実行される作業ディレクトリ。


#   Environment: 環境変数を設定。ここではPATHを設定し、kubectlコマンドが適切に見つかるようにする。


#   StandardOutput/StandardError: 標準出力と標準エラー出力の処理方法。journalに送ることで、journalctlでログを確認できる。


#   ProtectHome=true: ホームディレクトリへの書き込みを禁止し、セキュリティを強化。


#   ReadOnlyPaths=/etc/: /etcディレクトリへの書き込みを禁止。


#   ReadWritePaths=/tmp/: /tmpディレクトリへの書き込みを許可（一時ファイル用）。

[Service]
Type=oneshot
User=your_user # 実行する非特権ユーザーを指定。例: k8s-collector
Group=your_group # 実行するグループを指定。例: k8s-collector
ExecStart=/home/your_user/.local/bin/kubectl-get-node-info
WorkingDirectory=/home/your_user
Environment="PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/home/your_user/.local/bin"
StandardOutput=journal
StandardError=journal

# セキュリティ強化のための設定例 (必要に応じて調整)

#ProtectHome=true

#ReadOnlyPaths=/etc/

#ReadWritePaths=/tmp/

# [Install]セクション: サービスがどのように有効化されるかを定義。


#   WantedBy: このサービスが属するターゲットを指定。通常は multi-user.target (システムの通常起動時)

[Install]
WantedBy=multi-user.target

# /etc/systemd/system/k8s-node-info-collector.timer

#


# Description: Kubernetesノード情報収集サービスを定期実行するタイマー

#


# [Unit]セクション: タイマーの説明。

[Unit]
Description=Run Kubernetes Node Information Collector Plugin every 15 minutes

# [Timer]セクション: タイマーのスケジュールを定義。


#   OnCalendar: 特定の日時や間隔でタイマーを起動する。ここでは15分ごとに起動。


#   AccuracySec: イベントの精度。ここでは1分以内に起動するよう指定。これにより、システム負荷を軽減しつつ、おおよそOnCalendarのタイミングで実行される。


#   Persistent=true: タイマーが最後に起動した日時を記録し、システムが停止していた期間にスキップされたイベントを、起動後に実行しようとする。

[Timer]
OnCalendar=*:0/15:00 # 毎時0分、15分、30分、45分に実行
AccuracySec=1min    # 最大1分のずれを許容
Persistent=true     # 起動時に実行できなかったスケジュールを後で実行

# [Install]セクション: タイマーがどのように有効化されるかを定義。


#   WantedBy: timers.target は、システム起動時にすべてのタイマーユニットを有効にする。

[Install]
WantedBy=timers.target

# systemd設定をリロード

sudo systemctl daemon-reload

# タイマーを有効化し、即座に起動


# タイマーを有効化すると、関連するサービスも必要に応じて有効化される。

sudo systemctl enable --now k8s-node-info-collector.timer

# タイマーとサービスの状態を確認

systemctl status k8s-node-info-collector.timer
systemctl status k8s-node-info-collector.service

# サービスが実行されたことをログで確認

journalctl -u k8s-node-info-collector.service --since "1 hour ago"