Merge branch 'main' into feat-accept-external-stats

Author: Xetera

src/env.ts

@@ -14,6 +14,7 @@ const envSchema = z.object({
   GITHUB_TOKEN: z.string().optional(),
   LOG_PATH: z.string().optional(),
   POSTGRES_URL: z.string().optional(),
+  SOURCE_DATABASE_URL: z.string().optional(),
   DEBUG: z.stringbool().default(false),
   STATISTICS_PATH: z.string().optional(),
 

src/main.ts

@@ -59,6 +59,7 @@ async function runOutsideCI() {
     env.HOST,
     env.PORT,
     Connectable.fromString(env.POSTGRES_URL),
+    env.SOURCE_DATABASE_URL ? Connectable.fromString(env.SOURCE_DATABASE_URL) : undefined,
   );
 
   const shutdown = async () => {

src/server/http.ts

@@ -117,6 +117,7 @@ export async function createServer(
   hostname: string,
   port: number,
   targetDb?: Connectable,
+  sourceDb?: Connectable,
 ): Promise<FastifyInstance> {
   const fastify = Fastify({ logger: false });
 
@@ -171,13 +172,15 @@ export async function createServer(
   });
 
   if (remoteController) {
-    fastify.post("/postgres", async (request, reply) => {
-      log.info(`[POST] /postgres`, "http");
-      const result = await remoteController.onFullSync(
-        JSON.stringify(request.body),
-      );
-      return reply.status(result.status).send(result.body);
-    });
+    if (!sourceDb) {
+      fastify.post("/postgres", async (request, reply) => {
+        log.info(`[POST] /postgres`, "http");
+        const result = await remoteController.onFullSync(
+          JSON.stringify(request.body),
+        );
+        return reply.status(result.status).send(result.body);
+      });
+    }
 
     fastify.get("/postgres", async (request, reply) => {
       log.info(`[GET] /postgres`, "http");
@@ -223,6 +226,20 @@ export async function createServer(
   }
 
   await fastify.listen({ host: hostname, port });
+
+  if (remoteController && sourceDb) {
+    log.info(`SOURCE_DATABASE_URL set, triggering initial sync`, "http");
+    remoteController.onFullSync(JSON.stringify({ db: sourceDb.toString() })).then((result) => {
+      if (result.status >= 400) {
+        log.error(`Initial sync failed: ${JSON.stringify(result.body)}`, "http");
+        process.exit(1);
+      }
+    }).catch((error) => {
+      log.error(`Initial sync failed: ${error}`, "http");
+      process.exit(1);
+    });
+  }
+
   return fastify;
 }