114 lines
2.9 KiB
JavaScript
114 lines
2.9 KiB
JavaScript
import https from 'https';
|
|
import { logInfo, logError } from './logger.js';
|
|
import { getConfig } from './config.js';
|
|
|
|
const VERSION_URL = 'https://downloads.factory.ai/factory-cli/LATEST';
|
|
const USER_AGENT_PREFIX = 'factory-cli';
|
|
const CHECK_INTERVAL = 60 * 60 * 1000; // 1 hour
|
|
const RETRY_INTERVAL = 60 * 1000; // 1 minute
|
|
const MAX_RETRIES = 3;
|
|
|
|
let currentVersion = null;
|
|
let isUpdating = false;
|
|
|
|
function getDefaultVersion() {
|
|
const cfg = getConfig();
|
|
const userAgent = cfg.user_agent || 'factory-cli/0.40.2';
|
|
const match = userAgent.match(/\/(\d+\.\d+\.\d+)/);
|
|
return match ? match[1] : '0.40.2';
|
|
}
|
|
|
|
function initializeVersion() {
|
|
if (currentVersion === null) {
|
|
currentVersion = getDefaultVersion();
|
|
}
|
|
}
|
|
|
|
export function getCurrentUserAgent() {
|
|
initializeVersion();
|
|
return `${USER_AGENT_PREFIX}/${currentVersion}`;
|
|
}
|
|
|
|
function fetchLatestVersion() {
|
|
return new Promise((resolve, reject) => {
|
|
const request = https.get(VERSION_URL, (res) => {
|
|
let data = '';
|
|
|
|
if (res.statusCode !== 200) {
|
|
reject(new Error(`HTTP ${res.statusCode}`));
|
|
return;
|
|
}
|
|
|
|
res.on('data', (chunk) => {
|
|
data += chunk;
|
|
});
|
|
|
|
res.on('end', () => {
|
|
const version = data.trim();
|
|
if (version && /^\d+\.\d+\.\d+/.test(version)) {
|
|
resolve(version);
|
|
} else {
|
|
reject(new Error(`Invalid version format: ${version}`));
|
|
}
|
|
});
|
|
});
|
|
|
|
request.on('error', (err) => {
|
|
reject(err);
|
|
});
|
|
|
|
request.setTimeout(30000, () => {
|
|
request.destroy();
|
|
reject(new Error('Request timeout'));
|
|
});
|
|
});
|
|
}
|
|
|
|
async function updateVersionWithRetry(retryCount = 0) {
|
|
if (isUpdating) {
|
|
return;
|
|
}
|
|
|
|
isUpdating = true;
|
|
|
|
try {
|
|
const version = await fetchLatestVersion();
|
|
if (version !== currentVersion) {
|
|
const oldVersion = currentVersion;
|
|
currentVersion = version;
|
|
logInfo(`User-Agent version updated: ${oldVersion} -> ${currentVersion}`);
|
|
} else {
|
|
logInfo(`User-Agent version is up to date: ${currentVersion}`);
|
|
}
|
|
isUpdating = false;
|
|
} catch (error) {
|
|
// Silently fail on network issues, will retry later
|
|
if (retryCount < MAX_RETRIES - 1) {
|
|
setTimeout(() => {
|
|
updateVersionWithRetry(retryCount + 1);
|
|
}, RETRY_INTERVAL);
|
|
} else {
|
|
// Only log after all retries failed
|
|
logInfo(`Using default User-Agent version: ${currentVersion}`);
|
|
isUpdating = false;
|
|
}
|
|
}
|
|
}
|
|
|
|
export function initializeUserAgentUpdater() {
|
|
initializeVersion();
|
|
logInfo('Initializing User-Agent version updater...');
|
|
logInfo(`Default User-Agent from config: ${USER_AGENT_PREFIX}/${currentVersion}`);
|
|
|
|
// Fetch immediately on startup
|
|
updateVersionWithRetry();
|
|
|
|
// Schedule hourly checks
|
|
setInterval(() => {
|
|
logInfo('Running scheduled User-Agent version check...');
|
|
updateVersionWithRetry();
|
|
}, CHECK_INTERVAL);
|
|
|
|
logInfo(`User-Agent updater initialized. Will check every hour.`);
|
|
}
|