agent-tank 0.9.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Rinalds Uzkalns (https://propr.dev)
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,675 @@
+# Agent Tank
+
+![Agent Tank Web UI preview](./media/www-preview.png)
+
+Monitor and query usage limits for LLM CLI tools (Claude, Gemini, Codex) via a simple HTTP API.
+
+> **Note:** Agent Tank is designed for **AI coding agent subscriptions** such as Claude Code Pro/Max, Google Gemini CLI (with Advanced/Max plans), and ChatGPT Codex (with Plus/Pro plans). It tracks **active session and rate limit usage**—not API key consumption or pay-per-use billing. If you're using API keys with pay-as-you-go pricing, this tool won't help you track costs; check your provider's billing dashboard instead.
+
+## Features
+
+- **Privacy-first** - Runs entirely locally with no external data transmission
+- **Auto-discovery** - Automatically detects installed LLM CLI tools
+- **HTTP API** - Query usage limits via REST endpoints
+- **Unified Web UI** - All-in-one dashboard showing Claude, Gemini, and Codex usage at a glance
+- **Instant Tab Tracking** - Pin the dashboard to your browser tab and see live usage updates via favicon and title changes
+- **Lightweight** - Single dependency (node-pty)
+- **Multi-agent** - Monitor Claude, Gemini, and Codex simultaneously
+- **Secure by design** - No browser cookies, web scraping, or credential access
+
+## Usage
+
+### Basic Usage
+
+```bash
+# Auto-discover and monitor all available LLM agents
+agent-tank
+
+# Monitor specific agents only
+agent-tank --claude --gemini
+
+# Use a custom port (default: 3456)
+agent-tank --port 8080
+
+# Fetch usage once and exit (no HTTP server)
+agent-tank --once
+
+# Output pure JSON for scripting/piping
+agent-tank --once --json
+```
+
+### Command Line Options
+
+```
+Options:
+  --claude              Enable Claude monitoring
+  --gemini              Enable Gemini monitoring
+  --codex               Enable Codex monitoring
+  --port <port>         HTTP server port (default: 3456)
+  --host <host>         Bind address (default: 127.0.0.1)
+  --auth-user <user>    HTTP Basic Auth username
+  --auth-pass <pass>    HTTP Basic Auth password
+  --auth-token <token>  API key for Bearer token auth
+  --fresh-process       Spawn a new process per refresh (default: false)
+  --claude-api          Use direct Anthropic API for Claude usage (faster, 60s refresh)
+  --config, -c          Path to config file (JSON)
+  --auto-discover       Auto-discover available agents (default: true)
+  --auto-refresh        Enable/disable background auto-refresh (default: true)
+  --auto-refresh-mode <mode>         Refresh mode: none, interval, activity (default: activity)
+  --auto-refresh-interval <seconds>  Auto-refresh interval in seconds (default: 60)
+  --activity-debounce <ms>           Activity debounce interval in milliseconds (default: 5000)
+  --keepalive           Enable/disable session keepalive (default: true)
+  --keepalive-interval <seconds>     Session keepalive interval in seconds (default: 300)
+  --history-retention-days <days>    Days to retain usage history (default: 14)
+  --once                Fetch usage once and exit (no HTTP server)
+  --json                Output pure JSON (suppress logging, use with --once)
+  --help, -h            Show this help message
+```
+
+### Environment Variables
+
+Environment variables override CLI flags and config file settings:
+
+| Variable | Description |
+|----------|-------------|
+| `AGENT_TANK_USER` | Basic auth username (overrides `--auth-user`) |
+| `AGENT_TANK_PASS` | Basic auth password (overrides `--auth-pass`) |
+| `AGENT_TANK_TOKEN` | API key (overrides `--auth-token`) |
+| `AGENT_TANK_HOST` | Bind address (overrides `--host`) |
+| `AGENT_TANK_FRESH_PROCESS` | Use fresh process per refresh (`1` or `true`) |
+| `AGENT_TANK_CLAUDE_API` | Use direct Anthropic API for Claude usage (`1` or `true`) |
+| `AGENT_TANK_AUTO_REFRESH` | Enable/disable background auto-refresh (`1`/`true` or `0`/`false`) |
+| `AGENT_TANK_AUTO_REFRESH_MODE` | Refresh mode: `none`, `interval`, or `activity` (default: `activity`) |
+| `AGENT_TANK_AUTO_REFRESH_INTERVAL` | Auto-refresh interval in seconds |
+| `AGENT_TANK_ACTIVITY_DEBOUNCE` | Activity debounce interval in milliseconds (default: 5000) |
+| `AGENT_TANK_KEEPALIVE` | Enable/disable session keepalive (`1`/`true` or `0`/`false`) |
+| `AGENT_TANK_KEEPALIVE_INTERVAL` | Session keepalive interval in seconds (default: 300) |
+| `AGENT_TANK_HISTORY_RETENTION_DAYS` | Days to retain usage history (default: 14) |
+
+### Configuration File
+
+You can use a JSON configuration file:
+
+```json
+{
+  "claude": true,
+  "gemini": true,
+  "codex": false,
+  "port": 8080,
+  "history": {
+    "retentionDays": 7
+  }
+}
+```
+
+```bash
+agent-tank -c config.json
+```
+
+## Claude API Mode
+
+By default, Claude usage data is fetched by spawning a PTY session and running the `/usage` command. With `--claude-api`, Agent Tank fetches usage directly from the Anthropic OAuth API instead, which is faster and allows a 60-second refresh interval (vs 10 minutes for PTY).
+
+```bash
+agent-tank --claude --claude-api
+```
+
+### How It Works
+
+- Uses the OAuth token from `~/.claude/.credentials.json` (the same credentials Claude Code uses)
+- Calls the `api.anthropic.com/api/oauth/usage` endpoint with the `anthropic-beta: oauth-2025-04-20` header
+- Automatically refreshes expired tokens using the OAuth refresh token flow
+- Falls back to PTY mode if the API call fails
+
+### Configuration
+
+```json
+{
+  "claude": true,
+  "claudeApi": true
+}
+```
+
+Or via environment variable:
+
+```bash
+AGENT_TANK_CLAUDE_API=1 agent-tank --claude
+```
+
+## Activity-Based Polling
+
+Agent Tank supports intelligent activity-based polling that monitors local log directories for LLM CLI activity. Instead of polling on a fixed interval (which can waste resources during idle periods), activity mode only triggers usage refreshes when you're actively using the CLI tools.
+
+### How It Works
+
+1. **Log Directory Monitoring**: Agent Tank watches the following directories for file changes:
+   - Claude: `~/.config/claude/projects/`, `~/.claude/`
+   - Codex: `~/.codex/sessions/`, `~/.codex/`
+   - Gemini: `~/.config/gemini/`, `~/.gemini/`
+
+2. **Debounced Detection**: When activity is detected, a configurable debounce timer starts. This prevents excessive refreshes during bursts of activity.
+
+3. **On-Demand Refresh Cycles**: After the debounce period, a refresh cycle begins and continues at the configured interval while activity is ongoing.
+
+4. **Idle State**: When no more activity is detected, polling stops to conserve resources.
+
+### Auto-Refresh Modes
+
+Agent Tank supports three refresh modes:
+
+| Mode | Description |
+|------|-------------|
+| `activity` | (Default) Monitors log directories and refreshes when CLI activity is detected |
+| `interval` | Traditional interval-based polling at fixed intervals |
+| `none` | No automatic refresh; manual refresh only via `POST /refresh` |
+
+### Configuration
+
+```bash
+# Use activity-based polling (default)
+agent-tank
+
+# Use activity mode with custom debounce (wait 10 seconds after activity)
+agent-tank --activity-debounce 10000
+
+# Use traditional interval-based polling
+agent-tank --auto-refresh-mode interval
+
+# Disable all auto-refresh (manual only)
+agent-tank --auto-refresh-mode none
+
+# Via environment variables
+AGENT_TANK_AUTO_REFRESH_MODE=activity agent-tank
+AGENT_TANK_ACTIVITY_DEBOUNCE=10000 agent-tank
+```
+
+### Notes
+
+- Activity mode automatically falls back to interval mode if no log directories are found
+- The debounce interval is specified in milliseconds (default: 5000ms = 5 seconds)
+- During active usage, refreshes occur at the `--auto-refresh-interval` rate (default: 60 seconds)
+- Activity mode is disabled in one-shot mode (`--once`)
+
+## Session Keepalive
+
+Agent Tank includes an automatic session keepalive feature that prevents LLM CLI sessions from expiring due to inactivity. This is especially useful for Claude and Codex, which have session timeouts that require periodic activity.
+
+### How It Works
+
+The keepalive manager runs in the background and periodically sends lightweight "ping" commands to each agent's PTY session. This maintains the connection without triggering API calls or affecting rate limits.
+
+### Configuration
+
+```bash
+# Use default keepalive (every 5 minutes)
+agent-tank
+
+# Custom keepalive interval (every 10 minutes)
+agent-tank --keepalive-interval 600
+
+# Disable keepalive
+agent-tank --no-keepalive
+
+# Via environment variable
+AGENT_TANK_KEEPALIVE_INTERVAL=600 agent-tank
+```
+
+### Notes
+
+- Keepalive is automatically disabled in fresh process mode (`--fresh-process`) since there are no persistent sessions to maintain
+- Keepalive is also disabled in one-shot mode (`--once`)
+- The feature only affects PTY-based sessions; JSON-RPC mode (used by Codex when available) doesn't require keepalive
+
+## Usage History & Pace Evaluation
+
+Agent Tank automatically tracks usage history and evaluates whether you're burning through your rate limits faster than expected.
+
+### Features
+
+- **Historical Snapshots**: Usage percentages are stored in `~/.agent-tank/history.json` with timestamps
+- **Automatic Pruning**: Records older than the retention period (default: 14 days) are automatically removed
+- **Pace Evaluation**: Each metric includes a `paceEval` object that calculates:
+  - `expectedPercent`: What your usage should be based on linear pace
+  - `isBurningFast`: Whether you're using faster than the 1.2x threshold
+  - `etaSeconds`: Estimated seconds until you hit 100% (if burning fast)
+  - `deltaPercent`: Difference between actual and expected usage
+
+### Configuration
+
+```bash
+# Keep 7 days of history instead of default 14
+agent-tank --history-retention-days 7
+
+# Via environment variable
+AGENT_TANK_HISTORY_RETENTION_DAYS=7 agent-tank
+```
+
+### API Endpoints
+
+- `GET /history` - Get history statistics (total records, per-agent counts, date ranges)
+- `GET /history/:agent` - Get full history for a specific agent
+
+### Example Pace Evaluation Data
+
+When an agent is burning faster than expected, the `paceEval` object is attached to each metric:
+
+```json
+{
+  "session": {
+    "percent": 60,
+    "resetsInSeconds": 9000,
+    "paceEval": {
+      "expectedPercent": 50,
+      "isBurningFast": true,
+      "etaSeconds": 6000,
+      "paceRatio": 1.2,
+      "elapsedPercent": 50,
+      "deltaPercent": 10
+    }
+  }
+}
+```
+
+This example shows: 50% of the 5-hour session has elapsed, but 60% of usage is consumed. At this pace (1.2x), you'll hit 100% in approximately 6000 seconds (1h 40m).
+
+## Installation
+
+```bash
+npm install -g agent-tank
+```
+
+Or run directly with npx:
+
+```bash
+npx agent-tank
+```
+
+## HTTP API
+
+### Endpoints
+
+| Method | Endpoint | Description |
+|--------|----------|-------------|
+| GET | `/` | HTML status page |
+| GET | `/status` | JSON status for all agents |
+| GET | `/status/:agent` | JSON status for specific agent |
+| GET | `/config` | Auto-refresh and history configuration (JSON) |
+| GET | `/history` | Usage history statistics (JSON) |
+| GET | `/history/:agent` | Usage history for specific agent (JSON) |
+| POST | `/refresh` | Refresh all agents |
+| POST | `/refresh/:agent` | Refresh specific agent |
+
+### Example Responses
+
+#### GET /status
+
+```json
+{
+  "claude": {
+    "name": "claude",
+    "usage": {
+      "session": {
+        "label": "Current session",
+        "percent": 42,
+        "resetsAt": "10pm (Europe/London)",
+        "resetsIn": "12m",
+        "resetsInSeconds": 764
+      },
+      "weeklyAll": {
+        "label": "Current week (all models)",
+        "percent": 31,
+        "resetsAt": "Mar 13, 3am (Europe/London)",
+        "resetsIn": "4d 5h",
+        "resetsInSeconds": 364364
+      },
+      "weeklySonnet": {
+        "label": "Current week (Sonnet only)",
+        "percent": 5,
+        "resetsAt": "Mar 13, 10am (Europe/London)",
+        "resetsIn": "4d 12h",
+        "resetsInSeconds": 389564
+      }
+    },
+    "metadata": {
+      "sessionId": "0b34aa59-...",
+      "cwd": "/tmp",
+      "organization": "Your Organization",
+      "email": "user@example.com",
+      "version": "2.1.71"
+    },
+    "lastUpdated": "2026-03-08T21:47:15.090Z",
+    "error": null,
+    "isRefreshing": false
+  },
+  "gemini": {
+    "name": "gemini",
+    "usage": {
+      "models": [
+        {
+          "model": "gemini-2.5-flash",
+          "usageLeft": 100,
+          "resetsIn": "24h",
+          "percentUsed": 0,
+          "resetsInSeconds": 86400
+        },
+        {
+          "model": "gemini-2.5-pro",
+          "usageLeft": 99.5,
+          "resetsIn": "12h 16m",
+          "percentUsed": 0.5,
+          "resetsInSeconds": 44160
+        }
+      ]
+    },
+    "metadata": {
+      "version": "0.24.5",
+      "email": "user@example.com",
+      "authMethod": "OAuth",
+      "model": "auto-gemini-2.5",
+      "os": "linux",
+      "updateAvailable": {
+        "current": "0.24.5",
+        "latest": "0.32.1"
+      }
+    },
+    "lastUpdated": "2026-03-08T21:47:14.138Z",
+    "error": null,
+    "isRefreshing": false
+  },
+  "codex": {
+    "name": "codex",
+    "usage": {
+      "fiveHour": {
+        "percentLeft": 100,
+        "resetsAt": "02:44 on 9 Mar",
+        "label": "5h limit",
+        "percentUsed": 0,
+        "resetsIn": "4h 56m",
+        "resetsInSeconds": 17807
+      },
+      "weekly": {
+        "percentLeft": 100,
+        "resetsAt": "21:44 on 15 Mar",
+        "label": "Weekly limit",
+        "percentUsed": 0,
+        "resetsIn": "6d 23h",
+        "resetsInSeconds": 604607
+      },
+      "version": {
+        "current": "0.107.0",
+        "latest": "0.111.0"
+      },
+      "modelLimits": [
+        {
+          "name": "GPT-5.3-Codex-Spark",
+          "fiveHour": {
+            "percentLeft": 100,
+            "resetsAt": "02:46 on 9 Mar",
+            "label": "5h limit",
+            "percentUsed": 0,
+            "resetsIn": "4h 58m",
+            "resetsInSeconds": 17927
+          },
+          "weekly": {
+            "percentLeft": 100,
+            "resetsAt": "21:46 on 15 Mar",
+            "label": "Weekly limit",
+            "percentUsed": 0,
+            "resetsIn": "6d 23h",
+            "resetsInSeconds": 604727
+          }
+        }
+      ],
+      "model": "gpt-5.3-codex",
+      "account": "user@example.com"
+    },
+    "metadata": {
+      "directory": "/tmp",
+      "sessionId": "019ccf68-...",
+      "collaborationMode": "Default",
+      "model": "gpt-5.3-codex",
+      "email": "user@example.com"
+    },
+    "lastUpdated": "2026-03-08T21:47:12.648Z",
+    "error": null,
+    "isRefreshing": false
+  }
+}
+```
+
+#### GET /status/claude
+
+```json
+{
+  "name": "claude",
+  "usage": {
+    "session": {
+      "label": "Current session",
+      "percent": 42,
+      "resetsAt": "10pm (Europe/London)",
+      "resetsIn": "12m",
+      "resetsInSeconds": 764
+    },
+    "weeklyAll": {
+      "label": "Current week (all models)",
+      "percent": 31,
+      "resetsAt": "Mar 13, 3am (Europe/London)",
+      "resetsIn": "4d 5h",
+      "resetsInSeconds": 364364
+    },
+    "weeklySonnet": {
+      "label": "Current week (Sonnet only)",
+      "percent": 5,
+      "resetsAt": "Mar 13, 10am (Europe/London)",
+      "resetsIn": "4d 12h",
+      "resetsInSeconds": 389564
+    }
+  },
+  "metadata": {
+    "sessionId": "0b34aa59-...",
+    "cwd": "/tmp",
+    "organization": "Your Organization",
+    "email": "user@example.com",
+    "version": "2.1.71"
+  },
+  "lastUpdated": "2026-03-08T21:47:15.090Z",
+  "error": null,
+  "isRefreshing": false
+}
+```
+
+## Developer & Advanced Setup
+
+### Prerequisites
+
+#### Build Requirements
+
+The `node-pty` dependency requires native compilation. You'll need:
+
+- **Python 3.8+** (Python 3.11 recommended)
+- **C++ build tools** (gcc, g++, make)
+- **Node.js development headers**
+
+##### Linux (Ubuntu/Debian)
+```bash
+sudo apt-get install python3.11 python3.11-dev build-essential nodejs-dev
+```
+
+##### Linux (Fedora/RHEL/CentOS)
+```bash
+sudo dnf install python3.11 python3.11-devel gcc gcc-c++ make nodejs-devel
+```
+
+##### Linux (openSUSE)
+```bash
+sudo zypper install python311 python311-devel gcc gcc-c++ make nodejs20-devel
+```
+
+##### macOS
+```bash
+# Install Xcode Command Line Tools if not already installed
+xcode-select --install
+
+# Install Python 3.11
+brew install python@3.11
+```
+
+##### Windows
+- Install [Python 3.11+](https://www.python.org/downloads/)
+- Install [Visual Studio Build Tools](https://visualstudio.microsoft.com/downloads/#build-tools-for-visual-studio-2022)
+
+#### LLM CLI Tools
+
+You need at least one of these CLI tools installed and authenticated:
+
+- [Claude Code](https://claude.ai/download) (`claude`) - **Version 2.0+ required** for `/usage` command support
+- [Gemini CLI](https://github.com/anthropics/gemini-cli) (`gemini`) - **Version 0.24.5+ required** for `/stats` command support
+- [OpenAI Codex](https://platform.openai.com/docs/codex) (`codex`)
+
+**Version Requirements:**
+
+**Claude Code:** Version 1.x does not support the `/usage` command.
+```bash
+# Check version
+claude --version
+
+# Update to latest
+npm update -g @anthropic-ai/claude-code
+```
+
+**Gemini CLI:** Version 0.24.4 and below do not support the `/stats` command properly.
+```bash
+# Check version
+gemini --version
+
+# Update to latest
+npm update -g gemini
+```
+
+### Programmatic Usage
+
+```javascript
+const { AgentTank } = require('agent-tank');
+
+const watcher = new AgentTank({
+  agents: ['claude', 'gemini'], // or null for auto-discover
+  port: 3456,
+  autoDiscover: true
+});
+
+await watcher.start();
+
+// Get status programmatically
+const status = watcher.getStatus();
+console.log(status.claude.usage);
+
+// Refresh a specific agent
+await watcher.refreshAgent('claude');
+
+// Stop the server
+watcher.stop();
+```
+
+### How It Works: Privacy-First Local Execution
+
+Agent Tank is designed with privacy and security as core principles. Understanding how it collects usage data is essential for users evaluating the tool's security posture.
+
+#### Local Execution Architecture
+
+The tool operates entirely on your local machine using two different approaches depending on the CLI tool's capabilities:
+
+**JSON-RPC Mode (Codex)**
+
+For the Codex CLI, Agent Tank uses a structured JSON-RPC protocol when available:
+
+1. **Starts the app-server** - Launches `codex -s read-only -a untrusted app-server`
+2. **Sends JSON-RPC requests** - Calls `account/rateLimits/read` via stdin
+3. **Receives structured data** - Parses JSON responses with precise rate limit information
+4. **Falls back to PTY** - Automatically uses PTY mode if JSON-RPC is unavailable (older CLI versions)
+
+This approach provides more reliable data parsing and is the preferred method for Codex.
+
+**PTY-Based Mode (Claude, Gemini, Codex fallback)**
+
+For other CLI tools (and as a fallback for Codex), Agent Tank spawns instances within a pseudo-terminal (PTY). This approach is functionally equivalent to you opening a terminal window and typing commands yourself:
+
+1. **Spawns a local PTY** - Creates a pseudo-terminal session on your machine
+2. **Launches the CLI tool** - Starts the authenticated CLI (e.g., `claude`, `gemini`, or `codex`)
+3. **Sends usage commands** - Types the appropriate command (`/usage`, `/stats`, or `/status`)
+4. **Parses the output** - Reads and structures the text response from the terminal
+5. **Exposes via local HTTP** - Makes the parsed data available through a localhost API
+
+#### What Agent Tank Does NOT Do
+
+To be explicit about what this tool avoids:
+
+- **No browser cookie access** - Agent Tank never reads, parses, or transmits browser cookies
+- **No web scraping** - The tool does not access web interfaces or scrape HTML pages
+- **No credential extraction** - Your API keys, tokens, or passwords are never accessed or stored
+- **No network interception** - There is no proxy, MITM, or traffic inspection involved
+- **No external data transmission** - Usage data stays on your machine; nothing is sent to external servers
+
+#### Why This Approach?
+
+LLM usage data is sensitive—it can reveal work patterns, subscription tiers, and usage intensity. By operating through local CLI tools that you've already authenticated, Agent Tank inherits the security model you've already established with each provider. The tool acts as a local automation layer, not a data collection service.
+
+#### Supported Commands
+
+| Agent | Method | Command/RPC | Output |
+|-------|--------|-------------|--------|
+| Claude | PTY | `/usage` | Session %, Weekly % |
+| Gemini | PTY | `/stats` | Model-specific usage % |
+| Codex | JSON-RPC | `account/rateLimits/read` | 5h limit %, Weekly % |
+| Codex | PTY (fallback) | `/status` | 5h limit %, Weekly % |
+
+## Troubleshooting
+
+### Installation fails with "gyp ERR!"
+
+This indicates missing build dependencies. Make sure you have:
+
+1. **Python 3.8+**: Check with `python3 --version`. If you have multiple Python versions, set the PYTHON environment variable:
+   ```bash
+   PYTHON=/usr/bin/python3.11 npm install
+   ```
+
+2. **Build tools**: Install gcc, g++, and make for your platform (see Prerequisites above)
+
+3. **Node.js headers**: Install the nodejs-devel or nodejs-dev package for your distribution
+
+If you still have issues, try rebuilding:
+```bash
+npm run rebuild
+```
+
+### Agent shows "Timeout waiting for usage data"
+
+- Ensure the CLI tool is properly authenticated
+- Try running the CLI tool manually to verify it works
+- Check if there are any prompts requiring user input
+
+### Codex fails with cursor position error
+
+This is handled automatically. If you still see issues, ensure your terminal supports standard escape sequences.
+
+### Port already in use
+
+```bash
+agent-tank --port 8080
+```
+
+## Community
+
+[r/agenttank](https://www.reddit.com/r/agenttank)
+
+## Author
+
+[Rinalds Uzkalns](https://propr.dev)
+
+## License
+
+MIT
+
+## Contributing
+
+Contributions are welcome! Please feel free to submit a Pull Request.