94 lines
2.6 KiB
JavaScript
94 lines
2.6 KiB
JavaScript
#!/usr/bin/env node
|
||
// Claude Code Statusline
|
||
// Shows: <directory> | <git branch> | <model> | [===== ] 60%
|
||
// Git branch segment is omitted entirely when not in a git repository.
|
||
|
||
'use strict';
|
||
|
||
const { execSync } = require('child_process');
|
||
const os = require('os');
|
||
|
||
const R = '\x1b[0m';
|
||
const DIM = '\x1b[90m'; // separators
|
||
const CYAN = '\x1b[36m'; // directory
|
||
const GREEN = '\x1b[32m'; // git branch
|
||
const MAGENTA = '\x1b[35m'; // model
|
||
const BAR_OK = '\x1b[32m'; // usage < 70%
|
||
const BAR_WARN = '\x1b[33m'; // usage 70–89%
|
||
const BAR_CRIT = '\x1b[31m'; // usage >= 90%
|
||
|
||
const SEP = `${DIM} | ${R}`;
|
||
|
||
let input = '';
|
||
process.stdin.on('data', chunk => input += chunk);
|
||
process.stdin.on('end', () => {
|
||
let data = {};
|
||
try { if (input.trim()) data = JSON.parse(input); } catch (_) {}
|
||
|
||
// --- Directory ---
|
||
const cwd =
|
||
(data.workspace && data.workspace.current_dir) ||
|
||
data.cwd ||
|
||
process.env.PWD ||
|
||
process.cwd();
|
||
|
||
const home = os.homedir().replace(/\\/g, '/');
|
||
const cwdNorm = cwd.replace(/\\/g, '/');
|
||
const displayDir = cwdNorm.startsWith(home)
|
||
? '~' + cwdNorm.slice(home.length)
|
||
: cwdNorm;
|
||
|
||
// --- Git branch ---
|
||
let gitBranch = null;
|
||
try {
|
||
const branch = execSync('git rev-parse --abbrev-ref HEAD', {
|
||
cwd,
|
||
stdio: ['ignore', 'pipe', 'ignore'],
|
||
timeout: 2000,
|
||
env: { ...process.env, GIT_OPTIONAL_LOCKS: '0' },
|
||
}).toString().trim();
|
||
|
||
if (branch && branch !== 'HEAD') {
|
||
gitBranch = branch;
|
||
} else {
|
||
try {
|
||
gitBranch = execSync('git rev-parse --short HEAD', {
|
||
cwd,
|
||
stdio: ['ignore', 'pipe', 'ignore'],
|
||
timeout: 2000,
|
||
env: { ...process.env, GIT_OPTIONAL_LOCKS: '0' },
|
||
}).toString().trim() || 'HEAD';
|
||
} catch (_) { gitBranch = 'HEAD'; }
|
||
}
|
||
} catch (_) { gitBranch = null; }
|
||
|
||
// --- Model ---
|
||
const model =
|
||
(data.model && data.model.display_name) ||
|
||
(data.model && data.model.id) ||
|
||
'Claude';
|
||
|
||
// --- Context window usage ---
|
||
const pct = Math.round(
|
||
(data.context_window && data.context_window.used_percentage) || 0
|
||
);
|
||
|
||
const barColor = pct >= 90 ? BAR_CRIT : pct >= 70 ? BAR_WARN : BAR_OK;
|
||
|
||
const BAR_WIDTH = 10;
|
||
const filled = Math.round((pct / 100) * BAR_WIDTH);
|
||
const bar =
|
||
`${barColor}[` +
|
||
'='.repeat(filled) +
|
||
`${DIM}` + ' '.repeat(BAR_WIDTH - filled) +
|
||
`${barColor}] ${pct}%${R}`;
|
||
|
||
// --- Assemble ---
|
||
const segments = [`${CYAN}${displayDir}${R}`];
|
||
if (gitBranch !== null) segments.push(`${GREEN}${gitBranch}${R}`);
|
||
segments.push(`${MAGENTA}${model}${R}`);
|
||
segments.push(bar);
|
||
|
||
process.stdout.write(segments.join(SEP) + '\n');
|
||
});
|