mirror of
https://github.com/alkimake/paperclip.git
synced 2026-06-15 02:20:38 +09:00
41 lines
1.1 KiB
TypeScript
41 lines
1.1 KiB
TypeScript
|
|
import path from "node:path";
|
||
|
|
import {
|
||
|
|
TelemetryClient,
|
||
|
|
resolveTelemetryConfig,
|
||
|
|
loadOrCreateState,
|
||
|
|
} from "@paperclipai/shared/telemetry";
|
||
|
|
import { resolvePaperclipInstanceRoot } from "./config/home.js";
|
||
|
|
import { readConfig } from "./config/store.js";
|
||
|
|
import { cliVersion } from "./version.js";
|
||
|
|
|
||
|
|
let client: TelemetryClient | null = null;
|
||
|
|
|
||
|
|
export function initTelemetry(fileConfig?: { enabled?: boolean }): TelemetryClient | null {
|
||
|
|
if (client) return client;
|
||
|
|
|
||
|
|
const config = resolveTelemetryConfig(fileConfig);
|
||
|
|
if (!config.enabled) return null;
|
||
|
|
|
||
|
|
const stateDir = path.join(resolvePaperclipInstanceRoot(), "telemetry");
|
||
|
|
client = new TelemetryClient(config, () => loadOrCreateState(stateDir, cliVersion), cliVersion);
|
||
|
|
return client;
|
||
|
|
}
|
||
|
|
|
||
|
|
export function initTelemetryFromConfigFile(configPath?: string): TelemetryClient | null {
|
||
|
|
try {
|
||
|
|
return initTelemetry(readConfig(configPath)?.telemetry);
|
||
|
|
} catch {
|
||
|
|
return initTelemetry();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
export function getTelemetryClient(): TelemetryClient | null {
|
||
|
|
return client;
|
||
|
|
}
|
||
|
|
|
||
|
|
export async function flushTelemetry(): Promise<void> {
|
||
|
|
if (client) {
|
||
|
|
await client.flush();
|
||
|
|
}
|
||
|
|
}
|