mirror of
https://github.com/Unleash/unleash.git
synced 2024-10-18 20:09:08 +02:00
ff7be7696c
Co-authored-by: Ivar Conradi Østhus <ivarconr@gmail.com>
128 lines
3.4 KiB
TypeScript
128 lines
3.4 KiB
TypeScript
import fetch from 'node-fetch';
|
|
import { IUnleashStores } from '../types/stores';
|
|
import { IUnleashConfig } from '../types/option';
|
|
import version from '../util/version';
|
|
import { Logger } from '../logger';
|
|
import { ISettingStore } from '../types/stores/settings-store';
|
|
|
|
const TWO_DAYS = 48 * 60 * 60 * 1000;
|
|
|
|
export interface IVersionInfo {
|
|
oss: string;
|
|
enterprise?: string;
|
|
}
|
|
|
|
export interface IVersionHolder {
|
|
current: IVersionInfo;
|
|
latest: IVersionInfo | {};
|
|
isLatest: boolean;
|
|
instanceId: string;
|
|
}
|
|
|
|
export default class VersionService {
|
|
private logger: Logger;
|
|
|
|
private settingStore: ISettingStore;
|
|
|
|
private current: IVersionInfo;
|
|
|
|
private latest?: IVersionInfo;
|
|
|
|
private enabled: boolean;
|
|
|
|
private versionCheckUrl: string;
|
|
|
|
private instanceId?: string;
|
|
|
|
private isLatest: boolean;
|
|
|
|
private timer: NodeJS.Timeout;
|
|
|
|
constructor(
|
|
{ settingStore }: Pick<IUnleashStores, 'settingStore'>,
|
|
{
|
|
getLogger,
|
|
versionCheck,
|
|
enterpriseVersion,
|
|
}: Pick<
|
|
IUnleashConfig,
|
|
'getLogger' | 'versionCheck' | 'enterpriseVersion'
|
|
>,
|
|
) {
|
|
this.logger = getLogger('lib/services/version-service.js');
|
|
this.settingStore = settingStore;
|
|
this.current = {
|
|
oss: version,
|
|
enterprise: enterpriseVersion || '',
|
|
};
|
|
this.enabled = versionCheck.enable;
|
|
this.versionCheckUrl = versionCheck.url;
|
|
this.isLatest = true;
|
|
process.nextTick(() => this.setup());
|
|
}
|
|
|
|
async setup(): Promise<void> {
|
|
await this.setInstanceId();
|
|
await this.checkLatestVersion();
|
|
this.timer = setInterval(
|
|
async () => this.checkLatestVersion(),
|
|
TWO_DAYS,
|
|
);
|
|
this.timer.unref();
|
|
}
|
|
|
|
async setInstanceId(): Promise<void> {
|
|
try {
|
|
const { id } = await this.settingStore.get('instanceInfo');
|
|
this.instanceId = id;
|
|
} catch (err) {
|
|
this.logger.warn('Could not find instanceInfo');
|
|
}
|
|
}
|
|
|
|
async checkLatestVersion(): Promise<void> {
|
|
if (this.enabled) {
|
|
try {
|
|
const res = await fetch(this.versionCheckUrl, {
|
|
method: 'POST',
|
|
body: JSON.stringify({
|
|
versions: this.current,
|
|
instanceId: this.instanceId,
|
|
}),
|
|
headers: { 'Content-Type': 'application/json' },
|
|
});
|
|
if (res.ok) {
|
|
const data = await res.json();
|
|
this.latest = {
|
|
oss: data.versions.oss,
|
|
enterprise: data.versions.enterprise,
|
|
};
|
|
this.isLatest = data.latest;
|
|
} else {
|
|
this.logger.info(
|
|
`Could not check newest version. Status: ${res.status}`,
|
|
);
|
|
}
|
|
} catch (err) {
|
|
this.logger.info('Could not check newest version', err);
|
|
}
|
|
}
|
|
}
|
|
|
|
getVersionInfo(): IVersionHolder {
|
|
return {
|
|
current: this.current,
|
|
latest: this.latest || {},
|
|
isLatest: this.isLatest,
|
|
instanceId: this.instanceId,
|
|
};
|
|
}
|
|
|
|
destroy(): void {
|
|
clearInterval(this.timer);
|
|
this.timer = null;
|
|
}
|
|
}
|
|
|
|
module.exports = VersionService;
|