feat: Add update checker to notify users of new versions

Author: evanlong-me

bin/cli.js

@@ -11,6 +11,7 @@ const sorter = require('../lib/sorter');
 const aggregator = require('../lib/aggregator');
 const projectDetector = require('../lib/project-detector');
 const pricing = require('../lib/pricing');
+const updateChecker = require('../lib/update-checker');
 const { version } = require('../package.json');
 
 program
@@ -49,6 +50,9 @@ if (process.argv.slice(2).length === 0) {
 }
 
 async function showUsage(options) {
+  // Check for updates
+  await updateChecker.checkForUpdates();
+  
   try {
     const { messages } = await usage.getUsage();
 
@@ -220,6 +224,9 @@ async function showUsage(options) {
 }
 
 async function showProjects() {
+  // Check for updates
+  await updateChecker.checkForUpdates();
+  
   const spinner = createSpinner('Fetching project list...').start();
   try {
     const { projects, messageCount } = await usage.getProjects();
@@ -241,6 +248,9 @@ async function showProjects() {
 }
 
 async function showModels() {
+  // Check for updates
+  await updateChecker.checkForUpdates();
+  
   const spinner = createSpinner('Fetching model pricing data...').start();
   try {
     const pricingData = await pricing.fetchModelPricing();

lib/update-checker.js

@@ -0,0 +1,49 @@
+
+const https = require('https');
+const chalk = require('chalk');
+const { createSpinner } = require('nanospinner');
+const { name, version } = require('../package.json');
+
+const NPM_REGISTRY_URL = `https://registry.npmjs.org/${name}`;
+
+function fetchLatestVersion() {
+  return new Promise((resolve, reject) => {
+    https.get(NPM_REGISTRY_URL, (res) => {
+      let data = '';
+      res.on('data', (chunk) => {
+        data += chunk;
+      });
+      res.on('end', () => {
+        try {
+          const json = JSON.parse(data);
+          resolve(json['dist-tags'].latest);
+        } catch (e) {
+          reject(e);
+        }
+      });
+    }).on('error', (e) => {
+      reject(e);
+    });
+  });
+}
+
+async function checkForUpdates() {
+  const spinner = createSpinner('Checking for updates...').start();
+  
+  try {
+    const latestVersion = await fetchLatestVersion();
+    
+    if (version !== latestVersion) {
+      spinner.warn({ text: `New version available! Run: ${chalk.bgYellow.black.bold(` npm install -g ${name} `)}` });
+      console.log('');
+    } else {
+      spinner.success({ text: `You're using the latest version (${version})` });
+    }
+  } catch (e) {
+    spinner.stop();
+    // Silently ignore errors - we don't want to interrupt the main functionality
+  }
+}
+
+module.exports = { checkForUpdates };
+