wip on service ping

Author: brendan-kellam

.env.development

@@ -71,3 +71,5 @@ SOURCEBOT_TELEMETRY_DISABLED=true # Disables telemetry collection
 NODE_ENV=development
 
 DEBUG_WRITE_CHAT_MESSAGES_TO_FILE=true
+
+SOURCEBOT_LIGHTHOUSE_URL=http://localhost:3003

packages/shared/src/entitlements.ts

@@ -31,7 +31,6 @@ export type Plan = keyof typeof planLabels;
 const entitlements = [
     "search-contexts",
     "anonymous-access",
-    "multi-tenancy",
     "sso",
     "code-nav",
     "audit",

packages/shared/src/env.server.ts

@@ -266,6 +266,7 @@ const options = {
 
         SOURCEBOT_ENCRYPTION_KEY: z.string(),
         SOURCEBOT_INSTALL_ID: z.string().default("unknown"),
+        SOURCEBOT_LIGHTHOUSE_URL: z.string().url(),
 
         FALLBACK_GITHUB_CLOUD_TOKEN: z.string().optional(),
         FALLBACK_GITLAB_CLOUD_TOKEN: z.string().optional(),

packages/web/src/initialize.ts

@@ -1,5 +1,6 @@
 import { createGuestUser } from '@/lib/authUtils';
 import { __unsafePrisma } from "@/prisma";
+import { startServicePing } from '@/servicePing';
 import { OrgRole } from '@sourcebot/db';
 import { createLogger, env, hasEntitlement, loadConfig } from "@sourcebot/shared";
 import { SINGLE_TENANT_ORG_ID, SOURCEBOT_GUEST_USER_ID } from './lib/constants';
@@ -115,4 +116,5 @@ const init = async () => {
 
 (async () => {
     await init();
+    startServicePing();
 })();

packages/web/src/lib/utils.ts

@@ -589,4 +589,28 @@ export const isHttpError = (error: unknown, status: number): boolean => {
         && typeof error === 'object'
         && 'status' in error
         && error.status === status;
+}
+
+const sleep = (ms: number) => new Promise(resolve => setTimeout(resolve, ms));
+
+export const fetchWithRetry = async (
+    input: RequestInfo | URL,
+    init?: RequestInit,
+    { retries = 3, backoffMs = 1000 }: { retries?: number; backoffMs?: number } = {},
+): Promise<Response> => {
+    for (let attempt = 0; attempt <= retries; attempt++) {
+        try {
+            const response = await fetch(input, init);
+            if (response.ok || attempt === retries) {
+                return response;
+            }
+        } catch (error) {
+            if (attempt === retries) {
+                throw error;
+            }
+        }
+        await sleep(backoffMs * Math.pow(2, attempt));
+    }
+    // Unreachable, but TypeScript needs it
+    throw new Error('fetchWithRetry: exhausted retries');
 }

packages/web/src/servicePing.ts

@@ -0,0 +1,37 @@
+import { fetchWithRetry } from "@/lib/utils";
+import { createLogger, env, SOURCEBOT_VERSION } from "@sourcebot/shared";
+
+const logger = createLogger('service-ping');
+
+const ONE_DAY_MS = 24 * 60 * 60 * 1000;
+
+const sendPing = async () => {
+    const payload = {
+        installId: env.SOURCEBOT_INSTALL_ID,
+        version: SOURCEBOT_VERSION,
+    };
+
+    try {
+        const response = await fetchWithRetry(`${env.SOURCEBOT_LIGHTHOUSE_URL}/ping`, {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify(payload),
+        });
+
+        if (!response.ok) {
+            logger.error(`Service ping failed with status ${response.status}`);
+        } else {
+            logger.info(`Service ping sent successfully`);
+        }
+    } catch (error) {
+        logger.error(`Service ping failed: ${error}`);
+    }
+};
+
+export const startServicePing = () => {
+    // Fire immediately on startup
+    sendPing();
+
+    // Then repeat daily
+    setInterval(sendPing, ONE_DAY_MS);
+};