AIのトークン・クォータ (利用枠) 残量をグラフ化する 〜AntigravityとZabbixの場合〜

{
  "conversation_id": "",
  "status": "SUCCESS",
  "response": "Gemini Models\tWeekly Limit Remaining\t96%\t2026-11-14T02:40:30Z\nGemini Models\tFive Hour Limit Remaining\t95%\t2026-11-07T15:00:09Z\nClaude and GPT models\tWeekly Limit Remaining\t100%\t2026-11-14T11:01:57Z\nClaude and GPT models\tFive Hour Limit Remaining\t100%\t2026-11-07T16:01:57Z\n",
  "duration_seconds": 0,
  "num_turns": 0,
  "usage": {
    "input_tokens": 0,
    "output_tokens": 0,
    "thinking_tokens": 0,
    "cache_read_tokens": 0,
    "total_tokens": 0
  },
  "command": {
    "name": "usage",
    "data": {
      "description": "Within each group, models share a weekly limit and a 5-hour limit. Quota is consumed proportionally to the cost of the tokens. Thus, limits will last longer with shorter tasks or using more cost-effective models. The 5-hour limit smooths out aggregate demand to fairly distribute global capacity across all users, while your weekly limit is tied directly to your individual tier.",
      "groups": [
        {
          "name": "Gemini Models",
          "description": "Models within this group: Gemini Flash, Gemini Pro",
          "buckets": [
            {
              "id": "gemini-weekly",
              "name": "Weekly Limit Remaining",
              "description": "You have used some of your weekly limit, it will fully refresh in 6 days, 15 hours.",
              "window": "weekly",
              "remaining_fraction": 0.9619837999343872,
              "reset_time": "2026-11-14T02:40:30Z"
            },
            {
              "id": "gemini-5h",
              "name": "Five Hour Limit Remaining",
              "description": "You have used some of your 5-hour limit, it will fully refresh in 3 hours, 58 minutes.",
              "window": "5h",
              "remaining_fraction": 0.954495906829834,
              "reset_time": "2026-11-07T15:00:09Z"
            }
          ]
        },
        {
          "name": "Claude and GPT models",
          "description": "Models within this group: Claude Opus, Claude Sonnet, GPT-OSS",
          "buckets": [
            {
              "id": "3p-weekly",
              "name": "Weekly Limit Remaining",
              "window": "weekly",
              "remaining_fraction": 1,
              "reset_time": "2026-11-14T11:01:57Z"
            },
            {
              "id": "3p-5h",
              "name": "Five Hour Limit Remaining",
              "window": "5h",
              "remaining_fraction": 1,
              "reset_time": "2026-11-07T16:01:57Z"
            }
          ]
        }
      ]
    }
  }
}

#!/usr/bin/env bash

set -euo pipefail

# hook_export_quota.sh (Ver.20261107)

# 設定変数 (環境変数による上書き可)
readonly AGY_BIN="${AGY_BIN:-$(command -v agy 2>/dev/null || printf '%s/.local/bin/agy' "$HOME")}"
readonly OUTPUT_JSON="${OUTPUT_JSON:-/tmp/gemini_quota_hook.json}"
readonly LOCK_FILE="${LOCK_FILE:-/tmp/gemini_quota_hook.lock}"
readonly ZABBIX_SSH_TARGET="${ZABBIX_SSH_TARGET:-}"
readonly ZABBIX_HOST="${ZABBIX_HOST:-$(hostname -s 2>/dev/null || uname -n)}"

# 1. フックの入力を読み捨て、エージェントの停止を即座に許可する
if [[ ! -t 0 ]]; then
  cat > /dev/null || true
fi
printf '{"decision": "allow"}\n'

# 親子プロセスでの無限ループ防止ガード
if [[ "${GEMINI_QUOTA_HOOK_RUNNING:-0}" == "1" ]]; then
  exit 0
fi

# 実行コマンドの存在・実行権限チェック
if [[ ! -x "$AGY_BIN" ]]; then
  exit 0
fi

# 2. バックグラウンドでクォータJSONを取得して保存、SSH経由でZabbixへ送信する
(
  export GEMINI_QUOTA_HOOK_RUNNING=1

  # 別プロセス間での多重起動防止ロック
  exec 9>"$LOCK_FILE"
  if command -v flock >/dev/null 2>&1; then
    flock -n 9 || exit 0
  fi

  output_dir="$(dirname "$OUTPUT_JSON")"
  mkdir -p "$output_dir"

  tmp_file="$(mktemp "${OUTPUT_JSON}.tmp.XXXXXX")"
  trap 'rm -f "$tmp_file"' EXIT

  if "$AGY_BIN" -p "/usage" --output-format json < /dev/null 9>&- > "$tmp_file" 2>/dev/null && [[ -s "$tmp_file" ]]; then
    chmod 0644 "$tmp_file"
    mv -f "$tmp_file" "$OUTPUT_JSON"

    # Zabbixサーバーへ各バケットの残量(%)を送信
    if [[ -n "$ZABBIX_SSH_TARGET" ]] && command -v jq >/dev/null 2>&1; then
      jq -c --arg host "$ZABBIX_HOST" '{
        request: "sender data",
        data: [
          select(.status == "SUCCESS")
          | (.command.data.groups // [])[]
          | (.buckets // [])[]
          | {
              host: $host,
              key: "gemini.quota.remaining[\(.id)]",
              value: (((.remaining_fraction // 0) * 100) | tostring)
            }
        ]
      }' "$OUTPUT_JSON" | ssh -o BatchMode=yes -o ConnectTimeout=5 "$ZABBIX_SSH_TARGET" 'python3 -c "
import sys, socket, struct
payload = sys.stdin.read().encode()
if payload:
    packet = b\"ZBXD\x01\" + struct.pack(\"<Q\", len(payload)) + payload
    with socket.create_connection((\"127.0.0.1\", 10051), timeout=5) as s:
        s.sendall(packet)
        s.recv(1024)
"' >/dev/null 2>&1 || true
    fi
  fi
) </dev/null >/dev/null 2>&1 &

exit 0

{
  "quota-exporter": {
    "Stop": [
      {
        "type": "command",
        "command": "ZABBIX_SSH_TARGET=zabbix.example.com \"$HOME/.gemini/antigravity-cli/hook_export_quota.sh\"",
        "timeout": 15
      }
    ]
  }
}

*/15 * * * * ZABBIX_SSH_TARGET=zabbix.example.com /home/USER/.gemini/antigravity-cli/hook_export_quota.sh < /dev/null >/dev/null 2>&1