anything-llm/server/models/telemetry.js
Timothy Carambat 122f29069c
Enable optional Telemetry integration via posthog (#190)
* Enable optional Telemetry integration via posthog
2023-08-14 17:42:17 -07:00

66 lines
1.9 KiB
JavaScript

const { v4 } = require("uuid");
const { SystemSettings } = require("./systemSettings");
const Telemetry = {
// Write-only key. It can't read events or any of your other data, so it's safe to use in public apps.
pubkey: "phc_9qu7QLpV8L84P3vFmEiZxL020t2EqIubP7HHHxrSsqS",
stubDevelopmentEvents: true, // [DO NOT TOUCH] Core team only.
label: "telemetry_id",
id: async function () {
const result = await SystemSettings.get(`label = '${this.label}'`);
if (!!result?.value) return result.value;
return result?.value;
},
connect: async function () {
const client = this.client();
const distinctId = await this.findOrCreateId();
return { client, distinctId };
},
isDev: function () {
if (process.env.NODE_ENV === "development")
return this.stubDevelopmentEvents;
return false;
},
client: function () {
if (process.env.DISABLE_TELEMETRY === "true" || this.isDev()) return null;
const { PostHog } = require("posthog-node");
return new PostHog(this.pubkey);
},
sendTelemetry: async function (event, properties = {}) {
try {
const { client, distinctId } = await this.connect();
if (!client) return;
console.log(`\x1b[32m[TELEMETRY SENT]\x1b[0m`, {
event,
properties,
});
client.capture({
event,
distinctId,
properties,
});
} catch {
return;
}
},
flush: async function () {
const { client } = this.client();
if (!client) return;
await client.shutdownAsync();
return;
},
setUid: async function () {
const newId = v4();
await SystemSettings.updateSettings({ [this.label]: newId });
return newId;
},
findOrCreateId: async function () {
const currentId = await this.id();
if (!!currentId) return currentId;
const newId = await this.setUid();
return newId;
},
};
module.exports = { Telemetry };