gcusage 0.1.0 → 0.1.2

package/dist/core/range.js

@@ -0,0 +1,57 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.isPeriodProvided = isPeriodProvided;
+exports.resolveRange = resolveRange;
+exports.applyUntilClamp = applyUntilClamp;
+const time_1 = require("./utils/time");
+function isPeriodProvided(args) {
+    return args.some((arg) => arg === "--period" || arg.startsWith("--period="));
+}
+function resolveRange(period, periodWasProvided, sinceMs, untilMs) {
+    const now = new Date();
+    if (period === "session") {
+        const start = (0, time_1.startOfDay)(now).getTime();
+        const end = (0, time_1.endOfDay)(now).getTime();
+        return applyUntilClamp({ sinceMs: start, untilMs: end }, untilMs);
+    }
+    if (!periodWasProvided && sinceMs === null && untilMs === null) {
+        const end = (0, time_1.endOfDay)(now).getTime();
+        const startDate = new Date(now.getTime());
+        startDate.setDate(startDate.getDate() - 5);
+        const start = (0, time_1.startOfDay)(startDate).getTime();
+        return { sinceMs: start, untilMs: end };
+    }
+    if (period === "day") {
+        if (sinceMs === null && untilMs === null) {
+            const start = (0, time_1.startOfDay)(now).getTime();
+            const end = (0, time_1.endOfDay)(now).getTime();
+            return { sinceMs: start, untilMs: end };
+        }
+        return { sinceMs, untilMs };
+    }
+    if (period === "week") {
+        if (sinceMs !== null) {
+            const start = (0, time_1.startOfDay)(new Date(sinceMs)).getTime();
+            const end = (0, time_1.endOfDay)(new Date(start + 6 * 86400000)).getTime();
+            return applyUntilClamp({ sinceMs: start, untilMs: end }, untilMs);
+        }
+        const start = (0, time_1.startOfWeekMonday)(now).getTime();
+        const end = (0, time_1.endOfDay)(new Date(start + 6 * 86400000)).getTime();
+        return applyUntilClamp({ sinceMs: start, untilMs: end }, untilMs);
+    }
+    if (period === "month") {
+        const base = sinceMs !== null ? new Date(sinceMs) : now;
+        const start = new Date(base.getFullYear(), base.getMonth(), 1);
+        const end = new Date(base.getFullYear(), base.getMonth() + 1, 0);
+        return applyUntilClamp({ sinceMs: (0, time_1.startOfDay)(start).getTime(), untilMs: (0, time_1.endOfDay)(end).getTime() }, untilMs);
+    }
+    return { sinceMs, untilMs };
+}
+function applyUntilClamp(range, untilMs) {
+    if (untilMs === null)
+        return range;
+    if (range.untilMs === null || untilMs < range.untilMs) {
+        return { sinceMs: range.sinceMs, untilMs };
+    }
+    return range;
+}

package/dist/core/render/table.js

@@ -0,0 +1,192 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.renderTable = renderTable;
+exports.renderDailyTable = renderDailyTable;
+exports.renderSessionTable = renderSessionTable;
+exports.colorize = colorize;
+exports.colorizeLines = colorizeLines;
+exports.applyColor = applyColor;
+exports.renderTitleBlock = renderTitleBlock;
+exports.buildRangeText = buildRangeText;
+exports.resolveDateRange = resolveDateRange;
+const messages_1 = require("../../messages");
+const format_1 = require("../utils/format");
+const aggregate_1 = require("../aggregate");
+const time_1 = require("../utils/time");
+function renderTable(result, period) {
+    if (result.table.length === 0) {
+        console.log(messages_1.MESSAGES.NO_MATCHING);
+        return;
+    }
+    if (period === "session") {
+        const rows = result.table;
+        renderSessionTable(rows, result.range);
+        return;
+    }
+    const rows = result.table;
+    renderDailyTable(rows, result.range);
+}
+function renderDailyTable(rows, range) {
+    const headers = [
+        messages_1.MESSAGES.HEADER_DATE,
+        messages_1.MESSAGES.HEADER_MODELS,
+        messages_1.MESSAGES.HEADER_INPUT,
+        messages_1.MESSAGES.HEADER_OUTPUT,
+        messages_1.MESSAGES.HEADER_THOUGHT,
+        messages_1.MESSAGES.HEADER_CACHE,
+        messages_1.MESSAGES.HEADER_TOOL,
+        messages_1.MESSAGES.HEADER_TOTAL_TOKENS
+    ];
+    const widths = headers.map((h) => h.length);
+    const totals = { input: 0, output: 0, thought: 0, cache: 0, tool: 0 };
+    for (const row of rows) {
+        const modelsText = Array.from(row.models);
+        widths[0] = Math.max(widths[0], row.date.length);
+        widths[1] = Math.max(widths[1], (0, format_1.maxLineWidth)(modelsText));
+        widths[2] = Math.max(widths[2], (0, format_1.formatNumber)(row.input).length);
+        widths[3] = Math.max(widths[3], (0, format_1.formatNumber)(row.output).length);
+        widths[4] = Math.max(widths[4], (0, format_1.formatNumber)(row.thought).length);
+        widths[5] = Math.max(widths[5], (0, format_1.formatNumber)(row.cache).length);
+        widths[6] = Math.max(widths[6], (0, format_1.formatNumber)(row.tool).length);
+        widths[7] = Math.max(widths[7], (0, format_1.formatNumber)((0, aggregate_1.sumAll)(row)).length);
+        totals.input += row.input;
+        totals.output += row.output;
+        totals.thought += row.thought;
+        totals.cache += row.cache;
+        totals.tool += row.tool;
+    }
+    const lines = [];
+    lines.push(...renderTitleBlock(rows, range));
+    lines.push(colorize(headers, "header", widths));
+    lines.push("");
+    for (const row of rows) {
+        const modelLines = Array.from(row.models);
+        lines.push(...(0, format_1.formatRowMulti)([
+            [row.date],
+            modelLines.length > 0 ? modelLines : [""],
+            [(0, format_1.formatNumber)(row.input)],
+            [(0, format_1.formatNumber)(row.output)],
+            [(0, format_1.formatNumber)(row.thought)],
+            [(0, format_1.formatNumber)(row.cache)],
+            [(0, format_1.formatNumber)(row.tool)],
+            [(0, format_1.formatNumber)((0, aggregate_1.sumAll)(row))]
+        ], widths, [false, false, true, true, true, true, true, true]));
+        lines.push("");
+    }
+    const totalCells = [
+        [messages_1.MESSAGES.TOTAL_LABEL],
+        [""],
+        [(0, format_1.formatCompact)(totals.input)],
+        [(0, format_1.formatCompact)(totals.output)],
+        [(0, format_1.formatCompact)(totals.thought)],
+        [(0, format_1.formatCompact)(totals.cache)],
+        [(0, format_1.formatCompact)(totals.tool)],
+        [(0, format_1.formatCompact)((0, aggregate_1.sumTotals)(totals))]
+    ];
+    const totalLines = (0, format_1.formatRowMulti)(totalCells, widths, [false, false, true, true, true, true, true, true]);
+    lines.push(colorizeLines(totalLines, "total"));
+    lines.push("");
+    console.log(lines.join("\n"));
+}
+function renderSessionTable(rows, range) {
+    const headers = [
+        messages_1.MESSAGES.HEADER_DATE,
+        messages_1.MESSAGES.HEADER_SESSION,
+        messages_1.MESSAGES.HEADER_MODELS,
+        messages_1.MESSAGES.HEADER_INPUT,
+        messages_1.MESSAGES.HEADER_OUTPUT,
+        messages_1.MESSAGES.HEADER_THOUGHT,
+        messages_1.MESSAGES.HEADER_CACHE,
+        messages_1.MESSAGES.HEADER_TOOL,
+        messages_1.MESSAGES.HEADER_TOTAL_TOKENS
+    ];
+    const widths = headers.map((h) => h.length);
+    const totals = { input: 0, output: 0, thought: 0, cache: 0, tool: 0 };
+    for (const row of rows) {
+        const modelsText = Array.from(row.models);
+        widths[0] = Math.max(widths[0], row.date.length);
+        widths[1] = Math.max(widths[1], row.sessionId.length);
+        widths[2] = Math.max(widths[2], (0, format_1.maxLineWidth)(modelsText));
+        widths[3] = Math.max(widths[3], (0, format_1.formatNumber)(row.input).length);
+        widths[4] = Math.max(widths[4], (0, format_1.formatNumber)(row.output).length);
+        widths[5] = Math.max(widths[5], (0, format_1.formatNumber)(row.thought).length);
+        widths[6] = Math.max(widths[6], (0, format_1.formatNumber)(row.cache).length);
+        widths[7] = Math.max(widths[7], (0, format_1.formatNumber)(row.tool).length);
+        widths[8] = Math.max(widths[8], (0, format_1.formatNumber)((0, aggregate_1.sumAll)(row)).length);
+        totals.input += row.input;
+        totals.output += row.output;
+        totals.thought += row.thought;
+        totals.cache += row.cache;
+        totals.tool += row.tool;
+    }
+    const lines = [];
+    lines.push(...renderTitleBlock(rows, range));
+    lines.push(colorize(headers, "header", widths));
+    lines.push("");
+    for (const row of rows) {
+        const modelLines = Array.from(row.models);
+        lines.push(...(0, format_1.formatRowMulti)([
+            [row.date],
+            [row.sessionId],
+            modelLines.length > 0 ? modelLines : [""],
+            [(0, format_1.formatNumber)(row.input)],
+            [(0, format_1.formatNumber)(row.output)],
+            [(0, format_1.formatNumber)(row.thought)],
+            [(0, format_1.formatNumber)(row.cache)],
+            [(0, format_1.formatNumber)(row.tool)],
+            [(0, format_1.formatNumber)((0, aggregate_1.sumAll)(row))]
+        ], widths, [false, false, false, true, true, true, true, true, true]));
+        lines.push("");
+    }
+    const totalCells = [
+        [messages_1.MESSAGES.TOTAL_LABEL],
+        [""],
+        [""],
+        [(0, format_1.formatCompact)(totals.input)],
+        [(0, format_1.formatCompact)(totals.output)],
+        [(0, format_1.formatCompact)(totals.thought)],
+        [(0, format_1.formatCompact)(totals.cache)],
+        [(0, format_1.formatCompact)(totals.tool)],
+        [(0, format_1.formatCompact)((0, aggregate_1.sumTotals)(totals))]
+    ];
+    const totalLines = (0, format_1.formatRowMulti)(totalCells, widths, [false, false, false, true, true, true, true, true, true]);
+    lines.push(colorizeLines(totalLines, "total"));
+    lines.push("");
+    console.log(lines.join("\n"));
+}
+function colorize(headers, kind, widths) {
+    const raw = (0, format_1.formatRow)(headers, widths);
+    return applyColor(raw, kind);
+}
+function colorizeLines(lines, kind) {
+    return applyColor(lines.join("\n"), kind);
+}
+function applyColor(text, kind) {
+    const code = kind === "header" ? "\u001b[36m" : "\u001b[33m";
+    const reset = "\u001b[0m";
+    return `${code}${text}${reset}`;
+}
+function renderTitleBlock(rows, range) {
+    const rangeText = buildRangeText(rows, range);
+    const title = rangeText ? `${messages_1.MESSAGES.REPORT_TITLE} - ${rangeText}` : messages_1.MESSAGES.REPORT_TITLE;
+    const border = `+${"-".repeat(title.length + 2)}+`;
+    const line = `| ${title} |`;
+    return ["", applyColor(border, "header"), applyColor(line, "header"), applyColor(border, "header"), ""];
+}
+function buildRangeText(rows, range) {
+    const [start, end] = resolveDateRange(rows, range);
+    if (!start || !end)
+        return "";
+    return start === end ? start : `${start}${messages_1.MESSAGES.RANGE_SEP}${end}`;
+}
+function resolveDateRange(rows, range) {
+    if (range.sinceMs !== null && range.untilMs !== null) {
+        const start = (0, time_1.toDateKey)(range.sinceMs);
+        const end = (0, time_1.toDateKey)(range.untilMs);
+        return [start, end];
+    }
+    const dates = rows.map((r) => r.date).sort();
+    if (dates.length === 0)
+        return [null, null];
+    return [dates[0], dates[dates.length - 1]];
+}

package/dist/core/run.js

@@ -0,0 +1,65 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.run = run;
+const os_1 = __importDefault(require("os"));
+const path_1 = __importDefault(require("path"));
+const messages_1 = require("../messages");
+const aggregate_1 = require("./aggregate");
+const logs_1 = require("./logs");
+const range_1 = require("./range");
+async function run(activePeriod, periodWasProvided, rawSince, rawUntil, modelFilter, typeFilter) {
+    const logFiles = await (0, logs_1.findLogFiles)();
+    if (logFiles.length === 0) {
+        const logPath = path_1.default.join(os_1.default.homedir(), ".gemini", "telemetry.log");
+        console.error(`${messages_1.MESSAGES.NO_LOG_FOUND}${logPath}`);
+        return { json: [], table: [], range: { sinceMs: null, untilMs: null } };
+    }
+    const points = [];
+    for (const file of logFiles) {
+        const filePoints = await (0, logs_1.parseLogFile)(file);
+        points.push(...filePoints);
+    }
+    const range = (0, range_1.resolveRange)(activePeriod, periodWasProvided, rawSince, rawUntil);
+    const filteredByType = points.filter((p) => {
+        if (modelFilter && p.model !== modelFilter)
+            return false;
+        if (typeFilter && p.type !== typeFilter)
+            return false;
+        return true;
+    });
+    const summaries = (0, aggregate_1.buildSessionSummaries)(filteredByType).filter((s) => {
+        if (range.sinceMs !== null && s.sessionStartMs < range.sinceMs)
+            return false;
+        if (range.untilMs !== null && s.sessionStartMs > range.untilMs)
+            return false;
+        return true;
+    });
+    if (activePeriod === "session") {
+        const table = summaries.map((s) => (0, aggregate_1.toSessionTotals)(s));
+        const json = table.map((row) => ({
+            date: row.date,
+            session: row.sessionId,
+            models: Array.from(row.models),
+            input: Math.round(row.input),
+            output: Math.round(row.output),
+            thought: Math.round(row.thought),
+            cache: Math.round(row.cache),
+            tool: Math.round(row.tool)
+        }));
+        return { json, table, range };
+    }
+    const table = (0, aggregate_1.aggregateSessionsToDay)(summaries);
+    const json = table.map((row) => ({
+        date: row.date,
+        models: Array.from(row.models),
+        input: Math.round(row.input),
+        output: Math.round(row.output),
+        thought: Math.round(row.thought),
+        cache: Math.round(row.cache),
+        tool: Math.round(row.tool)
+    }));
+    return { json, table, range };
+}

package/dist/core/utils/format.js

@@ -0,0 +1,45 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.formatRow = formatRow;
+exports.formatRowMulti = formatRowMulti;
+exports.maxLineWidth = maxLineWidth;
+exports.formatNumber = formatNumber;
+exports.formatCompact = formatCompact;
+function formatRow(cells, widths, align) {
+    return cells
+        .map((cell, i) => {
+        const width = widths[i];
+        if (align && align[i]) {
+            return cell.padStart(width);
+        }
+        return cell.padEnd(width);
+    })
+        .join("  ");
+}
+function formatRowMulti(cells, widths, align) {
+    const height = cells.reduce((max, col) => Math.max(max, col.length), 1);
+    const lines = [];
+    for (let rowIndex = 0; rowIndex < height; rowIndex += 1) {
+        const rowCells = cells.map((col) => (rowIndex < col.length ? col[rowIndex] : ""));
+        lines.push(formatRow(rowCells, widths, align));
+    }
+    return lines;
+}
+function maxLineWidth(lines) {
+    if (lines.length === 0)
+        return 0;
+    return lines.reduce((max, line) => Math.max(max, line.length), 0);
+}
+function formatNumber(value) {
+    return Math.round(value).toLocaleString("en-US");
+}
+function formatCompact(value) {
+    const abs = Math.abs(value);
+    if (abs >= 1e9)
+        return `${(value / 1e9).toFixed(2)}B`;
+    if (abs >= 1e6)
+        return `${(value / 1e6).toFixed(2)}M`;
+    if (abs >= 1e3)
+        return `${(value / 1e3).toFixed(2)}k`;
+    return formatNumber(value);
+}

package/dist/core/utils/period.js

@@ -0,0 +1,8 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.normalizePeriod = normalizePeriod;
+function normalizePeriod(value) {
+    if (value === "day" || value === "week" || value === "month" || value === "session")
+        return value;
+    return null;
+}

package/dist/core/utils/time.js

@@ -0,0 +1,69 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.parseTimeSpec = parseTimeSpec;
+exports.startOfDay = startOfDay;
+exports.endOfDay = endOfDay;
+exports.toDateKey = toDateKey;
+exports.pad2 = pad2;
+exports.startOfWeekMonday = startOfWeekMonday;
+const messages_1 = require("../../messages");
+function parseTimeSpec(input, kind) {
+    if (typeof input !== "string")
+        return null;
+    const raw = input.trim().toLowerCase();
+    if (!raw)
+        return null;
+    const now = new Date();
+    if (raw === "today") {
+        return kind === "since" ? startOfDay(now).getTime() : endOfDay(now).getTime();
+    }
+    if (raw === "yesterday") {
+        const d = new Date(now.getTime() - 86400000);
+        return kind === "since" ? startOfDay(d).getTime() : endOfDay(d).getTime();
+    }
+    const relMatch = raw.match(/^(\d+)([dh])$/);
+    if (relMatch) {
+        const amount = Number(relMatch[1]);
+        const unit = relMatch[2];
+        if (!Number.isNaN(amount)) {
+            const delta = unit === "d" ? amount * 86400000 : amount * 3600000;
+            return now.getTime() - delta;
+        }
+    }
+    const parsed = new Date(raw);
+    if (!Number.isNaN(parsed.getTime())) {
+        if (/^\d{4}-\d{2}-\d{2}$/.test(raw)) {
+            return kind === "since" ? startOfDay(parsed).getTime() : endOfDay(parsed).getTime();
+        }
+        return parsed.getTime();
+    }
+    console.error((0, messages_1.buildInvalidTimeMessage)(input));
+    process.exit(1);
+}
+function startOfDay(d) {
+    const tmp = new Date(d.getTime());
+    tmp.setHours(0, 0, 0, 0);
+    return tmp;
+}
+function endOfDay(d) {
+    const tmp = new Date(d.getTime());
+    tmp.setHours(23, 59, 59, 999);
+    return tmp;
+}
+function toDateKey(timestampMs) {
+    const date = new Date(timestampMs);
+    const year = date.getFullYear();
+    const month = pad2(date.getMonth() + 1);
+    const day = pad2(date.getDate());
+    return `${year}-${month}-${day}`;
+}
+function pad2(n) {
+    return n < 10 ? `0${n}` : String(n);
+}
+function startOfWeekMonday(date) {
+    const d = new Date(date.getTime());
+    const day = (d.getDay() + 6) % 7;
+    d.setDate(d.getDate() - day);
+    d.setHours(0, 0, 0, 0);
+    return d;
+}