Skip to content

Show loading on start #95

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
rajbos merged 2 commits into from
Jan 4, 2026
Merged

Show loading on start #95

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
b7a9180
Loading message at startup
rajbos Jan 4, 2026
6453604
Show progress during loading
rajbos Jan 4, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
47 changes: 33 additions & 14 deletions src/extension.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -165,6 +165,7 @@ class CopilotTokenTracker implements vscode.Disposable {
100
);
this.statusBarItem.name = "GitHub Copilot Token Usage";
this.statusBarItem.text = "$(loading~spin) Copilot Tokens: Loading...";
this.statusBarItem.tooltip = "Daily and monthly GitHub Copilot token usage - Click to open details";
this.statusBarItem.command = 'copilot-token-tracker.showDetails';
this.statusBarItem.show();
Expand Down Expand Up @@ -213,10 +214,10 @@ class CopilotTokenTracker implements vscode.Disposable {
}, 5 * 1000);
} else if (!copilotExtension && !copilotChatExtension) {
this.log('No Copilot extensions found - starting immediate update');
this.updateTokenStats();
setTimeout(() => this.updateTokenStats(), 100);
} else {
this.log('Copilot extensions are active - starting immediate update');
this.updateTokenStats();
setTimeout(() => this.updateTokenStats(), 100);
}
}

Expand All @@ -241,10 +242,13 @@ class CopilotTokenTracker implements vscode.Disposable {
}
}

public async updateTokenStats(): Promise<void> {
public async updateTokenStats(): Promise<DetailedStats | undefined> {
try {
this.log('Updating token stats...');
const detailedStats = await this.calculateDetailedStats();
const detailedStats = await this.calculateDetailedStats((completed, total) => {
const percentage = Math.round((completed / total) * 100);
this.statusBarItem.text = `$(loading~spin) Analyzing Logs: ${percentage}%`;
});

this.statusBarItem.text = `$(symbol-numeric) ${detailedStats.today.tokens.toLocaleString()} | ${detailedStats.month.tokens.toLocaleString()}`;

Expand Down Expand Up @@ -285,10 +289,12 @@ class CopilotTokenTracker implements vscode.Disposable {
}

this.log(`Updated stats - Today: ${detailedStats.today.tokens}, Month: ${detailedStats.month.tokens}`);
return detailedStats;
} catch (error) {
this.error('Error updating token stats:', error);
this.statusBarItem.text = '$(error) Token Error';
this.statusBarItem.tooltip = 'Error calculating token usage';
return undefined;
}
}

Expand Down Expand Up @@ -333,7 +339,7 @@ class CopilotTokenTracker implements vscode.Disposable {
};
}

private async calculateDetailedStats(): Promise<DetailedStats> {
private async calculateDetailedStats(progressCallback?: (completed: number, total: number) => void): Promise<DetailedStats> {
const now = new Date();
const todayStart = new Date(now.getFullYear(), now.getMonth(), now.getDate());
const monthStart = new Date(now.getFullYear(), now.getMonth(), 1);
Expand All @@ -355,7 +361,13 @@ class CopilotTokenTracker implements vscode.Disposable {
let cacheHits = 0;
let cacheMisses = 0;

for (const sessionFile of sessionFiles) {
for (let i = 0; i < sessionFiles.length; i++) {
const sessionFile = sessionFiles[i];

if (progressCallback) {
progressCallback(i + 1, sessionFiles.length);
}

try {
const fileStats = fs.statSync(sessionFile);

Expand Down Expand Up @@ -524,7 +536,8 @@ class CopilotTokenTracker implements vscode.Disposable {

private async countInteractionsInSession(sessionFile: string): Promise<number> {
try {
const sessionContent = JSON.parse(fs.readFileSync(sessionFile, 'utf8'));
const fileContent = await fs.promises.readFile(sessionFile, 'utf8');
const sessionContent = JSON.parse(fileContent);

// Count the number of requests as interactions
if (sessionContent.requests && Array.isArray(sessionContent.requests)) {
Expand All @@ -543,7 +556,8 @@ class CopilotTokenTracker implements vscode.Disposable {
const modelUsage: ModelUsage = {};

try {
const sessionContent = JSON.parse(fs.readFileSync(sessionFile, 'utf8'));
const fileContent = await fs.promises.readFile(sessionFile, 'utf8');
const sessionContent = JSON.parse(fileContent);

if (sessionContent.requests && Array.isArray(sessionContent.requests)) {
for (const request of sessionContent.requests) {
Expand Down Expand Up @@ -886,7 +900,8 @@ class CopilotTokenTracker implements vscode.Disposable {

private async estimateTokensFromSession(sessionFilePath: string): Promise<number> {
try {
const sessionContent = JSON.parse(fs.readFileSync(sessionFilePath, 'utf8'));
const fileContent = await fs.promises.readFile(sessionFilePath, 'utf8');
const sessionContent = JSON.parse(fileContent);
let totalInputTokens = 0;
let totalOutputTokens = 0;

Expand Down Expand Up @@ -963,8 +978,11 @@ class CopilotTokenTracker implements vscode.Disposable {
return;
}

// Get detailed stats
const stats = await this.calculateDetailedStats();
// Get detailed stats (with progress in status bar)
const stats = await this.updateTokenStats();
if (!stats) {
return;
}

// Create a small webview panel
this.detailsPanel = vscode.window.createWebviewPanel(
Expand Down Expand Up @@ -1049,9 +1067,10 @@ class CopilotTokenTracker implements vscode.Disposable {
}

// Update token stats and refresh the webview content
await this.updateTokenStats();
const stats = await this.calculateDetailedStats();
this.detailsPanel.webview.html = this.getDetailsHtml(stats);
const stats = await this.updateTokenStats();
if (stats) {
this.detailsPanel.webview.html = this.getDetailsHtml(stats);
}
}

private async refreshChartPanel(): Promise<void> {
Expand Down